arXiv:2506.21408cs.LGcs.AI2025-06中稿 · UAI 2025被引 9

用低维子空间推断实现大模型贝叶斯适配,参数仅增1000个却可扩展至最大规模。

Scalable Bayesian Low-Rank Adaptation of Large Language Models via Stochastic Variational Subspace Inference

  • 在低秩适配参数的低维子空间中进行贝叶斯推断,用投影矩阵映射到完整权重空间。
  • 仅增加约1000个参数,即可达到顶尖方法性能,且支持4倍于以往的大模型规模。
  • 适合需要可信推理的高风险场景,如医疗、自动驾驶中的不确定性量化。

尽管大型语言模型(LLMs)应用广泛,但其常产生错误信息且校准不足,因此在自动驾驶、医疗等高风险领域,量化模型不确定性至关重要。先前工作通过在微调模型的低秩适配(LoRA)参数上执行贝叶斯推断来提升可行性,但这些方法因需额外参数而难以扩展至更大模型。本文提出可扩展的贝叶斯低秩适配方法(ScalaBL),在秩为 $r$ 的 $r$-维子空间中进行贝叶斯推断。通过将LoRA参数重用为投影矩阵,将子空间样本映射至大模型完整权重空间,并采用随机变分推断学习全部参数。尽管子空间维度低,仍实现与先进方法相当的性能,且仅需约1000个额外参数。该方法首次使贝叶斯大模型规模达到此前四倍之多。

原文摘要 · Abstract (English)

Despite their widespread use, large language models (LLMs) are known to hallucinate incorrect information and be poorly calibrated. This makes the uncertainty quantification of these models of critical importance, especially in high-stakes domains, such as autonomy and healthcare. Prior work has made Bayesian deep learning-based approaches to this problem more tractable by performing inference over the low-rank adaptation (LoRA) parameters of a fine-tuned model. While effective, these approaches struggle to scale to larger LLMs due to requiring further additional parameters compared to LoRA. In this work we present $\textbf{Scala}$ble $\textbf{B}$ayesian $\textbf{L}$ow-Rank Adaptation via Stochastic Variational Subspace Inference (ScalaBL). We perform Bayesian inference in an $r$-dimensional subspace, for LoRA rank $r$. By repurposing the LoRA parameters as projection matrices, we are able to map samples from this subspace into the full weight space of the LLM. This allows us to learn all the parameters of our approach using stochastic variational inference. Despite the low dimensionality of our subspace, we are able to achieve competitive performance with state-of-the-art approaches while only requiring ${\sim}1000$ additional parameters. Furthermore, it allows us to scale up to the largest Bayesian LLM to date, with four times as a many base parameters as prior work.

贝叶斯深度学习低秩适配不确定性量化大模型

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