From WSIs to Bulk, Spatial and Single-Cell Transcriptomics via Multi-Scale Graph Neural Networks
Organizations: Departamento de Engenharia Electrotécnica e de Computadores, Instituto Superior Técnico, Lisboa, Portugal · Champalimaud Foundation, Lisbon, Portugal
Abstract
Understanding the spatial distribution of gene expression is key to studying complex diseases like cancer. While Spatial Transcriptomics (ST) and Single-Cell Transcriptomics (scT) provide high-resolution insights, they remain expensive and scarce. In contrast, standard histopathology Whole-Slide Images (WSIs) paired with bulk RNA sequencing are widely available. This work introduces a Multi-Scale Graph Neural Network (GNN) capable of simultaneously predicting bulk, spatial, and single-cell transcriptomics directly from WSIs. By extracting morphological features using a Pathology Foundation Model (PFM) (UNI2-h) and constructing spatial graphs at both patch and cell levels, the model learns the hierarchical relationships within the tissue. The network is trained on a combined dataset of The Cancer Genome Atlas Program (TCGA), Visium, and Xenium samples across lung, kidney, and breast tissues. A hybrid training strategy is employed, leveraging direct supervision alongside weakly supervised consistency losses to ensure predictions align across biological scales. Results show the model accurately predicts gene expression at the slide level and moderately at the patch level (strongest on Xenium), and achieves moderate correlation at the challenging single-cell level. Furthermore, downstream tasks explore the clinical relevance of these predictions, demonstrating their utility in survival analysis, spatial hazard mapping, and tumor classification via molecular signatures.
Figures & tables
| Metric | TCGA | Visium | Xenium |
|---|---|---|---|
| Bulk PCC | |||
| Patch PCC | – | ||
| Cell PCC | – | – | |
| Patch AUROC | – | ||
| Cell AUROC | – | – |
| C-index | TCGA-LUAD | TCGA-KIRC | TCGA-BRCA |
|---|---|---|---|
| Real bulk | |||
| Predicted bulk | |||
| Visual features |
| Signature | Proposed Model | DeepSpot2Cell |
|---|---|---|
| Proliferation | ||
| Hypoxia | ||
| Apoptosis | ||
| Angiogenesis |
| Cohort / signature | AUROC | TPR | |
|---|---|---|---|
| LUAD: Prolif. | 431 | 0.711 | 1.412 |
| LUAD: Hypoxia | 431 | 0.593 | 1.131 |
| LUAD: Apopt. | 431 | 0.525 | 0.987 |
| LUAD: Angio. | 431 | 0.517 | 0.992 |
| KIRC: Prolif. | 416 | 0.648 | 1.193 |
| KIRC: Hypoxia | 416 | 0.776 | 1.444 |
| Cohort | Input | Bal. Acc. | F1-score | AUROC |
|---|---|---|---|---|
| LUAD | 50 signatures | 0.707 | 0.707 | 0.771 |
| Full transcript. | 0.684 | 0.679 | 0.742 | |
| Patch embeddings | 0.741 | 0.740 | 0.822 | |
| KIRC | 50 signatures | 0.857 | 0.861 | 0.932 |
| Full transcript. | 0.851 | 0.850 | 0.912 | |
| Patch embeddings | 0.862 | 0.861 | 0.942 |
| Cohort | Total | Bulk | Patch | Feat. Recon. | PS Cons. | CP Cons. | Patch | Signif. Genes (Patch) |
|---|---|---|---|---|---|---|---|---|
| PCC | PCC | Cos. Sim. | PCC | PCC | AUROC | (FDR < 0.05) | ||
| Overall | 105 | |||||||
| Breast | 7 | |||||||
| Kidney | 62 | |||||||
| Lung | 36 |
| Cohort / Subtype | Total | Bulk | Patch | Cell | Feat. Recon. | PS Cons. | CP Cons. | Patch | Signif. Genes (Patch) | Cell |
| PCC | PCC | PCC | Cos. Sim. | PCC | PCC | AUROC | (FDR < 0.05) | AUROC | ||
| Overall | 47 | |||||||||
| Connective | - | - | - | - | - | - | - | - | ||
| Dead | - | - | - | - | - | - | - | - | ||
| Epithelial | - | - | - | - | - | - | - | - | ||
| Inflammatory | - | - | - | - | - | - | - | - |
| Cohort | Total | Bulk | Feat. Recon. | PS Cons. | Bulk | Signif. Genes (Bulk) |
|---|---|---|---|---|---|---|
| PCC | Cos. Sim. | PCC | AUROC | (FDR < 0.05) | ||
| Overall | ||||||
| Breast | ||||||
| Kidney | ||||||
| Lung |
| Test Set | Average Accuracy | Average F1-score | Average AUROC |
|---|---|---|---|
| Visium true spot’s gene expression | |||
| Visium predicted spot’s gene expression | |||
| TCGA predicted spot’s gene expression |
| Test Set | Accuracy | F1-score | AUROC |
|---|---|---|---|
| Original spot’s embedding | |||
| Reconstructed spot’s embedding |
| Model | Total | Bulk | Patch | Feat. Recon. | PS Cons. | CP Cons. | Patch | Signif. Genes (Patch) |
|---|---|---|---|---|---|---|---|---|
| PCC | PCC | Cos. Sim. | PCC | PCC | AUROC | (FDR < 0.05) | ||
| Proposed Model (Full) | ||||||||
| Proposed Model (w/o Consistency) | ||||||||
| Proposed Model (w/o Feat. Recon.) | ||||||||
| Proposed Model (w/o TCGA) | ||||||||
| Proposed Model (w/o TCGA cells) |
| Model | Total | Bulk | Patch | Cell | Feat. Recon. | PS Cons. | CP Cons. | Patch | Signif. Genes (Patch) | Cell |
|---|---|---|---|---|---|---|---|---|---|---|
| PCC | PCC | PCC | Cos. Sim. | PCC | PCC | AUROC | AUROC | |||
| Proposed Model (Full) | ||||||||||
| Proposed Model (w/o Consistency) | ||||||||||
| Proposed Model (w/o Feat. Recon.) | ||||||||||
| Proposed Model (w/o TCGA) | ||||||||||
| Proposed Model (w/o TCGA cells) |
| Model | Total | Bulk | Feat. Recon. | PS Cons. | Bulk | Signif. Genes (Bulk) |
|---|---|---|---|---|---|---|
| PCC | Cos. Sim. | PCC | AUROC | (FDR 0.05) | ||
| Proposed Model (Full) | ||||||
| Proposed Model (w/o Consistency) | ||||||
| Proposed Model (w/o Feat. Recon.) | ||||||
| Proposed Model (w/o TCGA) | ||||||
| Proposed Model (w/o TCGA cells) |
| CellViT | Classpose | |
|---|---|---|
| Cell type | PQ / DQ | PQ / DQ |
| Neoplastic | / | / |
| Inflammatory | / | / |
| Connective | / | / |
| Dead | / | / |
| Epithelial | / | / |
Spatial Gene Expression Prediction
Spatial gene expression prediction infers tissue-resolved molecular activity from histology and spatial context.
See all 42 papers