Improving visual image reconstruction from human brain activity using latent diffusion models via multiple decoded inputs

Yu Takagi,Shinji Nishimoto
2023-06-20
Abstract:The integration of deep learning and neuroscience has been advancing rapidly, which has led to improvements in the analysis of brain activity and the understanding of deep learning models from a neuroscientific perspective. The reconstruction of visual experience from human brain activity is an area that has particularly benefited: the use of deep learning models trained on large amounts of natural images has greatly improved its quality, and approaches that combine the diverse information contained in visual experiences have proliferated rapidly in recent years. In this technical paper, by taking advantage of the simple and generic framework that we proposed (Takagi and Nishimoto, CVPR 2023), we examine the extent to which various additional decoding techniques affect the performance of visual experience reconstruction. Specifically, we combined our earlier work with the following three techniques: using decoded text from brain activity, nonlinear optimization for structural image reconstruction, and using decoded depth information from brain activity. We confirmed that these techniques contributed to improving accuracy over the baseline. We also discuss what researchers should consider when performing visual reconstruction using deep generative models trained on large datasets. Please check our webpage at <a class="link-external link-https" href="https://sites.google.com/view/stablediffusion-with-brain/" rel="external noopener nofollow">this https URL</a>. Code is also available at <a class="link-external link-https" href="https://github.com/yu-takagi/StableDiffusionReconstruction" rel="external noopener nofollow">this https URL</a>.
Neurons and Cognition,Artificial Intelligence,Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The paper primarily explores how to improve brain activity-based visual image reconstruction methods by combining multiple decoding techniques. The authors utilized a previously proposed simple and general framework and added three additional techniques: 1. **Using decoded captions from brain activity**: The authors employed an image captioning model (BLIP) to predict textual descriptions of images from brain activity, and then input these descriptions into the Stable Diffusion model to generate images. 2. **Combining nonlinear optimization with Generative Adversarial Networks (GANs)**: This method first decodes information from brain activity using a GAN algorithm, and then inputs the decoded results into an image encoder module to generate images. 3. **Utilizing depth information decoded from brain activity**: To further enrich the information sources for image reconstruction, the researchers also attempted to estimate depth information separately and used it along with semantic information for conditional image generation. By integrating these techniques, the research team evaluated their impact on the performance of visual experience reconstruction. The results showed that these techniques generally improved accuracy over the baseline level, but the performance varied among different subjects, and not all metrics showed consistent improvement. Additionally, the authors conducted control analyses to investigate whether there was any overlap between the dataset used to train the Stable Diffusion model and the images in the functional magnetic resonance imaging (fMRI) data, and whether this potential "data leakage" would affect reconstruction performance. The experimental results indicated that even if there was some data overlap, it did not significantly affect the validity of the research conclusions. In summary, this paper aims to explore how to improve the quality of brain activity-based visual image reconstruction by combining multiple decoding techniques and provides quantitative evaluations to support its findings.