JMLR

Wasserstein F-tests for Frechet regression on Bures-Wasserstein manifolds

Authors
Haoshu Xu Hongzhe Li
Research Topics
Machine Learning
Paper Information
  • Journal:
    Journal of Machine Learning Research
  • Added to Tracker:
    Jul 15, 2025
Abstract

This paper addresses regression analysis for covariance matrix-valued outcomes with Euclidean covariates, motivated by applications in single-cell genomics and neuroscience where covariance matrices are observed across many samples. Our analysis leverages Fr\'echet regression on the Bures-Wasserstein manifold to estimate the conditional Fr\'echet mean given covariates $x$. We establish a non-asymptotic uniform $\sqrt{n}$-rate of convergence (up to logarithmic factors) over covariates with $\|x\| \lesssim \sqrt{\log n}$ and derive a pointwise central limit theorem to enable statistical inference. For testing covariate effects, we devise a novel test whose null distribution converges to a weighted sum of independent chi-square distributions, with power guarantees against a sequence of contiguous alternatives. Simulations validate the accuracy of the asymptotic theory. Finally, we apply our methods to a single-cell gene expression dataset, revealing age-related changes in gene co-expression networks.

Author Details
Haoshu Xu
Author
Hongzhe Li
Author
Research Topics & Keywords
Machine Learning
Research Area
Citation Information
APA Format
Haoshu Xu & Hongzhe Li . Wasserstein F-tests for Frechet regression on Bures-Wasserstein manifolds. Journal of Machine Learning Research .
BibTeX Format
@article{JMLR:v26:24-0493,
  author  = {Haoshu Xu and Hongzhe Li},
  title   = {Wasserstein F-tests for Frechet regression on Bures-Wasserstein manifolds},
  journal = {Journal of Machine Learning Research},
  year    = {2025},
  volume  = {26},
  number  = {77},
  pages   = {1--123},
  url     = {http://jmlr.org/papers/v26/24-0493.html}
}
Related Papers