Abstract:Balancing training on long-tail data distributions remains a long-standing challenge in deep learning. While methods such as re-weighting and re-sampling help alleviate the imbalance issue, limited sample diversity continues to hinder models from learning robust and generalizable feature representations, particularly for tail classes. In contrast to existing methods, we offer a novel perspective on long-tail learning, inspired by an observation: datasets with finer granularity tend to be less affected by data imbalance. In this paper, we investigate this phenomenon through both quantitative and qualitative studies, showing that increased granularity enhances the generalization of learned features in tail categories. Motivated by these findings, we propose a method to increase dataset granularity through category extrapolation. Specifically, we introduce open-set auxiliary classes that are visually similar to existing ones, aiming to enhance representation learning for both head and tail classes. This forms the core contribution and insight of our approach. To automate the curation of auxiliary data, we leverage large language models (LLMs) as knowledge bases to search for auxiliary categories and retrieve relevant images through web crawling. To prevent the overwhelming presence of auxiliary classes from disrupting training, we introduce a neighbor-silencing loss that encourages the model to focus on class discrimination within the target dataset. During inference, the classifier weights for auxiliary categories are masked out, leaving only the target class weights for use. Extensive experiments and ablation studies on three standard long-tail benchmarks demonstrate the effectiveness of our approach, notably outperforming strong baseline methods that use the same amount of data. The code will be made publicly available.

To Balance or Not to Balance: A Simple-yet-Effective Approach for Learning with Long-Tailed Distributions

Exploring the auxiliary learning for long-tailed visual recognition

Long-Tailed Learning as Multi-Objective Optimization

Learning from Neighbors: Category Extrapolation for Long-Tail Learning

Feature Re-Balancing for Long-Tailed Visual Recognition.

Reviving Undersampling for Long-Tailed Learning

Towards Deeper Insights into Deep Learning from Imbalanced Data.

Open-Sampling: Exploring Out-of-Distribution Data for Re-balancing Long-tailed Datasets

Learning from Long-Tailed Noisy Data with Sample Selection and Balanced Loss.

Two-Stage Fine-Tuning: A Novel Strategy for Learning Class-Imbalanced Data

Learning with Imbalanced Noisy Data by Preventing Bias in Sample Selection

Trainable Undersampling for Class-Imbalance Learning.

Mixed Mutual Transfer for Long-Tailed Image Classification

Adaptive Sampling With Optimal Cost For Class-Imbalance Learning

How Re-sampling Helps for Long-Tail Learning?

A Normal Distribution-Based Over-Sampling Approach to Imbalanced Data Classification

Decoupling Representation and Classifier for Long-Tailed Recognition

Feature-Balanced Loss for Long-Tailed Visual Recognition

Rethinking Class-Balanced Methods for Long-Tailed Visual Recognition from a Domain Adaptation Perspective

Alleviating the Effect of Data Imbalance on Adversarial Training