Joint Flow Matching Enables Continuous Dose-Conditioned Cell Morphing

TL;DR - A joint flow-matching model continuously conditions predicted single-cell morphology on drug concentration, overcoming methods that ignore dosage or represent it as discrete classes. It also estimates concentration from morphology and generalizes to a held-out dose.

• Uses dual timesteps to jointly model cell latents and drug concentration.
• Flow invertibility enables dose-conditioned cell morphing and concentration estimation.
• Produces a monotonic dose-response geometry in latent space.
• Achieves competitive or improved per-concentration metrics on two compounds versus representative baselines.