Is H&E Image-to-Spatial Transcriptomics Simpler Than It Looks?
Organizations: VinUniversity, Hanoi, Vietnam · VinUni-Illinois Smart Health Center (VISHC), VinUniversity, Hanoi, Vietnam · Posts and Telecommunications Institute of Technology, Hanoi, Vietnam · National Center for Supercomputing Applications, University of Illinois Urbana-Champaign, Urbana, IL, USA
Abstract
Predicting spatial gene expression from routine H&E histology offers a scalable route toward spatial molecular profiling. Recent work has pursued increasingly sophisticated architectures to capture spatial context and richer expression structure. At the same time, simple estimators have shown strong performance in several studies, but what they already solve and where additional complexity is needed remain unclear. We study this behavior through the structure of prediction error under the mean-squared error (MSE) objective. Differences in average expression across genes can account for a substantial part of aggregate prediction performance, while a key unresolved error lies in recovering variation within each slide. Decomposing MSE into slide-level and within-slide components, we find that the within-slide component has lower residual-normalized parameter sensitivity in controlled neural experiments. This motivates Component-Guided Loss (CGL), which increases supervision of the within-slide component. CGL-Linear is a closed-form affine instantiation that achieves overall state-of-the-art performance across HEST-1k cohorts and gene-panel sizes. The same within-slide supervision improves existing neural models. These results suggest that substantial gains can come from aligning the training objective with prediction-error structure rather than increasing model complexity.
Figures & tables
| Method | Venue | ||||
|---|---|---|---|---|---|
| MeanOnly | – | N.A. | |||
| Linear | – | ||||
| HiST | ICML’26 | ||||
| HistoPrism | ICLR’26 | ||||
| HEXST | ICML’26 | ||||
| STFlow | ICML’25 |
Appendix figures & tables15 assets
Supplementary material from the paper’s appendix.
Appendix
| Cohort | Slides (train / val / test) | Spots (train / val / test) | ||||
| Brain | 9 | 2 | 3 | 23,780 | 4,891 | 4,219 |
| Colon | 33 | 4 | 4 | 94,871 | 6,359 | 15,138 |
| Heart | 26 | 7 | 8 | 70,802 | 14,100 | 14,195 |
| Kidney | 24 | 5 | 5 | 27,739 | 5,649 | 5,691 |
| Liver | 10 | 2 | 2 | 26,518 | 5,695 | 4,955 |
| Lung | 12 | 4 | 4 | 20,810 | 8,763 | 8,933 |
| Method | Epochs | Readout configuration | Optimization |
|---|---|---|---|
| HiST | 300 | Sparse hierarchy 111–21, window size 3, feed-forward ratio 4, three global tokens. | AdamW, LR , WD , cosine schedule, centered-target MSE. |
| HistoPrism | 300 | Hidden size 256, two Transformer layers, eight heads, dropout 0.1. | AdamW, LR , WD , linear decay with 5% warm-up, centered-target MSE. |
| HEXST | 300 | Hidden size 512, HexMSA/HexRoPE, 3,072-dimensional scFoundation alignment, dropout 0.2. | Adam, LR , WD , warm-up and cosine schedule, composite regression/correlation/deviation/alignment objective. |
| STFlow | 300 | Hidden size 128, four layers, four heads, eight neighbors, five sampling steps. | Adam, LR , gradient clipping at 1.0, no scheduler, conditional flow-matching objective. |
| BLEEP | 300 | Dual 256-dimensional projections, dropout 0.1, temperature 1, top-50 retrieval. | AdamW, LR , WD , batch size 512, symmetric contrastive cross-entropy. |
| SEPAL | Image readout followed by a six-neighbor graph head, relative positional encoding, two GCN layers, SAGPooling. | SGD, LR 0.0099913, momentum 0.9, batch size 32, MSE in both stages, frozen first-stage predictor during graph fitting. |
| Method | |
|---|---|
| HiST | 10 |
| HistoPrism | 15 |
| HEXST | 20 |
| STFlow | 10 |
| SEPAL | 5 |
| Cohort | HiST | HistoPrism | HEXST | STFlow | BLEEP | SEPAL | Linear | CGL-Linear |
|---|---|---|---|---|---|---|---|---|
| Brain | ||||||||
| Colon | ||||||||
| Heart | ||||||||
| Kidney | ||||||||
| Liver | ||||||||
| Cohort | HiST | HistoPrism | HEXST | STFlow | BLEEP | SEPAL | Linear | CGL-Linear |
|---|---|---|---|---|---|---|---|---|
| Brain | ||||||||
| Colon | ||||||||
| Heart | ||||||||
| Kidney | ||||||||
| Liver | ||||||||
| Cohort | HiST | HistoPrism | HEXST | STFlow | BLEEP | SEPAL | Linear | CGL-Linear |
|---|---|---|---|---|---|---|---|---|
| Brain | ||||||||
| Colon | ||||||||
| Heart | ||||||||
| Kidney | ||||||||
| Liver | ||||||||
| Cohort | HiST | HistoPrism | HEXST | STFlow | BLEEP | SEPAL | Linear | CGL-Linear |
|---|---|---|---|---|---|---|---|---|
| Brain | ||||||||
| Colon | ||||||||
| Heart | ||||||||
| Kidney | ||||||||
| Liver | ||||||||
| Cohort | HiST | HistoPrism | HEXST | STFlow | BLEEP | SEPAL | Linear | CGL-Linear |
|---|---|---|---|---|---|---|---|---|
| Brain | ||||||||
| Colon | ||||||||
| Heart | ||||||||
| Kidney | ||||||||
| Liver | ||||||||
| Cohort | HiST | HistoPrism | HEXST | STFlow | BLEEP | SEPAL | Linear | CGL-Linear |
|---|---|---|---|---|---|---|---|---|
| Brain | ||||||||
| Colon | ||||||||
| Heart | ||||||||
| Kidney | ||||||||
| Liver | ||||||||
| Cohort | HiST | HistoPrism | HEXST | STFlow | BLEEP | SEPAL | Linear | CGL-Linear |
|---|---|---|---|---|---|---|---|---|
| Brain | ||||||||
| Colon | ||||||||
| Heart | ||||||||
| Kidney | ||||||||
| Liver | ||||||||
| Cohort | HiST | HistoPrism | HEXST | STFlow | BLEEP | SEPAL | Linear | CGL-Linear |
|---|---|---|---|---|---|---|---|---|
| Brain | ||||||||
| Colon | ||||||||
| Heart | ||||||||
| Kidney | ||||||||
| Liver | ||||||||
| Cohort | HiST | HistoPrism | HEXST | STFlow | BLEEP | SEPAL | Linear | CGL-Linear |
|---|---|---|---|---|---|---|---|---|
| Brain | ||||||||
| Colon | ||||||||
| Heart | ||||||||
| Kidney | ||||||||
| Liver | ||||||||
| Cohort | HiST | HistoPrism | HEXST | STFlow | BLEEP | SEPAL | Linear | CGL-Linear |
|---|---|---|---|---|---|---|---|---|
| Brain | ||||||||
| Colon | ||||||||
| Heart | ||||||||
| Kidney | ||||||||
| Liver | ||||||||
| Cohort | HiST | HistoPrism | HEXST | STFlow | BLEEP | SEPAL | Linear | CGL-Linear |
|---|---|---|---|---|---|---|---|---|
| Brain | ||||||||
| Colon | ||||||||
| Heart | ||||||||
| Kidney | ||||||||
| Liver | ||||||||
| Cohort | HiST | HistoPrism | HEXST | STFlow | BLEEP | SEPAL | Linear | CGL-Linear |
|---|---|---|---|---|---|---|---|---|
| Brain | ||||||||
| Colon | ||||||||
| Heart | ||||||||
| Kidney | ||||||||
| Liver | ||||||||
Gene Expression Prediction from Histopathology Images
Gene expression prediction from histopathology images links H&E tissue morphology to molecular profiles, enabling virtual spatial transcriptomics without sequencing every sample.
See all 45 papers