Skip to yearly menu bar Skip to main content


Poster

Data Parameters: A New Family of Parameters for Learning a Differentiable Curriculum

Shreyas Saxena · Oncel Tuzel · Dennis DeCoste

East Exhibition Hall B, C #127

Keywords: [ Supervised Deep Networks ] [ Deep Learning ] [ Algorithms -> Classification; Applications ] [ Computer Vision ]


Abstract:

Recent works have shown that learning from easier instances first can help deep neural networks (DNNs) generalize better. However, knowing which data to present during different stages of training is a challenging problem. In this work, we address this problem by introducing data parameters. More specifically, we equip each sample and class in a dataset with a learnable parameter (data parameters), which governs their importance in the learning process. During training, at each iteration, as we update the model parameters, we also update the data parameters. These updates are done by gradient descent and do not require hand-crafted rules or design. When applied to image classification task on CIFAR10, CIFAR100,WebVision and ImageNet datasets, and object detection task on KITTI dataset, learning a dynamic curriculum via data parameters leads to consistent gains, without any increase in model complexity or training time. When applied to a noisy dataset, the proposed method learns to learn from clean images and improves over the state-of-the-art methods by 14%. To the best of our knowledge, our work is the first curriculum learning method to show gains on large scale image classification and detection tasks. Code is available at: https://github.com/apple/ml-data-parameters

Live content is unavailable. Log in and register to view live content