Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

DreamDiffusion: Generating High-Quality Images from Brain EEG Signals

About

This paper introduces DreamDiffusion, a novel method for generating high-quality images directly from brain electroencephalogram (EEG) signals, without the need to translate thoughts into text. DreamDiffusion leverages pre-trained text-to-image models and employs temporal masked signal modeling to pre-train the EEG encoder for effective and robust EEG representations. Additionally, the method further leverages the CLIP image encoder to provide extra supervision to better align EEG, text, and image embeddings with limited EEG-image pairs. Overall, the proposed method overcomes the challenges of using EEG signals for image generation, such as noise, limited information, and individual differences, and achieves promising results. Quantitative and qualitative results demonstrate the effectiveness of the proposed method as a significant step towards portable and low-cost ``thoughts-to-image'', with potential applications in neuroscience and computer vision. The code is available here \url{https://github.com/bbaaii/DreamDiffusion}.

Yunpeng Bai, Xintao Wang, Yan-pei Cao, Yixiao Ge, Chun Yuan, Ying Shan• 2023

Related benchmarks

TaskDatasetResultRank
Visual ReconstructionTHINGS-EEG (all subjects)
Pixel Correlation (PixCorr)0.206
8
Visual concept classificationImageNet-EEG (val)
Accuracy20.4
5
Visual concept classificationImageNet-EEG (test)
Accuracy19.2
5
EEG-Driven Image GenerationEEGCVPR40 Single
Accuracy45
4
Showing 4 of 4 rows

Other info

Follow for update