PulseExploreJournal ClubDebatesTrendingResearchersJournals
Instagram
HomeExploreJournal ClubTrending
Synapse
⌘+K
Synapse
April 26, 20260 citationsOpen Access

PDGMM-VAE: A Variational Autoencoder with Adaptive Per-Dimension Gaussian Mixture Model Priors for Nonlinear ICA

View Full Paper
YWYuan-Hao WEIYSYan-Jie Sun

Key Points

  • This work introduces PDGMM-VAE, which aims to enhance independent component analysis by using adaptive Gaussian mixture model priors for each latent dimension.
  • Developed a variational autoencoder architecture with individual source-specific Gaussian mixture model priors for latent dimensions.
  • Jointly learned parameters for the encoder, decoder, and adaptive prior during training.
  • Conducted experimental evaluations on linear and nonlinear signal mixing problems.
  • PDGMM-VAE effectively recovers latent source signals and fits non-Gaussian source marginals in varied scenarios.
  • The approach minimizes latent permutation symmetry compared to homogeneous priors, enhancing model performance.
  • KL regularization contributes to source-specific behavior, aiding in the specialization during training.

Abstract

Independent component analysis is a core framework within blind source separation for recovering latent source signals from observed mixtures under statistical independence assumptions. In this work, we propose PDGMM-VAE, a source-oriented variational autoencoder in which each latent dimension, interpreted explicitly as an individual source component, is assigned its own adaptive Gaussian mixture model prior. The proposed framework imposes heterogeneous per-dimension prior constraints, enabling different latent dimensions to model different non-Gaussian source marginals within a unified probabilistic encoder-decoder architecture. The parameters of these source-specific GMM priors are not fixed in advance, but are jointly learned together with the encoder and decoder under the overall training objective. Beyond the model construction itself, we provide a theoretical analysis clarifying why adaptive per-dimension prior design is meaningful in this setting. In particular, we show that heterogeneous per-dimension priors reduce latent permutation symmetry relative to homogeneous shared priors, and we further show that the KL regularization induced by the adaptive GMM prior creates source-specific attraction behavior that helps explain source-wise specialization during training. We also clarify the relation of the proposed model to the standard VAE and provide a weak recovery statement in an idealized linear low-noise regime. Experimental results on both linear and nonlinear mixing problems show that PDGMM-VAE can recover latent source signals and fit source-specific non-Gaussian marginals effectively. These results suggest that adaptive per-dimension mixture-prior design provides a principled and promising direction for VAE-based ICA and source-oriented generative modeling.

Ask AI
Helpful
Bookmark
Share
View Full Paper

Cite This Study

WEI et al. (2026) studied this question.

synapsesocial.com/papers/69edabb84a46254e215b3a01https://doi.org/10.5281/zenodo.19729584
Ask AI
Helpful
Bookmark
Share
View Full Paper