Skip to yearly menu bar Skip to main content


Poster

Poseidon: Efficient Foundation Models for PDEs

Maximilian Herde · Bogdan Raonic · Tobias Rohner · Roger Käppeli · Roberto Molinaro · Emmanuel de Bézenac · Siddhartha Mishra


Abstract:

We introduce Poseidon, a foundation model for learning the solution operators of PDEs. It is based on a multiscale operator transformer, with time-conditioned layer norms that enable continuous-in-time evaluations. A novel training strategy leveraging the semi-group property of time-dependent PDEs to allow for significant scaling-up of the training data is also proposed. Poseidon is pretrained on a diverse, large scale dataset for the governing equations of fluid dynamics. It is evaluated on a suite of 15 challenging downstream tasks that include a wide variety of PDE types and operators. We show that Poseidon exhibits excellent performance across the board by outperforming baselines significantly, both in terms of sample efficiency and accuracy. Poseidon also generalizes very well to new physics that is not seen during pretraining. Moreover, Poseidon scales with respect to model and data size, both for pretraining and for downstream tasks, demonstrating its potential as an efficient general purpose PDE foundation model.

Live content is unavailable. Log in and register to view live content