212 citations · 568 across the 48 of their papers we have counts for
13 papers · 1 filter
Taming Hyperparameter Tuning in Continuous Normalizing Flows Using the JKO Scheme
Alexander Vidal, Samy Wu Fung, Luis Tenorio +2
A normalizing flow (NF) is a mapping that transforms a chosen probability distribution to a normal distribution. Such flows are a common technique used for data generation and dens…
Improving Generative Flow Networks with Path Regularization
Anh Do, Duy Dinh, Tan Nguyen +3
Generative Flow Networks (GFlowNets) are recently proposed models for learning stochastic policies that generate compositional objects by sequences of actions with the probability…
Momentum Transformer: Closing the Performance Gap Between Self-attention and Its Linearization
Tan Nguyen, Richard G. Baraniuk, Robert M. Kirby +2
Transformers have achieved remarkable success in sequence modeling and beyond but suffer from quadratic computational and memory complexities with respect to the length of the inpu…
A primal-dual approach for solving conservation laws with implicit in time approximations
Siting Liu, Stanley Osher, Wuchen Li +1
In this work, we propose a novel framework for the numerical solution of time-dependent conservation laws with implicit schemes via primal-dual hybrid gradient methods. We solve an…
Multi-Agent Shape Control with Optimal Transport
Alex Tong Lin, Stanley J. Osher
We introduce a method called MASCOT (Multi-Agent Shape Control with Optimal Transport) to compute optimal control solutions of agents with shape/formation/density constraints. For…
Computational Mean-field Games on Manifolds
Jiajia Yu, Rongjie Lai, Wuchen Li +1
Conventional Mean-field games/control study the behavior of a large number of rational agents moving in the Euclidean spaces. In this work, we explore the mean-field games on Riema…