HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Dynamic Loss For Robust Learning

Shenwang Jiang Jianan Li Jizhou Zhang Ying Wang Tingfa Xu

Dynamic Loss For Robust Learning

Abstract

Label noise and class imbalance commonly coexist in real-world data. Previous works for robust learning, however, usually address either one type of the data biases and underperform when facing them both. To mitigate this gap, this work presents a novel meta-learning based dynamic loss that automatically adjusts the objective functions with the training process to robustly learn a classifier from long-tailed noisy data. Concretely, our dynamic loss comprises a label corrector and a margin generator, which respectively correct noisy labels and generate additive per-class classification margins by perceiving the underlying data distribution as well as the learning state of the classifier. Equipped with a new hierarchical sampling strategy that enriches a small amount of unbiased metadata with diverse and hard samples, the two components in the dynamic loss are optimized jointly through meta-learning and cultivate the classifier to well adapt to clean and balanced test data. Extensive experiments show our method achieves state-of-the-art accuracy on multiple real-world and synthetic datasets with various types of data biases, including CIFAR-10/100, Animal-10N, ImageNet-LT, and Webvision. Code will soon be publicly available.

Code Repositories

Benchmarks

BenchmarkMethodologyMetrics
image-classification-on-mini-webvision-1-0Dynamic Loss (Inception-ResNet-v2)
ImageNet Top-1 Accuracy: 74.76
ImageNet Top-5 Accuracy: 93.08
Top-1 Accuracy: 80.12
Top-5 Accuracy: 93.64
learning-with-noisy-labels-on-animalDynamic Loss
Accuracy: 86.5
ImageNet Pretrained: NO
Network: Vgg19-BN

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Dynamic Loss For Robust Learning | Papers | HyperAI