AITP
精选全部 AI 动态AI 日报Agent 接入我的简报我的追踪阅读偏好内容方法关于更新日志信源提报反馈
外观
登录 / 注册
AITOP

CMuon

共 1 条相关 AI 资讯
8月4日
11:32
11:32官方账号arXiv cs.AI@Chuyan Chen, Peng Sun, Kun Yuan
CMuon是一种针对Diffusion Transformer(DiT)训练的新优化器,通过将参数矩阵分块后再做正交化,解决了原版Muon在后期收敛停滞的问题。在ImageNet 256上,用CMuon训练的675M参数DiT仅用200个epoch就达到FID 1.18。相比AdamW,训练速度提升超过2倍。该方法能有效避免融合张量引发的隐式子空间耦合,保持更新方向准确。
论文CMuonMuonDiT

推荐理由:写了个新优化器CMuon,训练扩散Transformer比AdamW快两倍多,675M模型200轮就跑到FID 1.18,厉害得很。
原文
精选全部日报登录