Certification Frontiers for Gaussian LoRA: Independent Priors, Posterior Risk, and Prediction-Preserving Balancing
cs.LG
Submitted: 2026-09-26
Updated: 2026-09-26
Comments: 32 pages, 9 figures
License: http://creativecommons.org/licenses/by-nc-nd/4.0/
The gist: Post-hoc Bayesian fine-tuning places Gaussians around trained low-rank adapters, yet a calibrated posterior does not by itself yield a useful generalization certificate.
Terminology
Abstract
Post-hoc Bayesian fine-tuning places Gaussians around trained low-rank adapters, yet a calibrated posterior does not by itself yield a useful generalization certificate. Such a posterior admits an informative PAC-Bayes certificate only when both the loss of its sampled predictors and its KL divergence from an admissible prior are small. In this research, we characterize this certification frontier for Gaussian LoRA posteriors and separate three interventions: changing the prior, changing the stochastic predictor, and changing only how its complexity is counted. First, an exact isotropic KL envelope eliminates the prior scale and yields a width threshold that excludes posterior widths before sampling, while a zero-KL floor identifies targets that no complexity reduction can reach at a measured risk bound. Second, we minimize KL in closed form over the full GL(r) symmetry of the low-rank factors, leaving every sampled adapter product unchanged, and derive the noncentral objective required when the prior center is trained on an independent split. On a small-pool RoBERTa audit of 567 configurations, the recorded 64-draw summaries imply a certificate floor of 0.7298 even with zero KL and Chernoff accounting, so reducing complexity alone cannot certify these posteriors at the recorded budgets. For a stable posterior in a controlled Gaussian-factor task, Chernoff accounting certifies risk below 0.1 on 20 of 20 datasets with 1024 draws, whereas Hoeffding certifies none. On deliberately deformed synthetic rank-four factors, matrix balancing reduces KL by 29.3% on average beyond scalar balancing without changing any prediction. Numerical split-prior scenarios make the remaining risk and complexity budgets explicit.
Sources
- Fisher-Geometric Sharpness and the Implicit Bias of SGD toward Flat Minima
- PAC--Bayes Bounds on Quotient Parameter Spaces: Geometry-induced Implicit-Bias Priors
- Tight Sample Complexity for Low-Rank Adaptation: Matching Bounds and Rank Selection
- Improving LoRA with Variational Learning
- BaLoRA: Bayesian Low-Rank Adaptation of Large Scale Models
- Some theoretical improvements on the tightness of PAC-Bayes risk certificates for neural networks
- PAC-Bayes Beyond Parameter Space: Behavioral Equivalence, Z-Information, and Exact Complexity Decomposition
- Sharp Generalization Bounds for Foundation Models with Asymmetric Randomized Low-Rank Adapters
- Tuning without Peeking: Provable Generalization Bounds and Robust LLM Post-Training
- Smoothness-Based Derandomization of PAC-Bayes Bounds
- FragileFlow: Spectral Control of Correct-but-Fragile Predictions for Foundation Model Robustness
- Bayesian-LoRA: Probabilistic Low-Rank Adaptation of Large Language Models
- RoBERTa: A Robustly Optimized BERT Pretraining Approach
- Minimal Ranks, Maximum Confidence: Parameter-efficient Uncertainty Quantification for LoRA
- A Note on the PAC Bayesian Theorem
- Gaussian Stochastic Weight Averaging for Bayesian Low-Rank Adaptation of Large Language Models
- How good is PAC-Bayes at explaining generalisation?
- Non-Vacuous Generalization Bounds: Can Rescaling Invariances Help?
- Convergent Stochastic Training of Multi-Headed Attention and Understanding LoRA
- Non-vacuous Generalization Bounds for Deep Neural Networks without any modification to the trained models
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks