Predictive modeling for early detection of biliary atresia in infants with cholestasis: Insights from a machine learning study

Abstract

Cholestasis, characterized by the obstruction of bile flow, poses a significant concern in neonates and infants. It can result in jaundice, inadequate weight gain, and liver dysfunction. However, distinguishing between biliary atresia (BA) and non-biliary atresia in these young patients presenting with cholestasis poses a formidable challenge, given the similarity in their clinical manifestations. To this end, our study endeavors to construct a screening model aimed at prognosticating outcomes in cases of BA. Within this study, we introduce a wrapper feature selection model denoted as bWFMVO-SVM-FS, which amalgamates the water flow-based multi-verse optimizer (WFMVO) and support vector machine (SVM) technology. Initially, WFMVO is benchmarked against eleven state-of-the-art algorithms, with its efficiency in searching for optimized feature subsets within the model validated on IEEE CEC 2017 and IEEE CEC 2022 benchmark functions. Subsequently, the developed bWFMVO-SVM-FS model is employed to analyze a cohort of 870 consecutively registered cases of neonates and infants with cholestasis (diagnosed as either BA or non-BA) from Xinhua Hospital and Shanghai Children's Hospital, both affiliated with Shanghai Jiao Tong University. The results underscore the remarkable predictive capacity of the model, achieving an accuracy of 92.639 % and specificity of 88.865 %. Gamma-glutamyl transferase, triangular cord sign, weight, abnormal gallbladder, and stool color emerge as highly correlated with early symptoms in BA infants. Furthermore, leveraging these five significant features enhances the interpretability of the machine learning model's performance outcomes for medical professionals, thereby facilitating more effective clinical decision-making.

Keywords: Biliary atresia; Cholestasis; Feature selection; Infants; Machine learning; Multi-verse optimizer; Support vector machine.