Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper explores transferring a Continuum Dynamics Foundation Model (Walrus), pretrained on simulated physical systems, to predict Drosophila embryo tissue dynamics. Despite Fourier Neural Operators outperforming in-distribution, Walrus achieved superior zero-shot performance under unseen temperature and genetic perturbations, accurately capturing biologically meaningful velocity trajectories. This suggests that representations learned from diverse dynamical systems can generalize to experimental morphogenesis, particularly in low-data scenarios.
A foundation model pretrained on simulated physics predicts complex embryo development dynamics zero-shot, outperforming specialized models under significant environmental and genetic perturbations.
Morphogenesis is driven by the interplay of genetic regulation and tissue mechanics. In vivo video data is limited, particularly under genetic mutations and environmental perturbations, making generalization beyond the most representative conditions in the dataset challenging. However, at tissue scales, the dynamics of Drosophila can be described as a continuum flow, motivating the usage of representations learned across other physical systems as priors for developmental dynamics. We investigate finetuning Walrus, a Continuum Dynamics Foundation Model pretrained across a broad set of simulated continuous systems, on tissue velocity fields from only 12 wild-type Drosophila blastoderm videos. We evaluate autoregressive prediction both in-distribution and zero-shot under unseen temperature and genetic perturbations. Zero-shot refers to evaluation on perturbation conditions never seen during wild-type finetuning, without any additional training. In-distribution, a Fourier Neural Operator model trained from scratch achieves best performance on VRMSE. However, Walrus (CRPS) is the strongest model under the most severe distribution shift of unseen samples of mutated embryos and temperature conditions. Interestingly, predictions capture the shape of the mean surface tissue velocity trajectory and its separation between 17{\deg}C and 27{\deg}C, including the timing of the peak at Germ Band Extension onset, a biologically meaningful milestone of embryo development. Our results provide preliminary support that representations learned from different dynamical systems can transfer to experimental morphogenesis in developmental regimes, provided they admit a continuum description. More broadly, our results motivate further study of physics foundation models as a new complementary tool for the simulation of morphogenesis in low-data regimes.