三招提升脑电癫痫检测模型效率,适配可穿戴设备
Efficient EEG Seizure Detection Using INT8 Quantization, Channel Pruning, and Spiking Neural Networks

- 将CNN转为脉冲神经网络,实现稀疏时序活动
- 通道剪枝与2:4权值稀疏使模型体积减至0.44MB
- INT8量化降低64%能耗,推理速度提升2.8倍
癫痫的连续脑电图(EEG)监测受限于可穿戴和植入式设备的功耗与内存预算。深度神经网络虽能高精度检测发作,但计算开销大、模型尺寸大,难以部署。本文以CHB-MIT头皮EEG数据集上的单个一维CNN为基础,探索三种类脑高效策略:(i) 通过参数迁移将CNN转换为脉冲神经网络(SNN);(ii) 结合2:4结构化权值稀疏进行脑电通道剪枝;(iii) 使用FX与ONNX工作流进行INT8量化,包含量化感知训练和算子融合。量化后的CNN模型存储量从1.63MB降至0.44MB,单次推理能耗降低最高达64%,CPU延迟提速最多2.8倍,且准确率保持甚至小幅提升(AUC)。剪枝后输入通道数和非零权重减半,精度下降有限。SNN实现具有稀疏时间活动的脉冲计算。三者构成互补的高效方向。
原文摘要 · Abstract (English)
Continuous EEG monitoring for epilepsy is constrained by the limited power and memory budgets of wearable and implantable devices. Deep neural networks can detect seizures with high accuracy, but their computational cost and model size make them difficult to deploy on such platforms. In this work we use a single 1D CNN seizure detector on the CHB-MIT scalp EEG dataset as a common baseline, and then investigate three brain-inspired efficiency strategies: (i) conversion of the CNN into a spiking neural network (SNN) via parameter transfer, (ii) EEG channel pruning combined with 2:4 structured weight sparsity, and (iii) INT8 quantization using FX- and ONNX-based workflows, including quantization-aware training and operator fusion. The quantized CNN variants reduce stored model size from 1.63 MB to 0.44 MB, lower estimated energy per inference by up to 64%, and achieve as much as 2.8 times speedup in CPU latency while preserving, and in one case slightly improving, AUC. The pruned CNN halves the number of input channels and non-zero weights with only a modest accuracy drop, and the SNN conversion provides a spiking implementation with sparse temporal activity. Together, these experiments characterize three complementary efficiency directions for seizure detection.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。