no code implementations • 6 Mar 2024 • Nazar Buzun, Maksim Bobrin, Dmitry V. Dylov
We present a new extension for Neural Optimal Transport (NOT) training procedure, capable of accurately and efficiently estimating optimal transportation plan via specific regularisation on conjugate potentials.
no code implementations • 20 Feb 2024 • Maksim Bobrin, Nazar Buzun, Dmitrii Krylov, Dmitry V. Dylov
We report that AILOT outperforms state-of-the art offline imitation learning algorithms on D4RL benchmarks and improves the performance of other offline RL algorithms in the sparse-reward tasks.