arXiv Statistics ML

Prior Predictive Matching for Bayesian LoRA

arXiv Machine Learning
Jun 30

BaRA: Bayesian Adaptive Rank Allocation for Parameter-Efficient Fine-Tuning

arXiv:2606. 29184v1 Announce Type: new Abstract: While Low-rank adaptation (LoRA) enables highly efficient fine-tuning by constraining task-specific updates to fixed low-rank subspaces, this rigid design limits representational flexibility and often results in overconfident predictions and miscalibrated uncertainty, especially in low-data regimes.

By Zhibin Duan, Yuhong Wang, Jiahong Fu, Zongsheng Yue, Bo Chen, Zongben Xu
arXiv AI
Jul 28

Bayesian-LoRA: Probabilistic Low-Rank Adaptation of Large Language Models

arXiv:2601. 21003v3 Announce Type: replace Abstract: Large Language Models usually put more emphasis on accuracy and therefore, will guess even when not certain about the prediction, which is especially severe when fine-tuned on small datasets due to the inherent tendency toward miscalibration.

By Moule Lin, Shuhao Guan, Andrea Patane, David Gregg, Goetz Botterweck
arXiv AI
Sep 3

TaRA: Training-Aware Low-Rank Adaptation Initialization

TaRA: Training-Aware Low-Rank Adaptation Initialization proposes a new way to initialize LoRA by aligning the gradients of low‑rank factors with those of the full‑rank weight matrix. This approach directly incorporates training dynamics, improving gradient fidelity at the start of fine‑tuning while adding negligible computational cost. Experiments on a variety of challenging fine‑tuning tasks show that TaRA consistently outperforms existing state‑of‑the‑art initialization methods, offering a simple, robust, and scalable solution for effective LoRA initialization.

By Taehyeon Kim, Eunhyeok Park
arXiv Statistics ML
5d ago

LLM knowledge for improved Out-of-Distribution Tabular Prediction - A Cross-Validated Regularisation Approach

The study investigates whether coefficient priors derived from large language models (LLMs) can enhance out‑of‑distribution (OOD) tabular prediction by serving as a shrinkage target in regularised regression. Across five datasets, LLM‑elicited priors improved OOD mean squared error on four datasets by up to 7.9% and yielded better probabilistic accuracy in classification tasks, though the gains varied and were sometimes comparable to in‑domain improvements. Cross‑validation was used to select priors, but the variability among candidates meant that the method mainly avoided the worst priors rather than consistently finding the best. "whyItMatters":"The findings suggest that LLM‑derived priors can provide a scalable alternative to expert elicitation for improving model robustness to distribution shifts, though their effectiveness depends on dataset size and prior variability."

By S{\o}lve Langseth, Kristoffer Herland Hellton, Hugo L. Hammer