Multi-transport Distributional Regression
By: Yuanying Chen , Tongyu Li , Yang Bai and more
Potential Business Impact:
Combines many data shapes to make better predictions.
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.
Similar Papers
Neural Local Wasserstein Regression
Machine Learning (Stat)
Teaches computers to understand complex data patterns.
Wasserstein Regression as a Variational Approximation of Probabilistic Trajectories through the Bernstein Basis
Machine Learning (CS)
Teaches computers to predict patterns smoothly.
Wasserstein Distributionally Robust Nonparametric Regression
Machine Learning (Stat)
Makes computer predictions better even with bad data.