SpaGRA: graph augmentation facilitates domain identification for spatially resolved transcriptomics

J Genet Genomics. 2024 Oct 1:S1673-8527(24)00253-4. doi: 10.1016/j.jgg.2024.09.015. Online ahead of print.

Abstract

Recent advances in spatially resolved transcriptomics (SRT) have provided new opportunities for characterizing spatial structures of various tissues. Graph-based geometric deep learning have 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 demonstrates superior domain identification performance on multiple datasets generated from different protocols. Using SpaGRA, we analyzed the functional regions in the mouse hypothalamus, identified key genes related to heart development in mouse embryos, and observed cancer-associated fibroblasts enveloping cancer cells in the latest Visium HD data. Overall, SpaGRA can effectively characterize spatial structures across diverse SRT datasets.

Keywords: geometric contrastive learning; graph augmentation; multi-head graph attention networks; spatial domain identification; spatially resolved transcriptomics.