SearcharxivSearch

arXiv subjects

Borna Bateni

Publications and source records attributed to Borna Bateni.

2 recordsLinked to original sources

Distributional Treatment Effect Transportability across Heterogeneous Sites

We study distributional transportability of treatment effects in a ``cross-site, one-armed target" design, where both treated and control units are observed in a source site, but only control units are observed in a target site. Our object of interest is not estimating the average treatment effect, but recovering the full treated distribution in the target site using transfer knowledge from the source site, while allowing cross-site heterogeneity in measurement systems, observed features, outcome reporting, population composition, and latent contextual factors. We model cross-site heterogeneity through a transformation between the sites that transports the joint feature--outcome distributions. This transformation is learned from comparing the observed control samples in the source and target sites, using an optimal transport criterion. The learned transformation is then applied to the source treated sample to construct a synthetic sample from the target treated distribution. We establish convergence of the resulting synthetic empirical distribution to the target treated distribution. Simulation studies across multiple data-generating scenarios and a real-world application to patient-derived xenograft (PDX) data demonstrate that our framework recovers the full distributional properties of the target treated population.

stat.ME

Feed-Forward Panel Estimation for Discrete-time Survival Analysis of Recurrent Events with Frailty

In recurrent survival analysis where the event of interest can occur multiple times for each subject, frailty models play a crucial role by capturing unobserved heterogeneity at the subject level within a population. Frailty models traditionally face challenges due to the lack of a closed-form solution for the maximum likelihood estimation that is unconditional on frailty. In this paper, we propose a novel method: Feed-Forward Panel estimation for discrete-time Survival Analysis (FFPSurv). Our model uses variational Bayesian inference to sequentially update the posterior distribution of frailty as recurrent events are observed, and derives a closed form for the panel likelihood, effectively addressing the limitation of existing frailty models. We demonstrate the efficacy of our method through extensive experiments on numerical examples and real-world recurrent survival data. Furthermore, we mathematically prove that our model is identifiable under minor assumptions.

stat.ME