arrow
返回

Learning sample representativeness for class-imbalanced multi-label classification

delete2024-02-28
delete16
PRE
AI
张
张宇 (Yu Zhang) *
S
Sichen Cao
米思娅 封面图
米思娅 (Siya Mi)
Y
Yali Bian
DOI:10.1007/s10044-024-01209-8delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Class imbalance is a common problem that often occurs in multi-label image classification. In multi-label datasets, the co-occurrence of labels presents a unique set of difficulties, making it hard for traditional methods to produce satisfactory results, particularly on tail classes. Based on previous research and our investigation, we have found that the number of labels presents in a given sample can influence classification results. Nevertheless, it is worth noting that certain samples within the tail classes exhibit resistance to this influence, which is a critical aspect in the context of class-imbalanced multi-label classification. In this paper, we term these samples as representative samples. Highlighting representative samples during training can effectively address the above issues. Specifically, we propose a new method to learn sample representativeness, which is named Representativeness-Emphasizing Loss (REL). First, we use a new re-weighting form to rebalance the weights based on sample representativeness. Then, a modified focal loss dynamically assigns tailored parameters for each class in each sample to further emphasize the sample representativeness. Extensive experiments on two class-imbalanced datasets show that models trained with this new loss function achieve comparable performance to existing methods.
Keyword:
Multi-label classification
Class imbalance
Re-weighting representative samples

期刊

Pattern Analysis and Applications 封面图
Pattern Analysis and Applications
IF:
2
论文数:
1.9K
被引数:
1.9K

机构

I
intel usa
学者数:
736
论文数: 548
被引数: 1
S
southeast university - china
学者数:
5.3W
论文数: 4.9W
被引数: 57
引用论文

引用论文

err分享
err收藏
Cancer classification with data augmentation based on generative adversarial networks
err2021-09-09
err32
PREAI
errWei, Kaimin; Li, Tianqi; Huang, Feiran; Chen, Jinpeng; He, Zefan
err分享
err收藏
Quantum convolutional neural network for image classification用于图像分类的量子卷积神经网络
err2022-09-24
err55
PREAI
errChen, Guoming; Chen, Qiang; Long, Shun; Zhu, Weiheng; Yuan, Zeduo; Wu, Yilin
err分享
err收藏
Introduction
err2005-08-25
err0
PREAI
errMary Coleman; Catalina Betancur
err分享
err收藏
学者 查看更多内容