We address the interpretability of the machine learning algorithm in the context of the relevant problem of discriminating between patients with major depressive disorder (MDD) and healthy controls using functional networks derived from resting-state functional magnetic resonance imaging data. We applied linear discriminant analysis (LDA) to the data from 35 MDD patients and 50 healthy controls to discriminate between the two groups utilizing functional networks’ global measures as the features. We proposed the combined approach for feature selection based on statistical methods and the wrapper-type algorithm. This approach revealed that the groups are indistinguishable in the univariate feature space but become distinguishable in a three-dimensional feature space formed by the identified most important features: mean node strength, clustering coefficient, and the number of edges. LDA achieves the highest accuracy when considering the network with all connections or only the strongest ones. Our approach allowed us to analyze the separability of classes in the multidimensional feature space, which is critical for interpreting the results of machine learning models. We demonstrated that the parametric planes of the control and MDD groups rotate in the feature space with increasing the thresholding parameter and that their intersection increases with approaching the threshold of 0.45, for which classification accuracy is minimal. Overall, the combined approach for feature selection provides an effective and interpretable scenario for discriminating between MDD patients and healthy controls using measures of functional connectivity networks. This approach can be applied to other machine learning tasks to achieve high accuracy while ensuring the interpretability of the results.
Skip Nav Destination
Article navigation
June 2023
Research Article|
June 15 2023
Toward interpretability of machine learning methods for the classification of patients with major depressive disorder based on functional network measures
Andrey V. Andreev
;
Andrey V. Andreev
(Conceptualization, Investigation, Software, Validation, Visualization, Writing – original draft)
1
Baltic Center for Neurotechnology and Artificial Intelligence, Immanuel Kant Baltic Federal University
, 14, A. Nevskogo str., Kaliningrad 236016, Russia
Search for other works by this author on:
Semen A. Kurkin
;
Semen A. Kurkin
(Conceptualization, Formal analysis, Investigation, Methodology, Project administration, Validation, Writing – review & editing)
1
Baltic Center for Neurotechnology and Artificial Intelligence, Immanuel Kant Baltic Federal University
, 14, A. Nevskogo str., Kaliningrad 236016, Russia
Search for other works by this author on:
Drozdstoy Stoyanov
;
Drozdstoy Stoyanov
(Data curation, Investigation, Writing – review & editing)
2
Department of Psychiatry and Medical Psychology, Research Institute, Medical University Plovdiv
, 15A Vassil Aprilov Blvd., Plovdiv 4002, Bulgaria
Search for other works by this author on:
Artem A. Badarin
;
Artem A. Badarin
(Formal analysis, Software, Validation)
1
Baltic Center for Neurotechnology and Artificial Intelligence, Immanuel Kant Baltic Federal University
, 14, A. Nevskogo str., Kaliningrad 236016, Russia
Search for other works by this author on:
Rossitsa Paunova
;
Rossitsa Paunova
(Data curation, Investigation, Resources, Validation)
2
Department of Psychiatry and Medical Psychology, Research Institute, Medical University Plovdiv
, 15A Vassil Aprilov Blvd., Plovdiv 4002, Bulgaria
Search for other works by this author on:
Alexander E. Hramov
Alexander E. Hramov
a)
(Conceptualization, Formal analysis, Funding acquisition, Methodology, Supervision, Writing – review & editing)
1
Baltic Center for Neurotechnology and Artificial Intelligence, Immanuel Kant Baltic Federal University
, 14, A. Nevskogo str., Kaliningrad 236016, Russia
a)Author to whom correspondence should be addressed: aekhramov@kantiana.ru
Search for other works by this author on:
a)Author to whom correspondence should be addressed: aekhramov@kantiana.ru
Note: This paper is part of the Focus Issue on Nonlinear dynamics, synchronization and networks: Dedicated to Juergen Kurths' 70th birthday.
Chaos 33, 063140 (2023)
Article history
Received:
April 21 2023
Accepted:
May 24 2023
Citation
Andrey V. Andreev, Semen A. Kurkin, Drozdstoy Stoyanov, Artem A. Badarin, Rossitsa Paunova, Alexander E. Hramov; Toward interpretability of machine learning methods for the classification of patients with major depressive disorder based on functional network measures. Chaos 1 June 2023; 33 (6): 063140. https://doi.org/10.1063/5.0155567
Download citation file:
Sign in
Don't already have an account? Register
Sign In
You could not be signed in. Please check your credentials and make sure you have an active account and try again.
Pay-Per-View Access
$40.00
Citing articles via
Sex, ducks, and rock “n” roll: Mathematical model of sexual response
K. B. Blyuss, Y. N. Kyrychko
Nonlinear comparative analysis of Greenland and Antarctica ice cores data
Berenice Rojo-Garibaldi, Alberto Isaac Aguilar-Hernández, et al.
Focus on the disruption of networks and system dynamics
Peng Ji, Jan Nagler, et al.