ADHDeepNet From Raw EEG to Diagnosis: Improving ADHD Diagnosis through Temporal-Spatial Processing, Adaptive Attention Mechanisms, and Explainability in Raw EEG Signals
2509.08779v1
cs.LG, q-bio.NC
2025-09-12
Авторы:
Ali Amini, Mohammad Alijanpour, Behnam Latifi, Ali Motie Nasrabadi
Резюме на русском
## Контекст
Attention Deficit Hyperactivity Disorder (ADHD) является одной из наиболее распространенных неврологических заболеваний, влияющих на дети и взрослых, которое может оказывать влияние на социальную, академическую и профессиональную сферу жизни. Несмотря на то, что преждевременное выявление и лечение могут значительно улучшить качество жизни пациентов, диагностика ADHD обычно требует много времени и труда. Это связано с необходимостью анализа больших объемов данных, включающих электроэнцефалограммы (EEG), которые могут быть трудно интерпретировать с помощью традиционных методов. Задача авторов — разработать модель, которая улучшит точность диагностики и уменьшит время, затрачиваемое на процесс.
## Метод
ADHDeepNet — это новая Deep Learning (DL) модель, разработанная с целью улучшить точность диагностики ADHD. Она основывается на двух основных компонентах: **комплексной процедуре характеризации временно-пространственного процесса** и **адаптивных модулей внимания**. Модель обрабатывает непосредственно сырые EEG-сигналы, используя современные техники извлечения и очистки признаков. Уникальность модели заключается в использовании **трансформеров**, что позволяет обрабатывать данные с высокой точностью и интеллектуальным упорядочением. Для обеспечения эффективности, модель проходит процесс **гиперпараметрной оптимизации** в двух этапах: внутренний 2-fold cross-validation и внешний 10-fold cross-subject validation. Для разнообразия данных применяется Additive Gaussian Noise (AGN), что позволяет модели более хорошо адаптироваться к различным условиям.
## Результаты
На основе данных собранных у 121 участников (61 с диагнозом ADHD и 60 здоровых контрольных), ADHDeepNet достигла **100% точности классификации** на тестируемых данных. На множестве вариантов тестов (включая различные уровни шума и разнообразие в характеристиках сигналов), модель показала приемлемую высокую точность (99.17%). Эксперименты показали, что модель не только классифицирует заболевание с высокой точностью, но и может сфокусировать внимание на ключевых областях мозга и частотных бандах, которые играют ключевую роль в диагностике ADHD.
## Значимость
Результаты ADHDeepNet могут быть применены в различных областях. В первую очередь, это улучшение диагностики ADHD, что позволит сэкономить ресурсы здравоохранения и время пациентов. Благодаря интеллектуальной системе обработки данных, модель может быть использована в клинических условиях для легкого и автоматизированного диагностирования. Эта модель также может способствовать развитию исследований в области DL в здравоохранении и психиатрии, демонстрируя потенциал а
Abstract
Attention Deficit Hyperactivity Disorder (ADHD) is a common brain disorder in
children that can persist into adulthood, affecting social, academic, and
career life. Early diagnosis is crucial for managing these impacts on patients
and the healthcare system but is often labor-intensive and time-consuming. This
paper presents a novel method to improve ADHD diagnosis precision and
timeliness by leveraging Deep Learning (DL) approaches and electroencephalogram
(EEG) signals. We introduce ADHDeepNet, a DL model that utilizes comprehensive
temporal-spatial characterization, attention modules, and explainability
techniques optimized for EEG signals. ADHDeepNet integrates feature extraction
and refinement processes to enhance ADHD diagnosis. The model was trained and
validated on a dataset of 121 participants (61 ADHD, 60 Healthy Controls),
employing nested cross-validation for robust performance. The proposed
two-stage methodology uses a 10-fold cross-subject validation strategy.
Initially, each iteration optimizes the model's hyper-parameters with inner
2-fold cross-validation. Then, Additive Gaussian Noise (AGN) with various
standard deviations and magnification levels is applied for data augmentation.
ADHDeepNet achieved 100% sensitivity and 99.17% accuracy in classifying ADHD/HC
subjects. To clarify model explainability and identify key brain regions and
frequency bands for ADHD diagnosis, we analyzed the learned weights and
activation patterns of the model's primary layers. Additionally, t-distributed
Stochastic Neighbor Embedding (t-SNE) visualized high-dimensional data, aiding
in interpreting the model's decisions. This study highlights the potential of
DL and EEG in enhancing ADHD diagnosis accuracy and efficiency.
Ссылки и действия
Дополнительные ресурсы: