CellContrast: Reconstructing Spatial Relationships in Single-Cell RNA Sequencing Data via Deep Contrastive Learning

Abstract

A vast amount of single-cell RNA-seq (SC) data has been accumulated via various studies and consortiums, but the lack of spatial information limits its analysis of complex biological activities. To bridge this gap, we introduce cellContrast, a computational method for reconstructing spatial relationships among SC cells from spatial transcriptomics (ST) reference. By adopting a contrastive learning framework and training with ST data, cellContrast projects gene expressions into a hidden space where proximate cells share similar representation values. We performed extensive benchmarking on diverse platforms, including SeqFISH, Stereo-Seq, 10X Visium, and MERSCOPE, on mouse embryo and human breast cells. The results reveal that cellContrast substantially outperforms other related methods, facilitating accurate spatial reconstruction of SC. We further demonstrate cellContrasts utility by applying it to cell-type co-localization and cell-cell communication analysis with real-world SC samples, proving the recovered cell locations empower novel discoveries and mitigate potential false positives.