← Search

Stefan Lee

54 accepted papers

2025

Calibrating MLLM-as-a-judge via Multimodal Bayesian Prompt Ensembles

ICCV 2025poster

Multimodal large language models (MLLMs) are increasingly used to evaluate text-to-image (TTI) generation systems, providing automated judgments based on visual and textual context. However, these "judge" models often suffer from biases, overconfidence, and inconsistent performance across diverse im…

Cited by 0SourcePDFScholar
2025

Graph Neural Network Based Action Ranking for Planning

NeurIPS 2025poster

We propose a novel approach to learn relational policies for classical planning based on learning to rank actions. We introduce a new graph representation that explicitly captures action information and propose a Graph Neural Network (GNN) architecture augmented with Gated Recurrent Units (GRUs) to…

Cited by 0SourceScholar
2025

Harnessing Input-Adaptive Inference for Efficient VLN

ICCV 2025poster

An emerging paradigm in vision-and-language navigation (VLN) is the use of history-aware multi-modal transformer models. Given a language instruction, these models process observation and navigation history to predict the most appropriate action for an agent. While they have significantly improved p…

Cited by 0SourcePDFScholar
2025

Non-conflicting Energy Minimization in Reinforcement Learning based Robot Control

CoRL 2025oral

Efficient robot locomotion often requires balancing task performance with energy expenditure. A common approach in reinforcement learning (RL) is to penalize energy use directly in the reward function. This requires carefully weighting the reward terms to avoid undesirable trade-offs where energy mi…

Cited by 0SourceScholar
2024

AUTOSGM: A Unified Lowpass Regularization Framework for Accelerated Learning

ICASSP 2024accepted

This paper unifies commonly used accelerated stochastic gradient methods (Polyak’s Heavy Ball, Nesterov’s Accelerated Gradient and Adaptive Moment Estimation (Adam)) as specific cases of a general lowpass regularized learning framework, the Automatic Stochastic Gradient Method (AutoSGM). For AutoSGM…

Cited by 0SourceScholar
2024

FairDeDup: Detecting and Mitigating Vision-Language Fairness Disparities in Semantic Dataset Deduplication

CVPR 2024poster

Recent dataset deduplication techniques have demonstrated that content-aware dataset pruning can dramatically reduce the cost of training Vision-Language Pretrained (VLP) models without significant performance losses compared to training on the original dataset. These results have been based on prun…

Cited by 6SourcePDFScholar
2024

Language-Informed Beam Search Decoding for Multilingual Machine Translation

ACL 2024findings

Beam search decoding is the de-facto method for decoding auto-regressive Neural Machine Translation (NMT) models, including multilingual NMT where the target language is specified as an input. However, decoding multilingual NMT models commonly produces off-target translations – yielding translation…

2024

Point Cloud Models Improve Visual Robustness in Robotic Learners

ICRA 2024poster

Visual control policies can encounter significant performance degradation when visual conditions like lighting or camera position differ from those seen during training – often exhibiting sharp declines in capability even for minor differences. In this work, we examine robustness to a suite of these…

Cited by 3SourcecodeScholar
2024

Simple Masked Training Strategies Yield Control Policies That Are Robust to Sensor Failure

CoRL 2024poster

Sensor failure is common when robots are deployed in the real world, as sensors naturally wear out over time. Such failures can lead to catastrophic outcomes, including damage to the robot from unexpected robot behaviors such as falling during walking. Previous work has tried to address this problem…

Cited by 1SourceScholar
2023

Emergence of Maps in the Memories of Blind Navigation Agents

ICLR 2023top-5%

Animal navigation research posits that organisms build and maintain internal spa- tial representations, or maps, of their environment. We ask if machines – specifically, artificial intelligence (AI) navigation agents – also build implicit (or ‘mental’) maps. A positive answer to this question would…

Cited by 31SourcePDFScholar
2023

Iterative Vision-and-Language Navigation

CVPR 2023poster

We present Iterative Vision-and-Language Navigation (IVLN), a paradigm for evaluating language-guided agents navigating in a persistent environment over time. Existing Vision-and-Language Navigation (VLN) benchmarks erase the agent's memory at the beginning of every episode, testing the ability to p…

Cited by 22SourcePDFScholar
2023

Navigating to Objects Specified by Images

ICCV 2023poster

Images are a convenient way to specify which particular object instance an embodied agent should navigate to. Solving this task requires semantic visual reasoning and exploration of unknown environments. We present a system that can perform this task in both simulation and the real world. Our modula…

Cited by 41PDFScholar
2022

Sim-2-Sim Transfer for Vision-and-Language Navigation in Continuous Environments

ECCV 2022poster

"Recent work in Vision-and-Language Navigation (VLN) has presented two environmental paradigms with differing realism -- the standard VLN setting built on topological environments where navigation is abstracted away, and the VLN-CE setting where agents must navigate continuous 3D environments using…

2021

DeepAveragers: Offline Reinforcement Learning By Solving Derived Non-Parametric MDPs

ICLR 2021spotlight

We study an approach to offline reinforcement learning (RL) based on optimally solving finitely-represented MDPs derived from a static dataset of experience. This approach can be applied on top of any learned representation and has the potential to easily support multiple solution objective…

2021

Improving Multilingual Translation by Representation and Gradient Regularization

EMNLP 2021main

Multilingual Neural Machine Translation (NMT) enables one model to serve all translation directions, including ones that are unseen during training, i.e. zero-shot translation. Despite being theoretically attractive, current models often produce low quality translations – commonly failing to even pr…

2021

SOAT: A Scene- and Object-Aware Transformer for Vision-and-Language Navigation

NeurIPS 2021poster

Natural language instructions for visual navigation often use scene descriptions (e.g., bedroom) and object references (e.g., green chairs) to provide a breadcrumb trail to a goal location. This work presents a transformer-based vision-and-language navigation (VLN) agent that uses two different visu…

Cited by 64SourcePDFScholar
2021

Semantic MapNet: Building Allocentric Semantic Maps and Representations from Egocentric Views

AAAI 2021technical

We study the task of semantic mapping – specifically, an embodied agent (a robot or an egocentric AI assistant) is given a tour of a new environment and asked to build an allocentric top-down semantic map (‘what is where?’) from egocentric observations of an RGB-D camera with known pose (via localiz…

2021

THDA: Treasure Hunt Data Augmentation for Semantic Navigation

ICCV 2021poster

Can general-purpose neural models learn to navigate? For PointGoal navigation (""go to x, y""), the answer is a clear `yes' -- mapless neural models composed of task-agnostic components (CNNs and RNNs) trained with large-scale model-free reinforcement learning achieve near-perfect performance. Howev…

Cited by 91PDFScholar
2021

Waypoint Models for Instruction-Guided Navigation in Continuous Environments

ICCV 2021poster

Little inquiry has explicitly addressed the role of action spaces in language-guided visual navigation -- either in terms of its effect on navigation success or the efficiency with which a robotic agent could execute the resulting trajectory. Building on the recently released VLN-CE setting for inst…

Cited by 94PDFcodeScholar
2020

12-in-1: Multi-Task Vision and Language Representation Learning

CVPR 2020poster

Much of vision-and-language research focuses on a small but diverse set of independent tasks and supporting datasets often studied in isolation; however, the visually-grounded language understanding skills required for success at these tasks overlap significantly. In this work, we investigate these…

Cited by 575PDFcodeScholar
2020

Beyond the Nav-Graph: Vision-and-Language Navigation in Continuous Environments

ECCV 2020poster

We develop a language-guided navigation task set in a continuous 3D environment where agents must execute low-level actions to follow natural language navigation directions. By being situated in continuous environments, this setting lifts a number of assumptions implicit in prior work that represent…

2020

DD-PPO: Learning Near-Perfect PointGoal Navigators from 2.5 Billion Frames

ICLR 2020poster

We present Decentralized Distributed Proximal Policy Optimization (DD-PPO), a method for distributed reinforcement learning in resource-intensive simulated environments. DD-PPO is distributed (uses multiple machines), decentralized (lacks a centralized server), and synchronous (no computation is eve…

Cited by 542SourcecodeScholar
2020

Dialog without Dialog Data: Learning Visual Dialog Agents from VQA Data

NeurIPS 2020poster

Can we develop visually grounded dialog agents that can efficiently adapt to new tasks without forgetting how to talk to people? Such agents could leverage a larger variety of existing data to generalize to a new task, minimizing expensive data collection and annotation. In this work, we study a set…

2020

Improving Vision-and-Language Navigation with Image-Text Pairs from the Web

ECCV 2020poster

Following a navigation instruction such as 'Walk down the stairs and stop at the brown sofa' requires embodied AI agents to ground referenced scene elements referenced (e.g. 'stairs') to visual content in the environment (pixels corresponding to 'stairs'). We ask the following question -- can we lev…

2020

Integrating Egocentric Localization for More Realistic Point-Goal Navigation Agents

CoRL 2020

Recent work has presented embodied agents that can navigate to point-goal targets in novel indoor environments with near-perfect accuracy. However, these agents are equipped with idealized sensors for localization and take deterministic actions. This setting is practically sterile by comparison to t

Cited by 0SourcePDFScholar
2020

Language-Conditioned Imitation Learning for Robot Manipulation Tasks

NeurIPS 2020spotlight

Imitation learning is a popular approach for teaching motor skills to robots. However, most approaches focus on extracting policy parameters from execution traces alone (i.e., motion trajectories and perceptual data). No adequate communication channel exists between the human expert and the robot to…

2020

Sim-to-Real Transfer for Vision-and-Language Navigation

CoRL 2020

We study the challenging problem of releasing a robot in a previously unseen environment, and having it follow unconstrained natural language navigation instructions. Recent work on the task of Vision-and-Language Navigation (VLN) has achieved significant progress in simulation. To assess the implic

2020

Sim2Real Predictivity: Does Evaluation in Simulation Predict Real-World Performance?

RA-L 2020

Does progress in simulation translate to progress on robots? If one method outperforms another in simulation, how likely is that trend to hold in reality on a robot? We examine this question for embodied PointGoal navigation - developing engineering tools and a research paradigm for evaluating a sim

Cited by 252SourceScholar
2019

Audio Visual Scene-Aware Dialog

CVPR 2019poster

We introduce the task of scene-aware dialog. Our goal is to generate a complete and natural response to a question about a scene, given video and audio of the scene and the history of previous turns in the dialog. To answer successfully, agents must ground concepts from the question in the video whi…

Cited by 226PDFcodeScholar
2019

Chasing Ghosts: Instruction Following as Bayesian State Tracking

NeurIPS 2019poster

A visually-grounded navigation instruction can be interpreted as a sequence of expected observations and actions an agent following the correct trajectory would encounter and perform. Based on this intuition, we formulate the problem of finding the goal location in Vision-and-Language Navigation (VL…

2019

Embodied Question Answering in Photorealistic Environments With Point Cloud Perception

CVPR 2019oral

To help bridge the gap between internet vision-style problems and the goal of vision for embodied perception we instantiate a large-scale navigation task -- Embodied Question Answering [1] in photo-realistic environments (Matterport 3D). We thoroughly study navigation policies that utilize 3D poin…

Cited by 193PDFScholar
2019

Probabilistic Neural Symbolic Models for Interpretable Visual Question Answering

ICML 2019oral

We propose a new class of probabilistic neural-symbolic models, that have symbolic functional programs as a latent, stochastic variable. Instantiated in the context of visual question answering, our probabilistic formulation offers two key conceptual advantages over prior neural-symbolic models for…

Cited by 109SourcePDFScholar
2019

Taking a HINT: Leveraging Explanations to Make Vision and Language Models More Grounded

ICCV 2019poster

Many vision and language models suffer from poor visual grounding -- often falling back on easy-to-learn language priors rather than basing their decisions on visual concepts in the image. In this work, we propose a generic approach called Human Importance-aware Network Tuning (HINT) that effectivel…

Cited by 305PDFScholar
2019

Trainable Decoding of Sets of Sequences for Neural Sequence Models

ICML 2019oral

Many sequence prediction tasks admit multiple correct outputs and so, it is often useful to decode a set of outputs that maximize some task-specific set-level metric. However, retooling standard sequence prediction procedures tailored towards predicting the single best output leads to the decoding o…

Cited by 3SourcePDFScholar
2019

ViLBERT: Pretraining Task-Agnostic Visiolinguistic Representations for Vision-and-Language Tasks

NeurIPS 2019poster

We present ViLBERT (short for Vision-and-Language BERT), a model for learning task-agnostic joint representations of image content and natural language. We extend the popular BERT architecture to a multi-modal two-stream model, processing both visual and textual inputs in separate streams that inter…

Cited by 4490SourcePDFScholar
2019

nocaps: novel object captioning at scale

ICCV 2019poster

Image captioning models have achieved impressive results on datasets containing limited visual concepts and large amounts of paired image-caption training data. However, if these models are to ever function in the wild, a much larger variety of visual concepts must be learned, ideally from less supe…

Cited by 420PDFcodeScholar
2018

Choose Your Neuron: Incorporating Domain Knowledge through Neuron-Importance

ECCV 2018poster

Individual neurons in convolutional neural networks supervised for image-level classification tasks have been shown to implicitly learn semantically meaningful concepts ranging from simple textures and shapes to whole or partial objects – forming a “dictionary” of concepts acquired through the learn…

2018

Learn from Your Neighbor: Learning Multi-modal Mappings from Sparse Annotations

ICML 2018oral

Many structured prediction problems (particularly in vision and language domains) are ambiguous, with multiple outputs being ‘correct’ for an input {–} e.g. there are many ways of describing an image, multiple ways of translating a sentence; however, exhaustively annotating the applicability of all…

Cited by 6SourcePDFScholar
2018

Neural Modular Control for Embodied Question Answering

CoRL 2018

We present a modular approach for learning policies for navigation over long planning horizons from language input. Our hierarchical policy operates at multiple timescales, where the higher-level master policy proposes subgoals to be executed by specialized sub-policies. Our choice of subgoals is co

2018

Overcoming Language Priors in Visual Question Answering with Adversarial Regularization

NeurIPS 2018poster

Modern Visual Question Answering (VQA) models have been shown to rely heavily on superficial correlations between question and answer words learned during training -- \eg overwhelmingly reporting the type of room as kitchen or the sport being played as tennis, irrespective of the image. Most alarmin…

2018

Visual Curiosity: Learning to Ask Questions to Learn Visual Recognition

CoRL 2018

In an open-world setting, it is inevitable that an intelligent agent (e.g., a robot) will encounter visual objects, attributes or relationships it does not recognize. In this work, we develop an agent empowered with visual curiosity, i.e. the ability to ask questions to an Oracle (e.g., human) about

2017

Bidirectional Beam Search: Forward-Backward Inference in Neural Sequence Models for Fill-In-The-Blank Image Captioning

CVPR 2017poster

We develop the first approximate inference algorithm for 1-Best (and M-Best) decoding in bidirectional neural sequence models by extending Beam Search (BS) to reason about both forward and backward time dependencies. Beam Search (BS) is a widely used approximate inference algorithm for decoding se…

Cited by 56PDFScholar
2017

Learning Cooperative Visual Dialog Agents With Deep Reinforcement Learning

ICCV 2017oral

We introduce the first goal-driven training for visual question answering and dialog agents. Specifically, we pose a cooperative `image guessing' game between two agents -- Qbot and Abot -- who communicate in natural language dialog so that Qbot can select an unseen image from a lineup of images. We…

Cited by 493PDFcodeScholar
2016

Stochastic Multiple Choice Learning for Training Diverse Deep Ensembles

NeurIPS 2016poster

Many practical perception systems exist within larger processes which often include interactions with users or additional components that are capable of evaluating the quality of predicted solutions. In these contexts, it is beneficial to provide these oracle mechanisms with multiple highly likely h…

Cited by 233SourcePDFScholar
2015

Lending A Hand: Detecting Hands and Recognizing Activities in Complex Egocentric Interactions

ICCV 2015poster

Hands appear very often in egocentric video, and their appearance and pose give important cues about what people are doing and what they are paying attention to. But existing work in hand detection has made strong assumptions that work well in only simple scenarios, such as with limited interaction…

Cited by 541PDFScholar