← Search

Siddhant Jayakumar

6 accepted papers

2021

Powerpropagation: A sparsity inducing weight reparameterisation

NeurIPS 2021poster

The training of sparse neural networks is becoming an increasingly important tool for reducing the computational footprint of models at training and evaluation, as well enabling the effective scaling up of models. Whereas much work over the years has been dedicated to specialised pruning techniques,…

2020

Stabilizing Transformers for Reinforcement Learning

ICML 2020poster

Owing to their ability to both effectively integrate information over long time horizons and scale to massive amounts of data, self-attention architectures have recently shown breakthrough success in natural language processing (NLP). Harnessing the transformer’s ability to process long time horizon…

2020

Top-KAST: Top-K Always Sparse Training

NeurIPS 2020poster

Sparse neural networks are becoming increasingly important as the field seeks to improve the performance of existing models by scaling them up, while simultaneously trying to reduce power consumption and computational footprint. Unfortunately, most existing methods for inducing performant sparse mod…

Cited by 114SourcePDFScholar
2019

Distilling Policy Distillation

AISTATS 2019poster

The transfer of knowledge from one policy to another is an important tool in Deep Reinforcement Learning. This process, referred to as distillation, has been used to great success, for example, by enhancing the optimisation of agents, leading to stronger performance faster, on harder domains. Despit…

Cited by 179SourcePDFScholar
2018

Been There, Done That: Meta-Learning with Episodic Recall

ICML 2018oral

Meta-learning agents excel at rapidly learning new tasks from open-ended task distributions; yet, they forget what they learn about each task as soon as the next begins. When tasks reoccur {–} as they do in natural environments {–} meta-learning agents must explore again instead of immediately explo…

Cited by 114SourcePDFScholar
2018

Mix & Match Agent Curricula for Reinforcement Learning

ICML 2018oral

We introduce Mix and match (M&M) – a training framework designed to facilitate rapid and effective learning in RL agents that would be too slow or too challenging to train otherwise.The key innovation is a procedure that allows us to automatically form a curriculum over agents. Through such a curric…

Cited by 96SourcePDFScholar