用小块神经局部求解,拼出大域物理问题,通用性强且可扩展。
Neural-Schwarz Tiling for Geometry-Universal PDE Solving at Scale

- 在3×3×3小块上训练神经局部求解器,学习局部物理规律。
- 通过施瓦茨迭代和单位分解法,将局部解拼接成全局一致解。
- 支持大尺度、复杂几何的跨尺寸、跨边界条件泛化,适合大规模部署。
现有学习型偏微分方程(PDE)求解器多采用全局代理范式:神经算子在特定几何、边界条件和系数分布下,从完整问题描述映射到完整解场。这虽实现固定问题族内的快速推理,但限制了新领域复用,且大规模部署依赖昂贵的问题特异性数据生成。本文提出NEST(Neural-Schwarz Tiling),一种从全域求解算子转向可复用局部物理求解器的局部-全局框架。核心思想是:尽管全局解受几何、尺度和边界条件影响,但小邻域的物理响应可通过局部学习,并经经典域分解组合成全局解。NEST在包含多样局部几何与边界/界面数据的最小体素块(3×3×3)上训练神经算子。推理时,未见体素化域被划分为重叠块,局部求解器逐块应用,全局一致性通过施瓦茨迭代与单位分解组装强制满足。由此,泛化能力从单一神经模型转移至局部物理学习与算法级全局组装的结合。我们在可压缩neo-Hookean固体的非线性静力平衡问题上实例化NEST,评估其在远超训练块尺度的大型、几何复杂的三维域上的表现。结果表明,通过施瓦茨迭代耦合的局部神经构建块,为可扩展的、跨域尺寸、形状与边界条件配置的泛化求解器提供了可复用的局部训练路径。
原文摘要 · Abstract (English)
Most learned PDE solvers follow a global-surrogate paradigm: a neural operator is trained to map full problem descriptions to full solution fields for a prescribed distribution of geometries, boundary conditions, and coefficients. This has enabled fast inference within fixed problem families, but limits reuse across new domains and makes large-scale deployment dependent on expensive problem-specific data generation. We introduce $\textbf{NEST}$ ($\textbf{Ne}$ural-$\textbf{S}$chwarz $\textbf{T}$iling), a local-to-global framework that shifts learning from full-domain solution operators to reusable local physical solvers. The central premise is that, although global PDE solutions depend on geometry, scale, and boundary conditions, the physical response on small neighborhoods can be learned locally and composed into global solutions through classical domain decomposition. NEST learns a neural operator on minimal voxel patches ($3 \times 3 \times 3$) with diverse local geometries and boundary/interface data. At inference time, an unseen voxelized domain is tiled into overlapping patches, the learned local solver is applied patchwise, and global consistency is enforced through iterative Schwarz coupling with partition-of-unity assembly. In this way, generalization is shifted from a monolithic neural model to the combination of local physics learning and algorithmic global assembly. We instantiate NEST on nonlinear static equilibrium in compressible neo-Hookean solids and evaluate it on large, geometrically complex 3D domains far outside the scale of the training patches. Our results show that local neural building blocks, coupled through Schwarz iteration, offer a reusable local-training path toward scalable learned PDE solvers that generalize across domain size, shape, and boundary-condition configurations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。