Foundation models, such as large language models, have demonstrated success in addressing various language and image processing tasks. In this work, we introduce a multi-modal foundation model for scientific problems, named PROSE-PDE. Our model, designed for bi-modality to bi-modality learning, is a multi-operator learning approach which can predict future states of spatiotemporal systems while concurrently learning the underlying governing equations of the physical system. Specifically, we focus on multi-operator learning by training distinct one-dimensional time-dependent nonlinear constant coefficient partial differential equations, with potential applications to many physical applications including physics, geology, and biology. More importantly, we provide three extrapolation studies to demonstrate that PROSE-PDE can generalize physical features through the robust training of multiple operators and that the proposed model can extrapolate to predict PDE solutions whose models or data were unseen during the training. Furthermore, we show through systematic numerical experiments that the utilization of the symbolic modality in our model effectively resolves the well-posedness problems with training multiple operators and thus enhances our model's predictive capabilities.
翻译:基础模型(例如大型语言模型)已在处理各种语言和图像任务中展现出成功。在本工作中,我们针对科学问题引入了一种多模态基础模型,命名为PROSE-PDE。该模型专为双模态到双模态学习而设计,是一种多算子学习方法,能够在预测时空系统未来状态的同时,同步学习物理系统的潜在控制方程。具体而言,我们聚焦于多算子学习,通过训练不同的常系数非线性一维时间相关偏微分方程来实现,其潜在应用覆盖物理学、地质学和生物学等多个物理领域。更重要的是,我们提供了三项外推研究,以证明PROSE-PDE可通过多算子的鲁棒训练泛化物理特征,并且该模型能够外推预测其模型或数据在训练中未曾见过的偏微分方程解。此外,我们通过系统数值实验表明,模型中符号模态的利用有效解决了训练多个算子时的适定性问题,从而增强了模型的预测能力。