arXiv:2508.07253cs.LGeess.SP2025-08被引 1

一个可跨数据集通用的癫痫发作检测机器学习框架

PySeizure: A single machine learning classifier framework to detect seizures in diverse datasets

  • 用自动预处理和多模型投票提升检测鲁棒性
  • 在两个数据集上实现0.90以上AUC,跨数据集仍保持0.6以上性能
  • 开源可复现,适合临床部署与专家辅助诊断

可靠癫痫发作检测对癫痫诊疗至关重要,但临床仍依赖耗时的人工脑电图(EEG)解读。现有机器学习方法常需针对特定数据集优化,限制了实际应用与可复现性。本文提出一个开源的单分类器框架,可在不同临床数据集间实现稳健、泛化的癫痫发作检测。我们在两个公开的EEG数据集(CHB-MIT与TUSZ)上评估该方法,其在各自数据集内表现优异(CHB-MIT AUC 0.904±0.059,TUSZ AUC 0.864±0.060),跨数据集迁移性能也较强(反向迁移达0.762±0.175,正向为0.615±0.039),无需后处理。引入轻微后处理后,性能进一步提升至0.913±0.064和0.867±0.058,以及0.619±0.036和0.768±0.172。该框架具备高度可复现性,为构建无数据集依赖的临床可用系统提供基础。

原文摘要 · Abstract (English)

Reliable seizure detection is critical for diagnosing and managing epilepsy, yet clinical workflows remain dependent on time-consuming manual EEG interpretation. While machine learning has shown promise, existing approaches often rely on dataset-specific optimisations, limiting their real-world applicability and reproducibility. Here, we introduce an innovative, open-source machine-learning framework that enables robust and generalisable seizure detection across varied clinical datasets. We evaluate our approach on two publicly available EEG datasets that differ in patient populations and electrode configurations. To enhance robustness, the framework incorporates an automated pre-processing pipeline to standardise data and a majority voting mechanism, in which multiple models independently assess each second of EEG before reaching a final decision. We train, tune, and evaluate models within each dataset, assessing their cross-dataset transferability. Our models achieve high within-dataset performance (AUC 0.904+/-0.059 for CHB-MIT and 0.864+/-0.060 for TUSZ) and demonstrate strong generalisation across datasets despite differences in EEG setups and populations (AUC 0.615+/-0.039 for models trained on CHB-MIT and tested on TUSZ and 0.762+/-0.175 in the reverse case) without any post-processing. Furthermore, a mild post-processing improved the within-dataset results to 0.913+/-0.064 and 0.867+/-0.058 and cross-dataset results to 0.619+/-0.036 and 0.768+/-0.172. These results underscore the potential of, and essential considerations for, deploying our framework in diverse clinical settings. By making our methodology fully reproducible, we provide a foundation for advancing clinically viable, dataset-agnostic seizure detection systems. This approach has the potential for widespread adoption, complementing rather than replacing expert interpretation, and accelerating clinical integration.

癫痫检测机器学习EEG分析跨数据集

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