arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.17809cs.LG
收藏

校准适应:基于贝叶斯斯蒂夫尔流形先验的可靠参数高效微调

Calibrated Adaptation: Bayesian Stiefel Manifold Priors for Reliable Parameter-Efficient Fine-Tuning

Department of Computer Science, Iowa State University(计算机科学系,爱荷华州立大学);Department of Civil, Construction and Environmental Engineering, Iowa State University(土木、建设与环境工程系,爱荷华州立大学)

Ibne Farabi Shihab, Sanjeda Akter, Anuj Sharma

AI总结:

SBA通过在斯蒂夫尔流形上使用贝叶斯先验,提高参数高效微调的可靠性,减少校准误差并提升领域转移性能。

摘要:

参数高效微调方法如LoRA能够实现大型语言模型的实用适应,但并未提供原理上的不确定性估计,导致预测不准确且在领域转移时行为不可靠。我们引入了斯蒂夫尔-贝叶斯适配器(SBA),一种贝叶斯PEFT框架,它在斯蒂夫尔流形St上对正交适配因子放置矩阵兰格vin先验,并通过切空间拉普拉斯近似和测地线重构进行近似后验推断。与在平坦空间中投影到正交约束上的高斯先验不同,我们的流形先验自然编码了适配子空间应良好条件且正交的归纳偏差,而后验提供校准的预测不确定性而无需重新校准。我们正式证明切空间近似严格避免了从嵌入空间投影时固有的方差膨胀,建立了内在流形推断的严谨理论优势。在GLUE和SuperGLUE基准上,针对RoBERTa-large、LLaMA-2-7B、LLaMA-2-13B、Mistral-7B和Qwen2.5-7B,领域转移评估、选择性预测协议和抽象总结任务中,SBA在任务性能上与LoRA和DoRA相当,同时将期望校准误差降低18至34%超过确定性基线,在领域转移下改进选择性预测AUROC 12至25%,并且在OoD检测上以较少的参数成本优于五个LoRA模型的深度集成。我们的结果表明,在正确的几何结构上放置不确定性比简单地添加任何贝叶斯处理到适配器中更重要。

英文摘要:

Parameter-efficient fine-tuning methods such as LoRA enable practical adaptation of large language models but provide no principled uncertainty estimates, leading to poorly calibrated predictions and unreliable behavior under domain shift. We introduce Stiefel-Bayes Adapters (SBA), a Bayesian PEFT framework that places a Matrix Langevin prior over orthonormal adapter factors on the Stiefel manifold $\St$ and performs approximate posterior inference via tangent space Laplace approximation with geodesic retraction. Unlike Gaussian priors in flat space projected onto orthogonality constraints, our prior on the manifold naturally encodes the inductive bias that adapter subspaces should be well conditioned and orthogonal, while the posterior provides calibrated predictive uncertainty without recalibration. We prove formally that the tangent space approximation strictly avoids the structural variance inflation inherent in projecting from ambient space, establishing a rigorous theoretical advantage for intrinsic manifold inference. Across GLUE and SuperGLUE benchmarks on RoBERTa-large, LLaMA-2-7B, LLaMA-2-13B, Mistral-7B, and Qwen2.5-7B, domain shift evaluations, selective prediction protocols, and an abstractive summarization task, SBA achieves task performance comparable to LoRA and DoRA while reducing Expected Calibration Error by 18 to 34\% over deterministic baselines, improving selective prediction AUROC by 12 to 25\% under domain shift, and outperforming deep ensembles of five LoRA models on OOD detection at a fraction of the parameter cost. Our results demonstrate that where you place uncertainty, on the right geometric structure, matters more than simply adding any Bayesian treatment to adapters.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑