Cluster-Guided Contrastive Learning With Masked Autoencoder for Spatial Domain Identification Based on Spatial Transcriptomics.

Abstract

Recent advancements in spatial transcriptomics technology have enabled the capture of gene expression profiles while maintaining spatial information. Accurately identifying spatial clustering plays a pivotal role in analyzing spatial transcriptomics data and understanding tissue microenvironments. However, current spatial domain identification methods cannot explore the complex relationship of gene expression profiles and spatial topology. To alleviate this issue, we propose STMCCL, a novel self-supervised learning framework that jointly trains a masked autoencoder and cluster-guided contrastive learning. This framework extracts informative latent representations from gene expression profiles and spatial information. Specifically, we first use data augmentation strategies to build augmented views and employ a masked encoder to generate a feature view. Then, encoders are applied to learn view-unique embeddings of each view. Furthermore, we introduce a multiple cluster-perspectives module that considers both geometric and structural relationships between clusters to produce more reliable cluster assignments. Finally, to derive more discriminative positives and negatives, the cluster-guided contrastive module calculates the confidence of each sample based on the initial cluster. Comprehensive experiments on 7 public datasets demonstrate that STMCCL outperforms the state-of-the-art baselines with finer-scale spatial domain identification.