Back

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.

2025-06-23 epidemiology
10.1101/2025.06.21.25330041 medRxiv
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.

1
International Journal of Epidemiology
74 papers in training set
Top 0.1%
28.8%
2
International Journal of Obesity
25 papers in training set
Top 0.1%
12.8%
3
Scientific Reports
3102 papers in training set
Top 21%
5.0%
4
PLOS ONE
4510 papers in training set
Top 37%
3.7%
50% of probability mass above
5
BMC Medicine
163 papers in training set
Top 2%
3.0%
6
American Journal of Epidemiology
57 papers in training set
Top 0.4%
2.8%
7
PLOS Medicine
98 papers in training set
Top 1%
2.7%
8
eBioMedicine
130 papers in training set
Top 0.7%
2.2%
9
BMC Public Health
147 papers in training set
Top 3%
1.9%
10
BMC Medical Research Methodology
43 papers in training set
Top 0.7%
1.5%
11
Metabolism
14 papers in training set
Top 0.2%
1.4%
12
Journal of Clinical Medicine
91 papers in training set
Top 4%
1.3%
13
International Journal of Environmental Research and Public Health
124 papers in training set
Top 5%
1.3%
14
BMJ Open
554 papers in training set
Top 11%
0.9%
15
Frontiers in Public Health
140 papers in training set
Top 7%
0.9%
16
BMC Bioinformatics
383 papers in training set
Top 6%
0.9%
17
Journal of Affective Disorders
81 papers in training set
Top 1%
0.8%
18
Diabetes, Obesity and Metabolism
17 papers in training set
Top 0.5%
0.8%
19
International Journal of Medical Informatics
25 papers in training set
Top 2%
0.8%
20
Nature Human Behaviour
85 papers in training set
Top 4%
0.8%
21
PeerJ
261 papers in training set
Top 15%
0.7%
22
Frontiers in Pediatrics
29 papers in training set
Top 0.9%
0.7%
23
The Journal of Pediatrics
15 papers in training set
Top 0.7%
0.7%
24
Genome Medicine
154 papers in training set
Top 9%
0.7%
25
JAMA Network Open
127 papers in training set
Top 5%
0.7%
26
European Journal of Epidemiology
40 papers in training set
Top 0.9%
0.7%
27
Obesity
19 papers in training set
Top 0.8%
0.5%
28
JMIR Research Protocols
18 papers in training set
Top 2%
0.5%