An Ensemble of Light Gradient Boosting Machine and Adaptive Boosting for Prediction of Type-2 Diabetes

• Published in: International journal of computational intelligence systems 2023-02, Vol.16 (1), p.1-20, Article 14
• Main Authors: Sai, M. Jishnu, Chettri, Pratiksha, Panigrahi, Ranjit, Garg, Amik, Bhoi, Akash Kumar, Barsocchi, Paolo
• Format: Article
• Language: English
• Subjects: Artificial Intelligence; Classifier ensemble; Computational Intelligence; Control; Diabetes detection; Engineering; k-NN; Light GBM (Gradient Boosting Machine); Mathematical Logic and Foundations; Mechatronics; Naive Bayes (Gaussian); Random forest; Research Article; Robotics
• ISSN: 1875-6883; 1875-6883
• DOI: 10.1007/s44196-023-00184-y

Machine learning helps construct predictive models in clinical data analysis, predicting stock prices, picture recognition, financial modelling, disease prediction, and diagnostics. This paper proposes machine learning ensemble algorithms to forecast diabetes. The ensemble combines k-NN, Naive Bayes (Gaussian), Random Forest (RF), Adaboost, and a recently designed Light Gradient Boosting Machine. The proposed ensembles inherit detection ability of LightGBM to boost accuracy. Under fivefold cross-validation, the proposed ensemble models perform better than other recent models. The k -NN, Adaboost, and LightGBM jointly achieve 90.76% detection accuracy. The receiver operating curve analysis shows that k -NN, RF, and LightGBM successfully solve class imbalance issue of the underlying dataset.