# MIG-Vis: Uncovering Semantic Selectivity of Latent Groups in Higher Visual Cortex with Mutual Information-Guided Diffusion

This repository contains the code and dataset for probing group-wise neural latents using a structured VAE and Mutual Information (MI)-guided diffusion models.

## Project Structure

- `neural_vae/`: Neural VAE with group-wise disentangled penalty
- `ddim_models/`: DDIM and Classifier-guided diffusion implementation
- `diffusion_train_scripts/`: Training scripts for diffusion models
- `diffusion_sample_scripts/`: Sampling and probing scripts (e.g., w/ MI guidance)
- `datasets/`: Processed image and neural datasets used in the experiments

## Note

Also included as a ZIP file in the supplementary materials for reproducibility.
