DDSM

让扩散模型训练更快更省,按步骤智能分配学习资源

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

项目预览

DDSM 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类图像生成
开发团队EnVision-Research
定价模式🏷️完全免费
价格说明开源项目,MIT许可证,完全免费,无付费版本。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议MIT
主要语言Python
技术栈/模型diffusion-models,efficient-deep-learning,iclr2024,image-generation,model-pruning
GitHub 星标★ 61
上线时间2023-11-29 00:00:00
代码更新2024-02-06
仓库体积234 KB
本站数据更新2026-11-08
维护状态低维护
收录时间2026-08-09
浏览次数3

项目介绍

DDSM(Denoising Diffusion Step-aware Models)是ICLR 2024收录的扩散模型训练优化方法。它针对扩散模型训练中不同去噪步骤(timestep)的难度差异,提出了一种步骤感知的训练策略,通过动态调整各步骤的损失权重或网络容量,让模型更高效地学习。核心能力包括:基于步骤感知的损失加权、可选的模型剪枝技术,以及更快的收敛速度和更优的生成质量。项目提供PyTorch实现,代码结构清晰,适合研究者复现和扩展。它解决的是扩散模型训练效率低、收敛慢的问题,在保持生成质量的同时降低计算成本。

Star 增长趋势

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

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

核心亮点

  • 提出步骤感知训练策略,显著提升收敛速度,减少训练迭代次数
  • 支持模型剪枝,可进一步压缩模型大小,适合资源受限场景
  • ICLR 2024论文官方实现,算法有理论支撑,可靠性高

不足之处

  • 项目处于早期阶段(62星),文档和社区支持尚不完善
  • 仅提供基础实现,未包含预训练模型或完整训练配置,上手需自行配置

适用场景

  • 学术研究:复现ICLR 2024论文,探索扩散模型训练优化
  • 高效训练:在有限算力下训练高质量图像生成模型
  • 模型压缩:结合剪枝技术部署到边缘设备

替代项目

DPM-Solver、EDM (Elucidating Diffusion Models)、Progressive Distillation

同类项目推荐

evostra 开源

用 Python 快速实现进化策略,轻松训练强化学习模型

A fast Evolution Strategy implementation in Python

★ 271 2026-08-09
litgpt 开源

一条命令搞定大模型预训练、微调和部署,省时省力

20+ high-performance LLMs with recipes to pretrain, finetune and deploy at scale.

★ 13747 2026-08-09
Kardia-R1 开源

让AI学会共情,用强化学习提升情感支持回复质量

[WWW] Kardia-R1: Unleashing LLMs to Reason toward Understanding and Empathy for Emot···

★ 60 2026-08-09
Finetuning-Suite 开源

30秒微调 Hugging Face 任意模型,极速上手

Finetune any model on HF in less than 30 seconds

★ 59 2026-08-09
alignment-handbook 开源

一键复现主流 LLM 对齐实验,从 SFT 到 DPO 全流程开箱即用。

Robust recipes to align language models with human and AI preferences

★ 5711 2026-08-09
SuperPiano 开源

在 Colab 里用 Transformer 一键生成钢琴曲

Absolutely amazing SOTA Google Colab (Jupyter) Notebooks for creating/training SOTA ···

★ 91 2026-08-09
mlx-tune 开源

在 Mac 上本地微调大模型,多模态全支持,速度飞快

Fine-tune LLMs on your Mac with Apple Silicon. SFT, DPO, GRPO, Vision, TTS, STT, Emb···

★ 1456 2026-08-09
llamaduo 开源

把大模型能力搬回家,低成本打造专属小模型

[ACL'25] Official Code for LlamaDuo: LLMOps Pipeline for Seamless Migration from Ser···

★ 317 2026-08-09
meta-context-engineering 开源

让AI自动进化上下文策略,告别手写提示词

[ICML 2026] Meta Context Engineering via Agentic Skill Evolution

★ 184 2026-08-09