将基因调控先验知识融入Transformer注意力,提升单细胞测序分析的可解释性
Integrating gene regulatory priors into Transformer attention with scTransformer for interpretable scRNA-seq analysis

- 在注意力机制中引入已知基因调控结构,约束信息流动
- 分类准确率提升,细胞类型在嵌入空间中分离更清晰
- 适合关注生物可解释性的单细胞数据分析研究者
Transformer模型在大规模单细胞转录组分析中表现优异,但多数方法将基因视为独立特征,忽视生物先验知识,限制了可解释性与鲁棒性。本文提出scTransformer,首个将生物学机制先验嵌入Transformer注意力模式的框架。通过依据已知调控结构约束信息流,模型学习到更具生物学意义的表示。在疾病相关的单核RNA-seq数据集上,基于监督细胞类型分类评估显示,相比标准Transformer,scTransformer提升了分类准确率,增强了嵌入空间中细胞类型的分离度,并生成与已知调控程序一致的注意力模式。结果表明,将生物结构嵌入Transformer可增强可解释性而不牺牲性能,为构建生物学基础的单细胞组学大模型提供了原则性路径。
原文摘要 · Abstract (English)
Motivation: Transformer-based models are increasingly applied to large-scale single-cell transcriptomics, showing strong performance through self-supervised learning on millions of cells. However, most existing approaches treat genes as independent features, and largely ignore prior biological knowledge, which limits interpretability and robustness. In this paper, we explore whether explicitly incorporating gene regulatory information can improve both model performance and biological insight. Results: We present scTransformer, the first Transformer-based approach that builds a priori knowledge of biological mechanisms into the model's attention patterns. By constraining information flow according to known regulatory structures, the model learns representations that are more biologically meaningful. We evaluate scTransformer on a disease-relevant single-nucleus RNA-seq dataset using supervised cell-type classification. Compared to standard Transformers, our approach improves classification accuracy, enhances separation of cell types in embedding space, and produces attention patterns consistent with known regulatory programs. Overall, our results demonstrate that embedding biological structure into Transformer models can enhance interpretability without sacrificing performance, offering a principled step toward biologically grounded foundation models for single-cell omics.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。