arXiv:2508.08625cs.LG2025-08被引 2

动态调整训练秩,让低秩模型也能学好复杂模式。

Dynamic Rank Adjustment for Accurate and Efficient Neural Network Training

  • 在低秩训练中穿插全秩训练,防止权重秩坍缩。
  • 同等计算成本下,精度接近全参数模型。
  • 适合需要高效训练的大规模神经网络任务。

低秩训练通过矩阵分解(如SVD)减少可训练参数,但固定低秩结构会限制模型学习复杂模式的能力。此外,权重的有效秩在训练中会下降,且低秩重参数化会加速这一过程。本文提出一种动态秩训练框架,通过在低秩训练周期中穿插全秩训练周期,有效恢复权重秩。实验表明,该方法在保持与基于SVD的低秩训练相近计算成本的同时,在多个基准测试上实现了与全参数训练相当的精度。

原文摘要 · Abstract (English)

Low-rank training methods reduce the number of trainable parameters by re-parameterizing the weights with matrix decompositions (e.g., singular value decomposition). However, enforcing a fixed low-rank structure caps the rank of the weight matrices and can hinder the model's ability to learn complex patterns. Furthermore, the effective rank of the model's weights tends to decline during training, and this drop is accelerated when the model is reparameterized into a low-rank structure. In this study, we argue that strategically interleaving full-rank training epochs within low-rank training epochs can effectively restore the rank of the model's weights. Based on our findings, we propose a general dynamic-rank training framework that is readily applicable to a wide range of neural-network tasks. We first describe how to adjust the rank of weight matrix to alleviate the inevitable rank collapse that arises during training, and then present extensive empirical results that validate our claims and demonstrate the efficacy of the proposed framework. Our empirical study shows that the proposed method achieves almost the same computational cost as SVD-based low-rank training while achieving a comparable accuracy to full-rank training across various benchmarks.

低秩训练动态调整模型效率

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