arrow
返回

Parameter-Free Loss for Class-Imbalanced Deep Learning in Image Classification

delete2023-06-01
delete25
PRE
AI
J
Jie Du
周
周艳红 (Yanhong Zhou)
P
Peng Liu
C
Chi‐Man Vong *
汪天富 封面图
汪天富 (Tianfu Wang)
DOI:10.1109/TNNLS.2021.3110885delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Current state-of-the-art class-imbalanced loss functions for deep models require exhaustive tuning on hyperparameters for high model performance, resulting in low training efficiency and impracticality for nonexpert users. To tackle this issue, a parameter-free loss (PF-loss) function is proposed, which works for both binary and multiclass-imbalanced deep learning for image classification tasks. PF-loss provides three advantages: 1) training time is significantly reduced due to NO tuning on hyperparameter(s); 2) it dynamically pays more attention on minority classes (rather than outliers compared to the existing loss functions) with NO hyperparameters in the loss function; and 3) higher accuracy can be achieved since it adapts to the changes of data distribution in each mini-batch instead of the fixed hyperparameters in the existing methods during training, especially when the data are highly skewed. Experimental results on some classical image datasets with different imbalance ratios (IR, up to 200) show that PF-loss reduces the training time down to 1/148 of that spent by compared state-of-the-art losses and simultaneously achieves comparable or even higher accuracy in terms of both G-mean and area under receiver operating characteristic (ROC) curve (AUC) metrics, especially when the data are highly skewed.
Keyword:
Training
Tuning
Deep learning
Data models
Task analysis
Learning systems
Focusing
Class-imbalanced deep learning
dynamic changes
hyperparameters tuning
loss function
parameter-free

期刊

IEEE Transactions on Neural Networks and Learning Systems 封面图
IEEE Transactions on Neural Networks and Learning Systems
IF:
8.9
论文数:
7.6K
被引数:
7.2W

机构

S
shenzhen university
学者数:
4.6W
论文数: 3.4W
被引数: 72
U
University of Macau
学者数:
1.1W
论文数: 1.3W
被引数: 2.0W
引用论文

引用论文

err分享
err收藏
err分享
err收藏
err分享
err收藏
学者 查看更多内容