BMC Public Health (Jan 2024)
Machine-learning model predicting quality of life using multifaceted lifestyles in middle-aged South Korean adults: a cross-sectional study
Abstract
Abstract Background In the context of population aging, advances in healthcare technology, and growing interest in healthy aging and higher quality of life (QOL), have gained central focus in public health, particularly among middle-aged adults. Methods This study presented an optimal prediction model for QOL among middle-aged South Korean adults (N = 4,048; aged 30–55 years) using a machine-learning technique. Community-based South Korean population data were sampled through multistage stratified cluster sampling. Twenty-one variables related to individual factors and various lifestyle patterns were surveyed. QOL was assessed using the Short Form Health Survey (SF-12) and categorized into total QOL, physical component score (PCS), and mental component score (MCS). Seven machine-learning algorithms were used to predict QOL: decision tree, Gaussian Naïve Bayes, k-nearest neighbor, logistic regression, extreme gradient boosting, random forest, and support vector machine. Data imbalance was resolved with the synthetic minority oversampling technique (SMOTE). Random forest was used to compare feature importance and visualize the importance of each variable. Results For predicting QOL deterioration, the random forest method showed the highest performance. The random forest algorithm using SMOTE showed the highest area under the receiver operating characteristic (AUC) for total QOL (0.822), PCS (0.770), and MCS (0.786). Applying the data, SMOTE enhanced model performance by up to 0.111 AUC. Although feature importance differed across the three QOL indices, stress and sleep quality were identified as the most potent predictors of QOL. Random forest generated the most accurate prediction of QOL among middle-aged adults; the model showed that stress and sleep quality management were essential for improving QOL. Conclusion The results highlighted the need to develop a health management program for middle-aged adults that enables multidisciplinary management of QOL.
Keywords