Abstract
We present a unified framework for solving partial differential equations(PDEs) using video-inpainting diffusion transformer models. Unlike existingmethods that devise specialized strategies for either forward or inverseproblems under full or partial observation, our approach unifies these tasksunder a single, flexible generative framework. Specifically, we recastPDE-solving as a generalized inpainting problem, e.g., treating forwardprediction as inferring missing spatiotemporal information of future statesfrom initial conditions. To this end, we design a transformer-basedarchitecture that conditions on arbitrary patterns of known data to infermissing values across time and space. Our method proposes pixel-space videodiffusion models for fine-grained, high-fidelity inpainting and conditioning,while enhancing computational efficiency through hierarchical modeling.Extensive experiments show that our video inpainting-based diffusion modeloffers an accurate and versatile solution across a wide range of PDEs andproblem setups, outperforming state-of-the-art baselines.