Advanced machine learning model for predicting Crohn's disease with enhanced ant colony optimization

Abstract

Changes in human lifestyles have led to a dramatic increase in the incidence of Crohn's disease worldwide. Predicting the activity and remission of Crohn's disease has become an urgent research problem. In addition, the influence of each attribute in the test sample on the prediction results and the interpretability of the model still deserves further investigation. Therefore, in this paper, we proposed a wrapper feature selection classification model based on a combination of the improved ant colony optimization algorithm and the kernel extreme learning machine, called bIACOR-KELM-FS. IACOR introduces an evasive strategy and astrophysics strategy to balance the exploration and exploitation phases of the algorithm and enhance its optimization capabilities. The optimization capability of the proposed IACOR was validated on the IEEE CEC2017 benchmark test function. And the prediction was performed on Crohn's disease dataset. The results of the quantitative analysis showed that the prediction accuracy of bIACOR-KELM-FS for predicting the activity and remission of Crohn's disease reached 98.98%. The analysis of important attributes improved the interpretability of the model and provided a reference for the diagnosis of Crohn's disease. Therefore, the proposed model is considered a promising adjunctive diagnostic method for Crohn's disease.

Keywords: Ant colony optimization; Crohn's disease; Feature selection; Kernel extreme learning machine; Prediction.