Search papers, labs, and topics across Lattice.
This paper introduces MS-WDRO, a novel multi-source Wasserstein distributionally robust graph learning framework designed to effectively fuse heterogeneous data sources while preserving their intrinsic geometries. By leveraging the Wasserstein metric and constructing a weighted Wasserstein barycenter, the method addresses the challenges posed by inter-source divergence and minimizes worst-case risk through a regularized Laplacian estimator. Experimental results demonstrate that MS-WDRO significantly outperforms existing baselines in graph recovery and sample efficiency, particularly in scenarios with limited target-domain samples.
MS-WDRO outperforms traditional methods by leveraging the Wasserstein metric to fuse heterogeneous data sources, achieving superior graph recovery even in sample-scarce environments.
Network topology inference from graph signals is central to graph signal processing with applications in neuroscience, sensor, and social networks. In practice, target-domain samples are scarce while heterogeneous source-domain data are abundant. Fusing these sources is challenging: Euclidean averaging works for homogeneous sources but degrades sharply as inter-source divergence grows, collapsing distinct geometries into an inflated, biased consensus. We exploit the Wasserstein metric's distribution-preserving properties to counter heterogeneity while preserving each source's intrinsic geometry. We propose MS-WDRO, a multi-source Wasserstein distributionally robust graph learning framework that fuses heterogeneous sources via their weighted Wasserstein barycenter, a geometrically principled nominal distribution, then builds an ambiguity ball around it to hedge residual uncertainty. Minimizing worst-case risk yields a tractable regularized Laplacian estimator solved efficiently via a provably convergent ADMM scheme. We establish non-asymptotic guarantees: a finite-sample concentration bound for the empirical barycenter, a pooling bias lower bound proving naive aggregation is suboptimal, and an out-of-sample excess risk bound decaying at a parametric rate with only logarithmic dependence on source count. To calibrate hyperparameters governing robustness, sparsity, and source fusion, we unroll the solver into a differentiable architecture trained end-to-end, achieving data-adaptive calibration beyond cross-validation while retaining interpretability. Experiments on synthetic benchmarks and the multi-site ABIDE~I neuroimaging dataset show MS-WDRO consistently outperforms seven baselines in graph recovery, sample efficiency, and downstream diagnostic utility, with the largest gains in the sample-scarce regime.
