XGBoost-Based Framework for Smoking-Induced Noncommunicable Disease Prediction

• Main Authors: Khishigsuren Davagdorj, Van Huy Pham, Nipon Theera-Umpon, Keun Ho Ryu
• Format: Article
• Language: English
• Published: MDPI AG 2020-09-01
• Series: International Journal of Environmental Research and Public Health
• Subjects: smoking; noncommunicable disease; feature selection; extreme gradient boosting
• Online Access: https://www.mdpi.com/1660-4601/17/18/6513

Smoking-induced noncommunicable diseases (SiNCDs) have become a significant threat to public health and cause of death globally. In the last decade, numerous studies have been proposed using artificial intelligence techniques to predict the risk of developing SiNCDs. However, determining the most significant features and developing interpretable models are rather challenging in such systems. In this study, we propose an efficient extreme gradient boosting (XGBoost) based framework incorporated with the hybrid feature selection (HFS) method for SiNCDs prediction among the general population in South Korea and the United States. Initially, HFS is performed in three stages: (I) significant features are selected by t-test and chi-square test; (II) multicollinearity analysis serves to obtain dissimilar features; (III) final selection of best representative features is done based on least absolute shrinkage and selection operator (LASSO). Then, selected features are fed into the XGBoost predictive model. The experimental results show that our proposed model outperforms several existing baseline models. In addition, the proposed model also provides important features in order to enhance the interpretability of the SiNCDs prediction model. Consequently, the XGBoost based framework is expected to contribute for early diagnosis and prevention of the SiNCDs in public health concerns.