Shaha, Tumpa Rani and Begum, Momotaz and Uddin, Jia and Yélamos Torres, Vanessa and Alemany Iturriaga, Josep and Ashraf, Imran and Samad, Md. Abdus UNSPECIFIED, UNSPECIFIED, UNSPECIFIED, vanessa.yelamos@funiber.org, josep.alemany@uneatlantico.es, UNSPECIFIED, UNSPECIFIED (2024) Feature group partitioning: an approach for depression severity prediction with class balancing using machine learning algorithms. BMC Medical Research Methodology, 24 (1). ISSN 1471-2288
|
Text
s12874-024-02249-8.pdf Available under License Creative Commons Attribution. Download (2MB) | Preview |
Abstract
In contemporary society, depression has emerged as a prominent mental disorder that exhibits exponential growth and exerts a substantial influence on premature mortality. Although numerous research applied machine learning methods to forecast signs of depression. Nevertheless, only a limited number of research have taken into account the severity level as a multiclass variable. Besides, maintaining the equality of data distribution among all the classes rarely happens in practical communities. So, the inevitable class imbalance for multiple variables is considered a substantial challenge in this domain. Furthermore, this research emphasizes the significance of addressing class imbalance issues in the context of multiple classes. We introduced a new approach Feature group partitioning (FGP) in the data preprocessing phase which effectively reduces the dimensionality of features to a minimum. This study utilized synthetic oversampling techniques, specifically Synthetic Minority Over-sampling Technique (SMOTE) and Adaptive Synthetic (ADASYN), for class balancing. The dataset used in this research was collected from university students by administering the Burn Depression Checklist (BDC). For methodological modifications, we implemented heterogeneous ensemble learning stacking, homogeneous ensemble bagging, and five distinct supervised machine learning algorithms. The issue of overfitting was mitigated by evaluating the accuracy of the training, validation, and testing datasets. To justify the effectiveness of the prediction models, balanced accuracy, sensitivity, specificity, precision, and f1-score indices are used. Overall, comprehensive analysis demonstrates the discrimination between the Conventional Depression Screening (CDS) and FGP approach. In summary, the results show that the stacking classifier for FGP with SMOTE approach yields the highest balanced accuracy, with a rate of 92.81%. The empirical evidence has demonstrated that the FGP approach, when combined with the SMOTE, able to produce better performance in predicting the severity of depression. Most importantly the optimization of the training time of the FGP approach for all of the classifiers is a significant achievement of this research.
| Item Type: | Article |
|---|---|
| Uncontrolled Keywords: | Machine learning; Depression prediction; Class balancing; Oversampling; SMOTE; ADASYN; Stratified cross validation; Burn depression checklist; Feature group partitioning |
| Subjects: | Subjects > Engineering |
| Divisions: | Europe University of Atlantic > Research > Scientific Production Ibero-american International University > Research > Scientific Production Ibero-american International University > Research > Scientific Production Universidad Internacional do Cuanza > Research > Scientific Production University of La Romana > Research > Scientific Production |
| Depositing User: | Sr Bibliotecario |
| Date Deposited: | 17 Jun 2024 15:52 |
| Last Modified: | 17 Jun 2024 15:52 |
| URI: | http://repositorio.funiber.org/id/eprint/12751 |
Actions (login required)
 |
View Item |
