arXiv:2503.08251eess.SPcs.AI2025-03

提出轻量自适应模型MT-NAM,实现癫痫发作检测100倍提速

MT-NAM: An Efficient and Adaptive Model for Epileptic Seizure Detection

  • 基于神经加法模型构建轻量化树结构,提升推理效率
  • 在CHB-MIT数据集上保持85.3%敏感度,速度提升50倍以上
  • 引入测试时动态调整机制,实时适应脑电信号变化

提升神经接口系统中机器学习算法的准确率与效率,对发展下一代智能治疗设备至关重要。当前系统多采用基础机器学习模型,未能充分利用脑电信号的天然结构;且现有模型在推理阶段常表现低速低效。为此,本文提出基于神经加法模型(NAM)的微树结构轻量化模型MT-NAM。该模型在保持高精度的前提下,推理速度相比标准NAM提升100倍。我们在包含24名患者、多时段及多发作记录的CHB-MIT头皮脑电图数据集上进行评估:标准NAM达到85.3%窗口级敏感度和95%特异性;而MT-NAM仅损失2%敏感度。通过引入测试时模板调整器(T3A)作为在线更新机制,可补偿敏感度损失,在保持相同敏感度的同时实现约50倍的推理加速。

原文摘要 · Abstract (English)

Enhancing the accuracy and efficiency of machine learning algorithms employed in neural interface systems is crucial for advancing next-generation intelligent therapeutic devices. However, current systems often utilize basic machine learning models that do not fully exploit the natural structure of brain signals. Additionally, existing learning models used for neural signal processing often demonstrate low speed and efficiency during inference. To address these challenges, this study introduces Micro Tree-based NAM (MT-NAM), a distilled model based on the recently proposed Neural Additive Models (NAM). The MT-NAM achieves a remarkable 100$\times$ improvement in inference speed compared to standard NAM, without compromising accuracy. We evaluate our approach on the CHB-MIT scalp EEG dataset, which includes recordings from 24 patients with varying numbers of sessions and seizures. NAM achieves an 85.3\% window-based sensitivity and 95\% specificity. Interestingly, our proposed MT-NAM shows only a 2\% reduction in sensitivity compared to the original NAM. To regain this sensitivity, we utilize a test-time template adjuster (T3A) as an update mechanism, enabling our model to achieve higher sensitivity during test time by accommodating transient shifts in neural signals. With this online update approach, MT-NAM achieves the same sensitivity as the standard NAM while achieving approximately 50$\times$ acceleration in inference speed.

癫痫检测神经接口模型加速在线学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。