mlsanityData to 7 Oct 2026

Probabilistic Geometric Alignment via Bayesian Latent Transport for Domain-Adaptive Foundation Models

arXiv 2603.23783 ↗First posted 24 Mar 2026cs.LG cs.AI math.OCcited by 0

Not on the map. Why?

Abstract (arXiv)

Adapting large-scale foundation models to new domains with limited supervision remains a fundamental challenge due to latent distribution mismatch, unstable optimization dynamics, and miscalibrated uncertainty propagation. This paper introduces an uncertainty-aware probabilistic latent transport framework that formulates domain adaptation as a stochastic geometric alignment problem in representation space. A Bayesian transport operator is proposed to redistribute latent probability mass along Wasserstein-type geodesic trajectories, while a PAC-Bayesian regularization mechanism constrains posterior model complexity to mitigate catastrophic overfitting. The proposed formulation yields theoretical guarantees on convergence stability, loss landscape smoothness, and sample efficiency under distributional shift. Empirical analyses demonstrate substantial reduction in latent manifold discrepancy, accelerated transport energy decay, and improved covariance calibration compared with deterministic fine-tuning and adversarial domain adaptation baselines. Furthermore, bounded posterior uncertainty evolution indicates enhanced probabilistic reliability during cross-domain transfer. By establishing a principled connection between stochastic optimal transport geometry and statistical generalization theory, the proposed framework provides new insights into robust adaptation of modern foundation architectures operating in heterogeneous environments. These findings suggest that uncertainty-aware probabilistic alignment constitutes a promising paradigm for reliable transfer learning in next-generation deep representation systems.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 6 of its 7 references that are in the map. First posted on arXiv 24 Mar 2026.

Builds on

6 of 7 references.

References · 6

  1. 2023Domain-Indexing Variational Bayes: Interpretable Domain Index for Domain Adaptationcited by 19
  2. 2017PAC-Bayes and Domain Adaptationcited by 16
  3. 2022PAC-Bayesian Domain Adaptation Bounds for Multiclass Learnerscited by 13
  4. 2021A Variational Bayesian Approach to Learning Latent Variables for Acoustic Knowledge Transfercited by 2
  5. 2023Bayesian Domain Invariant Learning via Posterior Generalization of Parameter Distributionscited by 1
  6. 2024PAC-Bayesian Domain Adaptation Bounds for Multi-view learningcited by 1

Built on by

Not cited yet.