用图神经网络替代Transformer,高效处理单细胞转录组数据
Graph Neural Networks as a Substitute for Transformers in Single-Cell Transcriptomics
- 将图神经网络用于无位置依赖的单细胞数据建模
- 在大规模单细胞数据上性能媲美Transformer,计算成本更低
- 为非序列任务提供更优模型选择,适合生物信息研究者
图神经网络(GNNs)与变压器(Transformers)在节点特征交互策略上具有相似性,前者通过边传递信息,后者通过查询-键得分。相较于无法编码相对位置的GNN,Transformer利用动态注意力机制更优地表达相对关系,因此成为大规模序列预训练的标准骨干。然而,当位置信息不重要时,是否可用GNN替代Transformer?本文首先分析GNN与Transformer在相对位置表达上的异同,并设计合成实验验证二者在无位置依赖场景下的等价性。随后,在大规模无位置依赖的单细胞转录组数据集上进行广泛实验,结果表明GNN在性能上可媲美Transformer,同时消耗更少计算资源。该发现为单细胞转录组研究提供了新视角,挑战了Transformer始终最优的普遍认知。
原文摘要 · Abstract (English)
Graph Neural Networks (GNNs) and Transformers share significant similarities in their encoding strategies for interacting with features from nodes of interest, where Transformers use query-key scores and GNNs use edges. Compared to GNNs, which are unable to encode relative positions, Transformers leverage dynamic attention capabilities to better represent relative relationships, thereby becoming the standard backbones in large-scale sequential pre-training. However, the subtle difference prompts us to consider: if positions are no longer crucial, could we substitute Transformers with Graph Neural Networks in some fields such as Single-Cell Transcriptomics? In this paper, we first explore the similarities and differences between GNNs and Transformers, specifically in terms of relative positions. Additionally, we design a synthetic example to illustrate their equivalence where there are no relative positions between tokens in the sample. Finally, we conduct extensive experiments on a large-scale position-agnostic dataset-single-cell transcriptomics-finding that GNNs achieve competitive performance compared to Transformers while consuming fewer computation resources. These findings provide novel insights for researchers in the field of single-cell transcriptomics, challenging the prevailing notion that the Transformer is always the optimum choice.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。