arXiv:2609.08354cs.LG2026-09

在史蒂费尔流形上用随机变分梯度优化适配器,提升参数高效微调的校准性。

Geometry-Aware Bayesian Parameter-Efficient Fine-Tuning on the Stiefel Manifold via Stein Variational Gradient Descent

  • 基于奇异值分解构建适配器,在史蒂费尔流形上进行几何感知优化
  • 相比欧式空间方法,预测置信度更准确,准确率更高
  • 适合需要不确定性估计的高可靠性任务场景

针对大模型参数高效微调中的低秩适配器,现有几何感知方法通过正交约束利用低秩流形结构,提升子空间利用率并减少冗余。然而这类方法的预测可能过度自信。本文基于适配器的奇异值分解,提出一种在史蒂费尔流形上执行的斯坦因变分梯度下降(SVGD)框架,使低秩矩阵沿流形迁移以匹配目标分布,同时保持几何结构。该方法在推理时提供多个解,支持不确定性量化,在多个数据集上的实验表明,其校准性能优于传统欧式空间的SVGD及同类不确定性估计方法,且精度更高。

原文摘要 · Abstract (English)

Several geometry-aware approaches to low-rank adaptation have emerged for parameter-efficient fine-tuning of large pre-trained models. These methods aim to take full advantage of the geometric structure of low-rank manifolds for improving the efficiency in subspace utilization and reducing redundancy by enforcing orthogonality constraints during optimization. The strong empirical results of these techniques have motivated further study into whether predictions from such geometry-based adaptation methods could be overconfident. In this paper, we build on the singular value decomposition factorization of adapters to develop a framework based on Stein variational gradient descent (SVGD). In this formulation, the low-rank matrices are transported along the Stiefel manifold to match the targeted distributions while retaining their crucial geometric structure. Since this geometry-aware SVGD approach provides multiple solutions during inference, it supports uncertainty quantification and produces better-calibrated adapters on the Stiefel manifold. Extensive experiments show that our method delivers strong model calibration and attains higher prediction accuracy than SVGD and related uncertainty estimation methods that are formulated in Euclidean space.

参数高效贝叶斯方法几何优化不确定性

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