Nonlinear Feature Modeling Using Kolmogorov–Arnold Networks in Electroencephalography for Depression Classification: Deep Learning Study
ABSTRACT
Background:
Depression is a common mental disorder that seriously affects mental health and quality of life. Traditional depression diagnosis mainly relies on clinical interviews and subjective assessments, which may lead to variability in diagnostic outcomes. Electroencephalogram (EEG)-based computer-aided diagnosis has emerged as a promising objective approach for depression recognition. However, effectively modeling the complex nonlinear characteristics of EEG signals remains a major challenge for existing deep learning methods.
Objective:
This study aimed to investigate the applicability of Kolmogorov–Arnold Networks (KAN) for EEG-based depression recognition and to evaluate the effectiveness of KAN-based nonlinear modeling frameworks on frequency-domain EEG features.
Methods:
KAN was introduced into EEG-based depression recognition to enhance nonlinear feature representation of brain electrical signals. Two representative KAN-based convolutional architectures, ConvKAN and its lightweight variant ConvKAN Efficient, were systematically investigated for EEG classification tasks. These architectures incorporated KANLinear layers together with dynamic grid updating, enhanced weight initialization, and regularization strategies to improve nonlinear modeling capability while maintaining computational efficiency. Extensive experiments were conducted on the EDRA dataset, and the proposed framework was compared with conventional convolutional neural network (CNN)-based methods and traditional machine learning approaches. Model performance was evaluated using accuracy, precision, recall, and F1-score metrics.
Results:
Experimental results demonstrated that the KAN-based frameworks achieved superior EEG classification performance compared with conventional CNN-based methods and traditional machine learning approaches. Among the evaluated models, ConvKAN Efficient achieved the best overall performance, with accuracy, precision, recall, and F1-score all exceeding 94% on the EDRA dataset. The results indicate that KAN-based architectures can effectively capture the complex nonlinear characteristics of EEG signals while preserving high computational efficiency.
Conclusions:
This study systematically demonstrates the applicability of KAN-based nonlinear modeling frameworks to EEG-based depression recognition. The proposed framework provides a scalable and effective approach for EEG signal analysis and offers valuable technical support for both theoretical research and clinical applications in computer-aided depression recognition.
Citation
Request queued. Please wait while the file is being generated. It may take some time.
Copyright
© The authors. All rights reserved. This is a privileged document currently under peer-review/community review (or an accepted/rejected manuscript). Authors have provided JMIR Publications with an exclusive license to publish this preprint on it's website for review and ahead-of-print citation purposes only. While the final peer-reviewed paper may be licensed under a cc-by license on publication, at this stage authors and publisher expressively prohibit redistribution of this draft paper other than for review purposes.