← Search

Shentao Yang

5 accepted papers

2024

A Dense Reward View on Aligning Text-to-Image Diffusion with Preference

ICML 2024poster

Aligning text-to-image diffusion model (T2I) with preference has been gaining increasing research attention. While prior works exist on directly optimizing T2I by preference data, these methods are developed under the bandit assumption of a latent reward on the entire diffusion reverse chain, while…

2023

Fantastic Rewards and How to Tame Them: A Case Study on Reward Learning for Task-oriented Dialogue Systems

ICLR 2023poster

When learning task-oriented dialogue (ToD) agents, reinforcement learning (RL) techniques can naturally be utilized to train dialogue strategies to achieve user-specific goals. Prior works mainly focus on adopting advanced RL techniques to train the ToD agents, while the design of the reward functio…

2023

Preference-grounded Token-level Guidance for Language Model Fine-tuning

NeurIPS 2023poster

Aligning language models (LMs) with preferences is an important problem in natural language generation. A key challenge is that preferences are typically provided at the *sequence level* while LM training and generation both occur at the *token level*. There is, therefore, a *granularity mismatch* b…

2022

A Unified Framework for Alternating Offline Model Training and Policy Learning

NeurIPS 2022accept

In offline model-based reinforcement learning (offline MBRL), we learn a dynamic model from historically collected data, and subsequently utilize the learned model and fixed datasets for policy learning, without further interacting with the environment. Offline MBRL algorithms can improve the effici…

2022

Regularizing a Model-based Policy Stationary Distribution to Stabilize Offline Reinforcement Learning

ICML 2022spotlight

Offline reinforcement learning (RL) extends the paradigm of classical RL algorithms to purely learning from static datasets, without interacting with the underlying environment during the learning process. A key challenge of offline RL is the instability of policy training, caused by the mismatch be…