arXiv:2510.25244cs.LG2025-10EMNLP被引 4

通过分离参数更新的主次空间,实现训练加速与稳定。

BSFA: Leveraging the Subspace Dichotomy to Accelerate Neural Network Training

  • 区分损失海森矩阵的主导空间与非主导空间,分别调节更新强度。
  • 在预训练任务中实现约2倍速度提升,如LLaMA-72M和LLaMA-134M。
  • 适用于大模型,支持高效插件式部署,无需修改网络结构。

近期研究揭示深度学习优化中的基本二分现象:尽管损失海森矩阵的前主方向(Dom-space)承载了大部分参数更新幅度,但其对损失下降贡献甚微;相反,正交分量(Bulk-space)虽更新量小,却驱动主要学习进展。本文进一步理解该现象,并提出一种新型即插即用框架——批量空间滤波加速器(BSFA)。BSFA通过差异化缩放投影至不同子空间的更新分量,在抑制主导空间过激更新的同时增强非主导空间的贡献,从而兼顾稳定性与收敛速度。为确保在大型模型上的实用性与可扩展性,提出两项关键创新:基于历史更新的主成分分析(PCA)快速估计子空间,以及按参数块分组进行局部估计的块级策略。实验表明,BSFA在多种任务中显著加速训练,尤其在使用WikiText-103预训练LLaMA-72M及OpenWebText预训练LLaMA-134M时,相较原始AdamW实现约2×加速。

原文摘要 · Abstract (English)

Recent studies \citep{gur2018gradient,song2024does, wen2024understanding} highlight a fundamental dichotomy in deep learning optimization: Although parameter updates along the top eigendirections of the loss Hessian (Dom-space) capture most of the update magnitude, they often contribute minimally to loss reduction. In contrast, updates in the orthogonal component (Bulk-space) have smaller magnitudes but drive most learning progress. In this work, we further advance the understanding of this phenomenon and introduce the \textbf{Bulk-Space-Filtration-Accelerator (BSFA)}, a novel plug-and-play framework. BSFA accelerates training by differentially scaling update components projected onto these distinct subspaces, simultaneously enhancing stability by moderating updates in the dominant subspace and boosting convergence speed by amplifying those in the bulk-space. To ensure BSFA is both practical and scalable for contemporary large models, we introduce two key innovations: an efficient estimator using Principal Component Analysis (PCA) on historical updates for fast subspace estimation, and a block-wise strategy that applies this estimation on a per-parameter-block basis. These designs make BSFA computationally tractable and highly effective. We demonstrate BSFA's acceleration across various tasks, notably achieving approximately 2$\times$ speedup when pre-training LLaMA-72M on WikiText-103 and LLaMA-134M on OpenWebText compared to vanilla AdamW.

训练加速优化器改进子空间分析

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。