arXiv:2507.01841cs.LGcs.IT2025-07

用二阶信息自动确定LoRA适配的秩,提升模型精度与效率。

Automatic Rank Determination for Low-Rank Adaptation via Submodular Function Maximization

  • 基于子模函数最大化构建秩选择框架,利用海森矩阵捕捉复杂损失面。
  • 在求解偏微分方程任务中,相比现有方法显著提升性能,且保持计算高效。
  • 理论严谨,适合需要高精度微调的科研场景,如物理信息神经网络。

本文提出SubLoRA,一种基于子模函数最大化的低秩适配(LoRA)秩确定方法。与依赖一阶近似的传统方法(如AdaLoRA)不同,SubLoRA引入二阶信息,通过海森矩阵捕捉潜在复杂的损失景观。我们证明当LoRA参数已充分优化时,线性化近似会变得不准确且病态,因此亟需更可靠、细致的二阶建模。为此,我们将秩确定问题重构为具有二次目标的组合优化问题,但该问题通常为NP难。为克服计算挑战,我们引入子模函数最大化框架,并设计具有近似保证的贪心算法。我们推导出目标函数成为子模的充要条件,并构造出满足该条件的闭式海森投影,兼顾理论正确性与计算效率。此外,我们进一步将SubLoRA扩展至联合优化设置,在秩预算约束下交替更新LoRA参数与秩选择。在求解偏微分方程(PDEs)的物理信息神经网络(PINNs)微调任务上,大量实验验证了本方法的有效性:在秩确定与联合训练性能上均优于现有方法。

原文摘要 · Abstract (English)

In this paper, we propose SubLoRA, a rank determination method for Low-Rank Adaptation (LoRA) based on submodular function maximization. In contrast to prior approaches, such as AdaLoRA, that rely on first-order (linearized) approximations of the loss function, SubLoRA utilizes second-order information to capture the potentially complex loss landscape by incorporating the Hessian matrix. We show that the linearization becomes inaccurate and ill-conditioned when the LoRA parameters have been well optimized, motivating the need for a more reliable and nuanced second-order formulation. To this end, we reformulate the rank determination problem as a combinatorial optimization problem with a quadratic objective. However, solving this problem exactly is NP-hard in general. To overcome the computational challenge, we introduce a submodular function maximization framework and devise a greedy algorithm with approximation guarantees. We derive a sufficient and necessary condition under which the rank-determination objective becomes submodular, and construct a closed-form projection of the Hessian matrix that satisfies this condition while maintaining computational efficiency. Our method combines solid theoretical foundations, second-order accuracy, and practical computational efficiency. We further extend SubLoRA to a joint optimization setting, alternating between LoRA parameter updates and rank determination under a rank budget constraint. Extensive experiments on fine-tuning physics-informed neural networks (PINNs) for solving partial differential equations (PDEs) demonstrate the effectiveness of our approach. Results show that SubLoRA outperforms existing methods in both rank determination and joint training performance.

LoRA秩确定二阶优化子模优化

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