Frozen in Time: How a Physics Foundation Model Forgets Under Sim-to-Real Adaptation
Haaris Mian ⋅ Miles Cranmer ⋅ Alessandro Favero ⋅ Payel Mukhopadhyay
Abstract
Scientific foundation models (SciFMs) have achieved remarkable performance on continuum dynamics prediction across a wide range of PDEs, including the ability to efficiently adapt to unseen data distributions with additional finetuning. As with all neural architectures, finetuning such models comes with the risk of $\textit{catastrophic forgetting}$, whereby adapting to new tasks degrades prior knowledge of the model. Here, we consider Walrus, a FM for continuum dynamics, and investigate the extent to which forgetting occurs over the course of sim-to-real adaptation on OOD fluid dynamics datasets from the pretrained baseline. Moreover, we characterize the potential drivers of forgetting, including field and dataset dependence and architectural choices such as normalization. We observe rapid, catastrophic (up to 35$\times$) degradation of rollout performance across several pretraining datasets within the first few epochs of finetuning as well as significant forgetting of the simulated flows upon real data exposure. We study catastrophic forgetting in Walrus-1.3B, a SciFM for continuum dynamics, over the course of a two-stage sim-to-real adaptation: first finetuning on simulated flow around a cylinder and then adapting to experimental data of the same system. We find that finetuning on the simulations rapidly degrades rollout performance on the original pretraining datasets, with one-step errors increasing up to 35$\times$ after just 20 epochs. Subsequent adaptation to real data further erodes performance on the simulated task. We characterize the potential drivers of forgetting, such as dataset dependence and task representation similarity. We then show that replaying as little as 1\% simulated data during real-data finetuning recovers most of the lost simulation performance while retaining effective adaptation to the experimental dynamics. Our results identify catastrophic forgetting as a practical obstacle to sim-to-real adaptation of SciFMs and show that lightweight replay strategies significantly improve retention.
Chat is not available.
Successful Page Load