核方法求解器不再被预条件子内存卡住:分块草图让模型参数扩到千万级,精度还更高
核岭回归的预条件子内存不再随模型参数增长,Bloks 把参数规模推到千万级,在 Higgs 等大规模基准上取得与当前方法相当或更好的精度。
核岭回归的预条件子内存不再随模型参数增长,Bloks 把参数规模推到千万级,在 Higgs 等大规模基准上取得与当前方法相当或更好的精度。
作者提出 Clean 与低精度变体 Q-Clean:用随机 Nyström 方法近似 SOAP 的左右预条件子,把优化器状态内存从随层维度二次增长降为线性,并把低秩核心之外的 Main、A、B、C 四个正交块重新整合进更新;在 C4 上预训练 LLaMA-350M 与 1.3B 时,Clean 困惑度接近 SOAP 而优化器内存小于 AdamW,Q-Clean 相比 Muon 减少超过 50% 优化器内存,Clean 达到 AdamW 最终性能的墙钟时间快 26%,并首次使 13B 模型可在单张 80GB H100 上预训练。
已读完全部报道