LGAISPNov 25, 2024

Graph Adapter of EEG Foundation Models for Parameter Efficient Fine Tuning

arXiv:2411.16155v23 citationsh-index: 29
Originality Incremental advance
AI Analysis

This work addresses the problem of high computational costs and limited labeled data in EEG-based neurological disorder diagnosis for healthcare applications, representing an incremental improvement in fine-tuning efficiency.

The paper tackles the challenge of fine-tuning large foundation models for EEG data by proposing EEG-GraphAdapter, a parameter-efficient method that integrates a GNN module into a frozen temporal backbone, reducing computational costs and data needs. It achieves up to a 16.1% improvement in F1-score on tasks like Major Depressive Disorder detection compared to the baseline BENDR model.

In diagnosing neurological disorders from electroencephalography (EEG) data, foundation models such as Transformers have been employed to capture temporal dynamics. Additionally, Graph Neural Networks (GNNs) are critical for representing the spatial relationships among EEG sensors. However, fine-tuning these large-scale models for both temporal and spatial features can be prohibitively large in computational cost, especially under the limited availability of labeled EEG datasets. We propose EEG-GraphAdapter (EGA), a parameter-efficient fine-tuning (PEFT) approach designed to address these challenges. EGA is integrated into a pre-trained temporal backbone model as a GNN-based module, freezing the backbone and allowing only the adapter to be fine-tuned. This enables the effective acquisition of EEG spatial representations, significantly reducing computational overhead and data requirements. Experimental evaluations on two healthcare-related downstream tasks-Major Depressive Disorder (MDD) and Abnormality Detection (TUAB)-show that EGA improves performance by up to 16.1% in F1-score compared with the backbone BENDR model, highlighting its potential for scalable and accurate EEG-based predictions.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes