用黎曼几何优化低秩适配器,解决传统方法的参数歧义问题。
LoRA meets Riemannion: Muon Optimizer for Parametrization-independent Low-Rank Adapters
- 在固定秩流形上直接优化低秩适配器,避免参数依赖性
- 相比标准LoRA,收敛更快,任务性能提升明显
- 适合需要高效微调的大模型研究者
本文提出一种全新的全黎曼框架用于低秩适配(LoRA),将低秩适配器的优化过程几何化地置于固定秩流形上,从而消除标准欧氏优化器中存在的参数歧义。该框架包含三个关键组件:(1) 推导出一种新的黎曼优化器 Riemannion,它是近期提出的 Muon 优化器在固定秩矩阵流形上的推广;(2) 设计了一种基于黎曼梯度的 LoRA 初始化方法;(3) 实现了高效无显著开销的算法,利用自动微分计算几何操作,并遵循数值线性代数的最佳实践。在大语言模型和扩散模型架构上的全面实验表明,该方法在收敛速度和最终任务性能上均显著优于标准 LoRA 及其最先进的改进版本。
原文摘要 · Abstract (English)
This work presents a novel, fully Riemannian framework for Low-Rank Adaptation (LoRA) that geometrically treats low-rank adapters by optimizing them directly on the fixed-rank manifold. This formulation eliminates the parametrization ambiguity present in standard Euclidean optimizers. Our framework integrates three key components to achieve this: (1) we derive Riemannion, a new Riemannian optimizer on the fixed-rank matrix manifold that generalizes the recently proposed Muon optimizer; (2) we develop a Riemannian gradient-informed LoRA initialization, and (3) we provide an efficient implementation without prominent overhead that uses automatic differentiation to compute arising geometric operations while adhering to best practices in numerical linear algebra. Comprehensive experimental results on both LLM and diffusion model architectures demonstrate that our approach yields consistent and noticeable improvements in convergence speed and final task performance over both standard LoRA and its state-of-the-art modifications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。