← Search

Khimya Khetarpal

15 accepted papers

2026

Difference-Aware Retrieval Polices for Imitation Learning

ICLR 2026poster

Behavior cloning suffers from poor generalization to out-of-distribution states due to compounding errors during deployment. We present Difference-Aware Retrieval Polices for Imitation Learning (DARP), a novel nearest-neighbor-based imitation learning approach that addresses this limitation by repar…

Cited by 0SourceScholar
2026

Optimizing Return Distributions with Distributional Dynamic Programming

ICML 2026poster

We introduce distributional dynamic programming (DP) methods for optimizing statistical functionals of the return distribution, with standard reinforcement learning as a special case. Previous distributional DP methods could optimize the same class of expected utilities as classic DP. To go beyond, …

Cited by 0SourceScholar
2026

Self-Predictive Representations for Combinatorial Generalization in Behavioral Cloning

ICLR 2026poster

While goal-conditioned behavior cloning (GCBC) methods can perform well on in-distribution training tasks, they do not necessarily generalize zero-shot to tasks that require conditioning on novel state-goal pairs, i.e. combinatorial generalization. In part, this limitation can be attributed to a lac…

Cited by 0SourceScholar
2025

A Unifying Framework for Action-Conditional Self-Predictive Reinforcement Learning

AISTATS 2025poster

Learning a good representation is a crucial challenge for reinforcement learning (RL) agents. Self-predictive algorithms jointly learn a latent representation and dynamics model by bootstrapping from future latent representations (BYOL). Recent work has developed theoretical insights into these algo…

Cited by 0SourceScholar
2025

Long Range Navigator (LRN): Extending robot planning horizons beyond metric maps

CoRL 2025poster

A robot navigating an outdoor environment with no prior knowledge of the space must rely on its local sensing, which is in the form of a local metric map or local policy with some fixed horizon. A limited planning horizon can often result in myopic decisions leading the robot off course or worse, in…

Cited by 0SourceScholar
2025

Plasticity as the Mirror of Empowerment

NeurIPS 2025spotlight

Agents are minimally entities that are influenced by their past observations and act to influence future observations. This latter capacity is captured by empowerment, which has served as a vital framing concept across artificial intelligence and cognitive science. This former capacity, however, is…

Cited by 0SourceScholar
2024

Balancing Context Length and Mixing Times for Reinforcement Learning at Scale

NeurIPS 2024poster

Due to the recent remarkable advances in artificial intelligence, researchers have begun to consider challenging learning problems such as learning to generalize behavior from large offline datasets or learning online in non-Markovian environments. Meanwhile, recent advances in both of these areas h…

Cited by 3SourcePDFScholar
2024

Normalization and effective learning rates in reinforcement learning

NeurIPS 2024poster

Normalization layers have recently experienced a renaissance in the deep reinforcement learning and continual learning literature, with several works highlighting diverse benefits such as improving loss landscape conditioning and combatting overestimation bias. However, normalization brings with it…

Cited by 6SourcePDFScholar
2023

Discovering Object-Centric Generalized Value Functions From Pixels

ICML 2023poster

Deep Reinforcement Learning has shown significant progress in extracting useful representations from high-dimensional inputs albeit using hand-crafted auxiliary tasks and pseudo rewards. Automatically learning such representations in an object-centric manner geared towards control and fast adaptatio…

2021

Learning Robust State Abstractions for Hidden-Parameter Block MDPs

ICLR 2021poster

Many control tasks exhibit similar dynamics that can be modeled as having common latent structure. Hidden-Parameter Markov Decision Processes (HiP-MDPs) explicitly model this structure to improve sample efficiency in multi-task settings. However, this setting makes strong assumptions on the observab…

Cited by 57SourcePDFScholar
2021

Variance Penalized On-Policy and Off-Policy Actor-Critic

AAAI 2021technical

Reinforcement learning algorithms are typically geared towards optimizing the expected return of an agent. However, in many practical applications, low variance in the return is desired to ensure the reliability of an algorithm. In this paper, we propose on-policy and off-policy actor-critic algorit…

2020

Value Preserving State-Action Abstractions

AISTATS 2020poster

Abstraction can improve the sample efficiency of reinforcement learning. However, the process of abstraction inherently discards information, potentially compromising an agent’s ability to represent high-value policies. To mitigate this, we here introduce combinations of state abstractions and optio…

Cited by 71SourcePDFScholar
2020

What can I do here? A Theory of Affordances in Reinforcement Learning

ICML 2020poster

Reinforcement learning algorithms usually assume that all actions are always available to an agent. However, both people and animals understand the general link between the features of their environment and the actions that are feasible. Gibson (1977) coined the term "affordances" to describe the fa…