用贝叶斯重参数化提升低秩微调稳定性,仅增少量参数即显著改善大模型微调效果。
Robust and Efficient Fine-tuning of LLMs with Bayesian Reparameterization of Low-Rank Adaptation
- 引入蒙特卡洛估计实现低秩参数的无偏后验推断,降低方差
- 在理解任务上准确率提升0.5%,生成任务方差下降50%-62%
- 适合追求高效稳定微调的大模型应用开发者
大型语言模型微调因规模庞大而资源消耗高。尽管低秩适配是主流的参数高效微调方法,但其对超参数敏感,导致下游任务性能不稳定。本文强调低秩微调中有效参数化对降低估计方差、提升模型输出稳定性的关键作用。提出MonteCLoRA:通过蒙特卡洛估计学习低秩参数的无偏后验分布,以极低期望方差实现稳定微调,仅增加O(r)个额外参数(给定秩r)。在预训练的RoBERTa-base上,自然语言理解任务中准确率和鲁棒性分别提升0.5%和1.6%;在预训练的LLaMA-1-7B与LLaMA-3.2-3B-Instruct生成任务中,性能波动分别比当前高效微调方法降低50%和62%。理论与实证结果表明,参数化与超先验共同平衡低秩参数空间中的探索-利用关系,从而实现更优、更鲁棒的参数估计。
原文摘要 · Abstract (English)
Large Language Models (LLMs) are highly resource-intensive to fine-tune due to their enormous size. While low-rank adaptation is a prominent parameter-efficient fine-tuning approach, it suffers from sensitivity to hyperparameter choices, leading to instability in model performance on fine-tuning downstream tasks. This paper highlights the importance of effective parameterization in low-rank fine-tuning to reduce estimator variance and enhance the stability of final model outputs. We propose MonteCLoRA, an efficient fine-tuning technique that employs Monte Carlo estimation to learn an unbiased posterior estimation of low-rank parameters with low expected variance, stabilizing fine-tuned LLMs with only O(r) additional parameters, for a given rank r. MonteCLoRA shows 0.5% and 1.6% improvements in accuracy and robustness over unregularized low-rank adaptation method on natural language understanding tasks with pre-trained RoBERTa-base. Furthermore, in generative tasks with pre-trained LLaMA-1-7B and LLaMA-3.2-3B-Instruct, MonteCLoRA demonstrates robust performance with 50% and 62% lower spreads respectively than the contemporary efficient fine-tuning methods. The theoretical and empirical results presented in the paper underscore how parameterization and hyperpriors balance exploration-exploitation in the low-rank parametric space, therefore leading to more optimal and robust parameter estimation during efficient fine-tuning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。