arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2302.03038q-bio.GNcs.AIcs.LG
收藏

单细胞作为空间词元:用于空间转录组数据插补的Transformer

Single Cells Are Spatial Tokens: Transformers for Spatial Transcriptomic Data Imputation

Michigan State University(密歇根州立大学);Emory University(埃默里大学);TigerGraph Company(TigerGraph公司);MD Anderson Cancer Center(MD安德森癌症中心)

Hongzhi Wen, Wenzhuo Tang, Wei Jin, Jiayuan Ding, Renming Liu, Xinnan Dai, Feng Shi, Lulu Shang, Hui Liu, Yuying Xie

AI总结:

本文提出基于Transformer的空间转录组数据插补框架SpaFormer,将单细胞视为空间词元,通过研究细胞空间信息编码和模型训练方法,在三个大规模数据集上实现了优于现有最先进算法的性能并保持高计算效率。

摘要:

空间分辨转录组学通过提供物理位置和基因表达,为单细胞分析带来了令人兴奋的突破。然而,作为极高空间分辨率的代价,细胞级空间转录组数据受到缺失值的显著影响。虽然对缺失值进行插补是标准解决方案,但现有的大多数方法要么忽略了空间信息,要么仅结合局部空间上下文,而无法捕获长程空间信息。利用多头自注意力机制和位置编码,Transformer模型能够轻松掌握词元间的关系并编码位置信息。本文通过将单细胞视为空间词元,研究如何利用Transformer促进空间转录组学插补。具体而言,我们研究了以下两个关键问题:(1) $\ extit{如何在Transformer中编码细胞的空间信息}$,以及(2) $\ extit{如何为转录组插补训练Transformer}$。通过回答这两个问题,我们提出了一个基于Transformer的插补框架SpaFormer,用于细胞级空间转录组数据。大量实验表明,SpaFormer在三个大规模数据集上优于现有的最先进插补算法,同时保持了卓越的计算效率。

英文摘要:

Spatially resolved transcriptomics brings exciting breakthroughs to single-cell analysis by providing physical locations along with gene expression. However, as a cost of the extremely high spatial resolution, the cellular level spatial transcriptomic data suffer significantly from missing values. While a standard solution is to perform imputation on the missing values, most existing methods either overlook spatial information or only incorporate localized spatial context without the ability to capture long-range spatial information. Using multi-head self-attention mechanisms and positional encoding, transformer models can readily grasp the relationship between tokens and encode location information. In this paper, by treating single cells as spatial tokens, we study how to leverage transformers to facilitate spatial tanscriptomics imputation. In particular, investigate the following two key questions: (1) $\textit{how to encode spatial information of cells in transformers}$, and (2) $\textit{ how to train a transformer for transcriptomic imputation}$. By answering these two questions, we present a transformer-based imputation framework, SpaFormer, for cellular-level spatial transcriptomic data. Extensive experiments demonstrate that SpaFormer outperforms existing state-of-the-art imputation algorithms on three large-scale datasets while maintaining superior computational efficiency.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑