MAEST: accurately spatial domain detection in spatial transcriptomics with graph masked autoencoder.

Journal: Briefings in bioinformatics
PMID:

Abstract

Spatial transcriptomics (ST) technology provides gene expression profiles with spatial context, offering critical insights into cellular interactions and tissue architecture. A core task in ST is spatial domain identification, which involves detecting coherent regions with similar spatial expression patterns. However, existing methods often fail to fully exploit spatial information, leading to limited representational capacity and suboptimal clustering accuracy. Here, we introduce MAEST, a novel graph neural network model designed to address these limitations in ST data. MAEST leverages graph masked autoencoders to denoise and refine representations while incorporating graph contrastive learning to prevent feature collapse and enhance model robustness. By integrating one-hop and multi-hop representations, MAEST effectively captures both local and global spatial relationships, improving clustering precision. Extensive experiments across diverse datasets, including the human brain, mouse hippocampus, olfactory bulb, brain, and embryo, demonstrate that MAEST outperforms seven state-of-the-art methods in spatial domain identification. Furthermore, MAEST showcases its ability to integrate multi-slice data, identifying joint domains across horizontal tissue sections with high accuracy. These results highlight MAEST's versatility and effectiveness in unraveling the spatial organization of complex tissues. The source code of MAEST can be obtained at https://github.com/clearlove2333/MAEST.

Authors

  • Pengfei Zhu
    Department of Ultrasound, Affiliated Tumor Hospital of Nantong University, Jiangsu, Nantong, China.
  • Han Shu
    School of Computer Science, Northwestern Polytechnical University, 1 Dongxiang Road, Xi'an 710072, China.
  • Yongtian Wang
    Beijing Engineering Research Centre of Mixed Reality and Advanced Display, School of Optics and Photonics, Beijing Institute of Technology, 5 Zhongguancun South Street, Beijing, 100081, China.
  • XiaoFeng Wang
    Indiana University Bloomington.
  • Yuan Zhao
    Department of Neurobiology and Behavior, Stony Brook University, Stony Brook, NY.
  • Jialu Hu
    School of Computer Science, Northwestern Polytechnical University, West Youyi Road 127, Xi'an, 710072, China. jhu@nwpu.edu.cn.
  • Jiajie Peng
    School of Computer Science and Technology, Harbin Institute of Technology, Harbin, China. jiajiepeng@hit.edu.cn.
  • Xuequn Shang
  • Zhen Tian
    School of Computer Science and Technology, Harbin Institute of Technology, Harbin, 150001, People's Republic of China.
  • Jing Chen
    Department of Vascular Surgery, The First Affiliated Hospital of Guangxi Medical University, Nanning, Guangxi 530021, P.R. China.
  • Tao Wang
    Department of Urology, Tongji Hospital, Tongji Medical College, Huazhong University of Science and Technology, Wuhan, China.