EEG-based brain-computer interfaces (BCIs) have shown promise in various applications, such as motor imagery and cognitive state monitoring. However, decoding visual representations from EEG signals remains a significant challenge due to their complex and noisy nature. We thus propose a novel 5-stage framework for decoding visual representations from EEG signals: (1) an EEG encoder for concept classification, (2) cross-modal alignment of EEG and text embeddings in CLIP feature space, (3) caption refinement via re-ranking, (4) weighted interpolation of concept and caption embeddings for richer semantics, and (5) image generation using a pre-trained Stable Diffusion model. We enable context-aware EEG-to-image generation through cross-modal alignment and re-ranking. Experimental results demonstrate that our method generates high-quality images aligned with visual stimuli, outperforming SOTA approaches by \(27.08\%\) in Classification Accuracy, \(15.21\%\) in Generation Accuracy and reducing Fréchet Inception Distance by \(36.61\%\) , indicating superior semantic alignment and image quality. The code is available at https://github.com/CVLABLUMS/CATVis .

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

CATVis: Context-Aware Thought Visualization

  • Tariq Mehmood,
  • Hamza Ahmad,
  • Muhammad Haroon Shakeel,
  • Murtaza Taj

摘要

EEG-based brain-computer interfaces (BCIs) have shown promise in various applications, such as motor imagery and cognitive state monitoring. However, decoding visual representations from EEG signals remains a significant challenge due to their complex and noisy nature. We thus propose a novel 5-stage framework for decoding visual representations from EEG signals: (1) an EEG encoder for concept classification, (2) cross-modal alignment of EEG and text embeddings in CLIP feature space, (3) caption refinement via re-ranking, (4) weighted interpolation of concept and caption embeddings for richer semantics, and (5) image generation using a pre-trained Stable Diffusion model. We enable context-aware EEG-to-image generation through cross-modal alignment and re-ranking. Experimental results demonstrate that our method generates high-quality images aligned with visual stimuli, outperforming SOTA approaches by \(27.08\%\) in Classification Accuracy, \(15.21\%\) in Generation Accuracy and reducing Fréchet Inception Distance by \(36.61\%\) , indicating superior semantic alignment and image quality. The code is available at https://github.com/CVLABLUMS/CATVis .