SuperDisco: Super-Class Discovery Improves Visual Recognition for the Long-Tail

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Du, Yingjun, Shen, Jiayi, Zhen, Xiantong, Snoek, Cees G. M.
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916794395000832
author Du, Yingjun
Shen, Jiayi
Zhen, Xiantong
Snoek, Cees G. M.
author_facet Du, Yingjun
Shen, Jiayi
Zhen, Xiantong
Snoek, Cees G. M.
contents Modern image classifiers perform well on populated classes, while degrading considerably on tail classes with only a few instances. Humans, by contrast, effortlessly handle the long-tailed recognition challenge, since they can learn the tail representation based on different levels of semantic abstraction, making the learned tail features more discriminative. This phenomenon motivated us to propose SuperDisco, an algorithm that discovers super-class representations for long-tailed recognition using a graph model. We learn to construct the super-class graph to guide the representation learning to deal with long-tailed distributions. Through message passing on the super-class graph, image representations are rectified and refined by attending to the most relevant entities based on the semantic similarity among their super-classes. Moreover, we propose to meta-learn the super-class graph under the supervision of a prototype graph constructed from a small amount of imbalanced data. By doing so, we obtain a more robust super-class graph that further improves the long-tailed recognition performance. The consistent state-of-the-art experiments on the long-tailed CIFAR-100, ImageNet, Places and iNaturalist demonstrate the benefit of the discovered super-class graph for dealing with long-tailed distributions.
format Preprint
id arxiv_https___arxiv_org_abs_2304_00101
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle SuperDisco: Super-Class Discovery Improves Visual Recognition for the Long-Tail
Du, Yingjun
Shen, Jiayi
Zhen, Xiantong
Snoek, Cees G. M.
Computer Vision and Pattern Recognition
Modern image classifiers perform well on populated classes, while degrading considerably on tail classes with only a few instances. Humans, by contrast, effortlessly handle the long-tailed recognition challenge, since they can learn the tail representation based on different levels of semantic abstraction, making the learned tail features more discriminative. This phenomenon motivated us to propose SuperDisco, an algorithm that discovers super-class representations for long-tailed recognition using a graph model. We learn to construct the super-class graph to guide the representation learning to deal with long-tailed distributions. Through message passing on the super-class graph, image representations are rectified and refined by attending to the most relevant entities based on the semantic similarity among their super-classes. Moreover, we propose to meta-learn the super-class graph under the supervision of a prototype graph constructed from a small amount of imbalanced data. By doing so, we obtain a more robust super-class graph that further improves the long-tailed recognition performance. The consistent state-of-the-art experiments on the long-tailed CIFAR-100, ImageNet, Places and iNaturalist demonstrate the benefit of the discovered super-class graph for dealing with long-tailed distributions.
title SuperDisco: Super-Class Discovery Improves Visual Recognition for the Long-Tail
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2304.00101