Neuropsychiatric disorders constitute a substantial global health burden, affecting approximately 970 million individuals worldwide, with a exacerbation during the COVID-19 pandemic. This investigation presents a comparative analysis of three state-of-the-art artificial intelligence architectures–Multilayer Perceptron (ANN), TabNet, and FT-Transformer–for the automated classification of neuropsychiatric disorders using electroencephalographic (EEG) signals. The study employed a clinically validated dataset comprising 945 participants stratified across seven diagnostic categories: mood disorders, addictive disorders, trauma and stress-related disorders, schizophrenia, anxiety disorders, healthy controls, and obsessive-compulsive disorder. The neurophysiological data, already pre-processed in a previous study, were used in conjunction with the Synthetic Minority Over-sampling Technique (SMOTE) due to class imbalance. The dataset has the features: power spectral density (PSD) for six frequency bands, functional connectivity (FC), age, sex, and intelligence quotient (IQ). Experimental results demonstrated TabNet’s superior generalization capability across multiple psychiatric disorders, achieving notable performance in mood disorders (93.50% accuracy) and addictive disorders (96.00% accuracy). Our proposed ANN architecture exhibited exceptional discriminative power for schizophrenia classification (94.38% AUC) and obsessive-compulsive disorder detection (98.89% AUC). The FT-Transformer model yielded competitive performance, particularly in anxiety disorder identification. These findings underscore the critical importance of disorder-specific model selection and feature engineering, advancing the development of robust computational frameworks for neuropsychiatric diagnostic support systems.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

A Comparative Analysis of ANN, TabNet and FT-Transformer Models in EEG Classification of Neuropsychiatric Disorders

  • Mateus Balda Mota,
  • Alessandro Bof de Oliveira,
  • Patricia Bof,
  • Dante Augusto Couto Barone

摘要

Neuropsychiatric disorders constitute a substantial global health burden, affecting approximately 970 million individuals worldwide, with a exacerbation during the COVID-19 pandemic. This investigation presents a comparative analysis of three state-of-the-art artificial intelligence architectures–Multilayer Perceptron (ANN), TabNet, and FT-Transformer–for the automated classification of neuropsychiatric disorders using electroencephalographic (EEG) signals. The study employed a clinically validated dataset comprising 945 participants stratified across seven diagnostic categories: mood disorders, addictive disorders, trauma and stress-related disorders, schizophrenia, anxiety disorders, healthy controls, and obsessive-compulsive disorder. The neurophysiological data, already pre-processed in a previous study, were used in conjunction with the Synthetic Minority Over-sampling Technique (SMOTE) due to class imbalance. The dataset has the features: power spectral density (PSD) for six frequency bands, functional connectivity (FC), age, sex, and intelligence quotient (IQ). Experimental results demonstrated TabNet’s superior generalization capability across multiple psychiatric disorders, achieving notable performance in mood disorders (93.50% accuracy) and addictive disorders (96.00% accuracy). Our proposed ANN architecture exhibited exceptional discriminative power for schizophrenia classification (94.38% AUC) and obsessive-compulsive disorder detection (98.89% AUC). The FT-Transformer model yielded competitive performance, particularly in anxiety disorder identification. These findings underscore the critical importance of disorder-specific model selection and feature engineering, advancing the development of robust computational frameworks for neuropsychiatric diagnostic support systems.