VIOT · arXiv 2026

A Variational Optimal Transport Operator on Incompressible Flow

Georgia Institute of Technology

Green-to-gold fluid on a pale background sequentially forms the fourteen characters of the opening couplet of 春江花月夜.

Incompressible transport · Chinese poetry

Poetry in motion

春江潮水连海平,
海上明月共潮生。

Zhang Ruoxu · A Moonlit Night on the Spring River

A single trained operator transports density through the opening couplet, forming each character through a continuous sequence of fluid motion.

Full GIF

Abstract

We present the Variational Incompressible Optimal Transport (VIOT) operator, a generative neural operator for amortized incompressible density transport. Given a new source-target density pair, VIOT predicts a divergence-free velocity field and generates the full transport trajectory by feed-forward inference, replacing the hour-scale per-pair optimization used by adjoint fluid solvers and differentiable simulation baselines.

The system consists of three components: a stream-function or vector-potential representation that enforces incompressibility by construction, a regularized incompressible transport objective that balances endpoint accuracy and flow smoothness, and a Fourier Neural Operator backbone that amortizes the solve across new pairs and grid resolutions. Together, these components make incompressible transport a reusable neural operator that facilitates various transport processes. Further, the generative capability extends beyond the training distribution, with VIOT producing incompressible transports for user-drawn source-target pairs in a real-time interactive system.

We demonstrate VIOT on 2D and 3D density-transport benchmarks. Both 2D and 3D rollouts complete in seconds per pair, while per-instance baselines in our 2D comparisons optimize each new pair from scratch and require on the order of an hour, a roughly 104× online speedup.

From density pairs to fluid motion

VIOT pipeline: current density, target density and time enter an FNO; a predicted potential defines a divergence-free velocity, which advects the density.
The operator receives the current density, target density, and time. It predicts a stream function in 2D or a vector potential in 3D, from which a divergence-free velocity is constructed. Repeated operator queries and numerical advection produce the transport trajectory.

Transport in 2D and 3D

VIOT transports 2D digits and silhouettes, 3D letters, human poses, and a sphere into an airplane; red boxes mark target keyframes.
Examples generated by trained VIOT operators: 2D MNIST digits and MPEG-7 silhouettes, a 3D “SMOKE” sequence, a 3D human-pose sequence, and sphere-to-airplane transport. Red boxes mark target keyframes; insets show the ground truth.

BibTeX

@misc{he2026variational,
  title={A Variational Optimal Transport Operator on Incompressible Flow},
  author={Jinjin He and Shenyifan Lu and Sinan Wang and Zhiqi Li and Duowen Chen and Bo Zhu},
  year={2026},
  eprint={2609.13729},
  archivePrefix={arXiv},
  primaryClass={cs.LG},
  url={https://arxiv.org/abs/2609.13729}
}