Saved in:
Bibliographic Details
Main Authors: Li, Xinyi, Li, Zongyi, Kovachki, Nikola, Anandkumar, Anima
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2507.20065
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912504196628480
author Li, Xinyi
Li, Zongyi
Kovachki, Nikola
Anandkumar, Anima
author_facet Li, Xinyi
Li, Zongyi
Kovachki, Nikola
Anandkumar, Anima
contents We propose integrating optimal transport (OT) into operator learning for partial differential equations (PDEs) on complex geometries. Classical geometric learning methods typically represent domains as meshes, graphs, or point clouds. Our approach generalizes discretized meshes to mesh density functions, formulating geometry embedding as an OT problem that maps these functions to a uniform density in a reference space. Compared to previous methods relying on interpolation or shared deformation, our OT-based method employs instance-dependent deformation, offering enhanced flexibility and effectiveness. For 3D simulations focused on surfaces, our OT-based neural operator embeds the surface geometry into a 2D parameterized latent space. By performing computations directly on this 2D representation of the surface manifold, it achieves significant computational efficiency gains compared to volumetric simulation. Experiments with Reynolds-averaged Navier-Stokes equations (RANS) on the ShapeNet-Car and DrivAerNet-Car datasets show that our method achieves better accuracy and also reduces computational expenses in terms of both time and memory usage compared to existing machine learning models. Additionally, our model demonstrates significantly improved accuracy on the FlowBench dataset, underscoring the benefits of employing instance-dependent deformation for datasets with highly variable geometries.
format Preprint
id arxiv_https___arxiv_org_abs_2507_20065
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Geometric Operator Learning with Optimal Transport
Li, Xinyi
Li, Zongyi
Kovachki, Nikola
Anandkumar, Anima
Machine Learning
We propose integrating optimal transport (OT) into operator learning for partial differential equations (PDEs) on complex geometries. Classical geometric learning methods typically represent domains as meshes, graphs, or point clouds. Our approach generalizes discretized meshes to mesh density functions, formulating geometry embedding as an OT problem that maps these functions to a uniform density in a reference space. Compared to previous methods relying on interpolation or shared deformation, our OT-based method employs instance-dependent deformation, offering enhanced flexibility and effectiveness. For 3D simulations focused on surfaces, our OT-based neural operator embeds the surface geometry into a 2D parameterized latent space. By performing computations directly on this 2D representation of the surface manifold, it achieves significant computational efficiency gains compared to volumetric simulation. Experiments with Reynolds-averaged Navier-Stokes equations (RANS) on the ShapeNet-Car and DrivAerNet-Car datasets show that our method achieves better accuracy and also reduces computational expenses in terms of both time and memory usage compared to existing machine learning models. Additionally, our model demonstrates significantly improved accuracy on the FlowBench dataset, underscoring the benefits of employing instance-dependent deformation for datasets with highly variable geometries.
title Geometric Operator Learning with Optimal Transport
topic Machine Learning
url https://arxiv.org/abs/2507.20065