English

Heavy-Tailed Class-Conditional Priors for Long-Tailed Generative Modeling

Machine Learning 2026-04-16 v2 Artificial Intelligence Computer Vision and Pattern Recognition Machine Learning

Abstract

Variational Autoencoders (VAEs) with global priors trained under an imbalanced empirical class distribution can lead to underrepresentation of tail classes in the latent space. While t3t^3VAE improves robustness via heavy-tailed Student's tt-distribution priors, its single global prior still allocates mass proportionally to class frequency. We address this latent geometric bias by introducing C-t3t^3VAE, which assigns a per-class Student's tt joint prior over latent and output variables. This design promotes uniform prior mass across class-conditioned components. To optimize our model we derive a closed-form objective from the γ\gamma-power divergence, and we introduce an equal-weight latent mixture for class-balanced generation. On SVHN-LT, CIFAR100-LT, and CelebA datasets, C-t3t^3VAE consistently attains lower FID scores than t3t^3VAE and Gaussian-based VAE baselines under severe class imbalance while remaining competitive in balanced or mildly imbalanced settings. In per-class F1 evaluations, our model outperforms the conditional Gaussian VAE across highly imbalanced settings. Moreover, we identify the mild imbalance threshold ρ<5\rho < 5, for which Gaussian-based models remain competitive. However, for ρ5\rho \geq 5 our approach yields improved class-balanced generation and mode coverage.

Keywords

Cite

@article{arxiv.2509.02154,
  title  = {Heavy-Tailed Class-Conditional Priors for Long-Tailed Generative Modeling},
  author = {Aymene Mohammed Bouayed and Samuel Deslauriers-Gauthier and Adrian Iaccovelli and David Naccache},
  journal= {arXiv preprint arXiv:2509.02154},
  year   = {2026}
}
R2 v1 2026-07-01T05:17:02.801Z