vae-lagging-encoder

修复VAE后验坍缩,让生成模型更稳定多样

这是一个基于PyTorch的变分自编码器(VAE)改进实现,对应ICLR 2019论文《Lagging Inference Networks and Posterior Collapse in Variational Autoencoders》。该论文针对VAE训练中常见的后验坍缩问题(即解码器忽略潜变量,导致生成多样性下降)提出解决方案,通过引入滞后推理网络(lagging inference network)机制,在训练过程中动态调整编码器与解码器的更新节奏,从而有效缓解后验坍缩。项目提供了完整的模型定义、训练脚本和实验配置,支持图像生成和文本生成两种任务场景,并包含论文中的关键实验复现。核心能力包括:实现滞后更新策略、提供多种基准数据集(如MNIST、Yahoo等)的训练示例、支持自定义数据扩展。该项目是研究VAE稳定性训练和生成模型的重要参考实现,适合学术复现和二次开发。

开源 free 图像生成
访问官网 ↗ GitHub ↗ 文档 ↗
GitHub 星标 ★ 185
维护状态 低维护
是否开源
定价模式 free

项目数据

分类图像生成
开发团队jxhe
所属国家
官网地址
定价模式free
价格说明开源项目,MIT许可证,完全免费,可自行部署使用。
访问状态
是否开源
开源协议MIT
主要语言Python
技术栈/模型image-generation,posterior-collapse,text-generation,vae
GitHub 星标★ 185
30天Star增速
HF 下载量
上线时间2018-07-19 00:00:00
最近更新2026-06-15 00:00:00
维护状态低维护
中文支持
访问方式
移动端支持
综合评分
收录时间2026-08-09
浏览次数2

使用教程

核心亮点

  • 直接复现ICLR 2019经典论文,算法实现完整,适合学术研究
  • 同时覆盖图像和文本生成任务,验证了方法的通用性
  • 代码结构清晰,基于PyTorch,易于修改和扩展

不足之处

  • 项目维护较少,依赖库版本较老,可能需自行适配新环境
  • 文档/社区待观察

适用场景

  • 学术研究:复现论文实验或对比VAE改进方法
  • 生成模型开发:需要稳定训练VAE并避免后验坍缩的场景
  • 教学示例:讲解VAE训练难点及解决方案

替代项目

Variational Autoencoder (PyTorch官方示例)、beta-VAE、VQ-VAE

项目介绍

vae-lagging-encoder 是模型训练领域的开源项目,由 jxhe 开发,2018 年首次发布。

它收录于模型训练分类,该分类目前共有 518 个项目,GitHub 星标数为 185。

项目已超过三个月没有代码更新,维护节奏明显放缓,最近一次代码更新于 2026-06-15。MIT许可证,完全免费,可自行部署使用。

它主要面向的使用场景是:学术研究:复现论文实验或对比VAE改进方法。同类可对比的替代方案包括 Variational Autoencoder (PyTorch官方示例)、beta-VAE、VQ-VAE。

上一篇:PyTorch-CycleGAN

下一篇:denoising_diffusion

同类项目推荐

Long-RL 开源

让强化学习轻松驾驭超长序列,训练更稳更快

Long-RL: Scaling RL to Long Sequences (NeurIPS 2025)

★ 729 2026-08-09
TPA 开源

把注意力复杂度从平方降到线性,长序列不再卡顿

[NeurIPS 2025 Spotlight] TPA: Tensor ProducT ATTenTion Transformer (https://arxiv.or···

★ 463 2026-08-09
minimind 开源

两小时造出你的小模型,练手入门不求人

Train a 64M-parameter LLM from scratch in just 2h!

★ 62178 2026-08-09
pruna 开源

一键优化 AI 模型,推理提速又省资源。

Pruna is a model optimization framework built for developers, enabling you to delive···

★ 1286 2026-08-13