182 citations · 280 across the 14 of their papers we have counts for
18 papers · 1 filter
Offline Imitation Learning with Suboptimal Demonstrations via Relaxed Distribution Matching
Lantao Yu, Tianhe Yu, Jiaming Song +2
Offline imitation learning (IL) promises the ability to learn performant policies from pre-collected demonstrations without interactions with the environment. However, imitating be…
A General Recipe for Likelihood-free Bayesian Optimization
Jiaming Song, Lantao Yu, Willie Neiswanger +1
The acquisition function, a critical component in Bayesian optimization (BO), can often be written as the expectation of a utility function under a surrogate model. However, to ens…
GeoDiff: a Geometric Diffusion Model for Molecular Conformation Generation
Minkai Xu, Lantao Yu, Yang Song +3
Predicting molecular conformations from molecular graphs is a fundamental problem in cheminformatics and drug discovery. Recently, significant progress has been achieved with machi…
A Unified Framework for Multi-distribution Density Ratio Estimation
Lantao Yu, Yujia Jin, Stefano Ermon
Binary density ratio estimation (DRE), the problem of estimating the ratio given their empirical samples, provides the foundation for many state-of-the-art machine learni…
Pseudo-Spherical Contrastive Divergence
Lantao Yu, Jiaming Song, Yang Song +1
Energy-based models (EBMs) offer flexible distribution parametrization. However, due to the intractable partition function, they are typically trained via contrastive divergence fo…
Multi-Agent Imitation Learning with Copulas
Hongwei Wang, Lantao Yu, Zhangjie Cao +1
Multi-agent imitation learning aims to train multiple agents to perform tasks from demonstrations by learning a mapping between observations and actions, which is essential for und…