Multi-transport Distributional Regression

Read full text
Authors: Yuanying Chen, Tongyu Li, Yang Bai, Zhenhua Lin
Cited by: 0

Abstract

We study distribution-on-distribution regression problems in which a response distribution depends on multiple distributional predictors. Such settings arise naturally in applications where the outcome distribution is driven by several heterogeneous distributional sources, yet remain challenging due to the nonlinear geometry of the Wasserstein space. We propose an intrinsic regression framework that aggregates predictor-specific transported distributions through a weighted Fréchet mean in the Wasserstein space. The resulting model admits multiple distributional predictors, assigns interpretable weights quantifying their relative contributions, and defines a flexible regression operator that is invariant to auxiliary construction choices, such as the selection of a reference distribution. From a theoretical perspective, we establish identifiability of the induced regression operator and derive asymptotic guarantees for its estimation under a predictive Wasserstein semi-norm, which directly characterizes convergence of the composite prediction map. Extensive simulation studies and a real data application demonstrate the improved predictive performance and interpretability of the proposed approach compared with existing Wasserstein regression methods.

Fields of Study & Subjects

Statistical Methods and Inference Bayesian Methods and Mixture Models Statistical Methods and Bayesian Inference

Publication details

Journal / source
arXiv (Cornell University)
Publication date
7 January 2026
Document type
Preprint