Poisson Variational Autoencoder. 2024

Hadi Vafaii, and Dekel Galor, and Jacob L Yates
UC Berkeley.

Variational autoencoders (VAEs) employ Bayesian inference to interpret sensory inputs, mirroring processes that occur in primate vision across both ventral [1] and dorsal [2] pathways. Despite their success, traditional VAEs rely on continuous latent variables, which deviates sharply from the discrete nature of biological neurons. Here, we developed the Poisson VAE ( -VAE), a novel architecture that combines principles of predictive coding with a VAE that encodes inputs into discrete spike counts. Combining Poisson-distributed latent variables with predictive coding introduces a metabolic cost term in the model loss function, suggesting a relationship with sparse coding which we verify empirically. Additionally, we analyze the geometry of learned representations, contrasting the -VAE to alternative VAE models. We find that the -VAE encodes its inputs in relatively higher dimensions, facilitating linear separability of categories in a downstream classification task with a much better (5×) sample efficiency. Our work provides an interpretable computational framework to study brain-like sensory processing and paves the way for a deeper understanding of perception as an inferential process.

UI MeSH Term Description Entries

Related Publications

Hadi Vafaii, and Dekel Galor, and Jacob L Yates
February 2025, IEEE transactions on neural networks and learning systems,
Hadi Vafaii, and Dekel Galor, and Jacob L Yates
January 2020, IEEE transactions on neural networks and learning systems,
Hadi Vafaii, and Dekel Galor, and Jacob L Yates
March 2021, Accident; analysis and prevention,
Hadi Vafaii, and Dekel Galor, and Jacob L Yates
February 2024, Cell reports methods,
Hadi Vafaii, and Dekel Galor, and Jacob L Yates
January 2025, Proceedings of the ... AAAI Conference on Artificial Intelligence. AAAI Conference on Artificial Intelligence,
Hadi Vafaii, and Dekel Galor, and Jacob L Yates
August 2020, ACS omega,
Hadi Vafaii, and Dekel Galor, and Jacob L Yates
May 2022, IEEE transactions on cybernetics,
Hadi Vafaii, and Dekel Galor, and Jacob L Yates
August 2022, IEEE transactions on pattern analysis and machine intelligence,
Hadi Vafaii, and Dekel Galor, and Jacob L Yates
February 2022, Knowledge-based systems,
Hadi Vafaii, and Dekel Galor, and Jacob L Yates
January 2023, IEEE transactions on image processing : a publication of the IEEE Signal Processing Society,
Copied contents to your clipboard!