中文

Multivariate Varying-Coefficient BART with Graphical Horseshoe Priors

统计方法学 2026-06-27 v1 统计理论 机器学习

摘要

Modern multivariate regression problems involve several related outcomes whose regression effects are not only nonlinear, heterogeneous, and outcome-specific, but also where the residual dependence among outcomes is scientifically meaningful. Existing multivariate Bayesian tree-based methods typically address only part of this problem: some impose substantial sharing of tree architecture across outcomes, which is overly restrictive when responses depend on distinct predictors or effect modifiers, while others accommodate residual dependence but retain simpler mean structures. This paper develops multiVCBART, a multivariate varying-coefficient Bayesian additive regression tree framework that jointly models flexible outcome-specific coefficient surfaces and a sparse residual precision matrix. Each entry of the coefficient matrix B(x)B(x) is represented by an independent BART ensemble, allowing predictor effects to vary nonlinearly with modifiers xx across outcomes, while a Graphical Horseshoe prior on the precision matrix Ω\Omega captures parsimonious residual conditional dependence. To permit efficient computation, we introduce a sampler that reduces the multivariate Gaussian likelihood to a sequence of scalar pseudo-response updates, decoupling the tree backfitting from the Graphical Horseshoe step. Theoretically, we establish the first posterior contraction rates for a multivariate BART model with jointly estimated residual dependence, proving near-minimax adaptation to underlying smoothness and structural sparsity. Empirically, multiVCBART outperforms existing multivariate tree models and Bayesian SUR competitors on sparse, high-dimensional datasets. Finally, in a re-analysis of the Genomics of Drug Sensitivity in Cancer dataset, our method identifies distinct biomarker signals and recovers a coherent residual pharmacologic network.

引用

@article{arxiv.2606.29114,
  title  = {Multivariate Varying-Coefficient BART with Graphical Horseshoe Priors},
  author = {Soham Ghosh and Sameer K. Deshpande},
  journal= {arXiv preprint arXiv:2606.29114},
  year   = {2026}
}

备注

70 pages, 5 figures; includes supplementary material