cs.LGOct 10, 2026

Pro-t-VAE: Closed-Form Student's t Mixtures for Robust Prototype Discovery

Authors: Zhikang Xu, Jiarui Xing, Jian Wang

Organizations: Institute of Intelligent Information Processing, Shanxi University · School of Medicine, Yale University · Boston Children’s Hospital, Harvard Medical School

Abstract

We present a robust deep generative framework for unsupervised image prototype discovery, built on a heavy-tailed Student's t mixture prior, that bridges statistical robustness and mathematical tractability. Existing generative approaches typically rely on noise-prone Gaussian priors or computationally intensive diffusion models; neither simultaneously provides explicit latent structure nor exact analytic optimization. While diffusion models can yield strong priors, iterative sampling and intractable likelihoods limit scalability and interpretability for prototype discovery. We address these issues by constructing a Student's t latent space via a Bayesian scale-mixture formulation. This yields a fully closed-form augmented ELBO for unsupervised Student's t mixture variational autoencoders, which, to our knowledge, is the first such objective applied to robust image prototype discovery, avoiding biased variational approximations and high computational costs in prior heavy-tailed or diffusion-based frameworks. Treating precision as a latent variable automatically modulates each data point's influence, downweighting outliers to learn a data-dependent measure of trust. The result is improved robustness, accuracy, and efficiency, with well-separated latent representations and more stable manifold learning than state-of-the-art methods. We validate our model on standard vision benchmarks and challenging neuroimaging tasks. In brain MRI-driven neurodegenerative disease analysis, our model recovers clean, anatomically coherent prototypes where clinical noise often collapses rare subtypes. These results highlight the ability of our framework to preserve anatomy-associated patterns for precise discovery in clinical settings. Our code is available at: https://github.com/xuzhikangnba/torch-tvae.

Figures & tables

Explore similar work

CardsList
  1. Deep Image Prototype Learning with Geometric Heat-Kernel Priors

    Jun 17, 2026Jiarui Xing, Tal Zeevi, Nian Wu +1Medical Image AnalysisPrototype Learning

  2. Distribution Matching Variational AutoEncoder

    Dec 8, 2025Sen Ye, Jianning Pei, Mengde Xu +4Variational AutoencodersGenerative Modeling

  3. Function-Space Empirical Bayes Regularisation with Student's t Priors

    Feb 25, 2026Pengcheng Hao, Ercan Engin KuruogluBayesian Neural NetworksBayesian Inference