Task-guided cross-subject latent alignment: a multi-encoder-decoder VAE
Summary
A Multi-Encoder-Decoder Variational Autoencoder (MED-VAE) is introduced to align neural activity across subjects without requiring shared stimuli, addressing a limitation of traditional methods in naturalistic paradigms. MED-VAE anchors neural representations to a common scaffold provided by a pretrained Artificial Neural Network (ANN). Using the Natural Scenes Dataset, the model demonstrates superior semantic organization in its common latent spaces and achieves higher cross-subject alignment compared to existing techniques. It also maintains robust generalization to held-out stimuli, a scenario where traditional methods degrade. Furthermore, MED-VAE preserves stimulus-driven signal during reconstruction to original neural spaces and directly enables cross-subject neural prediction, exemplified by cross-subject image decoding for visual cortex responses to static images. This framework aims to identify generalizable common subspaces for various cross-subject predictions and downstream tasks.
Key takeaway
For Research Scientists developing neural decoders, if you face limitations due to requiring shared stimuli for cross-subject alignment, consider implementing the MED-VAE framework. This approach allows you to achieve superior semantic organization and robust generalization in common latent spaces without that constraint. You can directly enable cross-subject neural prediction and image decoding, expanding the applicability of your models to more naturalistic paradigms.
Key insights
MED-VAE aligns neural activity across subjects using a pretrained ANN scaffold, eliminating the need for shared stimuli.
Principles
- Cross-subject alignment is possible without shared stimuli.
- Pretrained ANNs can serve as common scaffolds for neural data.
- Generalizable common subspaces enable cross-subject predictions.
Method
MED-VAE uses multiple encoders and decoders, anchoring neural representations to a pretrained ANN's common scaffold to create aligned latent spaces.
In practice
- Apply MED-VAE for cross-subject neural prediction.
- Use MED-VAE to decode images from neural activity.
- Identify generalizable subspaces for downstream tasks.
Topics
- Multi-Encoder-Decoder VAE
- Cross-Subject Alignment
- Neural Activity
- Pretrained ANN Scaffolds
- Neural Prediction
- Image Decoding
Best for: AI Scientist, Machine Learning Engineer, Research Scientist
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.