← Search

Arnav Kumar Jain

6 accepted papers

2025

A Smooth Sea Never Made a Skilled SAILOR: Robust Imitation via Learning to Search

NeurIPS 2025spotlight

The fundamental limitation of the behavioral cloning (BC) approach to imitation learning is that it only teaches an agent what the expert did at states the expert visited. This means that when a BC agent makes a mistake which takes them out of the support of the demonstrations, they often don't know…

Cited by 0SourcecodeScholar
2025

Multi-Turn Code Generation Through Single-Step Rewards

ICML 2025spotlight

We address the problem of code generation from multi-turn execution feedback. Existing methods either generate code without feedback or use complex, hierarchical reinforcement learning to optimize multi-turn rewards. We propose a simple yet scalable approach, $\mu$CODE, that solves multi-turn code…

2025

Non-Adversarial Inverse Reinforcement Learning via Successor Feature Matching

ICLR 2025poster

In inverse reinforcement learning (IRL), an agent seeks to replicate expert demonstrations through interactions with the environment. Traditionally, IRL is treated as an adversarial game, where an adversary searches over reward models, and a learner optimizes the reward through repeated RL procedure…

2023

Maximum State Entropy Exploration using Predecessor and Successor Representations

NeurIPS 2023poster

Animals have a developed ability to explore that aids them in important tasks such as locating food, exploring for shelter, and finding misplaced items. These exploration skills necessarily track where they have been so that they can plan for finding items with relative efficiency. Contemporary expl…

Cited by 16SourcePDFScholar
2022

Learning Robust Dynamics through Variational Sparse Gating

NeurIPS 2022accept

Learning world models from their sensory inputs enables agents to plan for actions by imagining their future outcomes. World models have previously been shown to improve sample-efficiency in simulated environments with few objects, but have not yet been applied successfully to environments with many…

2020

Prior Guided GAN Based Semantic Inpainting

CVPR 2020poster

Contemporary deep learning based semantic inpainting can be approached from two directions. First, and the more explored, approach is to train an offline deep regression network over the masked pixels with an additional refinement by adversarial training. This approach requires a single feed-forward…

Cited by 124PDFScholar