AtlasOT - The Fused Unbalanced Gromov-Wasserstein for Multimodal Integration of Disease Atlases

Abstract

Single-cell and spatial multiomics technologies are transforming disease atlas construction, but computational integration of unpaired modalities remains challenging, as existing methods overlook two biological constraints of data with matching samples. First, cells should only be mapped across modalities within the same donor or biospecimen, and second, cell recovery frequently differs substantially between modalities. To address these gaps, we present AtlasOT, an optimal-transport framework based on the fused unbalanced Gromov-Wasserstein (FUGW) formulation for multi-modal integration. AtlasOT jointly models a shared cross-modality feature space and modality-specific geometric structures while restricting transport to within-sample cell pairs. Moreover, AtlasOT relaxes strict mass conservation of the balanced optimal transport optimization to accommodate unbalanced cell numbers. We benchmark AtlasOT against state-of-the-art methods on scRNA-scATAC and scRNA-spatial transcriptomics mapping scenarios. Our results indicate that AtlasOT outperforms baselines and state-of-the-art methods in all considered scenarios. Moreover, we demonstrate that AtlasOT's transport plan can be used to improve several relevant tasks such as the detection of rare cell populations, spot-level cell-type deconvolution, spatial gene imputation, and reconstruction of transcription-factor-driven spatial regulatory dynamics. These results establish AtlasOT as a unified, biologically constrained framework for multimodal integration in disease atlas studies, with broad applicability to label transfer, imputation, and regulatory network analysis.