Elucidating the Design Space of Flow Matching for Cellular Microscopy

View PDF HTML (experimental)

Abstract:Flow-matching generative models are increasingly used to simulate cell responses to biological perturbations. However, the design space for building such models is large and underexplored. We systematically analyse the design space of flow matching models for cell-microscopy images, finding that many popular techniques are unnecessary and can even hurt performance. We develop a simple, stable, and scalable recipe which we use to train our foundation model. We scale our model to two orders of magnitude larger than prior methods, achieving a two-fold FID and ten-fold KID improvement over prior methods. We then fine-tune our model with pre-trained molecular embeddings to achieve state-of-the-art performance simulating responses to unseen molecules. Code is available at this https URL

Subjects:

Computer Vision and Pattern Recognition (cs.CV)

Cite as: arXiv:2603.26790 [cs.CV]

(or arXiv:2603.26790v1 [cs.CV] for this version)

https://doi.org/10.48550/arXiv.2603.26790

arXiv-issued DOI via DataCite (pending registration)

Submission history

From: Jason Hartford [view email] [v1] Wed, 25 Mar 2026 15:18:37 UTC (14,452 KB)