activity
20172022
most citedOne Model To Learn Them All

258 citations · 1.1k across the 17 of their papers we have counts for

collaborators
Showing cs.LGShow all

13 papers · 1 filter

cs.LG2022★ 3 cited

Exploring Low Rank Training of Deep Neural Networks

Siddhartha Rao Kamalakara, Acyr Locatelli, Bharat Venkitesh +3

Training deep neural networks in low rank, i.e. with factorised layers, is of particular interest to the community: it offers efficiency over unfactorised training in terms of both…

cs.LG2022★ 19 cited

Prioritized Training on Points that are Learnable, Worth Learning, and Not Yet Learnt

Sören Mindermann, Jan Brauner, Muhammed Razzak +8

Training on web-scale data can take months. But most computation and time is wasted on redundant and noisy points that are already learnt or not learnable. To accelerate training,…

cs.LG2022★ 125 cited

Tranception: protein fitness prediction with autoregressive transformers and inference-time retrieval

Pascal Notin, Mafalda Dias, Jonathan Frazer +4

The ability to accurately model the fitness landscape of protein sequences is critical to a wide range of applications, from quantifying the effects of human variants on disease li…

cs.LG2021

Prioritized training on points that are learnable, worth learning, and not yet learned (workshop version)

Sören Mindermann, Muhammed Razzak, Winnie Xu +7

We introduce Goldilocks Selection, a technique for faster model training which selects a sequence of training points that are "just right". We propose an information-theoretic acqu…

cs.LG2021★ 44 cited

Self-Attention Between Datapoints: Going Beyond Individual Input-Output Pairs in Deep Learning

Jannik Kossen, Neil Band, Clare Lyle +3

We challenge a common assumption underlying most supervised deep learning: that a model makes a prediction depending only on its parameters and the features of a single input. To t…

cs.LG2021★ 9 cited

Robustness to Pruning Predicts Generalization in Deep Neural Networks

Lorenz Kuhn, Clare Lyle, Aidan N. Gomez +2

Existing generalization measures that aim to capture a model's simplicity based on parameter counts or norms fail to explain generalization in overparameterized deep neural network…

cs.LG2020★ 5 cited

Interlocking Backpropagation: Improving depthwise model-parallelism

Aidan N. Gomez, Oscar Key, Kuba Perlin +4

The number of parameters in state of the art neural networks has drastically increased in recent years. This surge of interest in large scale neural networks has motivated the deve…

cs.LG2020★ 1 cited

Improving compute efficacy frontiers with SliceOut

Pascal Notin, Aidan N. Gomez, Joanna Yoo +1

Pushing forward the compute efficacy frontier in deep learning is critical for tasks that require frequent model re-training or workloads that entail training a large number of mod…

cs.LG2019★ 30 cited

The Difficulty of Training Sparse Neural Networks

Utku Evci, Fabian Pedregosa, Aidan Gomez +1

We investigate the difficulties of training sparse neural networks and make new observations about optimization dynamics and the energy landscape within the sparse regime. Recent w…

cs.LG2019★ 74 cited

Learning Sparse Networks Using Targeted Dropout

Aidan N. Gomez, Ivan Zhang, Siddhartha Rao Kamalakara +4

Neural networks are easier to optimise when they have many more weights than are required for modelling the mapping from inputs to outputs. This suggests a two-stage learning proce…

cs.LG2018★ 14 cited

Tensor2Tensor for Neural Machine Translation

Ashish Vaswani, Samy Bengio, Eugene Brevdo +10

Tensor2Tensor is a library for deep learning models that is well-suited for neural machine translation and includes the reference implementation of the state-of-the-art Transformer…

cs.LG2018★ 24 cited

Unsupervised Cipher Cracking Using Discrete GANs

Aidan N. Gomez, Sicong Huang, Ivan Zhang +3

This work details CipherGAN, an architecture inspired by CycleGAN used for inferring the underlying cipher mapping given banks of unpaired ciphertext and plaintext. We demonstrate…

cs.LG2017★ 258 cited

One Model To Learn Them All

Lukasz Kaiser, Aidan N. Gomez, Noam Shazeer +4

Deep learning yields great results across many fields, from speech recognition, image classification, to translation. But for each problem, getting a deep model to work well involv…