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

post by Bogdan Ionut Cirstea (bogdan-ionut-cirstea) · 2024-03-10T01:30:46.477Z · LW · GW · 0 comments

This is a link post for https://openreview.net/forum?id=paqqd1O0Dl

Contents

  Abstract:
None
No comments

TL;DR: We show that accurate reconstructions of perception from fMRI brain activity are now possible from a single visit to the MRI facility using our diffusion-based approach that pretrains across other subjects' data using a novel alignment procedure.

Abstract:

Reconstructions of visual perception from brain activity have improved tremendously, but the practical utility of such methods has been limited. This is because such models are trained independently per subject where each subject requires dozens of hours of expensive fMRI training data to attain high-quality results. The present work showcases high-quality reconstructions using only 1 hour of fMRI training data. We pretrain our model across 7 subjects and then fine-tune on minimal data from a new subject. Our novel functional alignment procedure linearly maps all brain data to a shared-subject latent space, followed by a shared non-linear mapping to CLIP image space. We then map from CLIP space to pixel space by fine-tuning Stable Diffusion XL to accept CLIP latents as inputs instead of text. This approach improves out-of-subject generalization with limited training data and also attains state-of-the-art image retrieval and reconstruction metrics compared to single-subject approaches. MindEye2 demonstrates how accurate reconstructions of perception are possible from a single visit to the MRI facility.

0 comments

Comments sorted by top scores.