Study on Rheumatoid Arthritis Prediction Model With Machine Learning and Ensemble Learning Methods, and Demographic, Clinical, Laboratory Characteristics of These Patients
Abstract
The aim of this study is to develop a diagnostic and predictive model for rheumatoid arthritis (RA) with machine learning and ensemble learning methods using data from patients who applied to the rheumatology outpatient clinic with complaints of pain in the hand joints and to review the demographic, clinical and laboratory results of these RA patients. This study was conducted on demographic properties, clinical findings, and laboratory test results of a total of 421 patients, 260 with RA and 161 non-RA. Machine learning and ensemble learning algorithms were used for prediction. In this study, ensemble learning algorithms showed better performance than machine learning algorithms. With the Random Forest Classifier model 88% accuracy (95% confidence interval (CI), 81% - 94% ), 94% sensitivity (95% CI, 88% - 99%), 78% specificity (95% CI, 69% - 86%), 87% positive predictive value (PPV) (95% CI, 79% - 94%), 90% F1 score (95% CI, 84% - 96%), and 0.92 AUC (area under the curve) results were obtained. All three developed ensemble learning models showed similar performance in statistical metrics. These developed models can be improved with new multicenter studies with more patients and can be used in daily practice.
Keywords
Rheumatoid arthritis, Machine learning, Ensemble learning, Logistic regression, Support vector machines
Supporting Institution
Project Number
Ethical Statement
Thanks
References
- Abdelhafiz, D., Baker, T., Glascow, D., & Abdelhafiz, A. (2023). Biomarkers for the diagnosis and treatment of rheumatoid arthritis – a systematic review. Postgraduate Medicine, 135(3), 214–223. https://doi.org/10.1080/00325481.2022.2052626
- Ali, M. M., Paul, B. K., Ahmed, K., Bui, F. M., Quinn, J. M. W., & Moni, M. A. (2021). Heart disease prediction using supervised machine learning algorithms: Performance analysis and comparison. Computers in Biology and Medicine, 136, Article 104672. https://doi.org/10.1016/j.compbiomed.2021.104672
- Almutairi, K., Nossent, J., Preen, D., Keen, H., & Inderjeeth, C. (2021). The global prevalence of rheumatoid arthritis: A meta-analysis based on a systematic review. Rheumatology International, 41(5), 863–877. ttps://doi.org/10.1007/s00296-020-04731-0
- Azmi, S. S., & Baliga, S. (2020). An overview of boosting decision tree algorithms utilizing AdaBoost and XGBoost boosting strategies. International Research Journal of Engineering and Technology, 7(5), 6867–6870.
- Azur, M. J., Stuart, E. A., Frangakis, C., & Leaf, P. J. (2011). Multiple imputation by chained equations: What is it and how does it work? International Journal of Methods in Psychiatric Research, 20(1), 40–49. https://doi.org/10.1002/MPR.329
- Carvalho, D. V., Pereira, E. M., & Cardoso, J. S. (2019). Machine learning interpretability: A survey on methods and metrics. Electronics, 8(8), Article 832. https://doi.org/10.3390/electronics8080832
- Castiglioni, I., Rundo, L., Codari, M., Di Leo, G., Salvatore, C., Interlenghi, M., Gallivanone, F., Cozzi, A., D’Amico, N. C., & Sardanelli, F. (2021). AI applications to medical images: From machine learning to deep learning. Physica Medica, 83, 9–24. https://doi.org/10.1016/j.ejmp.2021.02.006
- Conigliaro, P., Chimenti, M. S., Triggianese, P., Sunzini, F., Novelli, L., Perricone, C., & Perricone, R. (2016). Autoantibodies in inflammatory arthritis. Autoimmunity Reviews, 15(7), 673–683. https://doi.org/10.1016/j.autrev.2016.03.003
- Conigliaro, P., D’Antonio, A., Pinto, S., Chimenti, M. S., Triggianese, P., Rotondi, M., & Perricone, R. (2020). Autoimmune thyroid disorders and rheumatoid arthritis: A bidirectional interplay. Autoimmunity Reviews, 19(6), Article 102529. https://doi.org/10.1016/j.autrev.2020.102529
- Dahir, A. M., & Thomsen, S. F. (2018). Comorbidities in vitiligo: Comprehensive review. International Journal of Dermatology, 57(10), 1157–1164. https://doi.org/10.1111/ijd.14055