← Search

Maithra Raghu

15 accepted papers

2022

Learning to Reason with Neural Networks: Generalization, Unseen Data and Boolean Measures

NeurIPS 2022accept

This paper considers the Pointer Value Retrieval (PVR) benchmark introduced in [ZRKB21], where a `reasoning' function acts on a string of digits to produce the label. More generally, the paper considers the learning of logical functions with gradient descent (GD) on neural networks. It is first show…

2021

Anatomy of Catastrophic Forgetting: Hidden Representations and Task Semantics

ICLR 2021poster

Catastrophic forgetting is a recurring challenge to developing versatile deep learning models. Despite its ubiquity, there is limited understanding of its connections to neural network (hidden) representations and task semantics. In this paper, we address this important knowledge gap. Through quanti…

Cited by 211SourcePDFScholar
2021

Do Vision Transformers See Like Convolutional Neural Networks?

NeurIPS 2021poster

Convolutional neural networks (CNNs) have so far been the de-facto model for visual data. Recent work has shown that (Vision) Transformer models (ViT) can achieve comparable or even superior performance on image classification tasks. This raises a central question: how are Vision Transformers solvin…

Cited by 1303SourcePDFScholar
2021

Do Wide and Deep Networks Learn the Same Things? Uncovering How Neural Network Representations Vary with Width and Depth

ICLR 2021poster

A key factor in the success of deep neural networks is the ability to scale models to improve performance by varying the architecture depth and width. This simple property of neural network design has resulted in highly effective architectures for a variety of tasks. Nevertheless, there is limited u…

2021

Teaching with Commentaries

ICLR 2021poster

Effective training of deep neural networks can be challenging, and there remain many open questions on how to best learn these models. Recently developed methods to improve neural network training examine teaching: providing learned information during the training process to improve downstream model…

2020

Rapid Learning or Feature Reuse? Towards Understanding the Effectiveness of MAML

ICLR 2020poster

An important research direction in machine learning has centered around developing meta-learning algorithms to tackle few-shot learning. An especially successful algorithm has been Model Agnostic Meta-Learning (MAML), a method that consists of two optimization loops, with the outer loop finding a me…

Cited by 809SourceScholar
2019

Direct Uncertainty Prediction for Medical Second Opinions

ICML 2019oral

The issue of disagreements amongst human experts is a ubiquitous one in both machine learning and medicine. In medicine, this often corresponds to doctor disagreements on a patient diagnosis. In this work, we show that machine learning models can be successfully trained to give uncertainty scores to…

Cited by 171SourcePDFScholar
2019

Transfusion: Understanding Transfer Learning for Medical Imaging

NeurIPS 2019poster

Transfer learning from natural image datasets, particularly ImageNet, using standard large models and corresponding pretrained weights has become a de-facto method for deep learning applications to medical imaging. However, there are fundamental differences in data sizes, features and task specific…

Cited by 1515SourcePDFScholar
2018

Can Deep Reinforcement Learning Solve Erdos-Selfridge-Spencer Games?

ICML 2018oral

Deep reinforcement learning has achieved many recent successes, but our understanding of its strengths and limitations is hampered by the lack of rich environments in which we can fully characterize optimal behavior, and correspondingly diagnose individual actions against such a characterization. He…

Cited by 43SourcePDFScholar
2018

Can Deep Reinforcement Learning solve Erdos-Selfridge-Spencer Games?

ICLR 2018workshop

Deep reinforcement learning has achieved many recent successes, but our understanding of its strengths and limitations is hampered by the lack of rich environments in which we can fully characterize optimal behavior, and correspondingly diagnose individual actions against such a characterization.…

Cited by 42SourceScholar
2018

Insights on representational similarity in neural networks with canonical correlation

NeurIPS 2018poster

Comparing different neural network representations and determining how representations evolve over time remain challenging open questions in our understanding of the function of neural networks. Comparing representations in neural networks is fundamentally difficult as the structure of representatio…

2017

On the Expressive Power of Deep Neural Networks

ICML 2017poster

We propose a new approach to the problem of neural network expressivity, which seeks to characterize how structural properties of a neural network family affect the functions it is able to compute. Our approach is based on an interrelated set of measures of expressivity, unified by the novel notion…

Cited by 1057SourcePDFScholar
2017

SVCCA: Singular Vector Canonical Correlation Analysis for Deep Learning Dynamics and Interpretability

NeurIPS 2017poster

We propose a new technique, Singular Vector Canonical Correlation Analysis (SVCCA), a tool for quickly comparing two representations in a way that is both invariant to affine transform (allowing comparison between different layers and networks) and fast to compute (allowing more comparisons to be ca…

2016

Exponential expressivity in deep neural networks through transient chaos

NeurIPS 2016poster

We combine Riemannian geometry with the mean field theory of high dimensional chaos to study the nature of signal propagation in deep neural networks with random weights. Our results reveal a phase transition in the expressivity of random deep networks, with networks in the chaotic phase computing n…