返回
Improving depression prediction using a novel feature selection algorithm coupled with context-aware analysis
DOI:10.1016/j.jad.2021.09.001.png)
摘要
En 中文
Background: Developing machine learning based depression prediction method with information from long-term recordings is important and challenging to clinical diagnosis of depression. Methods: We developed a novel two-stage feature selection algorithm conducted on the high-dimensional (over thirty thousand) features constructed by a context-aware analysis on the data set of DAIC-WOZ, including audio, video, and semantic features. The prediction performance was compared with seven reference models. The preferred topics and feature categories related to the retained features were also analyzed respectively. Results: Parsimonious subsets (tens of features) were selected by the proposed method in each case of prediction. We obtained the best performance in depression classification with F1-score as 0.96 (0.67), Precision as 1.00 (0.63), and Recall as 0.92 (0.71) on the development set (test set). We also achieved promising results in depression severity estimation with RMSE as 4.43 (5.11) and MAE as 3.22 (3.98), having a marginal difference with the best reference model (random forest with 'Selected-Text' features). Five most important topics related to depression were revealed. The audio features were predominant to the other feature categories in depression classification while the contributions of the three feature categories to severity estimation were almost equal. Limitations: More depression samples in the database we used should be further included. The second stage of feature selection is relatively time-consuming. Conclusion: This pipeline of depression recognition as well as the preferred topics and feature categories are expected to be useful in supporting the diagnosis of psychological distress conditions.
Keyword:
Depression prediction
Feature selection
Context-aware analysis
Maximal information coefficient
Support vector machine
AI总结
对已上传原文的论文进行重点信息的提取,主要内容包括:简要概述、研究摘要、背景介绍、关键亮点、图文解析、展望与总结。
期刊
IF:
4.9
论文数:
2.1W
被引数:
6.9W
机构
引用论文
Topic Modeling Analysis of Social Enterprises: Twitter Evidence社会企业主题建模分析: Twitter证据
SUSTAINABILITY
IF3.3
Fully component selection: An efficient combination of feature selection and principal component analysis to increase model performance完全组件选择: 特征选择和主成分分析的有效组合,以提高模型性能
The State of US Health, 1990-2010 Burden of Diseases, Injuries, and Risk Factors美国健康状况,1990 2010的疾病负担,伤害和危险因素
A Novel Method of Nonlinear Rapid Feature Selection for High Dimensional Data and Its Application in Peptide QSAR Modeling Based on Support Vector Machine
analytica chimica acta
IF13.5

