atlas

Research map / Image generation / Autoregressive and latent generation

MindEye2: Shared-Subject Models Enable fMRI-To-Image With 1 Hour of Data

ICML 2024 · Image generation › Autoregressive and latent generation · approach: Retrieval and matching with generative features · 5 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

MindEye2 pretrains a shared-subject fMRI-to-image model across 7 NSD subjects, then fine-tunes on 1 hour of a new subject's data, mapping fMRI to OpenCLIP ViT-bigG/14 and reconstructing images with a fine-tuned SDXL unCLIP model.

Problem

Brain-to-image reconstruction models are per-subject and need dozens of hours of fMRI data per person, limiting practical use.

Approach

A shared-subject functional alignment (subject-specific linear ridge layers into a shared 4096-dim latent, then a shared MLP backbone and diffusion prior) pretrained across subjects and fine-tuned on a held-out subject, with the CLIP-to-pixel decoder being SDXL fine-tuned to accept CLIP image tokens instead of text, plus retrieval and low-level submodules.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Replace the initial MLP with dropout by a linear voxel-to-latent mapping with L2 regularization (ablation)MindEye2 · NSD test set, subject 1, 10 sessionsPixCorr: 0.225 -> 0.292Helped
Map fMRI to OpenCLIP ViT-bigG/14 instead of CLIP ViT-L/14 (ablation)MindEye2 · NSD test set, subject 1, 10 sessionsPixCorr: 0.243 -> 0.292Helped
Add the low-level submodule to the prior-only model (ablation)MindEye2 · NSD test set, reduced-capacity ablationPixCorr: 0.155 -> 0.281Helped
Add the retrieval submodule to the prior-only model (ablation)MindEye2 · NSD test set, reduced-capacity ablationPixCorr: 0.155 -> 0.233Helped
Add the prior and low-level submodules to the retrieval-only model (ablation)MindEye2 · NSD test set, reduced-capacity ablationFwd retrieval: 96.5% -> 98.0%Helped
Fine-tune after multi-subject pretraining using 1 hour of target-subject dataMindEye2 · NSD test setImage retrievalHelped
Use predicted brain captions rather than empty text prompts for refinement (ablation)MindEye2 · NSD test set, subject 1, 1 sessionInception two-way identification: 82.67% -> 83.56%Helped
Use predicted brain captions rather than empty text prompts for refinement (ablation)MindEye2 · NSD test set, subject 1, 1 sessionCLIP two-way identification: 81.21% -> 80.75%Hurt
Reduce pretraining from the full available subject set to subjects 2, 5, and 7 (ablation)MindEye2 fine-tuned on subject 1 with 1 hour · NSD test set, subject 1PixCorr: 0.235 -> 0.234No clear effect
Skip base SDXL refinement (ablation)MindEye2 · NSD test setImage reconstruction qualityMixed
Replace weighted averaging with VAE-embedding conditioning (ablation)MindEye2 · NSD test setLow-level image metricsHurt

Surprising findings

Limitations

Experimental setup

Datasets: Natural Scenes Dataset COCO 2017 LAION-5B

Benchmarks: Natural Scenes Dataset shared1000 test set COCO 2017 (FID for unCLIP)

Metrics: PixCorr SSIM AlexNet two-way identification Inception CLIP two-way identification EfficientNet-B1 distance SwAV distance top-1 retrieval accuracy brain correlation caption metrics (METEOR, ROUGE, CLIP, Sentence)

Baselines: MindEye1 Brain Diffuser Takagi et al. UniBrain Versatile Diffusion

Builds on: MindEye1 Stable Diffusion XL OpenCLIP ViT-bigG/14 DALL-E 2 diffusion prior

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy