arXiv · 2512.01738
MSPT: Efficient Large-Scale Physical Modeling via Parallelized Multi-Scale Attention
Abstract
A key scalability challenge in neural solvers for industrial-scale physics simulations is efficiently capturing both fine-grained local interactions and long-range global dependencies across millions of spatial elements. We introduce the Multi-Scale Patch Transformer (MSPT), an architecture that combines local point attention within patches with global attention to coarse patch-level representations. To partition the input domain into spatially-coherent patches, we employ ball trees, which handle irregular geometries efficiently. This dual-scale design enables MSPT to scale to millions of points on a single GPU. We validate our method on standard PDE benchmarks (elasticity, plasticity, fluid dynamics, porous flow) and large-scale aerodynamic datasets (ShapeNet-Car, Ahmed-ML), achieving state-of-the-art accuracy with substantially lower memory footprint and computational cost.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Pedro M. P. Curvo, Jan-Willem van de Meent, Maksim Zhdanov. 2025-12-01. MSPT: Efficient Large-Scale Physical Modeling via Parallelized Multi-Scale Attention. https://arxiv.org/abs/2512.01738
Cite the original work for its findings. Save a collection to share your selection of sources.