Towards Linguistic Neural Representation Learning and Sentence Retrieval from Electroencephalogram Recordings

Jinzhao Zhou,Yiqun Duan,Ziyi Zhao,Yu-Cheng Chang,Yu-Kai Wang,Thomas Do,Chin-Teng Lin
2024-08-08
Abstract:Decoding linguistic information from non-invasive brain signals using EEG has gained increasing research attention due to its vast applicational potential. Recently, a number of works have adopted a generative-based framework to decode electroencephalogram (EEG) signals into sentences by utilizing the power generative capacity of pretrained large language models (LLMs). However, this approach has several drawbacks that hinder the further development of linguistic applications for brain-computer interfaces (BCIs). Specifically, the ability of the EEG encoder to learn semantic information from EEG data remains questionable, and the LLM decoder's tendency to generate sentences based on its training memory can be hard to avoid. These issues necessitate a novel approach for converting EEG signals into sentences. In this paper, we propose a novel two-step pipeline that addresses these limitations and enhances the validity of linguistic EEG decoding research. We first confirm that word-level semantic information can be learned from EEG data recorded during natural reading by training a Conformer encoder via a masked contrastive objective for word-level classification. To achieve sentence decoding results, we employ a training-free retrieval method to retrieve sentences based on the predictions from the EEG encoder. Extensive experiments and ablation studies were conducted in this paper for a comprehensive evaluation of the proposed approach. Visualization of the top prediction candidates reveals that our model effectively groups EEG segments into semantic categories with similar meanings, thereby validating its ability to learn patterns from unspoken EEG recordings. Despite the exploratory nature of this work, these results suggest that our method holds promise for providing more reliable solutions for converting EEG signals into text.
Computation and Language,Artificial Intelligence,Machine Learning
What problem does this paper attempt to address?
This paper attempts to solve several key problems encountered when decoding linguistic information from electroencephalogram (EEG) signals, especially during the process of converting EEG signals into sentences during silent reading tasks. Specifically: 1. **Insufficient Semantic Information Learning**: The existing EEG encoders are questionable in their ability to learn semantic information from EEG data. Traditional generation - based methods may not enable EEG encoders to truly learn to capture semantic patterns. 2. **The Influence of Pretrained Large Language Models (LLMs)**: Powerful pretrained language models (such as LLMs), when used as decoders, tend to generate sentences based on their training memory rather than relying on the input EEG signals. This leads to the neglect of EEG signals, causing EEG encoders to fail to effectively learn meaningful features. 3. **Data Sparsity and Noise Problems**: EEG signals in silent reading tasks have high data sparsity and noise, which increases the decoding difficulty. For example, participants may have inconsistent attention to different words when reading, thus affecting the decoding effect. To solve these problems, the authors propose a new two - step pipeline method, called EEG - to - Text Retrieval (ETER), aiming at: - **Verifying the Learning Ability of EEG Encoders**: Train a Conformer encoder to learn word - level semantic representations and optimize it using the masked contrastive loss function. This can ensure that EEG encoders can effectively extract semantic information from EEG data. - **Eliminating the Bias of Pretrained Language Models**: Adopt an untrained retrieval method to avoid the bias brought by relying on powerful pretrained language models. Specifically, find the most relevant sentences through the set of predicted keywords (SK) combined with the beam - search retrieval (BSR) method. - **Improving Decoding Accuracy**: Through a large number of experiments and ablation studies, the effectiveness of the proposed ETER method in word - level classification and sentence retrieval has been verified, achieving high precision and recall rates. In conclusion, the main contribution of this paper lies in providing a new and more reliable method for converting EEG signals into text, which solves several limitations in existing methods.