arXiv:2410.19815eess.SPcs.LG2024-10被引 1

BUNDL让深度学习模型识别脑电图标注模糊,提升癫痫发作检测鲁棒性。

Bayesian uncertainty-aware deep learning with noisy labels: Tackling annotation ambiguity in EEG seizure detection

  • 基于贝叶斯框架设计新损失函数,显式建模标签不确定性
  • 在三种基线模型上均提升抗噪声能力,准确率最高提升4.2%
  • 无需增加参数,可直接嵌入现有临床模型,适合医疗部署

深度学习正推动脑电图(EEG)自动化癫痫发作检测与发作起始区定位,但其性能高度依赖高质量标注数据。然而头皮EEG易受高噪声干扰,导致发作时间与特征标注不精确,形成“标签噪声”,严重挑战模型训练与泛化能力。本文提出贝叶斯不确定性感知深度学习(BUNDL),通过将领域知识融入贝叶斯框架,设计基于KL散度的新型损失函数,利用不确定性信息更有效地从头皮EEG中学习发作特征。BUNDL为训练带有噪声标签的深度神经网络提供了一种简单、模型无关的方法,不增加额外参数。我们还评估了改进检测系统对自动发作起始区定位的影响。在模拟EEG数据集及两个公开数据集(TUH和CHB-MIT)上验证表明,BUNDL能持续识别噪声标签,并在多种噪声条件下提升三个基础模型的鲁棒性。同时评估了跨中心泛化能力与计算开销。最终,BUNDL可无缝集成至现有临床深度模型,支持可信癫痫评估模型的训练。

原文摘要 · Abstract (English)

Deep learning is advancing EEG processing for automated epileptic seizure detection and onset zone localization, yet its performance relies heavily on high-quality annotated training data. However, scalp EEG is susceptible to high noise levels, which in turn leads to imprecise annotations of the seizure timing and characteristics. This "label noise" presents a significant challenge in model training and generalization. In this paper, we introduce Bayesian UncertaiNty-aware Deep Learning (BUNDL), a novel algorithm that informs a deep learning model of label ambiguities, thereby enhancing the robustness of seizure detection systems. By integrating domain knowledge into an underlying Bayesian framework, we derive a novel KL-divergence-based loss function that capitalizes on uncertainty to better learn seizure characteristics from scalp EEG. Thus, BUNDL offers a straightforward and model-agnostic method for training deep neural networks with noisy training labels that does not add any parameters to existing architectures. Additionally, we explore the impact of improved detection system on the task of automated onset zone localization. We validate BUNDL using a comprehensive simulated EEG dataset and two publicly available datasets collected by Temple University Hospital (TUH) and Boston Children's Hospital (CHB-MIT). Results show that BUNDL consistently identifies noisy labels and improves the robustness of three base models under various label noise conditions. We also evaluate cross-site generalizability and quantify computational cost of all methods. Ultimately, BUNDL presents as a reliable method that can be seamlessly integrated with existing deep models used in clinical practice, enabling the training of trustworthy models for epilepsy evaluation.

癫痫检测不确定性建模脑电图鲁棒学习

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