Group-level Brain Decoding with Deep Learning

Richard Csaky,Mats Van Es,Oiwi Parker Jones,Mark Woolrich
DOI: https://doi.org/10.1002/hbm.26500
2024-01-19
Abstract:Decoding brain imaging data are gaining popularity, with applications in brain-computer interfaces and the study of neural representations. Decoding is typicallysubject-specific and does not generalise well over subjects, due to high amounts ofbetween subject variability. Techniques that overcome this will not only providericher neuroscientific insights but also make it possible for group-level models to out-perform subject-specific models. Here, we propose a method that uses subjectembedding, analogous to word embedding in natural language processing, to learnand exploit the structure in between-subject variability as part of a decoding model,our adaptation of the WaveNet architecture for classification. We apply this to mag-netoencephalography data, where 15 subjects viewed 118 different images, with30 examples per image; to classify images using the entire 1 s window followingimage presentation. We show that the combination of deep learning and subjectembedding is crucial to closing the performance gap between subject- and group-level decoding models. Importantly, group models outperform subject models onlow-accuracy subjects (although slightly impair high-accuracy subjects) and can behelpful for initialising subject models. While we have not generally found group-levelmodels to perform better than subject-level models, the performance of groupmodelling is expected to be even higher with bigger datasets. In order to providephysiological interpretation at the group level, we make use of permutation featureimportance. This provides insights into the spatiotemporal and spectral informationencoded in the models. All code is available on GitHub (<a class="link-external link-https" href="https://github.com/ricsinaruto/MEG-group-decode" rel="external noopener nofollow">this https URL</a>).
Machine Learning,Signal Processing,Neurons and Cognition
What problem does this paper attempt to address?
The problem that this paper attempts to solve is how to overcome the challenge of inter - subject variability in brain - imaging data decoding in order to achieve a cross - subject (group - level) brain - activity decoding model. Specifically: 1. **Limitations of Subject - Specific Models**: Traditional brain - imaging decoding methods usually train a model for each subject separately. Although this method can well adapt to individual differences, it cannot use cross - subject data to improve the generalization ability of the model. In addition, each subject needs to train a model separately, which increases the computational burden and is prone to overfitting. 2. **Challenges of Cross - Subject Models**: Directly combining the data of multiple subjects to train a shared model (i.e., the "naive" group model) often performs poorly because the inter - subject variability is high, making it difficult for the model to capture effective features. 3. **Introducing Subject Embedding**: To overcome the above problems, this paper proposes a new method, which is to learn and utilize the structured variability between subjects by introducing subject embedding. Subject embedding is similar to word embedding in natural language processing and can capture the specific features of each subject, thereby improving the decoding performance in cross - subject models. 4. **Model Architecture**: The paper uses an improved WaveNet architecture for classification. Combined with subject embedding, it can improve the generalization ability of the model while maintaining individual differences. 5. **Experimental Verification**: By conducting experiments on MEG data of 15 subjects, the paper shows that after introducing subject embedding, the performance of the cross - subject model is significantly improved, especially more obvious on low - precision subjects. In addition, by fine - tuning the cross - subject model, the decoding performance can be further improved at the individual level. In conclusion, the main goal of this paper is to improve the performance of cross - subject brain - activity decoding models by introducing subject embedding, so that they can better generalize to new subjects and provide more abundant insights for neuroscience research.