DriftST: One-Step Generative Inference of Spatial Transcriptomics from H\&E Histology

2026-07-06Computer Vision and Pattern Recognition

Computer Vision and Pattern Recognition
AI summary

The authors developed DriftST, a new method to predict gene activity patterns in tissue using regular microscope images, avoiding the need for expensive gene-measuring technology. Their approach uses a special model that efficiently generates gene expression data while considering relationships between genes and their varying importance. DriftST works well for different scales, like groups of cells or single cells, and outperforms previous methods on multiple datasets.

Spatial TranscriptomicsHematoxylin and Eosin staininggene expression inferencegenerative modelsattention mechanismco-expressioncell-level resolutionspot-level resolutiongene panel
Authors
Yuhang Yang, Yonggan Bu, Shengyuan Zhou, Yiming Luo, Kai Zhang
Abstract
Spatial Transcriptomics (ST) measures gene expression while preserving spatial context, but its high cost and low throughput leave public datasets small. Inferring expression directly from widely available Hematoxylin and Eosin (H&E) stained histology offers a cost-effective alternative. However, existing approaches face several limitations: regression methods over-smooth toward the conditional mean, while generative methods are faithful but require slow multi-step inference; most methods treat genes as independent and equally important, ignoring inter-gene dependencies and heterogeneous gene informativeness; and most are tailored to a single resolution, either spot-level or cell-level. To address these issues, we propose DriftST, a unified framework for inferring spatially resolved gene expression from H&E images. DriftST builds on a Cellular Drifting generative model that learns a direct drift from a histology-conditioned source to the expression distribution, retaining generative expressiveness while enabling efficient one-step generation. To capture gene structure, we introduce the STransformer, which combines a co-expression attention module for inter-gene dependencies with a gene residual gate for differential gene importance. Operating on a generic gene-panel representation, DriftST applies directly to both spot-level and cell-level data in one framework, and extensive experiments across diverse tissues and platforms show that it achieves state-of-the-art performance at both resolutions.