用轻量Transformer+对抗训练,高效去除脑电中的肌电干扰。
Removing Neural Signal Artifacts with Autoencoder-Targeted Adversarial Transformers (AT-AT)
- 用自编码器定位关键时间点,让Transformer只在必要处工作。
- 模型体积减少90%以上,低信噪比下仍保持高重建相关性。
- 适合资源受限场景,如可穿戴设备实时脑电信号处理。
肌电噪声是脑电数据中主要的污染源,会干扰对脑区神经活动的准确分析。现有深度学习去噪方法计算开销大、训练耗时长。本文提出一种基于自编码器引导的对抗性变压器(AT-AT)系统,通过自编码器识别最优时间序列应用位置,使模型规模相比已有去噪模型减少超90%。对抗训练确保滤波后信号符合脑电基本特征。在67名受试者公开数据上训练,AT-AT在初始信噪比2 dB时平均重建相关系数超过0.95,在-7 dB时仍达0.70。附录还展示了其在真实场景中的部署结果,未来需在更大样本上验证泛化能力。
原文摘要 · Abstract (English)
Electromyogenic (EMG) noise is a major contamination source in EEG data that can impede accurate analysis of brain-specific neural activity. Recent literature on EMG artifact removal has moved beyond traditional linear algorithms in favor of machine learning-based systems. However, existing deep learning-based filtration methods often have large compute footprints and prohibitively long training times. In this study, we present a new machine learning-based system for filtering EMG interference from EEG data using an autoencoder-targeted adversarial transformer (AT-AT). By leveraging the lightweight expressivity of an autoencoder to determine optimal time-series transformer application sites, our AT-AT architecture achieves a >90% model size reduction compared to published artifact removal models. The addition of adversarial training ensures that filtered signals adhere to the fundamental characteristics of EEG data. We trained AT-AT using published neural data from 67 subjects and found that the system was able to achieve comparable test performance to larger models; AT-AT posted a mean reconstructive correlation coefficient above 0.95 at an initial signal-to-noise ratio (SNR) of 2 dB and 0.70 at -7 dB SNR. Further research generalizing these results to broader sample sizes beyond these isolated test cases will be crucial; while outside the scope of this study, we also include results from a real-world deployment of AT-AT in the Appendix.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。