arXiv:2502.00376cs.LGcs.HC2025-02被引 5

用自监督学习分析脑电数据,提升儿童多动症检测准确率。

SSRepL-ADHD: Adaptive Complex Representation Learning Framework for ADHD Detection from Visual Attention Tasks

  • 结合自监督与迁移学习,用LSTM和GRU捕捉脑电信号时序特征。
  • 在不平衡数据下达到81.11%的最高准确率。
  • 适合关注神经发育障碍早期筛查的研究者使用。

自监督表示学习(SSRepL)能够捕捉注意力缺陷多动障碍(ADHD)数据中的有意义且鲁棒的表征,具有提升下游多种神经发育障碍(NDD)检测性能的潜力。本文提出一种基于自监督学习与迁移学习(TL)的新框架,融合长短期记忆网络(LSTM)与门控循环单元(GRU)模型,通过视觉注意任务中提取的脑电图(EEG)信号实现儿童潜在多动症症状的检测。该方法通过归一化、滤波与数据平衡预处理提升脑电信号质量。实验采用三种模型:1)基于SSRepL与TL的LSTM-GRU模型(命名为SSRepL-ADHD),整合LSTM与GRU层以捕捉数据时序依赖性;2)轻量级自监督学习的DNN模型(LSSRepL-DNN);3)随机森林(RF)。模型在准确率、精确率、召回率与F1分数等指标上进行综合评估。结果表明,所提出的SSRepL-ADHD模型在数据不平衡与特征选择挑战下达到最高准确率81.11%。

原文摘要 · Abstract (English)

Self Supervised Representation Learning (SSRepL) can capture meaningful and robust representations of the Attention Deficit Hyperactivity Disorder (ADHD) data and have the potential to improve the model's performance on also downstream different types of Neurodevelopmental disorder (NDD) detection. In this paper, a novel SSRepL and Transfer Learning (TL)-based framework that incorporates a Long Short-Term Memory (LSTM) and a Gated Recurrent Units (GRU) model is proposed to detect children with potential symptoms of ADHD. This model uses Electroencephalogram (EEG) signals extracted during visual attention tasks to accurately detect ADHD by preprocessing EEG signal quality through normalization, filtering, and data balancing. For the experimental analysis, we use three different models: 1) SSRepL and TL-based LSTM-GRU model named as SSRepL-ADHD, which integrates LSTM and GRU layers to capture temporal dependencies in the data, 2) lightweight SSRepL-based DNN model (LSSRepL-DNN), and 3) Random Forest (RF). In the study, these models are thoroughly evaluated using well-known performance metrics (i.e., accuracy, precision, recall, and F1-score). The results show that the proposed SSRepL-ADHD model achieves the maximum accuracy of 81.11% while admitting the difficulties associated with dataset imbalance and feature selection.

多动症检测自监督学习脑电图时序建模

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