← Search

Mengjiao Yang

11 accepted papers

2022

Making Linear MDPs Practical via Contrastive Representation Learning

ICML 2022spotlight

It is common to address the curse of dimensionality in Markov decision processes (MDPs) by exploiting low-rank representations. This motivates much of the recent theoretical study on linear MDPs. However, most approaches require a given representation under unrealistic assumptions about the normaliz…

Cited by 55SourcePDFScholar
2022

Marginal Distribution Adaptation for Discrete Sets via Module-Oriented Divergence Minimization

ICML 2022spotlight

Distributions over discrete sets capture the essential statistics including the high-order correlation among elements. Such information provides powerful insight for decision making across various application domains, e.g., product assortment based on product distribution in shopping carts. While de…

Cited by 2SourcePDFScholar
2022

Offline Policy Selection under Uncertainty

AISTATS 2022poster

The presence of uncertainty in policy evaluation significantly complicates the process of policy ranking and selection in real-world settings. We formally consider offline policy selection as learning preferences over a set of policy prospects given a fixed experience dataset. While one can select o…

2021

Benchmarks for Deep Off-Policy Evaluation

ICLR 2021poster

Off-policy evaluation (OPE) holds the promise of being able to leverage large, offline datasets for both evaluating and selecting complex policies for decision making. The ability to learn offline is particularly important in many real-world domains, such as in healthcare, recommender systems, or ro…

2021

Combiner: Full Attention Transformer with Sparse Computation Cost

NeurIPS 2021spotlight

Transformers provide a class of expressive architectures that are extremely effective for sequence modeling. However, the key limitation of transformers is their quadratic memory and time complexity $\mathcal{O}(L^2)$ with respect to the sequence length in attention layers, which restricts applicati…

2021

Provable Representation Learning for Imitation with Contrastive Fourier Features

NeurIPS 2021poster

In imitation learning, it is common to learn a behavior policy to match an unknown target policy via max-likelihood training on a collected set of target demonstrations. In this work, we consider using offline experience datasets -- potentially far from the target distribution -- to learn low-dimens…

2021

Representation Matters: Offline Pretraining for Sequential Decision Making

ICML 2021spotlight

The recent success of supervised learning methods on ever larger offline datasets has spurred interest in the reinforcement learning (RL) field to investigate whether the same paradigms can be translated to RL algorithms. This research area, known as offline RL, has largely focused on offline policy…

2021

Towards Automatic Evaluation of Dialog Systems: A Model-Free Off-Policy Evaluation Approach

EMNLP 2021main

Reliable automatic evaluation of dialogue systems under an interactive environment has long been overdue. An ideal environment for evaluating dialog systems, also known as the Turing test, needs to involve human interaction, which is usually not affordable for large-scale experiments. Though researc…

2020

Off-Policy Evaluation via the Regularized Lagrangian

NeurIPS 2020poster

The recently proposed distribution correction estimation (DICE) family of estimators has advanced the state of the art in off-policy evaluation from behavior-agnostic data. While these estimators all perform some form of stationary distribution correction, they arise from different derivations and o…

Cited by 134SourcePDFScholar