SpaGRA: Graph augmentation facilitates domain identification for spatially resolved transcriptomics.

Abstract

Recent advances in spatially resolved transcriptomics (SRT) have provided new opportunities for characterizing spatial structures of various tissues. Graph-based geometric deep learning has gained widespread adoption for spatial domain identification tasks. Currently, most methods define adjacency relation between cells or spots by their spatial distance in SRT data, which overlooks key biological interactions like gene expression similarities, and leads to inaccuracies in spatial domain identification. To tackle this challenge, we propose a novel method, SpaGRA (https://github.com/sunxue-yy/SpaGRA), for automatic multi-relationship construction based on graph augmentation. SpaGRA uses spatial distance as prior knowledge and dynamically adjusts edge weights with multi-head graph attention networks (GATs). This helps SpaGRA to uncover diverse node relationships and enhance message passing in geometric contrastive learning. Additionally, SpaGRA uses these multi-view relationships to construct negative samples, addressing sampling bias posed by random selection. Experimental results show that SpaGRA presents superior domain identification performance on multiple datasets generated from different protocols. Using SpaGRA, we analyze the functional regions in the mouse hypothalamus, identify key genes related to heart development in mouse embryos, and observe cancer-associated fibroblasts enveloping cancer cells in the latest Visium HD data. Overall, SpaGRA can effectively characterize spatial structures across diverse SRT datasets.