← Search

Ashok Makkuva

3 accepted papers

2020

Optimal transport mapping via input convex neural networks

ICML 2020poster

In this paper, we present a novel and principled approach to learn the optimal transport between two distributions, from samples. Guided by the optimal transport theory, we learn the optimal Kantorovich potential which induces the optimal transport map. This involves learning two convex functions, b…

Cited by 240SourcePDFScholar
2019

Breaking the gridlock in Mixture-of-Experts: Consistent and Efficient Algorithms

ICML 2019oral

Mixture-of-Experts (MoE) is a widely popular model for ensemble learning and is a basic building block of highly successful modern neural networks as well as a component in Gated Recurrent Units (GRU) and Attention networks. However, present algorithms for learning MoE, including the EM algorithm an…

Cited by 33SourcePDFScholar