← Search

Nitarshan Rajkumar

4 accepted papers

2023

Metadata Archaeology: Unearthing Data Subsets by Leveraging Training Dynamics

ICLR 2023top-25%

Modern machine learning research relies on relatively few carefully curated datasets. Even in these datasets, and typically in `untidy' or raw data, practitioners are faced with significant issues of data quality and diversity which can be prohibitively labor intensive to address. Existing methods f…

Cited by 29SourcePDFScholar
2022

Myriad: a real-world testbed to bridge trajectory optimization and deep learning

NeurIPS 2022accept

We present Myriad, a testbed written in JAX which enables machine learning researchers to benchmark imitation learning and reinforcement learning algorithms against trajectory optimization-based methods in real-world environments. Myriad contains 17 optimal control problems presented in continuous t…

2021

Pretraining Representations for Data-Efficient Reinforcement Learning

NeurIPS 2021poster

Data efficiency is a key challenge for deep reinforcement learning. We address this problem by using unlabeled data to pretrain an encoder which is then finetuned on a small amount of task-specific data. To encourage learning representations which capture diverse aspects of the underlying MDP, we em…

2020

In search of robust measures of generalization

NeurIPS 2020poster

One of the principal scientific challenges in deep learning is explaining generalization, i.e., why the particular way the community now trains networks to achieve small training error also leads to small error on held-out data from the same population. It is widely appreciated that some worst-case…