arXiv Machine Learning

Where to Compute and How to Interact: Operator-Readable Adaptation with Gauge-Aware Transport

arXiv Machine Learning
3d ago

A Variational Optimal Transport Operator on Incompressible Flow

The paper introduces the Variational Incompressible Optimal Transport (VIOT) operator, a generative neural operator that predicts divergence‑free velocity fields for incompressible density transport. VIOT combines a stream‑function representation, a regularized transport objective, and a Fourier Neural Operator backbone to amortize the solve across new source‑target pairs and grid resolutions. Experiments on 2D and 3D benchmarks show that VIOT produces full transport trajectories in seconds, achieving roughly a $10^4 imes$ speedup over per‑instance baselines that require hours of optimization.

By Jinjin He, Shenyifan Lu, Sinan Wang, Zhiqi Li, Duowen Chen, Bo Zhu
arXiv Machine Learning
Sep 7

Solution-space heterogeneity shapes federated learning dynamics across partial differential equations

The paper introduces a new federated learning protocol for partial differential equations called solution-space PDE-Dirichlet, which transforms continuous supervised responses into reusable solution bins and measures client separation via optimal transport. It establishes an exact inverse relationship between population allocation heterogeneity and Dirichlet concentration, and shows how response heterogeneity can cause gradient disagreement, local-update dispersion, and parameter divergence. Experiments on seven PDE tasks, three neural-operator families, and five random seeds demonstrate that lower concentration consistently increases solution distance and optimization heterogeneity, with the most pronounced error increase observed in low-viscosity Burgers equations.

By Ping Luo, Jiahuan Wang, Ziqing Wen, Tao Sun, Dongsheng Li
arXiv AI
Aug 17

ArGEnT: Arbitrary Geometry-encoded Transformer for Operator Learning

arXiv:2602. 11626v3 Announce Type: replace-cross Abstract: Learning solution operators on arbitrary geometries remains a central challenge in scientific machine learning, especially for many-query simulation, physics-informed learning, and evolving geometries requiring accurate, geometry-aware predictions at arbitrary spatial locations.

By Wenqian Chen, Zhi-Feng Wei, Yucheng Fu, Michael Penwarden, Pratanu Roy, Panos Stinis