STAGATE software based on pyG (PyTorch Geometric) framework.
2022.08.23 The non-linear function for attention weight calculation is changed from LeakyRelu to Sigmoid, which is the same with the tensorflow version STAGATE.
2022.03.05 STAGATE based on pyG (PyTorch Geometric) framework is availble at STAGATE_pyG.
Benefit from the optimization of the pyG package for training graph neural networks, it is more than 10x faster than STAGATE based on the tensorflow1 framework, and can use a batch training strategy to deal with large-scale data.
The cell type-aware module has not been supported by STAGATE_pyG yet.
STAGATE is designed for spatial clustering and denoising expressions of spatial resolved transcriptomics (ST) data.
STAGATE learns low-dimensional latent embeddings with both spatial information and gene expressions via a graph attention auto-encoder. The method adopts an attention mechanism in the middle layer of the encoder and decoder, which adaptively learns the edge weights of spatial neighbor networks, and further uses them to update the spot representation by collectively aggregating information from its neighbors. The latent embeddings and the reconstructed expression profiles can be used to downstream tasks such as spatial domain identification, visualization, spatial trajectory inference, data denoising and 3D expression domain extraction.
STAGATE based on tensorflow framework can be found at here.
See Documentation and Tutorials.
scanpy
pytorch
pyG
cd STAGATE_pyG-main
python setup.py build
python setup.py install
Dong, Kangning, and Shihua Zhang. "Deciphering spatial domains from spatially resolved transcriptomics with an adaptive graph attention auto-encoder." Nature Communications 13.1 (2022): 1-12.