梯度压缩与适配器重参数化存在对偶关系,可统一高效训练方法。
On the Duality between Gradient Transformations and Adapters
- 用线性映射将梯度降维,降低内存占用
- 降维优化等价于只训练适配器参数
- 揭示了GaLore与LoRA的内在联系
我们研究了通过线性梯度变换实现神经网络(特别是语言模型)的内存高效优化,其中梯度被线性映射到低于完整参数空间的低维空间,从而节省梯度累积和优化器状态存储所需的内存。模型参数通过先在低维空间执行优化步骤,再通过线性映射的转置返回原始参数空间来更新。我们证明,在此变换空间中优化模型等价于通过线性适配器对原始模型进行加性重参数化,并仅优化适配器参数。当变换为Kronecker分解形式时,该方法建立了GaLore与单边LoRA之间的等价性。这一梯度变换与适配器重参数化的对偶关系,统一了现有的内存高效训练方法,并为提升训练效率和内存使用提供了新思路。
原文摘要 · Abstract (English)
We study memory-efficient optimization of neural networks (in particular language models) with linear gradient transformations, where the gradients are linearly mapped to a lower dimensional space than the full parameter space, thus saving memory required for gradient accumulation and optimizer state persistence. The model parameters are updated by first performing an optimization step in the lower dimensional space and then going back into the original parameter space via the linear map's transpose. We show that optimizing the model in this transformed space is equivalent to reparameterizing the original model through a linear adapter that additively modifies the model parameters, and then only optimizing the adapter's parameters. When the transformation is Kronecker-factored, this establishes an equivalence between GaLore and one-sided LoRA. We show that this duality between gradient transformations and adapter-based reparameterizations unifies existing approaches to memory-efficient training and suggests new techniques for improving training efficiency and memory use.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。