collaborators

5 papers

stat.ME2026

Double machine learning to estimate the effects of multiple treatments and their interactions

Qingyan Xiang, Yubai Yuan, Dongyuan Song +4

Causal inference literature has extensively focused on binary treatments, with relatively fewer methods developed for multi-valued treatments. In particular, methods for multiple s…

stat.ME2026

Individualized Dynamic Mediation Analysis Using Latent Factor Models

Yijiao Zhang, Yubai Yuan, Yuexia Zhang +2

Mediation analysis plays a crucial role in causal inference as it can investigate the pathways through which treatment influences outcome. Most existing mediation analysis assumes…

stat.ME2026

Scalable Sample-to-Population Estimation of Hyperbolic Space Models for Hypergraphs

Cornelius Fritz, Yubai Yuan, Michael Schweinberger

Hypergraphs are useful mathematical representations of overlapping and nested subsets of interacting units, including groups of genes or brain regions, economic cartels, political…

stat.ME2025

Distributional Treatment Effect Estimation across Heterogeneous Sites via Optimal Transport

Borna Bateni, Yubai Yuan, Qi Xu +1

We propose a novel framework for synthesizing counterfactual treatment group data in a target site by integrating full treatment and control group data from a source site with cont…

cs.SI2025

Inferring Diffusion Structures of Heterogeneous Network Cascade

Yubai Yuan, Siyu Huang, Abdul Basit Adeel

Network cascade refers to diffusion processes in which outcome changes within part of an interconnected population trigger a sequence of changes across the entire network. These ca…