Comparative analysis of transfer learning performance on generalised EEG data for use in a depression diagnosis task
The purpose of this work was to analyse the performance of different deep learning methods in the task of depression diagnosis based on bioelectrical brain activity data. In particular, to study the potential of transfer learning using an artificial neural network trained on a significant amount of...
Saved in:
| Main Author: | |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
Saratov State University
2025-01-01
|
| Series: | Известия высших учебных заведений: Прикладная нелинейная динамика |
| Subjects: | |
| Online Access: | https://andjournal.sgu.ru/sites/andjournal.sgu.ru/files/text-pdf/2025/01/and_2025-1_shusharina_100-122.pdf |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1832576642612985856 |
|---|---|
| author | Shusharina, Natalia Nikolaevna |
| author_facet | Shusharina, Natalia Nikolaevna |
| author_sort | Shusharina, Natalia Nikolaevna |
| collection | DOAJ |
| description | The purpose of this work was to analyse the performance of different deep learning methods in the task of depression diagnosis based on bioelectrical brain activity data. In particular, to study the potential of transfer learning using an artificial neural network trained on a significant amount of “generalised” electroencephalography data in the task of diagnosing depression from non-invasive electroencephalogram signals. Methods. Deep learning approaches such as transfer learning and contrastive learning were used in the present study. Artificial neural networks were trained on the public HBN EO/EC task dataset containing recordings of electroencephalogram signals. The 1D CNN and EEGNet architectures were used as auxiliary artificial networks for transfer learning. In order to test the quality of contrastive learning, the dataset was augmented and the following algorithms were selected as the donor network: SimCLR, MoCo, NNCLR, BarlowTwins, DINO. Results. It was found that the EEGNet architecture used as a auxiliary network, due to its small size, does not give the full potential of contrastive learning algorithms. Therefore, EEGNet was replaced by a 1D CNN architecture with a larger number of parameters, which led to an increase in the quality performance of the models. Conclusion. Although the considered method of transient learning looks promising, the specificity of electroencephalogram signals and problems solved on their basis requires large-scale adaptation of algorithms and contrastive optimisation techniques for effective training of the target task. It is also worth noting the crucial role of the representativeness of the data set for training the donor network, since it is the completeness of real observations that increases the effectiveness of augmentation, which leads to an increase in the number of “useful” features in the latent space of the network and the best conditions for transfer learning in the target task. If we talk about the diagnosis of depression, the data should maximally represent examples of electroencephalograms of depressed patients. |
| format | Article |
| id | doaj-art-b65bf57802194df59c7151085774af6b |
| institution | Kabale University |
| issn | 0869-6632 2542-1905 |
| language | English |
| publishDate | 2025-01-01 |
| publisher | Saratov State University |
| record_format | Article |
| series | Известия высших учебных заведений: Прикладная нелинейная динамика |
| spelling | doaj-art-b65bf57802194df59c7151085774af6b2025-01-31T04:16:45ZengSaratov State UniversityИзвестия высших учебных заведений: Прикладная нелинейная динамика0869-66322542-19052025-01-0133110012210.18500/0869-6632-003126Comparative analysis of transfer learning performance on generalised EEG data for use in a depression diagnosis taskShusharina, Natalia Nikolaevna0Immanuel Kant Baltic Federal University, 14 A. Nevskogo ul., Kaliningrad, 236041The purpose of this work was to analyse the performance of different deep learning methods in the task of depression diagnosis based on bioelectrical brain activity data. In particular, to study the potential of transfer learning using an artificial neural network trained on a significant amount of “generalised” electroencephalography data in the task of diagnosing depression from non-invasive electroencephalogram signals. Methods. Deep learning approaches such as transfer learning and contrastive learning were used in the present study. Artificial neural networks were trained on the public HBN EO/EC task dataset containing recordings of electroencephalogram signals. The 1D CNN and EEGNet architectures were used as auxiliary artificial networks for transfer learning. In order to test the quality of contrastive learning, the dataset was augmented and the following algorithms were selected as the donor network: SimCLR, MoCo, NNCLR, BarlowTwins, DINO. Results. It was found that the EEGNet architecture used as a auxiliary network, due to its small size, does not give the full potential of contrastive learning algorithms. Therefore, EEGNet was replaced by a 1D CNN architecture with a larger number of parameters, which led to an increase in the quality performance of the models. Conclusion. Although the considered method of transient learning looks promising, the specificity of electroencephalogram signals and problems solved on their basis requires large-scale adaptation of algorithms and contrastive optimisation techniques for effective training of the target task. It is also worth noting the crucial role of the representativeness of the data set for training the donor network, since it is the completeness of real observations that increases the effectiveness of augmentation, which leads to an increase in the number of “useful” features in the latent space of the network and the best conditions for transfer learning in the target task. If we talk about the diagnosis of depression, the data should maximally represent examples of electroencephalograms of depressed patients. https://andjournal.sgu.ru/sites/andjournal.sgu.ru/files/text-pdf/2025/01/and_2025-1_shusharina_100-122.pdfdiagnosisdepressionelectroencephalographyartificial neural networktransfer learningcontrastive learning |
| spellingShingle | Shusharina, Natalia Nikolaevna Comparative analysis of transfer learning performance on generalised EEG data for use in a depression diagnosis task Известия высших учебных заведений: Прикладная нелинейная динамика diagnosis depression electroencephalography artificial neural network transfer learning contrastive learning |
| title | Comparative analysis of transfer learning performance on generalised EEG data for use in a depression diagnosis task |
| title_full | Comparative analysis of transfer learning performance on generalised EEG data for use in a depression diagnosis task |
| title_fullStr | Comparative analysis of transfer learning performance on generalised EEG data for use in a depression diagnosis task |
| title_full_unstemmed | Comparative analysis of transfer learning performance on generalised EEG data for use in a depression diagnosis task |
| title_short | Comparative analysis of transfer learning performance on generalised EEG data for use in a depression diagnosis task |
| title_sort | comparative analysis of transfer learning performance on generalised eeg data for use in a depression diagnosis task |
| topic | diagnosis depression electroencephalography artificial neural network transfer learning contrastive learning |
| url | https://andjournal.sgu.ru/sites/andjournal.sgu.ru/files/text-pdf/2025/01/and_2025-1_shusharina_100-122.pdf |
| work_keys_str_mv | AT shusharinanatalianikolaevna comparativeanalysisoftransferlearningperformanceongeneralisedeegdataforuseinadepressiondiagnosistask |