Clean优化器:线性内存实现二阶LLM训练
原文标题:Clean: Second-order LLM Training at Linear Memory Cost via Nystr\"om Sketching
重要度 3/5 · AI 综合评估报道影响力与新鲜度,4–5 分为高价值资讯。
研究者提出Clean优化器,利用随机Nyström方法近似SOAP中的左右预条件子,将优化器内存复杂度从模型维度的二次方降至线性。其低精度变体Q-Clean在预训练LLaMA-1.3B时比Muon减少超50%优化器内存,且以比AdamW更小的状态占用在26%更短墙钟时间内达到同等性能。该方法还首次实现单张80GB GPU预训练130亿参数模型,为大规模模型优化提供更可扩展、易获取的路径。
- 问题
- 全曲率优化器(如 SOAP)内存开销过大,难以训练大模型。
- 方法
- 用随机 Nyström 方法近似 SOAP 左右预条件子,把优化器内存从二次降为线性,并回填子空间外曲率。
- 结果
- Q-Clean 预训练 LLaMA-1.3B 比 Muon 省超 50% 优化器内存,比 AdamW 快 26% 墙钟时间达到同等性能,单张 80GB GPU 可预训练 13B 模型。
- 影响
- 让全曲率优化在单卡上跑 13B 预训练成为可能,直接降低大模型训练的内存门槛。
深读 · 结论先行AI 生成展开看要点与影响 ↓
Clean 优化器把二阶预条件内存从二次方降到线性,单张 80GB GPU 可预训练 130 亿参数模型。
- •用随机 Nyström 方法近似 SOAP 的左右预条件子,内存复杂度由模型维度二次方降至线性
- •重新整合子空间外分量,在低秩近似之外保留曲率信息
- •低精度变体 Q-Clean 预训练 LLaMA-1.3B 时比 Muon 减少超 50% 优化器内存
- •优化器状态占用小于 AdamW,却以短 26% 的墙钟时间达到同等性能;首次支持单张 80GB GPU 预训练 13B 模型
为什么重要:同时压低显存与训练时间,让全曲率优化器从「用不起」变成中小团队可实际部署的选项。
以上深读由 AI 依据标题/摘要/正文节选生成,仅供参考;事实以原文为准。
AI 处理过程展开/收起
相关推荐
OpenAI 宣布数学、计算机科学、物理学三项新发现
AI 摘要OpenAI 宣布在调和分析、复杂性理论和量子场论三个方向取得新发现,并采用与学术界合作的新方法。公司邀请外部专家组织团队理解研究结果,参与者无需签署保密协议、可自由讨论并发表结论,且为自愿无报酬。OpenAI 表示不会为内部模型产生的发现寻求学术署名。
数学家称需数年消化 OpenAI 最新数学成果
AI 摘要OpenAI 本周突然发布近 400 项 AI 生成的数学成果,分布在 700 多份手稿中,覆盖组合数学、几何、数论等多个领域。多位数学家表示「震惊」「难以置信」,认为仅理解这些成果就需数年,且验证程度参差不齐。
OpenAI 公布数学难题突破成果
AI 摘要OpenAI 发布由未公开前沿模型生成的 722 篇数学手稿,覆盖 372 个成果族,声称解决了数百个长期未解问题,并公开 Lean 形式化证明与算力估算。此举延续了 AI 冲击数学界的势头,也引发研究伦理与学术规范的争议。独立顾问组织 AGMAI 呼吁实验室通过正规学术渠道发布成果,而非将其当作营销工具。
OpenAI 再抛数学突破成果引争议
AI 摘要OpenAI 公布 722 篇手稿,称未发布的前沿模型解决了数百个数学开放问题,平均每个结果消耗约三小时 ChatGPT Pro 级算力。成果虽令人震撼,但数学界对发布方式、研究伦理与学术规范提出质疑。AGMAI 建议实验室及时通过正规渠道披露模型、提示词与算力成本,避免把成果当作营销手段。