跳到正文
arXiv:cs.AI· Abdul Basit, Saim Rehman, Muhammad Shafique·· 13 小时前

ThinkNet:面向高效运动想象 EEG 分类的进化深度集成框架

ThinkNet: Evolutionary Deep Ensembles for Efficient Motor-Imagery EEG Classification

AI 导读

ThinkNet 是一个验证控制的框架,结合仅训练集归一化、验证引导的进化搜索和验证门控推理,用于跨被试运动想象 EEG 解码。在 BCI Competition IV-2a 四分类任务上,其固定基准中验证选出的紧凑解码器仅 4.9K 参数、19 KB FP32 权重,Orin CUDA 单样本推理 0.99 ms,准确率 44.35±15.41%;验证门控集成提升至 43.98±16.25%。

正文

View PDF HTML (experimental)

Abstract:Practical assistive and rehabilitative brain--computer interfaces require subject-independent motor-imagery EEG (MI-EEG) decoders that generalize to new users under limited target-user data and constrained compute. However, held-out-subject performance can be overstated when test-subject information influences preprocessing, model selection, or ensemble selection. We present \textit{ThinkNet}, a validation-controlled framework that combines train-only normalization, validation-guided evolutionary search, and validation-gated inference to identify compact decoders and inference policies for held-out subjects. We evaluate four-class BCI Competition IV-2a (session T) decoding with nine Leave-One-Subject-Out (LOSO) folds, three seeds, seven fixed decoder entries, and a broader search over ten representative decoder families; the held-out subject is never used for normalization, hyperparameter, architecture, or ensemble-policy selection. In the fixed benchmark, the validation-selected compact decoder achieved 44.35$\pm$15.41\% accuracy with 4.9K parameters, 19 KB FP32 weights, and 0.99 ms batch-1 Orin CUDA inference. Across the broader search, compact models ($\leq$25K parameters) achieved higher mean held-out accuracy than mid-size and large alternatives after selected retraining (40.10\% vs. 35.09\% and 34.78\%). Validation-gated ensembling improved over validation-selected single-model inference, reaching 43.98$\pm$16.25\% in the fixed benchmark and 43.31$\pm$15.88\% for the compact six-family ensemble. A non-deployable oracle analysis revealed a 6.1-point family-selection gap and near-zero validation--test correlation, showing that validation reliability remains a key bottleneck under subject shift. Thus, ThinkNet is a validation-controlled framework for compact MI-EEG model and inference-policy selection, rather than a single-architecture benchmark.
Comments: Accepted to IEEE-EMBS BHI'2026, 7 pages
Subjects: Human-Computer Interaction (cs.HC); Artificial Intelligence (cs.AI); Machine Learning (cs.LG); Signal Processing (eess.SP)
Cite as: arXiv:2609.33967 [cs.HC]
  (or arXiv:2609.33967v2 [cs.HC] for this version)
  https://doi.org/10.48550/arXiv.2609.33967

arXiv-issued DOI via DataCite

Submission history

From: Abdul Basit [view email]
[v1] Sun, 27 Sep 2026 22:11:26 UTC (3,392 KB)
[v2] Fri, 9 Oct 2026 15:00:12 UTC (3,393 KB)

来源:arXiv:cs.AI · arxiv.org