← Search

Pushmeet Kohli

69 accepted papers

2025

Scaling Wearable Foundation Models

ICLR 2025poster

Wearable sensors have become ubiquitous thanks to a variety of health tracking features. The resulting continuous and longitudinal measurements from everyday life generate large volumes of data. However, making sense of these observations for scientific and actionable insights is non-trivial. Inspir…

Cited by 6SourcePDFScholar
2025

SensorLM: Learning the Language of Wearable Sensors

NeurIPS 2025poster

We present SensorLM, a family of sensor-language foundation models that enable wearable sensor data understanding with natural language. Despite its pervasive nature, aligning and interpreting sensor data with language remains challenging due to the lack of paired, richly annotated sensor-text descr…

Cited by 0SourcecodeScholar
2022

Making Sense of Raw Input (Extended Abstract)

IJCAI 2022poster

How should a machine intelligence perform unsupervised structure discovery over streams of sensory input? One approach to this problem is to cast it as an apperception task. Here, the task is to construct an explicit interpretable theory that both explains the sensory sequence and also satisfies a s…

Cited by 0SourcePDFScholar
2021

Challenges in Detoxifying Language Models

EMNLP 2021finding

Large language models (LM) generate remarkably fluent text and can be efficiently adapted across NLP tasks. Measuring and guaranteeing the quality of generated text in terms of safety is imperative for deploying LMs in the real world; to this end, prior work often relies on automatic evaluation of L…

Cited by 256SourcePDFScholar
2021

Self-supervised Adversarial Robustness for the Low-label, High-data Regime

ICLR 2021poster

Recent work discovered that training models to be invariant to adversarial perturbations requires substantially larger datasets than those required for standard classification. Perhaps more surprisingly, these larger datasets can be "mostly" unlabeled. Pseudo-labeling, a technique simultaneously pio…

Cited by 49SourcePDFScholar
2020

A FRAMEWORK FOR ROBUSTNESS CERTIFICATION OF SMOOTHED CLASSIFIERS USING F-DIVERGENCES

ICLR 2020poster

Formal verification techniques that compute provable guarantees on properties of machine learning models, like robustness to norm-bounded adversarial perturbations, have yielded impressive results. Although most techniques developed so far require knowledge of the architecture of the machine learnin…

Cited by 64SourceScholar
2020

Achieving Robustness in the Wild via Adversarial Mixing With Disentangled Representations

CVPR 2020poster

Recent research has made the surprising finding that state-of-the-art deep learning models sometimes fail to generalize to small variations of the input. Adversarial training has been shown to be an effective approach to overcome this problem. However, its application has been limited to enforcing i…

Cited by 67PDFcodeScholar
2020

Adversarially Robust Representations with Smooth Encoders

ICLR 2020poster

This paper studies the undesired phenomena of over-sensitivity of representations learned by deep networks to semantically-irrelevant changes in data. We identify a cause for this shortcoming in the classical Variational Auto-encoder (VAE) objective, the evidence lower bound (ELBO). We show that the…

Cited by 35SourceScholar
2020

CLEVRER: Collision Events for Video Representation and Reasoning

ICLR 2020spotlight

The ability to reason about temporal and causal events from videos lies at the core of human intelligence. Most video reasoning benchmarks, however, focus on pattern recognition from complex visual and language input, instead of on causal structure. We study the complementary problem, exploring the…

Cited by 559SourceScholar
2020

Enabling certification of verification-agnostic networks via memory-efficient semidefinite programming

NeurIPS 2020poster

Convex relaxations have emerged as a promising approach for verifying properties of neural networks, but widely used using Linear Programming (LP) relaxations only provide meaningful certificates when networks are specifically trained to facilitate verification. This precludes many important applica…

2020

Lagrangian Decomposition for Neural Network Verification

UAI 2020poster

A fundamental component of neural network verification is the computation of bounds on the values their outputs can take. Previous methods have either used off-the-shelf solvers, discarding the problem structure, or relaxed the problem even further, making the bounds unnecessarily loose. We propose…

2020

Reinforced Genetic Algorithm Learning for Optimizing Computation Graphs

ICLR 2020poster

We present a deep reinforcement learning approach to minimizing the execution cost of neural network computation graphs in an optimizing compiler. Unlike earlier learning-based works that require training the optimizer on the same graph to be optimized, we propose a learning approach that trains an…

Cited by 77SourceScholar
2020

The Autoencoding Variational Autoencoder

NeurIPS 2020spotlight

Does a Variational AutoEncoder (VAE) consistently encode typical samples generated from its decoder? This paper shows that the perhaps surprising answer to this question is `No'; a (nominally trained) VAE does not necessarily amortize inference for typical samples that it is capable of generating. W…

Cited by 0SourcePDFScholar
2020

Toward Evaluating Robustness of Deep Reinforcement Learning with Continuous Control

ICLR 2020poster

Deep reinforcement learning has achieved great success in many previously difficult reinforcement learning tasks, yet recent studies show that deep RL agents are also unavoidably susceptible to adversarial perturbations, similar to deep neural networks in classification tasks. Prior works mostly foc…

Cited by 36SourceScholar
2020

Towards Robust Image Classification Using Sequential Attention Models

CVPR 2020poster

In this paper we propose to augment a modern neural-network architecture with an attention model inspired by human perception. Specifically, we adversarially train and analyze a neural model incorporating a human inspired, visual attention component that is guided by a recurrent top-down sequential…

Cited by 88PDFScholar
2020

Towards Verified Robustness under Text Deletion Interventions

ICLR 2020poster

Neural networks are widely used in Natural Language Processing, yet despite their empirical successes, their behaviour is brittle: they are both over-sensitive to small input changes, and under-sensitive to deletions of large fractions of input text. This paper aims to tackle under-sensitivity in th…

Cited by 3SourceScholar
2020

Training Generative Adversarial Networks by Solving Ordinary Differential Equations

NeurIPS 2020spotlight

The instability of Generative Adversarial Network (GAN) training has frequently been attributed to gradient descent. Consequently, recent methods have aimed to tailor the models and training procedures to stabilise the discrete updates. In contrast, we study the continuous-time dynamics induced by G…

2019

Adversarial Robustness through Local Linearization

NeurIPS 2019poster

Adversarial training is an effective methodology for training deep neural networks that are robust against adversarial, norm-bounded perturbations. However, the computational cost of adversarial training grows prohibitively as the size of the model and number of input dimensions increase. Further, t…

Cited by 367SourcePDFScholar
2019

Analysing Mathematical Reasoning Abilities of Neural Models

ICLR 2019poster

Mathematical reasoning---a core ability within human intelligence---presents some unique challenges as a domain: we do not come to understand and solve mathematical problems primarily on the back of experience and evidence, but on the basis of inferring, learning, and exploiting laws, axioms, and sy…

Cited by 496SourcePDFScholar
2019

Are Labels Required for Improving Adversarial Robustness?

NeurIPS 2019poster

Recent work has uncovered the interesting (and somewhat surprising) finding that training models to be invariant to adversarial perturbations requires substantially larger datasets than those required for standard classification. This result is a key hurdle in the deployment of robust machine learni…

2019

CompILE: Compositional Imitation Learning and Execution

ICML 2019oral

We introduce Compositional Imitation Learning and Execution (CompILE): a framework for learning reusable, variable-length segments of hierarchically-structured behavior from demonstration data. CompILE uses a novel unsupervised, fully-differentiable sequence segmentation module to learn latent encod…

2019

Efficient Neural Network Verification with Exactness Characterization

UAI 2019poster

Remarkable progress has been made on verification of neural networks, i.e., showing that neural networks are provably consistent with specifications encoding properties like adversarial robustness. Recent methods developed for scalable neural network verification are based on computing an upper bou…

Cited by 35SourcePDFScholar
2019

Graph Matching Networks for Learning the Similarity of Graph Structured Objects

ICML 2019oral

This paper addresses the challenging problem of retrieval and matching of graph structured objects, and makes two key contributions. First, we demonstrate how Graph Neural Networks (GNN), which have emerged as an effective model for various supervised prediction problems defined on structured data,…

Cited by 779SourcePDFScholar
2019

Knowing When to Stop: Evaluation and Verification of Conformity to Output-Size Specifications

CVPR 2019poster

Neural architectures able to generate variable-length outputs are extremely effective for applications like Machine Translation and Image Captioning. In this paper, we study the vulnerability of these models to attacks aimed at changing the output-size that can have undesirable consequences includin…

Cited by 13PDFScholar
2019

Learning to Understand Goal Specifications by Modelling Reward

ICLR 2019poster

Recent work has shown that deep reinforcement-learning agents can learn to follow language-like instructions from infrequent environment rewards. However, this places on environment designers the onus of designing language-conditional reward functions which may not be easily or tractably implemented…

Cited by 175SourcePDFScholar
2019

Rigorous Agent Evaluation: An Adversarial Approach to Uncover Catastrophic Failures

ICLR 2019poster

This paper addresses the problem of evaluating learning systems in safety critical domains such as autonomous driving, where failures can have catastrophic consequences. We focus on two problems: searching for scenarios when learned agents fail and assessing their probability of failure. The standar…

Cited by 91SourcePDFScholar
2019

Scalable Verified Training for Provably Robust Image Classification

ICCV 2019poster

Recent work has shown that it is possible to train deep neural networks that are provably robust to norm-bounded adversarial perturbations. Most of these methods are based on minimizing an upper bound on the worst-case loss over all possible adversarial perturbations. While these techniques show pro…

Cited by 214PDFScholar
2019

Structured agents for physical construction

ICML 2019oral

Physical construction—the ability to compose objects, subject to physical dynamics, to serve some function—is fundamental to human intelligence. We introduce a suite of challenging physical construction tasks inspired by how children play with blocks, such as matching a target configuration, stackin…

Cited by 126SourcePDFScholar
2019

The Neuro-Symbolic Concept Learner: Interpreting Scenes, Words, and Sentences From Natural Supervision

ICLR 2019oral

We propose the Neuro-Symbolic Concept Learner (NS-CL), a model that learns visual concepts, words, and semantic parsing of sentences without explicit supervision on any of them; instead, our model learns by simply looking at images and reading paired questions and answers. Our model builds an object…

2019

Value Propagation Networks

ICLR 2019poster

We present Value Propagation (VProp), a set of parameter-efficient differentiable planning modules built on Value Iteration which can successfully be trained using reinforcement learning to solve unseen tasks, has the capability to generalize to larger map sizes, and can learn to navigate in dynamic…

Cited by 38SourcePDFScholar
2019

Verification of Non-Linear Specifications for Neural Networks

ICLR 2019poster

Prior work on neural network verification has focused on specifications that are linear functions of the output of the network, e.g., invariance of the classifier output under adversarial perturbations of the input. In this paper, we extend verification algorithms to be able to certify richer proper…

Cited by 50SourcePDFScholar
2018

A Unified View of Piecewise Linear Neural Network Verification

NeurIPS 2018poster

The success of Deep Learning and its potential use in many safety-critical applications has motivated research on formal verification of Neural Network (NN) models. Despite the reputation of learned NN models to behave as black boxes and the theoretical hardness of proving their properties, re…

2018

Batched Large-scale Bayesian Optimization in High-dimensional Spaces

AISTATS 2018poster

Bayesian optimization (BO) has become an effective approach for black-box function optimization problems when function evaluations are expensive and the optimum can be achieved within a relatively small number of queries. However, many cases, such as the ones with high-dimensional inputs, may requir…

Cited by 0SourcePDFScholar
2018

Can Neural Networks Understand Logical Entailment?

ICLR 2018poster

We introduce a new dataset of logical entailments for the purpose of measuring models' ability to capture and exploit the structure of logical expressions against an entailment prediction task. We use this task to compare a series of architectures which are ubiquitous in the sequence-processing lite…

Cited by 154SourcePDFScholar
2018

Leveraging Grammar and Reinforcement Learning for Neural Program Synthesis

ICLR 2018poster

Program synthesis is the task of automatically generating a program consistent with a specification. Recent years have seen proposal of a number of neural approaches for program synthesis, many of which adopt a sequence generation paradigm similar to neural machine translation, in which sequence-to-…

Cited by 257SourcePDFScholar
2018

Neural-Symbolic VQA: Disentangling Reasoning from Vision and Language Understanding

NeurIPS 2018spotlight

We marry two powerful ideas: deep representation learning for visual recognition and language understanding, and symbolic program execution for reasoning. Our neural-symbolic visual question answering (NS-VQA) system first recovers a structural scene representation from the image and a program trace…

2018

Programmatically Interpretable Reinforcement Learning

ICML 2018oral

We present a reinforcement learning framework, called Programmatically Interpretable Reinforcement Learning (PIRL), that is designed to generate interpretable and verifiable agent policies. Unlike the popular Deep Reinforcement Learning (DRL) paradigm, which represents policies by neural networks, P…

Cited by 497SourcePDFScholar
2018

Semantic Code Repair using Neuro-Symbolic Transformation Networks

ICLR 2018workshop

We study the problem of semantic code repair, which can be broadly defined as automatically fixing non-syntactic bugs in source code. The majority of past work in semantic code repair assumed access to unit tests against which candidate repairs could be validated. In contrast, the goal here is to de…

Cited by 52SourceScholar
2017

Batched High-dimensional Bayesian Optimization via Structural Kernel Learning

ICML 2017poster

Optimization of high-dimensional black-box functions is an extremely challenging problem. While Bayesian optimization has emerged as a popular approach for optimizing black-box functions, its applicability has been limited to low-dimensional problems due to its computational and statistical challeng…

2017

DeepContext: Context-Encoding Neural Pathways for 3D Holistic Scene Understanding

ICCV 2017poster

3D context has been shown to be an extremely important cue for scene understanding, yet very little research has been done on integrating context information with deep models. This paper presents an approach to embed 3D context into the topology of a neural network trained to perform holistic scene…

Cited by 82PDFScholar
2017

Learning Continuous Semantic Representations of Symbolic Expressions

ICLR 2017workshop

The question of how procedural knowledge is represented and inferred is a fundamental problem in machine learning and artificial intelligence. Recent work on program induction has proposed neural architectures, based on abstractions like stacks, Turing machines, and interpreters, that operate on ab…

Cited by 128SourceScholar
2017

Learning Continuous Semantic Representations of Symbolic Expressions

ICML 2017poster

Combining abstract, symbolic reasoning with continuous neural reasoning is a grand challenge of representation learning. As a step in this direction, we propose a new architecture, called neural equivalence network, for the problem of learning continuous semantic representations of algebraic and log…

Cited by 128SourcePDFScholar
2017

Learning Disentangled Representations with Semi-Supervised Deep Generative Models

NeurIPS 2017poster

Variational autoencoders (VAEs) learn representations of data by jointly training a probabilistic encoder and decoder network. Typically these models encode all features of the data into a single variable. Here we are interested in learning disentangled representations that encode distinct aspects o…

2017

Learning to See Physics via Visual De-animation

NeurIPS 2017poster

We introduce a paradigm for understanding physical scenes without human annotations. At the core of our system is a physical world representation that is first recovered by a perception module and then utilized by physics and graphics engines. During training, the perception module and the generativ…

Cited by 238SourcePDFScholar
2017

Neuro-Symbolic Program Synthesis

ICLR 2017poster

Recent years have seen the proposal of a number of neural architectures for the problem of Program Induction. Given a set of input-output examples, these architectures are able to learn mappings that generalize to new test inputs. While achieving impressive results, these approaches have a number of…

Cited by 418SourceScholar
2017

Realistic Dynamic Facial Textures From a Single Image Using GANs

ICCV 2017poster

We present a novel method to realistically puppeteer and animate a face from a single RGB image using a source video sequence. We begin by fitting a multilinear PCA model to obtain the 3D geometry and a single texture of the target face. In order for the animation to be realistic, however, we need d…

Cited by 116PDFScholar
2017

RobustFill: Neural Program Learning under Noisy I/O

ICML 2017poster

The problem of automatically generating a computer program from some specification has been studied since the early days of AI. Recently, two competing approaches for `automatic program learning’ have received significant attention: (1) `neural program synthesis’, where a neural network is condition…

Cited by 483SourcePDFScholar
2017

Stabilising Experience Replay for Deep Multi-Agent Reinforcement Learning

ICML 2017poster

Many real-world problems, such as network packet routing and urban traffic control, are naturally modeled as multi-agent reinforcement learning (RL) problems. However, existing multi-agent RL methods typically scale poorly in the problem size. Therefore, a key challenge is to translate the success o…

Cited by 819SourcePDFScholar
2017

Support Regularized Sparse Coding and Its Fast Encoder

ICLR 2017poster

Sparse coding represents a signal by a linear combination of only a few atoms of a learned over-complete dictionary. While sparse coding exhibits compelling performance for various machine learning tasks, the process of obtaining sparse code with fixed dictionary is independent for each data point w…

Cited by 2SourceScholar
2017

Zero-Shot Task Generalization with Multi-Task Deep Reinforcement Learning

ICML 2017poster

As a step towards developing zero-shot task generalization capabilities in reinforcement learning (RL), we introduce a new RL problem where the agent should learn to execute sequences of instructions after learning useful skills that solve subtasks. In this problem, we consider two types of generali…

Cited by 333SourcePDFScholar
2016

Adaptive Neural Compilation

NeurIPS 2016poster

This paper proposes an adaptive neural-compilation framework to address the problem of learning efficient program. Traditional code optimisation strategies used in compilers are based on applying pre-specified set of transformations that make the code faster to execute without changing its semantics…

2016

Batched Gaussian Process Bandit Optimization via Determinantal Point Processes

NeurIPS 2016poster

Gaussian Process bandit optimization has emerged as a powerful tool for optimizing noisy black box functions. One example in machine learning is hyper-parameter optimization where each evaluation of the target function may require training a model which may involve days or even weeks of computation.…

2016

PerforatedCNNs: Acceleration through Elimination of Redundant Convolutions

NeurIPS 2016poster

We propose a novel approach to reduce the computational cost of evaluation of convolutional neural networks, a factor that has hindered their deployment in low-power devices such as mobile phones. Inspired by the loop perforation technique from source code optimization, we speed up the bottleneck co…

2015

Consensus Message Passing for Layered Graphical Models

AISTATS 2015poster

Generative models provide a powerful framework for probabilistic reasoning. However, in many domains their use has been hampered by the practical difficulties of inference. This is particularly the case in computer vision, where models of the imaging process tend to be large, loopy and layered. For…

Cited by 5SourcePDFScholar
2015

Deep Convolutional Inverse Graphics Network

NeurIPS 2015spotlight

This paper presents the Deep Convolution Inverse Graphics Network (DC-IGN), a model that aims to learn an interpretable representation of images, disentangled with respect to three-dimensional scene structure and viewing transformations such as depth rotations and lighting variations. The DC-IGN mod…

Cited by 1155SourcePDFScholar
2015

Efficient Non-greedy Optimization of Decision Trees

NeurIPS 2015poster

Decision trees and randomized forests are widely used in computer vision and machine learning. Standard algorithms for decision tree induction optimize the split functions one node at a time according to some splitting criteria. This greedy procedure often leads to suboptimal trees. In this paper, w…

Cited by 146SourcePDFScholar
2015

Opening the Black Box: Hierarchical Sampling Optimization for Estimating Human Hand Pose

ICCV 2015oral

We address the problem of hand pose estimation, formulated as an inverse problem. Typical approaches optimize an energy function over pose parameters using a `black box' image generation procedure. This procedure knows little about either the relationships between the parameters or the form of the…

Cited by 170PDFScholar
2015

Picture: A Probabilistic Programming Language for Scene Perception

CVPR 2015poster

Recent progress on probabilistic modeling and statistical learning, coupled with the availability of large training datasets, has led to remarkable progress in computer vision. Generative probabilistic models, or analysis-by-synthesis approaches, can capture rich scene structure but have been less w…

Cited by 250SourcePDFScholar