论文

ThinkNet:面向跨被试MI-EEG解码的紧凑模型选择与验证门控集成框架

ThinkNet: Compact Architecture Selection and Validation-Gated Ensembles for Subject-Independent MI-EEG Decoding

精选理由

做脑机接口或EEG应用的朋友可以看看,ThinkNet用验证门控选模型,4.9K参数就能跑出44%的四分类准确率,还比大模型泛化更好。

论文提出ThinkNet框架,结合仅训练集归一化、验证集引导的进化搜索和验证门控推理,为运动想象脑机接口筛选紧凑解码器。在BCI Competition IV-2a数据集的九折LOSO评测中,验证选出的紧凑解码器仅4.9K参数、19 KB FP32权重,批推理耗时0.99 ms(Orin CUDA),准确率44.35±15.41%。更广搜索显示,≤25K参数的小模型经再训练后平均跨被试准确率达40.10%,高于中型(35.09%)和大型模型(34.78%)。验证门控集成在固定基准上进一步提升至43.98±16.25%。oracle分析发现6.1个百分点的家族选择差距,且验证集与测试集相关性接近零,说明被试偏移下的验证可靠性仍是瓶颈。

原文 · arXiv cs.AI

ThinkNet: Compact Architecture Selection and Validation-Gated Ensembles for Subject-Independent MI-EEG Decoding

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.