Go to ICLR 2026 Workshop Gen2 homepageGenerative Modeling of Spatial Transcriptomics via Gaussian Mixture Flow Matching
Track: Tiny / short paper (2-4 pages)
Keywords: Flow Matching, Generative Modeling, Gaussian Mixture Model, Spatial Transcriptomics, Spatially Aware Embeddings, Embryonic Development
TL;DR: We propose a Gaussian-Mixture based Flow Matching for modeling gene expression in spatial transcriptomics data to explicitly model the distribution of cell types in the data generating process.
Abstract: Spatial transcriptomics enables the joint analysis of cellular gene expression and spatial organization, offering new insights into tissue development and function. Existing graph- and embedding-based methods capture spatial patterns but remain descriptive and non-generative. Spatial transcriptomics is still an expensive and resource-intensive assay, limiting its widespread application. Consequently, there is a growing need for generative models capable of accurately simulating gene expression within spatial contexts, particularly in settings where experimental data acquisition is impractical or cost-prohibitive. In this work, we revisit the generative modeling of spatially informed cell embeddings derived from gene expression and spatial information and apply a Gaussian Mixture Flow model (GMFlow) to explicitly model the multinomial nature of cell type distributions during generation. Using a mouse embryonic development dataset, we find that both single-Gaussian conditional flow and GMFlow models learn cell type distributions over different developmental timepoints with comparable accuracy, while GMFlow results in generations that are closer to ground truth cell representations, suggesting that GMFlow may enable more realistic simulation of cellular landscape in spatial transcriptomics data.
Anonymization: This submission has been anonymized for double-blind review via the removal of identifying information such as names, affiliations, and identifying URLs.
Submission Number: 75
Loading