← Search

Unnat Jain

26 accepted papers

2026

Constructive Distortion: Improving MLLMs with Attention-Guided Image Warping

ICLR 2026poster

Multimodal large language models (MLLMs) often miss small details and spatial relations in cluttered scenes, leading to errors in fine-grained perceptual grounding. We introduce AttWarp, a lightweight method that allocates more resolution to query-relevant content while compressing less informative…

Cited by 0SourceScholar
2026

Robotic Manipulation by Imitating Generated Videos Without Physical Demonstrations

ICLR 2026poster

This work introduces Robots Imitating Generated Videos (RIGVid), a system that enables robots to perform complex manipulation tasks—such as pouring, wiping, and mixing—purely by imitating AI-generated videos, without requiring any physical demonstrations or robot-specific training. Given a language…

Cited by 0SourcecodeScholar
2026

Unifying Stacking and Cascading for Efficient Ensemble Inference

ICML 2026poster

We introduce LazyStack, a method for efficient model ensemble inference. The core idea is intuitive: after each model executes, we check whether accumulated evidence is sufficient to exit confidently. Sometimes one model suffices; other times we aggregate predictions from several models via trained …

Cited by 0SourceScholar
2025

An Image is Worth More Than 16x16 Patches: Exploring Transformers on Individual Pixels

ICLR 2025poster

This work does not introduce a new method. Instead, we present an interesting finding that questions the necessity of the inductive bias of locality in modern computer vision architectures. Concretely, we find that vanilla Transformers can operate by directly treating each individual pixel as a toke…

Cited by 13SourcePDFScholar
2024

Exploitation-Guided Exploration for Semantic Embodied Navigation

ICRA 2024poster

In the recent progress in embodied navigation and sim-to-robot transfer, modular policies have emerged as a de facto framework. However, there is more to compositionality beyond the decomposition of the learning load into modular components. In this work, we investigate a principled way to syntactic…

Cited by 4SourcecodeScholar
2024

Habitat 3.0: A Co-Habitat for Humans, Avatars, and Robots

ICLR 2024poster

We present Habitat 3.0: a simulation platform for studying collaborative human-robot tasks in home environments. Habitat 3.0 offers contributions across three dimensions: (1) Accurate humanoid simulation: addressing challenges in modeling complex deformable bodies and diversity in appearance and mot…

Cited by 111SourcePDFScholar
2023

Adaptive Coordination in Social Embodied Rearrangement

ICML 2023poster

We present the task of "Social Rearrangement", consisting of cooperative everyday tasks like setting up the dinner table, tidying a house or unpacking groceries in a simulated multi-agent environment. In Social Rearrangement, two robots coordinate to complete a long-horizon task, using onboard sensi…

Cited by 12SourcePDFScholar
2023

Affordances From Human Videos as a Versatile Representation for Robotics

CVPR 2023poster

Building a robot that can understand and learn to interact by watching humans has inspired several vision problems. However, despite some successful results on static datasets, it remains unclear how current models can be used on a robot directly. In this paper, we aim to bridge this gap by leveragi…

Cited by 165SourcePDFScholar
2023

Pretrained Language Models as Visual Planners for Human Assistance

ICCV 2023poster

In our pursuit of advancing multi-modal AI assistants capable of guiding users to achieve complex multi-step goals, we propose the task of 'Visual Planning for Assistance (VPA)'. Given a succinct natural language goal, e.g., "make a shelf", and a video of the user's progress so far, the aim of VPA i…

Cited by 43PDFcodeScholar
2022

Last-Mile Embodied Visual Navigation

CoRL 2022poster

Realistic long-horizon tasks like image-goal navigation involve exploratory and exploitative phases. Assigned with an image of the goal, an embodied agent must explore to discover the goal, i.e., search efficiently using learned priors. Once the goal is discovered, the agent must accurately calibrat…

Cited by 45SourcecodeScholar
2022

Learning State-Aware Visual Representations from Audible Interactions

NeurIPS 2022accept

We propose a self-supervised algorithm to learn representations from egocentric video data. Recently, significant efforts have been made to capture humans interacting with their own environments as they go about their daily activities. In result, several large egocentric datasets of interaction-rich…

2021

Bridging the Imitation Gap by Adaptive Insubordination

NeurIPS 2021poster

In practice, imitation learning is preferred over pure reinforcement learning whenever it is possible to design a teaching agent to provide expert supervision. However, we show that when the teaching agent makes decisions with access to privileged information that is unavailable to the student, this…

Cited by 41SourcePDFScholar
2021

Cooperative Exploration for Multi-Agent Deep Reinforcement Learning

ICML 2021oral

Exploration is critical for good results in deep reinforcement learning and has attracted much attention. However, existing multi-agent deep reinforcement learning algorithms still use mostly noise-based techniques. Very recently, exploration methods that consider cooperation among multiple agents h…

Cited by 137SourcePDFScholar
2021

GridToPix: Training Embodied Agents With Minimal Supervision

ICCV 2021poster

While deep reinforcement learning (RL) promises freedom from hand-labeled data, great successes, especially for Embodied AI, require significant work to create supervision via carefully shaped rewards. Indeed, without shaped rewards, i.e., with only terminal rewards, present-day Embodied AI results…

Cited by 24PDFcodeScholar
2021

Interpretation of Emergent Communication in Heterogeneous Collaborative Embodied Agents

ICCV 2021poster

Communication between embodied AI agents has received increasing attention in recent years. Despite its use, it is still unclear whether the learned communication is interpretable and grounded in perception. To study the grounding of emergent forms of communication, we first introduce the collaborat…

Cited by 39PDFScholar
2021

Language-Aligned Waypoint (LAW) Supervision for Vision-and-Language Navigation in Continuous Environments

EMNLP 2021main

In the Vision-and-Language Navigation (VLN) task an embodied agent navigates a 3D environment, following natural language instructions. A challenge in this task is how to handle ‘off the path’ scenarios where an agent veers from a reference path. Prior work supervises the agent with actions based on…

2020

A Cordial Sync: Going Beyond Marginal Policies for Multi-Agent Embodied Tasks

ECCV 2020poster

Autonomous agents must learn to collaborate. It is not scalable to develop a new centralized agent every time a task’s difficulty outpaces a single agent’s abilities. While multi-agent collaboration research has flourished in gridworld-like environments, relatively little work has considered visuall…

2020

MultiON: Benchmarking Semantic Map Memory using Multi-Object Navigation

NeurIPS 2020poster

Navigation tasks in photorealistic 3D environments are challenging because they require perception and effective planning under partial observability. Recent work shows that map-like memory is useful for long-horizon navigation tasks. However, a focused investigation of the impact of maps on navigat…

Cited by 120SourcePDFScholar
2020

SoundSpaces: Audio-Visual Navigation in 3D Environments

ECCV 2020poster

Moving around in the world is naturally a multi-sensory experience, but today's embodied agents are deaf - restricted to solely their visual perception of the environment. We introduce audio-visual navigation for complex, acoustically and visually realistic 3D environments. By both seeing and hearin…

2019

Two Body Problem: Collaborative Visual Task Completion

CVPR 2019oral

Collaboration is a necessary skill to perform tasks that are beyond one agent's capabilities. Addressed extensively in both conventional and modern AI, multi-agent collaboration has often been studied in the context of simple grid worlds. We argue that there are inherently visual aspects to collabor…

Cited by 98PDFScholar
2018

Two Can Play This Game: Visual Dialog With Discriminative Question Generation and Answering

CVPR 2018poster

Human conversation is a complex mechanism with subtle nuances. It is hence an ambitious goal to develop artificial intelligence agents that can participate fluently in a conversation. While we are still far from achieving this goal, recent progress in visual question answering, image captioning, and…

Cited by 101SourcePDFScholar