vae-lagging-encoder

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

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 图像生成 🍴 复刻 33 🐛 待处理 2 👀 关注 3
GitHub ↗
GitHub 星标 ★ 185
维护状态 低维护
是否开源 开源
定价模式 🏷️完全免费 开源项目,MIT许可证,完全免费,可自行部署使用。

项目预览

vae-lagging-encoder 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类图像生成
开发团队jxhe
定价模式🏷️完全免费
价格说明开源项目,MIT许可证,完全免费,可自行部署使用。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议MIT
主要语言Python
技术栈/模型image-generation,posterior-collapse,text-generation,vae
GitHub 星标★ 185
上线时间2018-07-19 00:00:00
代码更新2020-12-15
仓库体积855 KB
本站数据更新2026-11-10
维护状态低维护
收录时间2026-08-09
浏览次数3

项目介绍

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

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 185

核心亮点

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

不足之处

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

适用场景

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

替代项目

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

上一篇:PyTorch-CycleGAN

下一篇:denoising_diffusion

同类项目推荐

OpenRLHF 开源

基于 Ray 的强化学习框架,让大规模 RLHF 训练又快又稳

An Easy-to-use, Scalable and High-performance Agentic RL Framework based on Ray (PPO···

★ 10146 2026-08-09
SLED 开源

不训练不微调,用模型自己纠正自己,让回答更靠谱

SLED: Self Logits Evolution Decoding for Improving Factuality in Large Language Mode···

★ 123 2026-08-09
Awesome-Model-Quantization 开源

一页看全模型量化论文、工具与部署方案

A curated collection of papers, benchmarks, surveys, and tools for model quantizatio···

★ 2465 2026-10-17
runbooks 开源

在K8s上声明式微调LLM,一键跑通训练全流程

Finetune LLMs on K8s by using Runbooks

★ 169 2026-08-09
rlix 开源

让 RL 实验排队更聪明,GPU 等待时间更短

Run more RL experiments. Wait less for GPUs.

★ 300 2026-08-09
HoliTom 开源

合并视频令牌,让视频大模型推理快数倍

[NeurIPS 2025] HoliTom: Holistic Token Merging for Fast Video Large Language Models

★ 86 2026-08-09