正交约束对语言模型预训练帮助有限:系统性研究指向球面优化
Do Orthogonality Constraints Help Language-Model Training Beyond Sphere Optimization?
研究者在 200B tokens 规模上复检了正交约束的各种玩法,结论是真正起作用的只是球面优化,还顺手拆穿了 5 个先前方法的增益来源。
研究团队在 360M 到 3.1B 参数的 decoder-only 模型上用近 200B tokens 系统测试了 14 种正交训练变体。结果显示块大小 d>1 的分块正交始终不如 d=1,唯一有微小增益的 Q/K 投影对称正定因子也随模型变大而消失。d=1 时约束退化为球面优化,其本身已优于调参后的 Muon。团队还复检了 5 个声称正交带来提升的既有方法,发现增益实际来自数值精度、训练时长或逐行范数控制等因素。
Do Orthogonality Constraints Help Language-Model Training Beyond Sphere Optimization?
A growing body of work reports benefits from orthogonality constraints in fine-tuning large language models (LLMs). These results raise an open question: do orthogonality constraints also help LLM pre-training? We conduct a systematic empirical study of decoder-only LLM pre-training with models ranging from 360M to 3.1B parameters, trained on nearly 200B tokens. We examine fourteen variants of orthogonality training: some constrain the weight matrix itself, while others use orthogonal factors to reparameterize it. Because a strictly orthogonal square matrix loses half of its degrees of freedom, we also study block-wise orthogonality, where every $d$ rows form a row-orthonormal (Stiefel) matrix and the block size $d$ is a hyperparameter. Among block sizes ranging from 1 to the shorter side of the matrix, no $d>1$ used throughout the network reliably outperforms $d=1$, and large blocks are clearly worse; the one placement that helps, symmetric positive-definite factors on the query and key projections, loses its small gain as models grow. At $d=1$, the constraint reduces to sphere optimization, with each row direction on its own unit sphere, which by itself already improves on tuned Muon. The same two findings hold for a diffusion transformer trained on ImageNet, and after fine-tuning, downstream accuracy does not separate the parameterizations. Our conclusion is that, across the realizations we tried, orthogonality between rows does not help language-model pre-training beyond sphere optimization. Since this appears to conflict with several prior results, we re-examine five prior methods in pre-training and fine-tuning that report orthogonality improvements. In each case the reported gain is attributable to numerical precision, training horizon, optimizer realization, or per-row norm control rather than to the orthogonality the methods impose.