Score: 0

Multi-transport Distributional Regression

Published: January 7, 2026 | arXiv ID: 2601.03674v1

By: Yuanying Chen , Tongyu Li , Yang Bai and more

Potential Business Impact:

Combines many data shapes to make better predictions.

Business Areas:
A/B Testing Data and Analytics

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.

Country of Origin
🇸🇬 Singapore

Page Count
33 pages

Category
Statistics:
Methodology