arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2604.09258cs.LG
收藏

Nexus: 相同预训练损失,通过公共极小值实现更好的下游泛化

Nexus: Same Pretraining Loss, Better Downstream Generalization via Common Minima

Tsinghua University(清华大学)

Huanran Chen, Huaqing Zhang, Xiao Li, Yinpeng Dong, Ke Shen, Jun Zhu

AI总结:

本文提出Nexus优化器,通过最大化梯度相似性促使不同数据源的损失函数极小值靠近,在保持相同预训练损失的情况下显著提升下游泛化性能。

摘要:

大型语言模型的基础能力是在互联网规模、高度异构的数据混合上进行预训练时获得的。在这项工作中,我们研究了关于预训练收敛状态的一个有趣的几何问题:模型是否收敛到所有数据源的公共极小值(例如,图\cref{fig:cwa_illustration:close}),还是仅仅收敛到总损失的极小值(例如,图\cref{fig:cwa_illustration:distant})?我们假设任务特定极小值的几何“接近度”与下游泛化内在相关。我们发现标准优化器(例如AdamW)通常收敛到任务特定极小值彼此远离的点。为了解决这个问题,我们提出了Nexus优化器,它通过在优化过程中最大化梯度相似性来鼓励这些极小值的接近。在从130M到3B参数的各种模型、多种数据混合和超参数调度下的实验表明,Nexus在实现相同预训练损失的情况下显著提升了下游性能(见图\cref{fig:demo:benchmark})。值得注意的是,在3B模型上,Nexus将分布外损失降低了0.012,并在复杂推理任务(例如GSM8k)上带来了高达15.0%的准确率提升。这一发现挑战了将预训练损失作为模型评估唯一代理的依赖,并展示了隐式偏好在解锁下游泛化中的重要性。

英文摘要:

The foundational capabilities of large language models are acquired during pretraining on internet-scale, highly heterogeneous data mixtures. In this work, we investigate an interesting geometric question regarding the converged state of pretraining: Does the model converge to a common minimizer across all data sources (e.g., Fig. 2b), or merely a minimizer of the summed loss (e.g., Fig. 2a)? We hypothesize that the geometric "closeness" of task-specific minima is intrinsically linked to downstream generalization. We reveal that standard optimizers (e.g., AdamW) often converge to points where task-specific minima are distant from each other. To address this, we propose the Nexus optimizer, which encourages the closeness of these minima by maximizing gradient similarity during optimization. Experiments across models ranging from 130M to 3B parameters, various data mixtures and hyperparameter schedules, show that Nexus significantly boosts downstream performance, despite achieving the same pretraining loss (see Fig. 1). Notably, on the 3B model, Nexus reduces the out-of-distribution loss by 0.012 and yields up to a 15.0% accuracy improvement on complex reasoning tasks (e.g., GSM8k). This finding challenges the reliance on pretraining loss as the sole proxy for model evaluation and demonstrates the importance of implicit biases in unlocking downstream generalization.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑