Abstract
Major Depressive Disorder (MDD) is the most common disorder worldwide. Accurate detection of depression is a challenging problem. Machine learning-based automated depression detection provides useful assistance to the clinicians for effective depression diagnosis. One of the most fundamental steps in any automated detection is feature selection and investigation of the most relevant features. Studies show that regional volumes of the brain are affected in response to depression. Regional volumes are considered as features. The gray matter volumes’ correlation with depression and the most effective gray volumes for depression detection is investigated in this study. Various feature selection techniques are studied, along with the investigation on the importance of resampling to handle imbalanced data, which is typically the case for depression detection, as the number of depressed instances is commonly a fraction of the entire data size. Experimental results using Random Forests (RF) and support vector machines (SVM) with a Gaussian kernel (RBF) as classifiers show that feature selection followed by data resampling gives superior performance measured by Area Under the ROC Curve (AUC) as well as prediction accuracy, and RF outperforms SVM for the depression detection task.
Keywords
This is a preview of subscription content, log in via an institution.
Buying options
Tax calculation will be finalised at checkout
Purchases are for personal use only
Learn about institutional subscriptionsReferences
Gong, Q., et al.: Prognostic prediction of therapeutic response in depression using high-field MR imaging. NeuroImage 55, 1497–1503 (2011)
Kipli, K., Kouzani, A.: Degree of contribution (DoC) feature selection algorithm for structural brain MRI volumetric features in depression detection. Int. J. Comput. Assist. Radiol. Surg. 10, 1003–1016 (2014)
Patel, M., Khalaf, A., Aizenstein, H.: Studying depression using imaging and machine learning methods. NeuroImage: Clin. 10, 115–123 (2016)
Costafreda, S., Chu, C., Ashburner, J., Fu, C.: Prognostic and diagnostic potential of the structural neuroanatomy of depression. PLoS ONE 4, e6353 (2009)
Fcon_1000.projects.nitrc.org. Accessed 28 Mar 2018
Mwangi, B., Matthews, K., Steele, J.: Prediction of illness severity in patients with major depression using structural MR brain scans. J. Magn. Reson. Imaging 35, 64–71 (2011)
Mwangi, B., Tian, T., Soares, J.: A review of feature reduction techniques in neuroimaging. Neuroinformatics 12, 229–244 (2013)
Hira, Z., Gillies, D.: A review of feature selection and feature extraction methods applied on microarray data. Adv. Bioinf. 2015, 1–13 (2015)
Feature Selection/Extraction Dimensionality Reduction. http://vision.psych.umn.edu/users/schrater/schrater_lab/courses/PattRecog09/Lec17PattRec09.pdf. Accessed 28 Mar 2018
Hemphill, E., Lindsay, J., Lee, C., Măndoiu, I., Nelson, C.: Feature selection and classifier performance on diverse bio- logical datasets. BMC Bioinf. 15, S4 (2014)
Geng, X., Xu, J.: Application of autoencoder in depression diagnosis. In: 3rd International Conference on Computer Science and Mechanical Automation, CSMA (2017)
Bosch, N., Paquette, L.: Unsupervised Deep Autoencoders for Feature Extraction with Educational Data (2018)
Unbalanced data and cross-validation|Kaggle. https://www.kaggle.com/questions-and-answers/27589. Accessed 28 Mar 2018
Scikit-learn-contrib/imbalanced-learn. https://github.com/scikit-learn-contrib/imbalanced-learn. Accessed 28 Mar 2018
Malone, I., et al.: Accurate automatic estimation of total intracranial volume: a nuisance variable with less nuisance. NeuroImaging 104, 366–372 (2015)
Amami, R., Ayed, D.B., Ellouze, N.: Practical selection of SVM supervised parameters with different feature representations for vowel recognition. arXiv preprint arXiv:1507.06020 (2015)
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2018 Springer Nature Switzerland AG
About this paper
Cite this paper
Mousavian, M., Chen, J., Greening, S. (2018). Feature Selection and Imbalanced Data Handling for Depression Detection. In: Wang, S., et al. Brain Informatics. BI 2018. Lecture Notes in Computer Science(), vol 11309. Springer, Cham. https://doi.org/10.1007/978-3-030-05587-5_33
Download citation
DOI: https://doi.org/10.1007/978-3-030-05587-5_33
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-030-05586-8
Online ISBN: 978-3-030-05587-5
eBook Packages: Computer ScienceComputer Science (R0)