← Search

Nan Rosemary Ke

17 accepted papers

2023

Learning to Induce Causal Structure

ICLR 2023poster

The fundamental challenge in causal induction is to infer the underlying graph structure given observational and/or interventional data. Most existing causal induction algorithms operate by generating candidate graphs and evaluating them using either score-based methods (including continuous optimiz…

Cited by 59SourcePDFScholar
2022

Coordination Among Neural Modules Through a Shared Global Workspace

ICLR 2022oral

Deep learning has seen a movement away from representing examples with a monolithic hidden state towards a richly structured state. For example, Transformers segment by position, and object-centric architectures decompose images into entities. In all these architectures, interactions between differe…

Cited by 108SourcePDFScholar
2022

On the Convergence of Continuous Constrained Optimization for Structure Learning

AISTATS 2022poster

Recently, structure learning of directed acyclic graphs (DAGs) has been formulated as a continuous optimization problem by leveraging an algebraic characterization of acyclicity. The constrained problem is solved using the augmented Lagrangian method (ALM) which is often preferred to the quadratic p…

2022

Retrieval-Augmented Reinforcement Learning

ICML 2022spotlight

Most deep reinforcement learning (RL) algorithms distill experience into parametric behavior policies or value functions via gradient updates. While effective, this approach has several disadvantages: (1) it is computationally expensive, (2) it can take many updates to integrate experiences into the…

Cited by 59SourcePDFScholar
2022

Temporal Latent Bottleneck: Synthesis of Fast and Slow Processing Mechanisms in Sequence Learning

NeurIPS 2022accept

Recurrent neural networks have a strong inductive bias towards learning temporally compressed representations, as the entire history of a sequence is represented by a single vector. By contrast, Transformers have little inductive bias towards learning temporally compressed representations, as they…

Cited by 14SourcePDFScholar
2021

Fast And Slow Learning Of Recurrent Independent Mechanisms

ICLR 2021poster

Decomposing knowledge into interchangeable pieces promises a generalization advantage when there are changes in distribution. A learning agent interacting with its environment is likely to be faced with situations requiring novel combinations of existing pieces of knowledge. We hypothesize that such…

Cited by 55SourcePDFScholar
2021

Neural Production Systems

NeurIPS 2021poster

Visual environments are structured, consisting of distinct objects or entities. These entities have properties---visible or latent---that determine the manner in which they interact with one another. To partition images into entities, deep-learning researchers have proposed structural inductive bia…

Cited by 96SourcePDFScholar
2021

Systematic Evaluation of Causal Discovery in Visual Model Based Reinforcement Learning

NeurIPS 2021poster

Inducing causal relationships from observations is a classic problem in machine learning. Most work in causality starts from the premise that the causal variables themselves are observed. However, for AI agents such as robots trying to make sense of their environment, the only observables are low-l…

Cited by 52SourcecodeScholar
2020

A Meta-Transfer Objective for Learning to Disentangle Causal Mechanisms

ICLR 2020poster

We propose to use a meta-learning objective that maximizes the speed of transfer on a modified distribution to learn how to modularize acquired knowledge. In particular, we focus on how to factor a joint distribution into appropriate conditionals, consistent with the causal directions. We explain wh…

Cited by 438SourceScholar
2019

Modeling the Long Term Future in Model-Based Reinforcement Learning

ICLR 2019poster

In model-based reinforcement learning, the agent interleaves between model learning and planning. These two components are inextricably intertwined. If the model is not able to provide sensible long-term prediction, the executed planer would exploit model flaws, which can yield catastrophic failur…

Cited by 42SourcePDFScholar
2019

h-detach: Modifying the LSTM Gradient Towards Better Optimization

ICLR 2019poster

Recurrent neural networks are known for their notorious exploding and vanishing gradient problem (EVGP). This problem becomes more evident in tasks where the information needed to correctly solve them exist over long time scales, because EVGP prevents important gradient components from being back-pr…

2018

Focused Hierarchical RNNs for Conditional Sequence Processing

ICML 2018oral

Recurrent Neural Networks (RNNs) with attention mechanisms have obtained state-of-the-art results for many sequence processing tasks. Most of these models use a simple form of encoder with attention that looks over the entire sequence and assigns a weight to each token independently. We present a me…

Cited by 28SourcePDFScholar
2018

Sparse Attentive Backtracking: Temporal Credit Assignment Through Reminding

NeurIPS 2018spotlight

Learning long-term dependencies in extended temporal sequences requires credit assignment to events far back in the past. The most common method for training recurrent neural networks, back-propagation through time (BPTT), requires credit information to be propagated backwards through every single s…

Cited by 113SourcePDFScholar
2018

Twin Networks: Matching the Future for Sequence Generation

ICLR 2018poster

We propose a simple technique for encouraging generative RNNs to plan ahead. We train a ``backward'' recurrent network to generate a given sequence in reverse order, and we encourage states of the forward model to predict cotemporal states of the backward model. The backward network is used only dur…

2017

Variational Walkback: Learning a Transition Operator as a Stochastic Recurrent Net

NeurIPS 2017poster

We propose a novel method to {\it directly} learn a stochastic transition operator whose repeated application provides generated samples. Traditional undirected graphical models approach this problem indirectly by learning a Markov chain model whose stationary distribution obeys detailed balance wit…

2017

Z-Forcing: Training Stochastic Recurrent Networks

NeurIPS 2017poster

Many efforts have been devoted to training generative latent variable models with autoregressive decoders, such as recurrent neural networks (RNN). Stochastic recurrent models have been successful in capturing the variability observed in natural sequential data such as speech. We unify successful id…

2017

Zoneout: Regularizing RNNs by Randomly Preserving Hidden Activations

ICLR 2017poster

We propose zoneout, a novel method for regularizing RNNs. At each timestep, zoneout stochastically forces some hidden units to maintain their previous values. Like dropout, zoneout uses random noise to train a pseudo-ensemble, improving generalization. But by preserving instead of dropping hidden un…

Cited by 397SourcecodeScholar