cs.LGSep 23, 2026

Pheno-GS: Phenoscape-scale Geodesic Sinkhorn

Authors: Alistair WilkinsonChristopher J. TapeSmita Krishnaswamy

Abstract

High-throughput single-cell data is now collected across large patient cohorts. Understanding patient-level heterogeneity from cellular-level data motivates phenoscaping: embedding each single-cell distribution as a "datapoint," with distances given by optimal transport (OT). Computing geometry-aware OT at this scale, between all pairs of patient datasets, remains an open challenge, since existing methods either rely on Euclidean ground metrics that distort manifold structure or fail under sparse, unevenly sampled, or large-scale data. We present \textbf{Pheno-GS} (Phenoscape-scale Geodesic Sinkhorn), which computes accurate, scalable geodesic transport distances under noisy, unbalanced, large-scale settings via three components: (11) graph connectivity regularization for well-defined geodesics on sparse/disconnected manifolds; (22) an unbalanced OT formulation via KL marginal penalties; and (33) a batched matrix algorithm computing all pairwise distances in one heat diffusion (over 200×200 \times faster than Geodesic Sinkhorn for 500500 distributions). We validate Pheno-GS on synthetic benchmarks and a CyTOF perturbation dataset.

Explore similar work

CardsList