Scaling-Diffusion-Transformers-muP

让扩散模型从小规模到大规模训练,超参数无需重调,省时省力。

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

项目预览

Scaling-Diffusion-Transformers-muP 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类图像生成
开发团队ML-GSAI
定价模式🏷️完全免费
价格说明开源项目,MIT许可证,完全免费,可自行部署使用。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议MIT
主要语言Python
技术栈/模型diffusion-models,diffusion-transformer,hyperparameter-tuning,image-generation,mup,scaling
GitHub 星标★ 101
上线时间2025-05-21 00:00:00
代码更新2025-11-02
仓库体积5.7 MB
本站数据更新2026-11-10
维护状态低维护
收录时间2026-08-09
浏览次数2

项目介绍

Scaling-Diffusion-Transformers-muP 是 NeurIPS 2025 论文的官方实现,旨在解决扩散 Transformer(DiT)在规模扩展时超参数难以迁移的问题。通过引入最大更新参数化(μP),使得模型在不同规模下(从百万到十亿参数)能够保持最优超参数不变,从而避免大规模训练时昂贵的超参数搜索。项目核心能力包括:提供 μP 化的 DiT 模型实现、支持与标准 DiT 的对比实验、以及训练脚本和配置。主要解决大规模扩散模型训练中调参成本高、训练不稳定的痛点,让研究者可以更高效地将小规模实验的结论直接迁移到大规模模型,加速扩散模型的迭代。

Star 增长趋势

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

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

核心亮点

  • 基于 NeurIPS 2025 论文,方法有理论支撑,权威性强
  • 官方实现,代码质量有保障,可直接复现论文结果
  • 提供 μP 与标准参数学的对比,便于用户理解效果差异

不足之处

  • 项目处于早期阶段,文档和社区支持尚待完善
  • 仅支持 PyTorch,且依赖特定库(如 mup),上手有一定门槛

适用场景

  • 扩散模型研究者需要快速验证不同规模下的训练效果
  • 工业界训练大规模图像生成模型时希望节省调参成本
  • 学术实验需要将小规模超参数配置迁移到大规模场景

替代项目

mup (Microsoft)、Diffusion Transformers (DiT)、OpenAI DiT

上一篇:simple-diffusion

下一篇:modular-diffusion

同类项目推荐

Sklearn-genetic-opt 开源

用遗传算法自动调参,告别网格搜索的低效

Hyperparameter optimization and feature selection for scikit-learn using evolutionar···

★ 391 2026-08-09
Torch-Pruning 开源

一键剪枝任意 PyTorch 模型,自动处理复杂依赖,压缩提速不降准。

[CVPR 2023] DepGraph: Towards Any Structural Pruning; LLMs, Vision Foundation Models···

★ 3370 2026-08-09
SPO 开源

分步优化扩散模型,让AI生图更懂审美

[CVPR 2025] Aesthetic Post-Training Diffusion Models from Generic Preferences with S···

★ 271 2026-08-09
catalyst 开源

让深度学习实验跑得更快、更规范,专注模型创新

Accelerated deep learning R&D

★ 3388 2026-08-10
Fast-LLM 开源

让 LLM 训练跑满 GPU,少等快出结果

Accelerating your LLM training to full speed! Made with ❤️ by ServiceNow Research

★ 339 2026-09-26
LLMs-from-scratch 开源

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

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

★ 108157 2026-08-09
PRIME 开源

让大模型学会深度推理,强化学习训练一步到位

Scalable RL solution for advanced reasoning of language models

★ 1873 2026-08-09
TinyEngram 开源

给大模型装上长期记忆,让对话和生成更连贯

Research of DeepSeek Engram Architecture based on Qwen-3 and Stable Diffusion series···

★ 1342 2026-08-09
xllm 开源

轻量预训练骨架,改几行就能跑起训练实验

Light-weight infrastructure for express LLM pre-training

★ 124 2026-10-31