arXiv:2510.01938cs.LG2025-10NeurIPS被引 12

通过流形约束提升低秩微调性能,实现更优的子空间学习。

StelLA: Subspace Learning in Low-rank Adaptation using Stiefel Manifold

  • 采用三因子分解并约束子空间在Stiefel流形上保持正交
  • 在多个任务中超越现有LoRA方法,最高提升12.3%准确率
  • 兼容现有训练流程,适合需要高效微调的研究者

低秩适应(LoRA)是微调大规模预训练模型的常用参数高效技术,但其性能仍逊于全量微调,部分原因在于未能充分挖掘低秩流形的几何结构。本文提出一种几何感知的LoRA扩展方法,采用三因子分解 $U ext{SV}^ op$,将适配器的输入与输出子空间 $V$、$U$ 与缩放因子 $S$ 分离。通过将 $U$ 与 $V$ 约束在Stiefel流形上,确保训练过程中始终保持正交性。为在流形上优化,设计了一种灵活模块化的几何优化方案,可将任意欧氏优化器转换为黎曼优化器,实现高效子空间学习且兼容现有微调流程。在常识推理、数学与代码生成、图像分类及图像生成等多类下游任务上的实验表明,该方法显著优于近期最先进的LoRA变体。代码已开源:https://github.com/SonyResearch/stella。

原文摘要 · Abstract (English)

Low-rank adaptation (LoRA) has been widely adopted as a parameter-efficient technique for fine-tuning large-scale pre-trained models. However, it still lags behind full fine-tuning in performance, partly due to its insufficient exploitation of the geometric structure underlying low-rank manifolds. In this paper, we propose a geometry-aware extension of LoRA that uses a three-factor decomposition $U\!SV^\top$. Analogous to the structure of singular value decomposition (SVD), it separates the adapter's input and output subspaces, $V$ and $U$, from the scaling factor $S$. Our method constrains $U$ and $V$ to lie on the Stiefel manifold, ensuring their orthonormality throughout the training. To optimize on the Stiefel manifold, we employ a flexible and modular geometric optimization design that converts any Euclidean optimizer to a Riemannian one. It enables efficient subspace learning while remaining compatible with existing fine-tuning pipelines. Empirical results across a wide range of downstream tasks, including commonsense reasoning, math and code generation, image classification, and image generation, demonstrate the superior performance of our approach against the recent state-of-the-art variants of LoRA. Code is available at https://github.com/SonyResearch/stella.

低秩微调流形优化子空间学习

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