HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Weight-guided class complementing for long-tailed image recognition

{Bingfeng Zhang Hui Li Siyue Yu Jimin Xiao Xinqiao Zhao}

Abstract

Real-world data are often long-tailed distributed and have plenty classes. This characteristic leads to a significant performance drop for various models. One reason behind that is the gradient shift caused by unsampled classes in each training iteration. In this paper, we propose a Weight-Guided Class Complementing framework to address this issue. Specifically, this framework first complements the unsampled classes in each training iteration by using a dynamic updated data slot. Then, considering the over-fitting issue caused by class complementing, we utilize the classifier weights as learned knowledge and encourage the model to discover more class specific characteristics. Finally, we design a weight refining scheme to deal with the long-tailed bias existing in classifier weights. Experimental results show that our framework can be implemented upon different existing approaches effectively, achieving consistent improvements on various benchmarks with new state-of-the-art performances.

Benchmarks

BenchmarkMethodologyMetrics
long-tail-learning-on-cifar-10-lt-r-100NCL* + WGCC (ensemble)
Error Rate: 15.4
long-tail-learning-on-cifar-100-lt-r-100LDAM-DRW + WGCC
Error Rate: 56.4
long-tail-learning-on-cifar-100-lt-r-100NCL* + WGCC (ensemble)
Error Rate: 44.9

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
Weight-guided class complementing for long-tailed image recognition | Papers | HyperAI