arXiv:2506.11152q-bio.GNcs.LG2025-06被引 17

构建可同时处理基因与蛋白空间数据的图神经网络模型,实现细胞微环境适应性分析。

HEIST: A Graph Foundation Model for Spatial Transcriptomics and Proteomics Data

  • 将组织建模为细胞层级图,细胞内部用基因共表达网络表示
  • 在124个组织2230万细胞上预训练,支持跨技术泛化且无需重新训练
  • 能发现传统方法遗漏的空间特异性细胞亚群,适用于临床预测与基因补全

单细胞转录组与蛋白质组已成为生物数据驱动洞察的重要来源,使深度学习能够解析细胞异质性与基因表达。随着空间组学数据的出现,我们有望在组织环境中表征细胞,因其同时提供空间坐标与细胞内转录或蛋白计数。蛋白质组学通过直接测量蛋白质,提供了细胞功能执行者和关键治疗靶点的互补视角。然而,现有模型或忽略空间信息,或无法捕捉细胞内复杂的遗传与蛋白调控程序,因而难以推断细胞内部调控如何响应微环境信号。此外,这些模型通常依赖固定基因词典,限制了对未见基因的泛化能力。本文提出HEIST,一种用于空间转录组与蛋白质组的分层图转换器基础模型。HEIST将组织建模为分层图:高层为空间细胞图,每个细胞则由其低层的基因共表达网络图表示。通过执行层内与跨层消息传递,利用层次结构进行嵌入学习,从而可在不重新训练的情况下泛化至新数据类型(如空间蛋白质组)。HEIST在来自15个器官的124个组织中2230万细胞上,使用空间感知对比学习与掩码自编码目标进行预训练。无监督分析显示,HEIST嵌入揭示了先前模型遗漏的空间相关细胞亚群。下游评估表明,其在蛋白质组数据泛化、临床结果预测、细胞类型注释与多技术基因补全任务中均达到领先性能。

原文摘要 · Abstract (English)

Single-cell transcriptomics and proteomics have become a great source for data-driven insights into biology, enabling the use of advanced deep learning methods to understand cellular heterogeneity and gene expression at the single-cell level. With the advent of spatial-omics data, we have the promise of characterizing cells within their tissue context as it provides both spatial coordinates and intra-cellular transcriptional or protein counts. Proteomics offers a complementary view by directly measuring proteins, which are the primary effectors of cellular function and key therapeutic targets. However, existing models either ignore the spatial information or the complex genetic and proteomic programs within cells. Thus they cannot infer how cell internal regulation adapts to microenvironmental cues. Furthermore, these models often utilize fixed gene vocabularies, hindering their generalizability unseen genes. In this paper, we introduce HEIST, a hierarchical graph transformer foundation model for spatial transcriptomics and proteomics. HEIST models tissues as hierarchical graphs. The higher level graph is a spatial cell graph, and each cell in turn, is represented by its lower level gene co-expression network graph. HEIST achieves this by performing both intra-level and cross-level message passing to utilize the hierarchy in its embeddings and can thus generalize to novel datatypes including spatial proteomics without retraining. HEIST is pretrained on 22.3M cells from 124 tissues across 15 organs using spatially-aware contrastive and masked autoencoding objectives. Unsupervised analysis of HEIST embeddings reveals spatially informed subpopulations missed by prior models. Downstream evaluations demonstrate generalizability to proteomics data and state-of-the-art performance in clinical outcome prediction, cell type annotation, and gene imputation across multiple technologies.

空间组学图神经网络蛋白质组基础模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。