Explainable AI to predict a complex multifactorial outcome, childhood obesity: Application to clinical epidemiology
Chen, F.; Melton, P.; Vinsen, K.; Mori, T. A.; Beilin, L.; Huang, R.-C.
Show abstract
BackgroundChildhood obesity, driven by genetic and epidemiological factors, poses significant health risks, yet traditional machine learning models lack interpretability for clinical use. ObjectiveThis study aims to apply Kolmogorov-Arnold Networks (KAN), an explainable machine learning model, to predict body mass index (BMI) at age 8 as an indicator of obesity risk and to develop a publicly accessible prediction tool. MethodsWe utilized the Raine Study Gen2 cohort (n=2,868) to train KAN and traditional models (such as Random Forest, Gradient Boosting, Lasso, and Multi-Layer Perceptron) using perinatal, early-life, and polygenic risk score (PGS) data collected before age 5. Feature importance was analyzed across all the models. A publicly accessible online calculator was developed for practical use. ResultsKAN achieved an R2 of 0.81, outperforming traditional models. Key predictors included Year 5 BMI z-score, mid-arm circumference, occupation of mother, and PGS. The online calculator supports predictions without PGS, maintaining an R2 of 0.81. ConclusionsKANs transparent formulas enhance interpretability, offering a practical approach to predicting childhood obesity. The freely accessible tool enables clinicians to implement personalized prevention strategies, advancing precision medicine. Graphical abstract O_FIG O_LINKSMALLFIG WIDTH=186 HEIGHT=200 SRC="FIGDIR/small/25330041v3_ufig1.gif" ALT="Figure 1"> View larger version (59K): org.highwire.dtl.DTLVardef@1551d55org.highwire.dtl.DTLVardef@f8e337org.highwire.dtl.DTLVardef@d5447org.highwire.dtl.DTLVardef@11841f8_HPS_FORMAT_FIGEXP M_FIG KAN model predicts childhood obesity (BMI at age 8), showcasing key features, top performance, and accurate formularised results with epidemiological and genetic factors. Online calculator is available at https://bmi-y8-calc.onrender.com/. C_FIG
Matching journals
The top 4 journals account for 50% of the predicted probability mass.