跳到正文
原文
arXiv · Machine Learning Theory· Vishrut Goyal, Rohan Ramkumar·· 3 小时前AI 评分37

Spectrally Targeted Muon 优化器:只对部分奇异值正交化

Spectrally Targeted Muon

AI 导读

研究者提出 Spectrally Targeted Muon 优化器,仅对高于或低于阈值 τ 的奇异值做正交化,通过在平移 Gram 矩阵上运行 Newton-Schulz 迭代隔离相关奇异子空间,无需 SVD,并可在 normalized SGD 与 Muon 之间插值。

来源:arXiv · Machine Learning Theory · arxiv.org