← Search

Mohammad Gheshlaghi Azar

16 accepted papers

2025

Self-Improving Robust Preference Optimization

ICLR 2025poster

Online and offline $\mathtt{RLHF}$ methods, such as $\mathtt{PPO}$ and $\mathtt{DPO}$, have been highly successful in aligning AI with human preferences. Despite their success, however, these methods suffer from fundamental limitations: $\mathbf{(a)}$ Models trained with $\mathtt{RLHF}$ can learn fr…

Cited by 7SourcePDFScholar
2024

A General Theoretical Paradigm to Understand Learning from Human Preferences

AISTATS 2024poster

The prevalent deployment of learning from human preferences through reinforcement learning (RLHF) relies on two important approximations: the first assumes that pairwise preferences can be substituted with pointwise rewards. The second assumes that a reward model trained on these pointwise rewards c…

2024

Contrastive Policy Gradient: Aligning LLMs on sequence-level scores in a supervised-friendly fashion

EMNLP 2024main

Reinforcement Learning (RL) has been used to finetune Large Language Models (LLMs) using a reward model trained from preference data, to better align with human judgment. The recently introduced direct alignment methods, which are often simpler, more stable, and computationally lighter, can more dir…

Cited by 3SourcePDFScholar
2024

Nash Learning from Human Feedback

ICML 2024spotlight

Reinforcement learning from human feedback (RLHF) has emerged as the main paradigm for aligning large language models (LLMs) with human preferences. Traditionally, RLHF involves the initial step of learning a reward model from pairwise human feedback, i.e., expressed as preferences between pairs of…

Cited by 129SourcePDFScholar
2023

Regularization and Variance-Weighted Regression Achieves Minimax Optimality in Linear MDPs: Theory and Practice

ICML 2023poster

Mirror descent value iteration (MDVI), an abstraction of Kullback-Leibler (KL) and entropy-regularized reinforcement learning (RL), has served as the basis for recent high-performing practical RL algorithms. However, despite the use of function approximation in practice, the theoretical understandin…

2023

Understanding Self-Predictive Learning for Reinforcement Learning

ICML 2023poster

We study the learning dynamics of self-predictive learning for reinforcement learning, a family of algorithms that learn representations by minimizing the prediction error of their own future latent representations. Despite its recent empirical success, such algorithms have an apparent defect: trivi…

Cited by 34SourcePDFScholar
2022

BYOL-Explore: Exploration by Bootstrapped Prediction

NeurIPS 2022accept

We present BYOL-Explore, a conceptually simple yet general approach for curiosity-driven exploration in visually complex environments. BYOL-Explore learns the world representation, the world dynamics and the exploration policy all-together by optimizing a single prediction loss in the latent space w…

Cited by 76SourcePDFScholar
2022

Large-Scale Representation Learning on Graphs via Bootstrapping

ICLR 2022poster

Self-supervised learning provides a promising path towards eliminating the need for costly label information in representation learning on graphs. However, to achieve state-of-the-art performance, methods often need large numbers of negative examples and rely on complex augmentations. This can be…

2021

Drop, Swap, and Generate: A Self-Supervised Approach for Generating Neural Activity

NeurIPS 2021oral

Meaningful and simplified representations of neural activity can yield insights into how and what information is being processed within a neural circuit. However, without labels, finding representations that reveal the link between the brain and behavior can be challenging. Here, we introduce a nove…

Cited by 31SourcePDFScholar
2020

Bootstrap Latent-Predictive Representations for Multitask Reinforcement Learning

ICML 2020poster

Learning a good representation is an essential component for deep reinforcement learning (RL). Representation learning is especially important in multitask and partially observable settings where building a representation of the unknown environment is crucial to solve the tasks. Here we introduce Pr…

Cited by 173SourcePDFScholar
2020

Bootstrap Your Own Latent - A New Approach to Self-Supervised Learning

NeurIPS 2020oral

We introduce Bootstrap Your Own Latent (BYOL), a new approach to self-supervised image representation learning. BYOL relies on two neural networks, referred to as online and target networks, that interact and learn from each other. From an augmented view of an image, we train the online network to p…

2020

Fast computation of Nash Equilibria in Imperfect Information Games

ICML 2020poster

We introduce and analyze a class of algorithms, called Mirror Ascent against an Improved Opponent (MAIO), for computing Nash equilibria in two-player zero-sum games, both in normal form and in sequential form with imperfect information. These algorithms update the policy of each player with a mirror…

Cited by 12SourcePDFScholar
2019

Hindsight Credit Assignment

NeurIPS 2019spotlight

We consider the problem of efficient credit assignment in reinforcement learning. In order to efficiently and meaningfully utilize new data, we propose to explicitly assign credit to past decisions based on the likelihood of them having led to the observed outcome. This approach uses new information…

2018

Noisy Networks For Exploration

ICLR 2018poster

We introduce NoisyNet, a deep reinforcement learning agent with parametric noise added to its weights, and show that the induced stochasticity of the agent’s policy can be used to aid efficient exploration. The parameters of the noise are learned with gradient descent along with the remaining networ…

Cited by 1259SourcePDFScholar
2018

The Reactor: A fast and sample-efficient Actor-Critic agent for Reinforcement Learning

ICLR 2018poster

In this work we present a new agent architecture, called Reactor, which combines multiple algorithmic and architectural contributions to produce an agent with higher sample-efficiency than Prioritized Dueling DQN (Wang et al., 2016) and Categorical DQN (Bellemare et al., 2017), while giving better r…

Cited by 116SourcePDFScholar