Search papers, labs, and topics across Lattice.
This paper introduces a density-reweighted framework for Entropic Optimal Transport (EOT) that mitigates the influence of sampling density on dataset alignment, which is crucial when datasets originate from geometrically similar structures but differ significantly in sampling density. By decoupling geometry from sampling density, the proposed method allows for more accurate correspondences between datasets, ensuring that the transport plan prioritizes geometric proximity over density biases. Simulations demonstrate that this approach yields geometrically faithful correspondences, outperforming existing EOT methods in scenarios with substantial sampling density disparities.
Density-reweighted EOT can recover geometrically faithful correspondences even when datasets have drastically different sampling densities, challenging traditional assumptions in dataset alignment.
Dataset alignment is a central step in data analysis across science and engineering, where the goal is to match observations between datasets. Entropic Optimal Transport (EOT) offers a computationally tractable framework for this task by encoding cross-dataset affinities in a transport plan. However, when two datasets are sampled from geometrically similar low-dimensional structures with substantially different sampling densities, the EOT plan may match points by relative sampling density rather than geometric proximity, yielding geometrically misleading correspondences. To address this issue, we propose a density-reweighted EOT framework in which the influence of sampling density on the transport plan can be discounted to a desired degree, ranging from standard EOT to alignment driven purely by underlying geometry. Under suitable regularity conditions, we establish convergence of the reweighted EOT plan to a family of population-level plans whose dependence on sampling density is made explicit. Through simulations, we show that our approach recovers geometrically faithful correspondences, improving over related EOT-based frameworks when datasets exhibit substantial sampling density disparity.
