IEFS-GMB: Gradient Memory Bank-Guided Feature Selection Based on Information Entropy for EEG Classification of Neurological Disorders

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Zhang, Liang, Dong, Hanyang, Gao, Jia-Hong, Sun, Yi, Xiao, Kuntao, Yang, Wanli, Lv, Zhao, Sheng, Shurong
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866909797124669440
author Zhang, Liang
Dong, Hanyang
Gao, Jia-Hong
Sun, Yi
Xiao, Kuntao
Yang, Wanli
Lv, Zhao
Sheng, Shurong
author_facet Zhang, Liang
Dong, Hanyang
Gao, Jia-Hong
Sun, Yi
Xiao, Kuntao
Yang, Wanli
Lv, Zhao
Sheng, Shurong
contents Deep learning-based EEG classification is crucial for the automated detection of neurological disorders, improving diagnostic accuracy and enabling early intervention. However, the low signal-to-noise ratio of EEG signals limits model performance, making feature selection (FS) vital for optimizing representations learned by neural network encoders. Existing FS methods are seldom designed specifically for EEG diagnosis; many are architecture-dependent and lack interpretability, limiting their applicability. Moreover, most rely on single-iteration data, resulting in limited robustness to variability. To address these issues, we propose IEFS-GMB, an Information Entropy-based Feature Selection method guided by a Gradient Memory Bank. This approach constructs a dynamic memory bank storing historical gradients, computes feature importance via information entropy, and applies entropy-based weighting to select informative EEG features. Experiments on four public neurological disease datasets show that encoders enhanced with IEFS-GMB achieve accuracy improvements of 0.64% to 6.45% over baseline models. The method also outperforms four competing FS techniques and improves model interpretability, supporting its practical use in clinical settings.
format Preprint
id arxiv_https___arxiv_org_abs_2509_15259
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle IEFS-GMB: Gradient Memory Bank-Guided Feature Selection Based on Information Entropy for EEG Classification of Neurological Disorders
Zhang, Liang
Dong, Hanyang
Gao, Jia-Hong
Sun, Yi
Xiao, Kuntao
Yang, Wanli
Lv, Zhao
Sheng, Shurong
Machine Learning
Artificial Intelligence
Deep learning-based EEG classification is crucial for the automated detection of neurological disorders, improving diagnostic accuracy and enabling early intervention. However, the low signal-to-noise ratio of EEG signals limits model performance, making feature selection (FS) vital for optimizing representations learned by neural network encoders. Existing FS methods are seldom designed specifically for EEG diagnosis; many are architecture-dependent and lack interpretability, limiting their applicability. Moreover, most rely on single-iteration data, resulting in limited robustness to variability. To address these issues, we propose IEFS-GMB, an Information Entropy-based Feature Selection method guided by a Gradient Memory Bank. This approach constructs a dynamic memory bank storing historical gradients, computes feature importance via information entropy, and applies entropy-based weighting to select informative EEG features. Experiments on four public neurological disease datasets show that encoders enhanced with IEFS-GMB achieve accuracy improvements of 0.64% to 6.45% over baseline models. The method also outperforms four competing FS techniques and improves model interpretability, supporting its practical use in clinical settings.
title IEFS-GMB: Gradient Memory Bank-Guided Feature Selection Based on Information Entropy for EEG Classification of Neurological Disorders
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2509.15259