English

Bayesian Hybrid Machine Learning of Gallstone Risk

Applications 2025-06-18 v1 Machine Learning

Abstract

Gallstone disease is a complex, multifactorial condition with significant global health burdens. Identifying underlying risk factors and their interactions is crucial for early diagnosis, targeted prevention, and effective clinical management. Although logistic regression remains a standard tool for assessing associations between predictors and gallstone status, it often underperforms in high-dimensional settings and may fail to capture intricate relationships among variables. To address these limitations, we propose a hybrid machine learning framework that integrates robust variable selection with advanced interaction detection. Specifically, Adaptive LASSO is employed to identify a sparse and interpretable subset of influential features, followed by Bayesian Additive Regression Trees (BART) to model nonlinear effects and uncover key interactions. Selected interactions are further characterized by physiological knowledge through differential equation-informed interaction terms, grounding the model in biologically plausible mechanisms. The insights gained from these steps are then integrated into a final logistic regression model within a Bayesian framework, providing a balance between predictive accuracy and clinical interpretability. This proposed framework not only enhances prediction but also yields actionable insights, offering a valuable support tool for medical research and decision-making.

Keywords

Cite

@article{arxiv.2506.14561,
  title  = {Bayesian Hybrid Machine Learning of Gallstone Risk},
  author = {Chitradipa Chakraborty and Nayana Mukherjee},
  journal= {arXiv preprint arXiv:2506.14561},
  year   = {2025}
}

Comments

25 pages, 5 figures

R2 v1 2026-07-01T03:21:57.772Z