arXiv:2604.03388cs.LGstat.ML2026-04被引 1

提升大模型不确定性量化能力,解决微调后过度自信问题。

Scalable Variational Bayesian Fine-Tuning of LLMs via Orthogonalized Low-Rank Adapters

  • 用正交化低秩适配器增强表达力,稳定微调过程。
  • 结合变分推断实现高效贝叶斯微调,推理开销可控。
  • 适合对可靠性要求高的安全关键场景应用。

在将大语言模型(LLM)应用于安全关键任务时,不确定性量化(UQ)对于评估决策可靠性至关重要。然而,下游领域微调后,尤其在数据有限的情况下,模型常出现过度自信问题。现有方法或依赖拉普拉斯近似后的后处理框架,校准效果受训练轨迹影响;或采用变分贝叶斯训练,需多次完整前向传播以进行蒙特卡洛估计,部署时面临可扩展性挑战。为此,本文基于贝叶斯最后一层(BLL)模型,将确定性特征提取器与随机最后一层参数结合以实现不确定性建模。针对现有低秩适配器(LoRA)因秩坍缩导致表达力不足的问题,提出极坐标分解的低秩适配器表示(PoLAR),结合黎曼优化实现更稳定、更丰富的参数化。在此基础上,采用变分推断框架,通过交替优化联合学习PoLAR参数与最后一层参数的近似后验,构建可扩展的贝叶斯微调方法(PoLAR-VBLL)。该框架融合架构增强优化与高效贝叶斯推断,显著提升模型泛化与不确定性估计能力,在多种常识推理任务的分布内和分布外数据上均表现优异。

原文摘要 · Abstract (English)

When deploying large language models (LLMs) to safety-critical applications, uncertainty quantification (UQ) is of utmost importance to self-assess the reliability of the LLM-based decisions. However, such decisions typically suffer from overconfidence, particularly after parameter-efficient fine-tuning (PEFT) for downstream domain-specific tasks with limited data. Existing methods to alleviate this issue either rely on Laplace approximation based post-hoc framework, which may yield suboptimal calibration depending on the training trajectory, or variational Bayesian training that requires multiple complete forward passes through the entire LLM backbone at inference time for Monte Carlo estimation, posing scalability challenges for deployment. To address these limitations, we build on the Bayesian last layer (BLL) model, where the LLM-based deterministic feature extractor is followed by random last layer parameters for uncertainty reasoning. Since existing low-rank adapters (LoRA) for PEFT have limited expressiveness due to rank collapse, we address this with Polar-decomposed Low-rank Adapter Representation (PoLAR), an orthogonalized parameterization paired with Riemannian optimization to enable more stable and expressive adaptation. Building on this PoLAR-BLL model, we leverage the variational (V) inference framework to put forth a scalable Bayesian fine-tuning approach which jointly seeks the PoLAR parameters and approximate posterior of the last layer parameters via alternating optimization. The resulting PoLAR-VBLL is a flexible framework that nicely integrates architecture-enhanced optimization with scalable Bayesian inference to endow LLMs with well-calibrated UQ. Our empirical results verify the effectiveness of PoLAR-VBLL in terms of generalization and uncertainty estimation on both in-distribution and out-of-distribution data for various common-sense reasoning tasks.

大模型不确定性贝叶斯微调

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