Back

stFormer: a foundation model for spatial transcriptomics

Cao, S.; Yuan, Y.

2024-10-17 bioinformatics
10.1101/2024.09.27.615337 bioRxiv
Show abstract

Recent foundation models for single-cell transcriptomics data generate informative, context-aware representations for genes and cells. The Spatial Transcriptomics (ST) data offer extra positional insights, which were not considered by these single-cell models. Here, we introduce stFormer, a transformer model tailored for ST data. stFormer employs the cross-attention module to incorporate spatial ligand genes into the transformer encoder of single-cell transcriptomics. To unify different ST technologies with trade-off between resolution and gene coverage, we propose a biased cross-attention method that enables the model to do learning with single-cell resolution on whole-transcriptome but low-resolution Visium data. We collected human Visium datasets from a public ST database and performed cell type deconvolution, generating ~4.1 million pretraining samples. As a foundation model for ST data, stFormer improved performance upon the state-of-the-art single-cell foundation model, scFoundation, across a variety of tasks, including cell clustering, batch effect correction, cell type prediction, and gene function prediction. stFormer also revealed intercellular ligand-receptor signaling responses via in silico perturbation.

Matching journals

The top 4 journals account for 50% of the predicted probability mass.

50% of probability mass above

"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.