Phase-Type Variational Autoencoders for Heavy-Tailed Data
Abdelhakim Ziani, Andras Horvath, Paolo Ballarini
Abstract
Heavy-tailed distributions are ubiquitous in real-world data, where rare but extreme events dominate risk and variability. However, standard Variational Autoencoders (VAEs) employ simple decoder distributions (e.g., Gaussian) that fail to capture heavy-tailed behavior, while existing heavy-tail-aware extensions remain restricted to predefined parametric families whose tail behavior is fixed a priori. We propose the Phase-Type Variational Autoencoder (PH-VAE), whose decoder distribution is a latent-conditioned Phase-Type (PH) distribution—defined as the absorption time of a continuous-time Markov chain (CTMC). This formulation composes multiple exponential time scales, yielding a flexible, analytically tractable decoder that adapts its finite-range tail behavior directly from the observed data. Experiments on synthetic and real-world benchmarks demonstrate that PH-VAE accurately approximates diverse heavy-tailed distributions, significantly outperforming Gaussian, Student-t, and extreme-value-based VAE decoders in modeling observed tail behavior and extreme quantiles. In multivariate settings, PH-VAE captures realistic cross-dimensional tail dependence through its shared latent representation. To our knowledge, this is the first work to integrate Phase-Type distributions into deep generative modeling, bridging applied probability and representation learning.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on1
Related papers
- -Variational Autoencoder: Learning Heavy-tailed Data with Student's t and Power DivergenceJuno Kim, Jaehyuk Kwon, Mincheol Cho, Hyunjong Lee et al.ICLR 2024 · 11 citations
- A Variational Autoencoder for Neural Temporal Point Processes with Dynamic Latent GraphsSikun Yang, Hongyuan ZhaAAAI 2024 · 7 citations
- VAEM: a Deep Generative Model for Heterogeneous Mixed Type DataChao Ma, Sebastian Tschiatschek, Richard E. Turner, José Miguel Hernández-Lobato et al.NeurIPS 2020 · 105 citations
- Pareto GAN: Extending the Representational Power of GANs to Heavy-Tailed DistributionsTodd Huster, Jeremy E. J. Cohen, Zinan Lin, Kevin Chan et al.ICML 2021 · 37 citations
- Community-Aware Variational Autoencoder for Continuous Dynamic NetworksJunwei Cheng, Chaobo He, Pengxing Feng, Weixiong Liu et al.AAAI 2025 · 4 citations
