Bayesian Fine-tuning in Projected Subspaces

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Dubovik, Viktar, Marszałek, Patryk, Tabor, Jacek, Kuśmierczyk, Tomasz
Format: Preprint
Publié: 2026
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866913103478784000
author Dubovik, Viktar
Marszałek, Patryk
Tabor, Jacek
Kuśmierczyk, Tomasz
author_facet Dubovik, Viktar
Marszałek, Patryk
Tabor, Jacek
Kuśmierczyk, Tomasz
contents Low-Rank Adaptation (LoRA) enables parameter-efficient fine-tuning of large models by decomposing weight updates into low-rank matrices, significantly reducing storage and computational overhead. While effective, standard LoRA lacks mechanisms for uncertainty quantification, leading to overconfident and poorly calibrated models. Bayesian variants of LoRA address this limitation, but at the cost of a significantly increased number of trainable parameters, partially offsetting the original efficiency gains. Additionally, these models are harder to train and may suffer from unstable convergence. In this work, we propose a novel framework for parameter-efficient Bayesian fine-tuning, demonstrating that effective uncertainty quantification can be achieved in very low-dimensional parameter spaces. The proposed method achieves strong performance with improved calibration and generalization while maintaining computational efficiency. Our empirical findings show that, with the appropriate projection of the weight space uncertainty can be effectively modeled in a low-dimensional space, and weight covariances exhibit low ranks.
format Preprint
id arxiv_https___arxiv_org_abs_2605_07706
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Bayesian Fine-tuning in Projected Subspaces
Dubovik, Viktar
Marszałek, Patryk
Tabor, Jacek
Kuśmierczyk, Tomasz
Machine Learning
Low-Rank Adaptation (LoRA) enables parameter-efficient fine-tuning of large models by decomposing weight updates into low-rank matrices, significantly reducing storage and computational overhead. While effective, standard LoRA lacks mechanisms for uncertainty quantification, leading to overconfident and poorly calibrated models. Bayesian variants of LoRA address this limitation, but at the cost of a significantly increased number of trainable parameters, partially offsetting the original efficiency gains. Additionally, these models are harder to train and may suffer from unstable convergence. In this work, we propose a novel framework for parameter-efficient Bayesian fine-tuning, demonstrating that effective uncertainty quantification can be achieved in very low-dimensional parameter spaces. The proposed method achieves strong performance with improved calibration and generalization while maintaining computational efficiency. Our empirical findings show that, with the appropriate projection of the weight space uncertainty can be effectively modeled in a low-dimensional space, and weight covariances exhibit low ranks.
title Bayesian Fine-tuning in Projected Subspaces
topic Machine Learning
url https://arxiv.org/abs/2605.07706