DiffusionOPSD

让扩散模型自我蒸馏,少步采样也能生成高质量图像

DiffusionOPSD 是一个针对扩散模型的在线策略自蒸馏(On-Policy Self-Distillation)训练框架,旨在解决扩散模型在生成质量和训练效率上的平衡问题。它通过让模型在训练过程中自我蒸馏,利用当前策略生成的样本作为监督信号,从而无需额外教师模型即可持续提升生成性能。项目基于 PyTorch 实现,支持图像生成和编辑任务,核心能力包括高效的采样策略、稳定的训练流程以及灵活的配置接口。该框架特别适用于需要高保真图像输出且计算资源有限的场景,能够在减少采样步数的同时保持生成质量,为扩散模型的研究和应用提供了一个轻量级但有效的优化工具。

开源 free 模型训练
访问官网 ↗ GitHub ↗ 文档 ↗
GitHub 星标 ★ 161
维护状态 活跃
是否开源
定价模式 free

项目数据

分类模型训练
开发团队worldbench
所属国家
定价模式free
价格说明开源项目,免费使用,定价信息待确认
访问状态
是否开源
开源协议Apache-2.0
主要语言Python
技术栈/模型aigc,diffusion-models,generative-ai,image-editing,image-generation,on-policy-distillation,on-policy-self-distillation,post-training,reinforcement-learning,rl,t2i,text-to-image,text-to-image-generation
GitHub 星标★ 161
30天Star增速
HF 下载量
上线时间2026-07-29 00:00:00
最近更新2026-08-28 00:00:00
维护状态活跃
中文支持
访问方式
移动端支持
综合评分
收录时间2026-08-29
浏览次数0

核心亮点

  • 在线自蒸馏机制,无需预训练教师模型,降低训练成本
  • 支持少步采样,显著提升推理速度,适合实时生成场景
  • 代码结构清晰,基于 PyTorch,易于集成到现有扩散模型流程

不足之处

  • 项目处于早期阶段,文档和社区支持尚不完善
  • 仅支持图像生成/编辑,未覆盖音频、视频等多模态任务

适用场景

  • 图像生成任务中需要加速采样且保持画质的场景
  • 计算资源有限,希望减少训练开销的扩散模型研究
  • 图像编辑应用中需要快速迭代生成结果的产品开发

替代项目

Progressive Distillation、Consistency Models、Adversarial Diffusion Distillation

项目介绍

DiffusionOPSD 是一个模型训练领域的开源项目,官方简介:On-Policy Self-Distillation in Diffusion Models。项目使用 Python 开发,在 GitHub 上获得 161 星标。

上一篇:lightly-train

下一篇:没有了!

同类项目推荐

pruna 开源

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

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

★ 1273 2026-08-13
minimind 开源

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

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

★ 55113 2026-08-09
LLMs-from-scratch 开源

手把手从零造出 ChatGPT,原理代码一次搞懂

Implement a ChatGPT-like LLM in PyTorch from scratch, step by step

★ 103950 2026-08-09
petals 开源

在家用 BitTorrent 方式跑大模型,推理微调快 10 倍

Run LLMs at home, BitTorrent-style. Fine-tuning and inference up to 10x faster than ···

★ 10526 2026-08-09