← Search

Amrit Bedi

13 accepted papers

2024

Closing the Gap: Achieving Global Convergence (Last Iterate) of Actor-Critic under Markovian Sampling with Neural Network Parametrization

ICML 2024spotlight

The current state-of-the-art theoretical analysis of Actor-Critic (AC) algorithms significantly lags in addressing the practical aspects of AC implementations. This crucial gap needs bridging to bring the analysis in line with practical implementations of AC. To address this, we advocate for conside…

Cited by 2SourcePDFScholar
2024

FACT or Fiction: Can Truthful Mechanisms Eliminate Federated Free Riding?

NeurIPS 2024poster

Standard federated learning (FL) approaches are vulnerable to the free-rider dilemma: participating agents can contribute little to nothing yet receive a well-trained aggregated model. While prior mechanisms attempt to solve the free-rider dilemma, none have addressed the issue of truthfulness. In p…

2024

MaxMin-RLHF: Alignment with Diverse Human Preferences

ICML 2024poster

Reinforcement Learning from Human Feedback (RLHF) aligns language models to human preferences by employing a singular reward model derived from preference data. However, the single reward model overlooks the rich diversity of human preferences inherent in data collected from multiple users. In this…

Cited by 82SourcePDFScholar
2024

PARL: A Unified Framework for Policy Alignment in Reinforcement Learning from Human Feedback

ICLR 2024poster

We present a novel unified bilevel optimization-based framework, \textsf{PARL}, formulated to address the recently highlighted critical issue of policy alignment in reinforcement learning using utility or preference-based feedback. We identify a major gap within current algorithmic designs for solvi…

Cited by 29SourcePDFScholar
2024

PIPER: Primitive-Informed Preference-based Hierarchical Reinforcement Learning via Hindsight Relabeling

ICML 2024poster

In this work, we introduce PIPER: Primitive-Informed Preference-based Hierarchical reinforcement learning via Hindsight Relabeling, a novel approach that leverages preference-based learning to learn a reward model, and subsequently uses this reward model to relabel higher-level replay buffers. Since…

2024

Position: On the Possibilities of AI-Generated Text Detection

ICML 2024poster

Our study addresses the challenge of distinguishing human-written text from Large Language Model (LLM) outputs. We provide evidence that this differentiation is consistently feasible, except when human and machine text distributions are indistinguishable across their entire support. Employing inform…

Cited by 4SourcePDFScholar
2024

Towards Global Optimality for Practical Average Reward Reinforcement Learning without Mixing Time Oracles

ICML 2024poster

In the context of average-reward reinforcement learning, the requirement for oracle knowledge of the mixing time, a measure of the duration a Markov chain under a fixed policy needs to achieve its stationary distribution, poses a significant challenge for the global convergence of policy gradient me…

Cited by 2SourcePDFScholar
2024

Transfer Q-star : Principled Decoding for LLM Alignment

NeurIPS 2024poster

Aligning foundation models is essential for their safe and trustworthy deployment. However, traditional fine-tuning methods are computationally intensive and require updating billions of model parameters. A promising alternative, alignment via decoding, adjusts the response distribution directly wit…

Cited by 20SourcePDFScholar
2023

Beyond Exponentially Fast Mixing in Average-Reward Reinforcement Learning via Multi-Level Monte Carlo Actor-Critic

ICML 2023poster

Many existing reinforcement learning (RL) methods employ stochastic gradient iteration on the back end, whose stability hinges upon a hypothesis that the data-generating process mixes exponentially fast with a rate parameter that appears in the step-size selection. Unfortunately, this assumption is…

Cited by 14SourcePDFScholar
2023

Intent-Aware Planning in Heterogeneous Traffic via Distributed Multi-Agent Reinforcement Learning

CoRL 2023oral

Navigating safely and efficiently in dense and heterogeneous traffic scenarios is challenging for autonomous vehicles (AVs) due to their inability to infer the behaviors or intentions of nearby drivers. In this work, we introduce a distributed multi-agent reinforcement learning (MARL) algorithm for…

Cited by 10SourceScholar
2023

STEERING : Stein Information Directed Exploration for Model-Based Reinforcement Learning

ICML 2023poster

Directed Exploration is a crucial challenge in reinforcement learning (RL), especially when rewards are sparse. Information-directed sampling (IDS), which optimizes the information ratio, seeks to do so by augmenting regret with information gain. However, estimating information gain is computational…

Cited by 8SourcePDFScholar
2023

SWIFT: Rapid Decentralized Federated Learning via Wait-Free Model Communication

ICLR 2023poster

The decentralized Federated Learning (FL) setting avoids the role of a potentially unreliable or untrustworthy central host by utilizing groups of clients to collaboratively train a model via localized training and model/gradient sharing. Most existing decentralized FL algorithms require synchroniza…

2022

HTRON: Efficient Outdoor Navigation with Sparse Rewards via Heavy Tailed Adaptive Reinforce Algorithm

CoRL 2022poster

We present a novel approach to improve the performance of deep reinforcement learning (DRL) based outdoor robot navigation systems. Most, existing DRL methods are based on carefully designed dense reward functions that learn the efficient behavior in an environment.  We circumvent this issue by work…

Cited by 13SourceScholar