Abstract:Visual brain decoding aims to decode visual information from human brain activities. Despite the great progress, one critical limitation of current brain decoding research lies in the lack of generalization capability to unseen subjects. Prior works typically focus on decoding brain activity of individuals based on the observation that different subjects exhibit different brain activities, while it remains unclear whether brain decoding can be generalized to unseen subjects. This study aims to answer this question. We first consolidate an image-fMRI dataset consisting of stimulus-image and fMRI-response pairs, involving 177 subjects in the movie-viewing task of the Human Connectome Project (HCP). This dataset allows us to investigate the brain decoding performance with the increase of participants. We then present a learning paradigm that applies uniform processing across all subjects, instead of employing different network heads or tokenizers for individuals as in previous methods, which can accommodate a large number of subjects to explore the generalization capability across different subjects. A series of experiments are conducted and we have the following findings. First, the network exhibits clear generalization capabilities with the increase of training subjects. Second, the generalization capability is common to popular network architectures (MLP, CNN and Transformer). Third, the generalization performance is affected by the similarity between subjects. Our findings reveal the inherent similarities in brain activities across individuals. With the emerging of larger and more comprehensive datasets, it is possible to train a brain decoding foundation model in the future. Codes and models can be found at <a class="link-external link-https" href="https://github.com/Xiangtaokong/TGBD" rel="external noopener nofollow">this https URL</a>.
What problem does this paper attempt to address?
The paper attempts to address the issue of insufficient generalization ability of Visual Brain Decoding across different individuals. Specifically, current brain decoding research mainly focuses on decoding brain activity of specific individuals, but these models are difficult to apply to new, unseen individuals. The paper aims to answer the following question: Can brain decoding models generalize to unseen individuals?
### Background of the Paper
The goal of visual brain decoding is to decode visual information from human brain activity, including tasks such as image classification, retrieval, and reconstruction. Although significant progress has been made in existing research, a key limitation is the lack of generalization ability to unseen individuals. Most existing methods typically decode for specific individuals, assuming that brain activity varies greatly between different individuals, making it difficult to generalize to new individuals.
### Main Contributions
1. **Dataset Integration**: The authors constructed an image-fMRI dataset of a movie-watching task involving 177 participants, with each participant providing 3,127 image-fMRI pairs. This dataset allows researchers to explore brain decoding performance as the number of participants increases.
2. **Learning Paradigm**: A new learning paradigm is proposed, which applies a unified processing method to all participants, rather than designing different network heads or encoders for each individual as in previous methods. This allows the model to handle a large number of participants without increasing model complexity and parameter count.
3. **Experimental Results**: Through a series of experiments, the authors found that:
- The generalization ability of the model significantly improves as the number of training participants increases. For example, when the number of training participants increased from 1 to 167, the Top-1 accuracy for unseen individuals increased from 2% to 45%.
- Different network architectures (MLP, CNN, and Transformer) all exhibit generalization ability, but MLP performs the best.
- Generalization performance is influenced by the similarity of participants. For example, using participants of the same gender during model training results in better generalization performance.
### Experimental Details
- **Dataset**: The HCP dataset was used, involving 177 participants, with each participant providing 3,127 image-fMRI pairs.
- **Model Architecture**: Mainly used MLP as the backbone network and validated the generalization ability of other network architectures (such as 1D CNN, 3D CNN, and Transformer).
- **Evaluation Metrics**: The generalization performance of the model on unseen individuals was evaluated using Top-1 and Top-3 accuracy.
### Conclusion
Through a large-scale dataset and a unified learning paradigm, the paper demonstrates that visual brain decoding models can generalize to unseen individuals. This finding reveals the inherent similarity in brain activity between different individuals, providing the possibility for further training of larger-scale foundational models.