DeepAI AI Chat
Log In Sign Up

Physics Informed Convex Artificial Neural Networks (PICANNs) for Optimal Transport based Density Estimation

by   Amanpreet Singh, et al.

Optimal Mass Transport (OMT) is a well studied problem with a variety of applications in a diverse set of fields ranging from Physics to Computer Vision and in particular Statistics and Data Science. Since the original formulation of Monge in 1781 significant theoretical progress been made on the existence, uniqueness and properties of the optimal transport maps. The actual numerical computation of the transport maps, particularly in high dimensions, remains a challenging problem. By Brenier's theorem, the continuous OMT problem can be reduced to that of solving a non-linear PDE of Monge-Ampere type whose solution is a convex function. In this paper, building on recent developments of input convex neural networks and physics informed neural networks for solving PDE's, we propose a Deep Learning approach to solve the continuous OMT problem. To demonstrate the versatility of our framework we focus on the ubiquitous density estimation and generative modeling tasks in statistics and machine learning. Finally as an example we show how our framework can be incorporated with an autoencoder to estimate an effective probabilistic generative model.


page 8

page 11

page 13


Optimal transport mapping via input convex neural networks

In this paper, we present a novel and principled approach to learn the o...

A scalable deep learning approach for solving high-dimensional dynamic optimal transport

The dynamic formulation of optimal transport has attracted growing inter...

Optimal Transport Based Refinement of Physics-Informed Neural Networks

In this paper, we propose a refinement strategy to the well-known Physic...

A Distributed Framework for the Construction of Transport Maps

The need to reason about uncertainty in large, complex, and multi-modal ...

Metasurfaces and Optimal transport

This paper provides a theoretical and numerical approach to show existen...

Taming Hyperparameter Tuning in Continuous Normalizing Flows Using the JKO Scheme

A normalizing flow (NF) is a mapping that transforms a chosen probabilit...

Learning Transport Processes with Machine Intelligence

We present a machine learning based approach to address the study of tra...