Search papers, labs, and topics across Lattice.
This paper introduces a joint flow matching approach that models cellular perturbations by simultaneously handling cell latents and drug concentrations in a dual-timestep framework. This method allows for continuous dose-conditioned morphing of single cells, overcoming limitations of existing models that either ignore concentration nuances or treat them as discrete labels. The results show that the proposed approach not only improves per-concentration metrics on two compounds compared to baseline methods but also enables concentration estimation from cell morphology and generalizes to unseen doses.
Continuous dose control in cellular modeling is now achievable, unlocking new avenues for precision in drug response predictions.
Generative modeling has shown increasing promise for predicting cellular perturbation effects under chemical compound treatments. Existing approaches either model perturbation as a distribution-to-distribution mapping without explicit concentration handling, or treat concentration as a discrete class label, precluding continuous dose control. We introduce a joint flow matching approach that simultaneously models cell latents and drug concentration via a dual-timestep formulation, enabling dose-conditioned single-cell morphing through the invertibility of flow matching. The joint formulation induces a monotonic dose-response geometry in latent space and additionally supports concentration estimation from cell morphology. As proof of concept, we further demonstrate generalization to an unseen dose held out during training. Empirically, our method achieves competitive or improved per-concentration metrics on two compounds compared with representative baselines, while enabling capabilities structurally unavailable to discrete-class methods.
