Youjiang Xu Linchao Zhu Lu Jiang Yi Yang

Abstract
It has been shown that deep neural networks are prone to overfitting on biased training data. Towards addressing this issue, meta-learning employs a meta model for correcting the training bias. Despite the promising performances, super slow training is currently the bottleneck in the meta learning approaches. In this paper, we introduce a novel Faster Meta Update Strategy (FaMUS) to replace the most expensive step in the meta gradient computation with a faster layer-wise approximation. We empirically find that FaMUS yields not only a reasonably accurate but also a low-variance approximation of the meta gradient. We conduct extensive experiments to verify the proposed method on two tasks. We show our method is able to save two-thirds of the training time while still maintaining the comparable or achieving even better generalization performance. In particular, our method achieves the state-of-the-art performance on both synthetic and realistic noisy labels, and obtains promising performance on long-tailed recognition on standard benchmarks.
Code Repositories
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| image-classification-on-cifar-10-40-symmetric | FaMUS | Percentage correct: 95.37 |
| image-classification-on-cifar-10-40-symmetric | MentorMix | Percentage correct: 94.2 |
| image-classification-on-cifar-10-60-symmetric | FaMUS | Percentage correct: 26.42 |
| image-classification-on-cifar-10-60-symmetric | MentorMix | Percentage correct: 91.3 |
| image-classification-on-cifar-100-40 | FaMUS | Percentage correct: 75.91 |
| image-classification-on-cifar-100-40 | MentorMix | Percentage correct: 71.3 |
| image-classification-on-cifar-100-60 | MentorMix | Percentage correct: 64.6 |
| image-classification-on-mini-webvision-1-0 | FaMUS | ImageNet Top-1 Accuracy: 77 ImageNet Top-5 Accuracy: 92.76 Top-1 Accuracy: 79.4 Top-5 Accuracy: 92.80 |
| image-classification-on-red-miniimagenet-20 | FaMUS | Accuracy: 51.42 |
| image-classification-on-red-miniimagenet-40 | FaMUS | Accuracy: 48.06 |
| image-classification-on-red-miniimagenet-60 | FaMUS | Accuracy: 45.1 |
| image-classification-on-red-miniimagenet-80 | FaMUS | Accuracy: 35.5 |
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.