Wasserstein F-tests for Frechet regression on Bures-Wasserstein manifolds
Authors
Research Topics
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
AuthorHongzhe Li
AuthorResearch Topics & Keywords
Machine Learning
Research AreaCitation 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}
}