arXiv Science⌕ Search
Home/Archive/arXiv

arXiv · 2610.27425

A Bayesian machine learning approach to direct and spillover effects in cluster-randomized trials with noncompliance

Saurabh Bhandari·Fan Li·Michael J. Daniels
View original source ↗Read source document ↗Export metadata ↓

Abstract

We develop a Bayesian machine learning approach for estimating direct and spillover effects in cluster-randomized trials with treatment noncompliance. To flexibly model outcomes and treatment receipt while accounting for within-cluster dependence, we combine Bayesian additive regression trees with cluster-level random effects. We augment the outcome regression with a peer-treatment clever covariate, defined as the predicted mean proportion of treated peers from a preliminary receipt model, to mitigate regularization-induced distortion of the effect decomposition. Within a Bayesian g-computation framework, we estimate cluster-average intent-to-treat effects and cluster-average local average treatment effects and examine sensitivity to the cross-world restrictions underlying their decomposition. Simulation studies show reduced bias under model misspecification, while the correctly specified parametric models show expected frequentist properties. We find that the clever covariate improves the decomposition of the total effect into direct and spillover components under moderate or strong spillover, with similar gains in smaller clusters. Its benefits vary with spillover strength, and the $95\%$ credible intervals remain conservative. An application to the Primary School Deworming Project in Kenya suggests reduced infection risk, with a larger estimated contribution from individual treatment receipt than from peer receipt. Overall, our approach provides a flexible Bayesian estimation of direct and spillover effect decomposition in the presence of treatment noncompliance.

Explore related subjects

stat.MEstat.APstat.ML

Keep this discovery

Explore connections, maps & timelines

BibTeXRIS

Saurabh Bhandari, Fan Li, Michael J. Daniels. 2026-10-20. A Bayesian machine learning approach to direct and spillover effects in cluster-randomized trials with noncompliance. https://arxiv.org/abs/2610.27425

Cite the original work for its findings. Save a collection to share your selection of sources.

KEEP EXPLORING

Related papers

arXiv2026

On the Spherical Dirichlet Distribution: Corrections and Results

This article revisits the spherical-Dirichlet distribution (SDD) introduced by Guardiola (2020) and clarifies the correction to its density presented in an earlier version of this preprint. The principal issue concerns Equation (3) of the original publication, which expresses the density in local Euclidean coordinates rather than with respect to spherical surface-area measure. We derive the corrected intrinsic density and revisit its moments, covariance matrix, mode, and limiting properties. We show that the two density representations describe the same probability distribution, preserving the moment formulas and likelihood-based parameter estimation. However, the mode and uniformity conditions depend on the reference measure and require corresponding clarification. Reported simulation and text mining estimates are revisited and updated. The resulting presentation reconciles the original construction with its intrinsic spherical properties.

stat.ME↗
arXiv2026

A sensitivity analysis for the average derivative effect

In observational studies, exposures are often continuous rather than binary or discrete. At the same time, sensitivity analysis is an important tool that can help determine the robustness of a causal conclusion to a certain level of unmeasured confounding, which can never be ruled out in an observational study. Sensitivity analysis approaches for continuous exposures have now been proposed for several causal estimands. In this article, we focus on the average derivative effect (ADE). We obtain closed-form bounds for the ADE under a sensitivity model that constrains the odds ratio (at any two dose levels) between the latent and observed generalized propensity score. We propose flexible, efficient estimators for the bounds that are compatible with machine learning algorithms as well as point-wise and simultaneous (over the sensitivity parameter) confidence intervals. We examine the finite sample performance of the methods through simulations and illustrate the methods on a study assessing the effect of parental income on educational attainment and a study assessing the price elasticity of petrol.

stat.ME↗
arXiv2026

Graphical Model-based Inference on Persistent Homology

Persistent homology is a cornerstone of topological data analysis, offering a multiscale summary of topology with robustness to nuisance transformations, such as rotations and small deformations. Persistent homology has seen broad use across domains such as computer vision and neuroscience. Most statistical treatments, however, use homology primarily as a feature extractor, relying on statistical distance-based tests or simple time-to-event models for inferential tasks. While these approaches can detect global differences, they rarely localize the source of those differences. We address this gap by taking a graphical model-based approach: we associate each vertex with a population latent position in a conic space and model each bar's key events (birth and death times) using an exponential distribution, whose rate is a transformation of the latent positions according to an event occurring on the graph. The low-dimensional bars have simple graph-event representations, such as the formation of a minimum spanning tree or the triangulation of a loop, and thus enjoy tractable likelihoods. Taking a Bayesian approach, we infer latent positions and enable model extensions such as hierarchical models that allow borrowing strength across groups. Applications to a neuroimaging study of Alzheimer's disease demonstrate that our method localizes sources of difference and provides interpretable, model-based analyses of topological structure in complex data. The code is provided and maintained at https://github.com/zitian-wu/graphPH.

stat.ME↗