arXiv Machine Learning

Discovery and inference beyond linearity for epidemiological data by integrating Bayesian regression, tree ensembles and Shapley values

arXiv:2505. 00571v3 Announce Type: replace-cross Abstract: Machine Learning (ML) is gaining popularity in epidemiology and healthcare studies for hypothesis-free discovery of risk and protective factors.

arXiv Statistics ML
Oct 26

Bayesian Propensity Score Predictive Inference for Extending Trial Findings to Target Populations

The paper introduces Bayesian Propensity Score Predictive Inference (PSPI), an interaction‑based outcome modeling framework that uses Bayesian additive regression trees and natural cubic splines of the sampling propensity score to extend randomized trial results to broader target populations. PSPI includes variants that apply splines to the treatment effect component alone or to both prognostic and treatment effect components, enabling separate regularization of selection‑related variation and explicit extrapolation beyond boundary knots. Simulation studies demonstrate that spline‑based PSPI models outperform competing methods, achieving lower root mean squared errors and near‑nominal coverage, especially under substantial covariate shift, and in the CODIACS Vanguard trial application PSPI estimates larger intervention benefits than the unadjusted trial estimate.

By Jungang Zou, Joseph E. Schwartz, Roderick J. A. Little, Nathalie Moise, Qixuan Chen
arXiv Machine Learning
Aug 11

Demystifying Prediction Powered Inference

arXiv:2601. 20819v2 Announce Type: replace-cross Abstract: Machine learning predictions are increasingly used to supplement incomplete or costly-to-measure outcomes in fields such as biomedical research, environmental science, and social science.

By Yilin Song, Dan M. Kluger, Harsh Parikh, Tian Gu
arXiv AI
Jul 2

LLM-Guided ODE Discovery and Parameter Inference from Small-Cohort Aggregate Data

arXiv:2607. 00733v1 Announce Type: cross Abstract: Mechanistic modeling via ordinary differential equations (ODEs) provides interpretable descriptions of complex dynamics and enables inference of underlying mechanisms, which is particularly valuable in clinical settings.

By Hanning Yang, Meropi Karakioulaki, Lennart Purucker, Tim Litwin, Cristina Has, Moritz Hess
arXiv Statistics ML
Oct 19

Empowering Bayes with sources of uncertainty: from confidence modelling to sparse estimation

The paper introduces Sources of Uncertainty (SoU) inference, a framework that extends Bayesian inference by allowing heterogeneous confidence to be encoded across model components. This selective treatment controls the uncertainty retained in each component and its propagation, yielding Empirical Bayes and profiling-type inference as special cases. The authors also present a general method for inducing sparsity, demonstrating its application in linear and logistic regression and in Bayesian neural networks for self‑pruning.

By Rafael Mouallem Rosa, Julyan Arbel, Hien Duy Nguyen
arXiv Machine Learning
Oct 16

Clinically Interpretable Biological Age Estimation Using CatBoost and SHAP with Continuous and Categorical Features from NHANES

The study presents a clinically interpretable machine‑learning framework that estimates biological age (BA) using a CatBoost regression model trained on 44 diverse predictors from NHANES data. SHAP values are leveraged not only for visual interpretation but also through an automated quantitative procedure that identifies feature‑value ranges linked to the lowest and highest BA contributions, including zero‑crossing thresholds. The resulting BA estimates correlate strongly with chronological age and highlight key predictors such as blood urea nitrogen, serum glucose, and vigorous activity, offering transparent, reproducible insights for precision health research.

By Yanni Sun, Zheng Qu, Kevin Ren, Zhanglin Lin Cui