← Search

Ruslan salakhutdinov

131 accepted papers

2026

Accelerating Diffusion Planners in Offline RL via Reward-Aware Consistency Trajectory Distillation

ICLR 2026poster

Although diffusion models have achieved strong results in decision-making tasks, their slow inference speed remains a key limitation. While consistency models offer a potential solution, existing applications to decision-making either struggle with suboptimal demonstrations under behavior cloning or…

Cited by 0SourceScholar
2026

Act2See: Emergent Active Visual Perception for Video Reasoning

CVPR 2026

Vision-Language Models (VLMs) typically rely on static initial frames for video reasoning, restricting their ability to incorporate essential dynamic information as the reasoning process evolves. Existing methods that augment Chain-of-Thought (CoT) with additional frame information often exhibit sub

Cited by 0SourcecodeScholar
2026

Joint Distillation for Fast Likelihood Evaluation and Sampling in Flow-based Models

ICLR 2026poster

Log-likelihood evaluation enables important capabilities in generative models, including model comparison, certain fine-tuning objectives, and many downstream applications. Yet paradoxically, some of today's best generative models -- diffusion and flow-based models -- still require hundreds to thous…

Cited by 0SourceScholar
2026

RLAD: Training LLMs to Discover Abstractions for Solving Reasoning Problems

ICLR 2026poster

Reasoning requires going beyond pattern matching or memorization of solutions to identify and implement algorithmic procedures that can be used to deduce answers to hard problems. Doing so requires reusing primitives, intermediate results, or procedures across multiple problems. While RL post-traini…

Cited by 0SourceScholar
2025

AgentDAM: Privacy Leakage Evaluation for Autonomous Web Agents

NeurIPS 2025poster

Autonomous AI agents that can follow instructions and perform complex multi-step tasks have tremendous potential to boost human productivity. However, to perform many of these tasks, the agents need access to personal information from their users, raising the question of whether they are capable of…

Cited by 0SourcecodeScholar
2025

Local Policies Enable Zero-Shot Long-Horizon Manipulation

ICRA 2025

Sim2real for robotic manipulation is difficult due to the challenges of simulating complex contacts and generating realistic task distributions. To tackle the latter problem, we introduce ManipGen, which leverages a new class of policies for sim2real transfer: local policies. Locality enables a vari

Cited by 31SourcecodeScholar
2025

Neural MP: A Neural Motion Planner

IROS 2025

The current paradigm for motion planning generates solutions from scratch for every new problem, which consumes significant amounts of time and computational resources. For complex, cluttered scenes, motion planning approaches can often take minutes to produce a solution, while humans are able to ac

Cited by 0SourcecodeScholar
2025

Optimizing Test-Time Compute via Meta Reinforcement Finetuning

ICML 2025poster

Training models to efficiently use test-time compute is crucial for improving the reasoning performance of LLMs. While current methods mostly do so via fine-tuning on search traces or running RL against the 0/1 outcome reward, do these approaches efficiently utilize test-time compute? Would these ap…

Cited by 21SourcePDFScholar
2025

Retrospective In-Context Learning for Temporal Credit Assignment with Large Language Models

NeurIPS 2025poster

Learning from self-sampled data and sparse environmental feedback remains a fundamental challenge in training self-evolving agents. Temporal credit assignment mitigates this issue by transforming sparse feedback into dense supervision signals. However, previous approaches typically depend on domain-…

Cited by 0SourceScholar
2024

Confronting Reward Model Overoptimization with Constrained RLHF

ICLR 2024spotlight

Large language models are typically aligned with human preferences by optimizing reward models (RMs) fitted to human feedback. However, human preferences are multi-faceted, and it is increasingly common to derive reward from a composition of simpler reward models which each capture a different aspec…

2024

Effective Data Augmentation With Diffusion Models

ICLR 2024poster

Data augmentation is one of the most prevalent tools in deep learning, underpinning many recent advances, including those from classification, generative models, and representation learning. The standard approach to data augmentation combines simple transformations like rotations and flips to genera…

2024

Manifold Preserving Guided Diffusion

ICLR 2024poster

Despite the recent advancements, conditional image generation still faces challenges of cost, generalizability, and the need for task-specific training. In this paper, we propose Manifold Preserving Guided Diffusion (MPGD), a training-free conditional generation framework that leverages pretrained d…

Cited by 50SourcePDFScholar
2024

OmniACT: A Dataset and Benchmark for Enabling Multimodal Generalist Autonomous Agents for Desktop and Web

ECCV 2024poster

"* These authors contributed equally. The order is determined by dice rolling. For decades, human-computer interaction has fundamentally been manual. Even today, almost all productive work done on the computer necessitates human input at every step. Autonomous virtual agents represent an exciting st…

2024

Plan-Seq-Learn: Language Model Guided RL for Solving Long Horizon Robotics Tasks

ICLR 2024poster

Large Language Models (LLMs) are highly capable of performing planning for long-horizon robotics tasks, yet existing methods require access to a pre-defined skill library (*e.g.* picking, placing, pulling, pushing, navigating). However, LLM planning does not address how to design or learn those beha…

2024

Situated Instruction Following

ECCV 2024poster

"Language is never spoken in a vacuum. It is expressed, comprehended, and contextualized within the holistic backdrop of the speaker’s history, actions, and environment. Since humans are used to communicating efficiently with situated language, the practicality of robotic assistants hinge on their a…

2024

Stabilizing Contrastive RL: Techniques for Robotic Goal Reaching from Offline Data

ICLR 2024spotlight

Robotic systems that rely primarily on self-supervised learning have the potential to decrease the amount of human annotation and engineering effort required to learn control strategies. In the same way that prior robotic systems have leveraged self-supervised techniques from computer vision (CV) an…

2023

A Connection between One-Step RL and Critic Regularization in Reinforcement Learning

ICML 2023poster

As with any machine learning problem with limited data, effective offline RL algorithms require careful regularization to avoid overfitting. One class of methods, known as one-step RL, perform just one step of policy improvement. These methods, which include advantage-weighted regression and conditi…

Cited by 4SourcePDFScholar
2023

A Simple Approach for Visual Room Rearrangement: 3D Mapping and Semantic Search

ICLR 2023poster

Physically rearranging objects is an important capability for embodied agents. Visual room rearrangement evaluates an agent's ability to rearrange objects in a room to a desired goal based solely on visual input. We propose a simple yet effective method for this problem: (1) search for and map which…

Cited by 4SourcePDFScholar
2023

Cross-modal Attention Congruence Regularization for Vision-Language Relation Alignment

ACL 2023long

Despite recent progress towards scaling up multimodal vision-language models, these models are still known to struggle on compositional generalization benchmarks such as Winoground. We find that a critical component lacking from current vision-language models is relation-level alignment: the ability…

Cited by 13SourcePDFScholar
2023

Grounding Language Models to Images for Multimodal Inputs and Outputs

ICML 2023poster

We propose an efficient method to ground pretrained text-only language models to the visual domain, enabling them to process arbitrarily interleaved image-and-text data, and generate text interleaved with retrieved images. Our method leverages the abilities of language models learnt from large scale…

2023

Imitating Task and Motion Planning with Visuomotor Transformers

CoRL 2023poster

Imitation learning is a powerful tool for training robot manipulation policies, allowing them to learn from expert demonstrations without manual programming or trial-and-error. However, common methods of data collection, such as human supervision, scale poorly, as they are time-consuming and labor-i…

Cited by 56SourcecodeScholar
2023

MultiViz: Towards Visualizing and Understanding Multimodal Models

ICLR 2023poster

The promise of multimodal models for real-world applications has inspired research in visualizing and understanding their internal mechanics with the end goal of empowering stakeholders to visualize model behavior, perform model debugging, and promote trust in machine learning models. However, moder…

2023

Nano: Nested Human-in-the-Loop Reward Learning for Few-shot Language Model Control

ACL 2023findings

Pretrained language models have demonstrated extraordinary capabilities in language generation. However, real-world tasks often require controlling the distribution of generated text in order to mitigate bias, promote fairness, and achieve personalization. Existing techniques for controlling the dis…

2023

Quantifying & Modeling Multimodal Interactions: An Information Decomposition Framework

NeurIPS 2023poster

The recent explosion of interest in multimodal applications has resulted in a wide selection of datasets and methods for representing and integrating information from different modalities. Despite these empirical advances, there remain fundamental research questions: How can we quantify the interact…

2023

SPRING: Studying Papers and Reasoning to play Games

NeurIPS 2023poster

Open-world survival games pose significant challenges for AI algorithms due to their multi-tasking, deep exploration, and goal prioritization requirements. Despite reinforcement learning (RL) being popular for solving games, its high sample complexity limits its effectiveness in complex open-world g…

2023

Scenario-based Question Answering with Interacting Contextual Properties

ICLR 2023poster

In the scenario-based Question Answering (QA) task, models are asked to find answers that are appropriate to the user scenarios associated with the question and identify information that is missing from the scenarios but is necessary for the answers to hold. Scenarios commonly include multiple prope…

Cited by 0SourcePDFScholar
2023

Self-Supervised Object Goal Navigation with In-Situ Finetuning

IROS 2023poster

A household robot should be able to navigate to target objects without requiring users to first annotate everything in their home. Most current approaches to object navigation do not test on real robots and rely solely on reconstructed scans of houses and their expensively labeled semantic 3D meshes…

Cited by 7SourceScholar
2022

C-Planning: An Automatic Curriculum for Learning Goal-Reaching Tasks

ICLR 2022poster

Goal-conditioned reinforcement learning (RL) has shown great success recently at solving a wide range of tasks(e.g., navigation, robotic manipulation). However, learning to reach distant goals remains a central challenge to the field, and the task is particularly hard without any offline data, exper…

Cited by 22SourcePDFScholar
2022

Conditional Contrastive Learning with Kernel

ICLR 2022poster

Conditional contrastive learning frameworks consider the conditional sampling procedure that constructs positive or negative data pairs conditioned on specific variables. Fair contrastive learning constructs negative pairs, for example, from the same gender (conditioning on sensitive information), w…

2022

ConditionalQA: A Complex Reading Comprehension Dataset with Conditional Answers

ACL 2022long

We describe a Question Answering (QA) dataset that contains complex questions with conditional answers, i.e. the answers are only applicable when certain conditions apply. We call this dataset ConditionalQA. In addition to conditional answers, the dataset also features:(1) long context documents wit…

2022

Contrastive Learning as Goal-Conditioned Reinforcement Learning

NeurIPS 2022accept

In reinforcement learning (RL), it is easier to solve a task if given a good representation. While deep RL should automatically acquire such good representations, prior work often finds that learning representations in an end-to-end fashion is unstable and instead equip RL algorithms with additional…

Cited by 162SourcePDFScholar
2022

Don’t Copy the Teacher: Data and Model Challenges in Embodied Dialogue

EMNLP 2022main

Embodied dialogue instruction following requires an agent to complete a complex sequence of tasks from a natural language exchange. The recent introduction of benchmarks raises the question of how best to train and evaluate models for this multi-turn, multi-agent, long-horizon task. This paper contr…

2022

FILM: Following Instructions in Language with Modular Methods

ICLR 2022poster

Recent methods for embodied instruction following are typically trained end-to-end using imitation learning. This often requires the use of expert trajectories and low-level language instructions. Such approaches assume that neural states will integrate multimodal semantics to perform state tracking…

2022

FewNLU: Benchmarking State-of-the-Art Methods for Few-Shot Natural Language Understanding

ACL 2022long

The few-shot natural language understanding (NLU) task has attracted much recent attention. However, prior methods have been evaluated under a disparate set of protocols, which hinders fair comparison and measuring the progress of the field. To address this issue, we introduce an evaluation framewor…

2022

Imitating Past Successes can be Very Suboptimal

NeurIPS 2022accept

Prior work has proposed a simple strategy for reinforcement learning (RL): label experience with the outcomes achieved in that experience, and then imitate the relabeled experience. These outcome-conditioned imitation learning methods are appealing because of their simplicity, strong performance, an…

Cited by 19SourcePDFScholar
2022

Learning Weakly-supervised Contrastive Representations

ICLR 2022poster

We argue that a form of the valuable information provided by the auxiliary information is its implied data clustering information. For instance, considering hashtags as auxiliary information, we can hypothesize that an Instagram image will be semantically more similar with the same hashtags. With th…

2022

Mismatched No More: Joint Model-Policy Optimization for Model-Based RL

NeurIPS 2022accept

Many model-based reinforcement learning (RL) methods follow a similar template: fit a model to previously observed data, and then use data from that model for RL or planning. However, models that achieve better training performance (e.g., lower MSE) are not necessarily better for control: an RL agen…

2022

PACS: A Dataset for Physical Audiovisual Commonsense Reasoning

ECCV 2022poster

"In order for AI to be safely deployed in real-world scenarios such as hospitals, schools, and the workplace, it must be able to robustly reason about the physical world. Fundamental to this reasoning is physical common sense: understanding the physical properties and affordances of available object…

2022

Paraphrasing Is All You Need for Novel Object Captioning

NeurIPS 2022accept

Novel object captioning (NOC) aims to describe images containing objects without observing their ground truth captions during training. Due to the absence of caption annotation, captioning models cannot be directly optimized via sequence-to-sequence training or CIDEr optimization. As a result, we pr…

Cited by 5SourcePDFScholar
2022

Recurrent Model-Free RL Can Be a Strong Baseline for Many POMDPs

ICML 2022spotlight

Many problems in RL, such as meta-RL, robust RL, generalization in RL, and temporal credit assignment, can be cast as POMDPs. In theory, simply augmenting model-free RL with memory-based architectures, such as recurrent neural networks, provides a general approach to solving all types of POMDPs. How…

2022

The Information Geometry of Unsupervised Reinforcement Learning

ICLR 2022oral

How can a reinforcement learning (RL) agent prepare to solve downstream tasks if those tasks are not known a priori? One approach is unsupervised skill discovery, a class of algorithms that learn a set of policies without access to a reward function. Such algorithms bear a close resemblance to repre…

2022

Uncertainty Quantification with Pre-trained Language Models: A Large-Scale Empirical Analysis

EMNLP 2022finding

Pre-trained language models (PLMs) have gained increasing popularity due to their compelling prediction performance in diverse natural language processing (NLP) tasks. When formulating a PLM-based prediction pipeline for NLP tasks, it is also crucial for the pipeline to minimize the calibration erro…

2021

Accelerating Robotic Reinforcement Learning via Parameterized Action Primitives

NeurIPS 2021poster

Despite the potential of reinforcement learning (RL) for building general-purpose robotic systems, training RL agents to solve robotics tasks still remains challenging due to the difficulty of exploration in purely continuous action spaces. Addressing this problem is an active area of research with…

Cited by 115SourcePDFScholar
2021

C-Learning: Learning to Achieve Goals via Recursive Classification

ICLR 2021poster

We study the problem of predicting and controlling the future state distribution of an autonomous agent. This problem, which can be viewed as a reframing of goal-conditioned reinforcement learning (RL), is centered around learning a conditional probability density function over future states. Instea…

Cited by 93SourcePDFScholar
2021

Focused Attention Improves Document-Grounded Generation

NAACL 2021long

Document grounded generation is the task of using the information provided in a document to improve text generation. This work focuses on two different document grounded generation tasks: Wikipedia Update Generation task and Dialogue response generation. Our work introduces two novel adaptations of…

2021

Hubert: How Much Can a Bad Teacher Benefit ASR Pre-Training?

ICASSP 2021accepted

Compared to vision and language applications, self-supervised pre-training approaches for ASR are challenged by three unique problems: (1) There are multiple sound units in each input utterance, (2) With audio-only pre-training, there is no lexicon of sound units, and (3) Sound units have variable l…

Cited by 0SourceScholar
2021

Information Obfuscation of Graph Neural Networks

ICML 2021spotlight

While the advent of Graph Neural Networks (GNNs) has greatly improved node and graph representation learning in many applications, the neighborhood aggregation scheme exposes additional vulnerabilities to adversaries seeking to extract node-level information about sensitive attributes. In this paper…

2021

Instabilities of Offline RL with Pre-Trained Neural Representation

ICML 2021spotlight

In offline reinforcement learning (RL), we seek to utilize offline data to evaluate (or learn) policies in scenarios where the data are collected from a distribution that substantially differs from that of the target policy to be evaluated. Recent theoretical advances have shown that such sample-eff…

Cited by 56SourcePDFScholar
2021

Learning Language and Multimodal Privacy-Preserving Markers of Mood from Mobile Data

ACL 2021long

Mental health conditions remain underdiagnosed even in countries with common access to advanced medical care. The ability to accurately and efficiently predict mood from easily collectible data has several important implications for the early detection, intervention, and treatment of mental health d…

Cited by 21SourcePDFScholar
2021

Learning To Hallucinate Examples From Extrinsic and Intrinsic Supervision

ICCV 2021poster

Learning to hallucinate additional examples has recently been shown as a promising direction to address few-shot learning tasks. This work investigates two important yet overlooked natural supervision signals for guiding the hallucination process -- (i) extrinsic: classifiers trained on hallucinated…

Cited by 8PDFScholar
2021

Off-Dynamics Reinforcement Learning: Training for Transfer with Domain Classifiers

ICLR 2021poster

We propose a simple, practical, and intuitive approach for domain adaptation in reinforcement learning. Our approach stems from the idea that the agent's experience in the source domain should look similar to its experience in the target domain. Building off of a probabilistic view of RL, we achieve…

Cited by 105SourcePDFScholar
2021

On Proximal Policy Optimization’s Heavy-tailed Gradients

ICML 2021spotlight

Modern policy gradient algorithms such as Proximal Policy Optimization (PPO) rely on an arsenal of heuristics, including loss clipping and gradient clipping, to ensure successful learning. These heuristics are reminiscent of techniques from robust statistics, commonly used for estimation in outlier-…

Cited by 15SourcePDFScholar
2021

Reasoning Over Virtual Knowledge Bases With Open Predicate Relations

ICML 2021spotlight

We present the Open Predicate Query Language (OPQL); a method for constructing a virtual KB (VKB) trained entirely from text. Large Knowledge Bases (KBs) are indispensable for a wide-range of industry applications such as question answering and recommendation. Typically, KBs encode world knowledge i…

Cited by 29SourcePDFScholar
2021

Replacing Rewards with Examples: Example-Based Policy Search via Recursive Classification

NeurIPS 2021oral

Reinforcement learning (RL) algorithms assume that users specify tasks by manually writing down a reward function. However, this process can be laborious and demands considerable technical expertise. Can we devise RL algorithms that instead enable users to specify tasks simply by providing examples…

2021

SEAL: Self-supervised Embodied Active Learning using Exploration and 3D Consistency

NeurIPS 2021poster

In this paper, we explore how we can build upon the data and models of Internet images and use them to adapt to robot vision without requiring any extra labels. We present a framework called Self-supervised Embodied Active Learning (SEAL). It utilizes perception models trained on internet images to…

Cited by 96SourcePDFScholar
2021

Self-supervised Learning from a Multi-view Perspective

ICLR 2021poster

As a subset of unsupervised representation learning, self-supervised representation learning adopts self-defined signals as supervision and uses the learned representation for downstream tasks, such as object detection and image captioning. Many proposed approaches for self-supervised learning follo…

2021

Self-supervised Representation Learning with Relative Predictive Coding

ICLR 2021poster

This paper introduces Relative Predictive Coding (RPC), a new contrastive representation learning objective that maintains a good balance among training stability, minibatch size sensitivity, and downstream task performance. The key to the success of RPC is two-fold. First, RPC introduces the relati…

2021

StylePTB: A Compositional Benchmark for Fine-grained Controllable Text Style Transfer

NAACL 2021long

Text style transfer aims to controllably generate text with targeted stylistic changes while maintaining core meaning from the source sentence constant. Many of the existing style transfer benchmarks primarily focus on individual high-level semantic changes (e.g. positive to negative), which enable…

2021

Towards Understanding and Mitigating Social Biases in Language Models

ICML 2021spotlight

As machine learning methods are deployed in real-world settings such as healthcare, legal systems, and social science, it is crucial to recognize how they shape social biases and stereotypes in these sensitive decision-making processes. Among such real-world deployments are large-scale pretrained la…

2021

Uncertainty Weighted Actor-Critic for Offline Reinforcement Learning

ICML 2021spotlight

Offline Reinforcement Learning promises to learn effective policies from previously-collected, static datasets without the need for exploration. However, existing Q-learning and actor-critic based off-policy RL algorithms fail when bootstrapping from out-of-distribution (OOD) actions or states. We h…

2020

A Closer Look at Accuracy vs. Robustness

NeurIPS 2020poster

Current methods for training robust networks lead to a drop in test accuracy, which has led prior works to posit that a robustness-accuracy tradeoff may be inevitable in deep learning. We take a closer look at this phenomenon and first show that real image datasets are actually separated. With this…

2020

Capsules with Inverted Dot-Product Attention Routing

ICLR 2020poster

We introduce a new routing algorithm for capsule networks, in which a child capsule is routed to a parent based only on agreement between the parent's state and the child's vote. The new mechanism 1) designs routing via inverted dot-product attention; 2) imposes Layer Normalization as normalization…

Cited by 115SourceScholar
2020

Complex Transformer: A Framework for Modeling Complex-Valued Sequence

ICASSP 2020accepted

While deep learning has received a surge of interest in a variety of fields in recent years, major deep learning models barely use complex numbers. However, speech, signal and audio data are naturally complex-valued after Fourier Transform, and studies have shown a potentially richer representation…

Cited by 0SourceScholar
2020

Differentiable Reasoning over a Virtual Knowledge Base

ICLR 2020talk

We consider the task of answering complex multi-hop questions using a corpus as a virtual knowledge base (KB). In particular, we describe a neural module, DrKIT, that traverses textual data like a KB, softly following paths of relations between mentions of entities in the corpus. At each step the mo…

Cited by 103SourceScholar
2020

Embodied Multimodal Multitask Learning

IJCAI 2020poster

Visually-grounded embodied language learning models have recently shown to be effective at learning multiple multimodal tasks such as following navigational instructions and answering questions. In this paper, we address two key limitations of these models, (a) the inability to transfer the grounded…

Cited by 0SourcePDFScholar
2020

Harnessing the Power of Infinitely Wide Deep Nets on Small-data Tasks

ICLR 2020spotlight

Recent research shows that the following two models are equivalent: (a) infinitely wide neural networks (NNs) trained under l2 loss by gradient descent with infinitesimally small learning rate (b) kernel regression with respect to so-called Neural Tangent Kernels (NTKs) (Jacot et al., 2018). An effi…

Cited by 202SourcecodeScholar
2020

Learning To Explore Using Active Neural SLAM

ICLR 2020poster

This work presents a modular and hierarchical approach to learn policies for exploring 3D environments, called `Active Neural SLAM'. Our approach leverages the strengths of both classical and learning-based methods, by using analytical path planners with learned SLAM module, and global and local pol…

Cited by 649SourcecodeScholar
2020

Neural Methods for Point-wise Dependency Estimation

NeurIPS 2020spotlight

Since its inception, the neural estimation of mutual information (MI) has demonstrated the empirical success of modeling expected dependency between high-dimensional random variables. However, MI is an aggregate statistic and cannot be used to measure point-wise dependency between different events.…

2020

Object Goal Navigation using Goal-Oriented Semantic Exploration

NeurIPS 2020poster

This work studies the problem of object goal navigation which involves navigating to an instance of the given object category in unseen environments. End-to-end learning-based navigation methods struggle at this task as they are ineffective at exploration and long-term planning. We propose a modular…

2020

On Reward-Free Reinforcement Learning with Linear Function Approximation

NeurIPS 2020poster

Reward-free reinforcement learning (RL) is a framework which is suitable for both the batch RL setting and the setting where there are many reward functions of interest. During the exploration phase, an agent collects samples without using a pre-specified reward function. After the exploration phase…

Cited by 135SourcePDFScholar
2020

Planning with General Objective Functions: Going Beyond Total Rewards

NeurIPS 2020poster

Standard sequential decision-making paradigms aim to maximize the cumulative reward when interacting with the unknown environment., i.e., maximize $\sum_{h = 1}^H r_h$ where $H$ is the planning horizon. However, this paradigm fails to model important practical applications, e.g., safe control that a…

Cited by 24SourcePDFScholar
2020

Reinforcement Learning with General Value Function Approximation: Provably Efficient Approach via Bounded Eluder Dimension

NeurIPS 2020poster

Value function approximation has demonstrated phenomenal empirical success in reinforcement learning (RL). Nevertheless, despite a handful of recent progress on developing theory for RL with linear function approximation, the understanding of \emph{general} function approximation schemes largely rem…

Cited by 192SourcePDFScholar
2020

Rewriting History with Inverse RL: Hindsight Inference for Policy Improvement

NeurIPS 2020oral

Multi-task reinforcement learning (RL) aims to simultaneously learn policies for solving many tasks. Several prior works have found that relabeling past experience with different reward functions can improve sample efficiency. Relabeling methods typically pose the question: if, in hindsight, we assu…

2020

Weakly-Supervised Reinforcement Learning for Controllable Behavior

NeurIPS 2020poster

Reinforcement learning (RL) is a powerful framework for learning to take actions to solve tasks. However, in many settings, an agent must winnow down the inconceivably large space of all possible tasks to the single task that it is currently being asked to solve. Can we instead constrain the space o…

2019

AutoLoss: Learning Discrete Schedule for Alternate Optimization

ICLR 2019poster

Many machine learning problems involve iteratively and alternately optimizing different task objectives with respect to different sets of parameters. Appropriately scheduling the optimization of a task objective or a set of parameters is usually crucial to the quality of convergence. In this paper,…

Cited by 51SourcePDFScholar
2019

Deep Gamblers: Learning to Abstain with Portfolio Theory

NeurIPS 2019poster

We deal with the selective classification problem (supervised-learning problem with a rejection option), where we want to achieve the best performance at a certain level of coverage of the data. We transform the original $m$-class classification problem to (m+1)-class where the (m+1)-th class repres…

2019

Deep Neural Networks with Multi-Branch Architectures Are Intrinsically Less Non-Convex

AISTATS 2019poster

Several recently proposed architectures of neural networks such as ResNeXt, Inception, Xception, SqueezeNet and Wide ResNet are based on the designing idea of having multiple branches and have demonstrated improved performance in many applications. We show that one cause for such success is due to t…

Cited by 48SourcePDFScholar
2019

Graph Neural Tangent Kernel: Fusing Graph Neural Networks with Graph Kernels

NeurIPS 2019poster

While graph kernels (GKs) are easy to train and enjoy provable theoretical guarantees, their practical performances are limited by their expressive power, as the kernel function often depends on hand-crafted combinatorial features of graphs. Compared to graph kernels, graph neural networks (GNNs) us…

2019

Learning Data Manipulation for Augmentation and Weighting

NeurIPS 2019poster

Manipulating data, such as weighting data examples or augmenting with new instances, has been increasingly used to improve model training. Previous work has studied various rule- or learning-based approaches designed for specific types of data manipulation. In this work, we propose a new method that…

2019

Learning Factorized Multimodal Representations

ICLR 2019poster

Learning multimodal representations is a fundamentally complex research problem due to the presence of multiple heterogeneous sources of information. Although the presence of multiple modalities provides additional valuable information, there are two key challenges to address when learning from mult…

2019

Learning Neural Networks with Adaptive Regularization

NeurIPS 2019poster

Feed-forward neural networks can be understood as a combination of an intermediate representation and a linear hypothesis. While most previous works aim to diversify the representations, we explore the complementary direction by performing an adaptive and data-dependent regularization motivated by t…

2019

Mixtape: Breaking the Softmax Bottleneck Efficiently

NeurIPS 2019poster

The softmax bottleneck has been shown to limit the expressiveness of neural lan- guage models. Mixture of Softmaxes (MoS) is an effective approach to address such a theoretical limitation, but are expensive compared to softmax in terms of both memory and time. We propose Mixtape, an output layer tha…

Cited by 16SourcePDFScholar
2019

On Exact Computation with an Infinitely Wide Neural Net

NeurIPS 2019spotlight

How well does a classic deep net architecture like AlexNet or VGG19 classify on a standard dataset such as CIFAR-10 when its “width”— namely, number of channels in convolutional layers, and number of nodes in fully-connected internal layers — is allowed to increase to infinity? Such questions have c…

2019

Post Selection Inference with Incomplete Maximum Mean Discrepancy Estimator

ICLR 2019poster

Measuring divergence between two distributions is essential in machine learning and statistics and has various applications including binary classification, change point detection, and two-sample test. Furthermore, in the era of big data, designing divergence measure that is interpretable and can ha…

Cited by 28SourcePDFScholar
2019

Search on the Replay Buffer: Bridging Planning and Reinforcement Learning

NeurIPS 2019poster

The history of learning for control has been an exciting back and forth between two broad classes of algorithms: planning and reinforcement learning. Planning algorithms effectively reason over long horizons, but assume access to a local policy and distance metric over collision-free paths. Reinforc…

2019

Video Relationship Reasoning Using Gated Spatio-Temporal Energy Graph

CVPR 2019poster

Visual relationship reasoning is a crucial yet challenging task for understanding rich interactions across visual concepts. For example, a relationship \ man, open, door\ involves a complex relation \ open\ between concrete entities \ man, door\ . While much of the existing work has studied this p…

Cited by 127PDFcodeScholar
2019

XLNet: Generalized Autoregressive Pretraining for Language Understanding

NeurIPS 2019oral

With the capability of modeling bidirectional contexts, denoising autoencoding based pretraining like BERT achieves better performance than pretraining approaches based on autoregressive language modeling. However, relying on corrupting the input with masks, BERT neglects dependency between the mask…

2018

A Generic Approach for Escaping Saddle points

AISTATS 2018poster

A central challenge to using first-order methods for optimizing nonconvex problems is the presence of saddle points. First-order methods often get stuck at saddle points, greatly deteriorating their performance. Typically, to escape from saddles one has to use second-order methods. However, most wor…

Cited by 0SourcePDFScholar
2018

Breaking the Softmax Bottleneck: A High-Rank RNN Language Model

ICLR 2018oral

We formulate language modeling as a matrix factorization problem, and show that the expressiveness of Softmax-based models (including the majority of neural language models) is limited by a Softmax bottleneck. Given that natural language is highly context-dependent, this further implies that in prac…

2018

Deep Generative Models with Learnable Knowledge Constraints

NeurIPS 2018poster

The broad set of deep generative models (DGMs) has achieved remarkable advances. However, it is often difficult to incorporate rich structured domain knowledge with the end-to-end DGMs. Posterior regularization (PR) offers a principled framework to impose structured constraints on probabilistic mode…

Cited by 99SourcePDFScholar
2018

GLoMo: Unsupervised Learning of Transferable Relational Graphs

NeurIPS 2018poster

Modern deep transfer learning approaches have mainly focused on learning generic feature vectors from one task that are transferable to other tasks, such as word embeddings in language and pretrained convolutional features in vision. However, these approaches usually transfer unary features and larg…

Cited by 27SourcePDFScholar
2018

How Many Samples are Needed to Estimate a Convolutional Neural Network?

NeurIPS 2018poster

A widespread folklore for explaining the success of Convolutional Neural Networks (CNNs) is that CNNs use a more compact representation than the Fully-connected Neural Network (FNN) and thus require fewer training samples to accurately estimate their parameters. We initiate the study of rigorously c…

Cited by 88SourcePDFScholar
2018

Transformation Autoregressive Networks

ICML 2018oral

The fundamental task of general density estimation $p(x)$ has been of keen interest to machine learning. In this work, we attempt to systematically characterize methods for density estimation. Broadly speaking, most of the existing methods can be categorized into either using:

2017

Deep Determinantal Point Process for Large-Scale Multi-Label Classification

ICCV 2017poster

We study large-scale multi-label classification (MLC) on two recently released datasets: Youtube-8M and Open Images that contain millions of data instances and thousands of classes. The unprecedented problem scale poses great challenges for MLC. First, finding out the correct label subset out of exp…

Cited by 33PDFScholar
2017

Deep Sets

NeurIPS 2017oral

We study the problem of designing models for machine learning tasks defined on sets. In contrast to the traditional approach of operating on fixed dimensional vectors, we consider objective functions defined on sets and are invariant to permutations. Such problems are widespread, ranging from the es…

2017

Good Semi-supervised Learning That Requires a Bad GAN

NeurIPS 2017poster

Semi-supervised learning methods based on generative adversarial networks (GANs) obtained strong empirical results, but it is not clear 1) how the discriminator benefits from joint training with a generator, and 2) why good semi-supervised classification performance and a good generator cannot be ob…

2017

Improved Variational Autoencoders for Text Modeling using Dilated Convolutions

ICML 2017poster

Recent work on generative text modeling has found that variational autoencoders (VAE) with LSTM decoders perform worse than simpler LSTM language models (Bowman et al., 2015). This negative result is so far poorly understood, but has been attributed to the propensity of LSTM decoders to ignore condi…

Cited by 485SourcePDFScholar
2017

On the Quantitative Analysis of Decoder-Based Generative Models

ICLR 2017poster

The past several years have seen remarkable progress in generative models which produce convincing samples of images and other modalities. A shared component of some popular models such as generative adversarial networks and generative moment matching networks, is a decoder network, a parametric dee…

Cited by 285SourcecodeScholar
2017

Spatially Adaptive Computation Time for Residual Networks

CVPR 2017poster

This paper proposes a deep learning architecture based on Residual Network that dynamically adjusts the number of executed layers for the regions of the image. This architecture is end-to-end trainable, deterministic and problem-agnostic. It is therefore applicable without any modifications to a wid…

Cited by 431PDFcodeScholar
2017

Toward Controlled Generation of Text

ICML 2017poster

Generic generation and manipulation of text is challenging and has limited success compared to recent deep generative modeling in visual domain. This paper aims at generating plausible text sentences, whose attributes are controlled by learning disentangled latent representations with designated sem…

2017

Transfer Learning for Sequence Tagging with Hierarchical Recurrent Networks

ICLR 2017poster

Recent papers have shown that neural networks obtain state-of-the-art performance on several different sequence tagging tasks. One appealing property of such systems is their generality, as excellent performance can be achieved with a unified architecture and without task-specific feature engineerin…

Cited by 432SourcecodeScholar
2017

Words or Characters? Fine-grained Gating for Reading Comprehension

ICLR 2017poster

Previous work combines word-level and character-level representations using concatenation or scalar weighting, which is suboptimal for high-level tasks like reading comprehension. We present a fine-grained gating mechanism to dynamically combine word-level and character-level representations based o…

Cited by 100SourcecodeScholar
2016

Architectural Complexity Measures of Recurrent Neural Networks

NeurIPS 2016poster

In this paper, we systematically analyze the connecting architectures of recurrent neural networks (RNNs). Our main contribution is twofold: first, we present a rigorous graph-theoretic framework describing the connecting architectures of RNNs in general. Second, we propose three architecture comple…

2016

Iterative Refinement of the Approximate Posterior for Directed Belief Networks

NeurIPS 2016poster

Variational methods that rely on a recognition network to approximate the posterior of directed graphical models offer better inference and learning than previous methods. Recent advances that exploit the capacity and flexibility in this approach have expanded what kinds of models can be trained. Ho…

2016

On Multiplicative Integration with Recurrent Neural Networks

NeurIPS 2016poster

We introduce a general simple structural design called “Multiplicative Integration” (MI) to improve recurrent neural networks (RNNs). MI changes the way of how the information flow gets integrated in the computational building block of an RNN, while introducing almost no extra parameters. The new st…

2016

Path-Normalized Optimization of Recurrent Neural Networks with ReLU Activations

NeurIPS 2016poster

We investigate the parameter-space geometry of recurrent neural networks (RNNs), and develop an adaptation of path-SGD optimization method, attuned to this geometry, that can learn plain RNNs with ReLU activations. On several datasets that require capturing long-term dependency structure, we show th…

Cited by 38SourcePDFScholar
2016

Review Networks for Caption Generation

NeurIPS 2016poster

We propose a novel extension of the encoder-decoder framework, called a review network. The review network is generic and can enhance any existing encoder- decoder model: in this paper, we consider RNN decoders with both CNN and RNN encoders. The review network performs a number of review steps with…

2015

Accurate and conservative estimates of MRF log-likelihood using reverse annealing

AISTATS 2015poster

Markov random fields (MRFs) are difficult to evaluate as generative models because computing the test log-probabilities requires the intractable partition function. Annealed importance sampling (AIS) is widely used to estimate MRF partition functions, and often yields quite accurate results. However…

Cited by 79SourcePDFScholar
2015

Aligning Books and Movies: Towards Story-Like Visual Explanations by Watching Movies and Reading Books

ICCV 2015oral

Books are a rich source of both fine-grained information, how a character, an object or a scene looks like, as well as high-level semantics, what someone is thinking, feeling and how these states evolve through a story. This paper aims to align books to their movie releases in order to provide rich…

Cited by 3512PDFScholar
2015

Learning Wake-Sleep Recurrent Attention Models

NeurIPS 2015spotlight

Despite their success, convolutional neural networks are computationally expensive because they must examine all image locations. Stochastic attention-based models have been shown to improve computational efficiency at test time, but they remain difficult to train because of intractable posterior in…

Cited by 77SourcePDFScholar
2015

Path-SGD: Path-Normalized Optimization in Deep Neural Networks

NeurIPS 2015poster

We revisit the choice of SGD for training deep neural networks by reconsidering the appropriate geometry in which to optimize the weights. We argue for a geometry invariant to rescaling of weights that does not affect the output of the network, and suggest Path-SGD, which is an approximate steepest…

2015

Predicting Deep Zero-Shot Convolutional Neural Networks Using Textual Descriptions

ICCV 2015poster

One of the main challenges in Zero-Shot Learning of visual categories is gathering semantic attributes to accompany images. Recent work has shown that learning from textual descriptions, such as Wikipedia articles, avoids the problem of having to explicitly define these attributes. We present a new…

Cited by 527PDFScholar
2015

Skip-Thought Vectors

NeurIPS 2015poster

We describe an approach for unsupervised learning of a generic, distributed sentence encoder. Using the continuity of text from books, we train an encoder-decoder model that tries to reconstruct the surrounding sentences of an encoded passage. Sentences that share semantic and syntactic properties a…

2015

segDeepM: Exploiting Segmentation and Context in Deep Neural Networks for Object Detection

CVPR 2015poster

In this paper, we propose an approach that exploits object segmentation in order to improve the accuracy of object detection. We frame the problem as inference in a Markov Random Field, in which each detection hypothesis scores object appearance as well as contextual information using Convolutional…

Cited by 211SourcePDFScholar