Abstract:Modern deep neural networks can easily overfit to biased training data containing corrupted labels or class imbalance. Sample re-weighting methods are popularly used to alleviate this data bias issue. Most current methods, however, require to manually pre-specify the weighting schemes relying on the characteristics of the investigated problem and training data. This makes them fairly hard to be generally applied in practical scenarios, due to their significant complexities and inter-class variations of data bias. To address this issue, we propose a meta-model capable of adaptively learning an explicit weighting scheme directly from data. Specifically, by seeing each training class as a separate learning task, our method aims to extract an explicit weighting function with sample loss and task/class feature as input, and sample weight as output, expecting to impose adaptively varying weighting schemes to different sample classes based on their own intrinsic bias characteristics. Extensive experiments substantiate the capability of our method on achieving proper weighting schemes in various data bias cases, like class imbalance, feature-independent and dependent label noises, and more complicated bias scenarios beyond conventional cases. Besides, the task-transferability of the learned weighting scheme is also substantiated, by readily deploying the weighting function learned on relatively smaller-scale CIFAR-10 dataset on much larger-scale full WebVision dataset. The general availability of our method for multiple robust deep learning issues, including partial-label learning, semi-supervised learning and selective classification, has also been validated. Code for reproducing our experiments is available at https://github.com/xjtushujun/CMW-Net.

Adaptive Class-Balanced Loss Based on Re-Weighting

Revisiting the Effective Number Theory for Imbalanced Learning

Value-Aware Resampling and Loss for Imbalanced Classification

Adaptive Weight Optimization for Classification of Imbalanced Data.

Inverse Weight-Balancing for Deep Long-Tailed Learning

Improving the Accuracy of Learning Example Weights for Imbalance Classification

Towards Deeper Insights into Deep Learning from Imbalanced Data.

Margin-Aware Adaptive-Weighted-Loss for deep learning based imbalanced data classification

Learning to Reweight Samples with Offline Loss Sequence

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

CMW-Net: Learning a Class-Aware Sample Weighting Mapping for Robust Deep Learning

Neural Collapse Inspired Attraction-Repulsion-Balanced Loss for Imbalanced Learning

Meta-Weight-Net: Learning an Explicit Mapping For Sample Weighting.

Fairness with Adaptive Weights.

AutoBalance: Optimized Loss Functions for Imbalanced Data

Adaptive Sampling With Optimal Cost For Class-Imbalance Learning

Learning with Imbalanced Noisy Data by Preventing Bias in Sample Selection

Sampling + Reweighting: Boosting the Performance of AdaBoost on Imbalanced Datasets

Group & Reweight: A Novel Cost-Sensitive Approach to Mitigating Class Imbalance in Network Traffic Classification

Meta-Weight-Net: Learning an Explicit Mapping for Sample Weighting

Class-Wise Difficulty-Balanced Loss for Solving Class-Imbalance