/spatialID

Primary LanguagePython

spatial-ID

python >3.8.8

Spatial-ID: a cell typing method for spatially resolved transcriptomics via transfer learning and spatial embedding

Spatially resolved transcriptomics (SRT) provides the opportunity to investigate the gene expression profiles and the spatial context of cells in naive state. Cell type annotation is a crucial task in the spatial transcriptome analysis of cell and tissue biology. In this study, we propose Spatial-ID, a supervision-based cell typing method, for high-throughput cell-level SRT datasets that integrates transfer learning and spatial embedding. Spatial-ID effectively incorporates the existing knowledge of reference scRNA-seq datasets and the spatial information of SRT datasets.

Dependences

numpy-1.21.3 pandas-1.2.4 scanpy-1.8.1 torch-1.8.1 torch__geometric-1.7.2

Datasets

Usage

  • Run cell_type_annotation_for_merfish.py to annotate cells in MERFISH dataset.
  • Run cell_type_annotation_for_hyp3d.py to annotate cells in MERFISH-3D dataset.
  • Run cell_type_annotation_for_slideseq.py to annotate cells in Slide-seq dataset.
  • Run cell_type_annotation_for_nanostring.py to annotate cells in NanoString dataset.

p.s. You may need to unzip dnn_model/checkpoint_Slide-seq_DM1.t7 first before running cell_type_annotation_for_slideseq.py.

[!!!] Note: An AttributeError saying that 'GELU' object has no attribute 'approximate' may occurs if your pytorch version is higher than 1.10.0. You can simply downgrade pytorch to 1.8.1 or modify the source code of pytorch temporarily.

Example

  1. Put downloaded MERFISH data (e.g. mouse1_sample1.h5ad) in "dataset/MERFISH/" (as in Line 30 of cell_type_annotation_for_merfish.py).
  2. Run cell_type_annotation_for_merfish.py to annotate cells of mouse1_sample1 data.
  3. 4 files can be found in "result/MERFISH/" (as in Line 31 of cell_type_annotation_for_merfish.py):
    • spatialID-mouse1_sample1.t7: Checkpoint of the self-supervised model in Stage 2.
    • spatialID-mouse1_sample1.h5ad: Updated H5AD file with annotation result stored.
    • spatialID-mouse1_sample1.csv: Annotation results with column "cell" representing cell IDs and "celltype_pred" representing annotated cell types.
    • spatialID-mouse1_sample1.pdf: Visualization of annotation results as shown below.

Disclaimer

This tool is for research purpose and not approved for clinical use.

This is not an official Tencent product.

Coypright

This tool is developed in Tencent AI Lab.

The copyright holder for this project is Tencent AI Lab.

All rights reserved.