Optimizing Depression Prediction in Older Adults : A Comparative Study of Feature Extraction and Machine Learning Models

Depression emerged as a major public health concern in older adults, and timely prediction of depression has become a difficult problem in medical informatics. The latest studies have attentiveed on feature transformation and selection for better depression prediction. In this study, we assess the performance of various feature extraction algorithms, including principal component analysis (PCA), independent component analysis (ICA), locally linear Embedding (LLE), and t-distributed stochastic neighbor embedding (TSNE). These algorithms are combined with machine learning (ML) classifier algorithms such as Gaussian Naive Bayes (GNB), Logistic Regression (LR), K-nearest-neighbor (KNN), and Decision Tree (DT) to enhance depression prediction. In total, sixteen automated integrated systems are constructed based on the above-mentioned feature extraction methods and ML classifiers. The performance of all of these integrated models is assessed using data from the Swedish National Study on Aging and Care (SNAC). According to the experimental results, the PCA algorithm combined with the Logistic Regression (LR) model provides 89.04% depression classification accuracy. As a result, it is demonstrated that the PCA is a more suitable feature extraction method for depression data than ICA, LLE, and TSNE. © 2024 IEEE.
SNAC