Residential College | false |
Status | 已發表Published |
Parameter-Free Loss for Class-Imbalanced Deep Learning in Image Classification | |
Du, Jie1,2,3; Zhou, Yanhong1,2,3; Liu, Peng4; Vong, Chi Man4![]() ![]() | |
2021-09-15 | |
Source Publication | IEEE Transactions on Neural Networks and Learning Systems
![]() |
ISSN | 2162-237X |
Volume | 34Issue:6Pages:3234 - 3240 |
Abstract | 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 | Class-imbalanced Deep Learning Dynamic Changes Hyperparameters Tuning Loss Function Parameter-free |
DOI | 10.1109/TNNLS.2021.3110885 |
URL | View the original |
Indexed By | SCIE |
Language | 英語English |
WOS Research Area | Computer Science ; Engineering |
WOS Subject | Computer Science, Artificial Intelligence ; Computer Science, Hardware & Architecture ; Computer Science, Theory & Methods ; Engineering, Electrical & Electronic |
WOS ID | WOS:000732928900001 |
Publisher | IEEE-INST ELECTRICAL ELECTRONICS ENGINEERS INC, 445 HOES LANE, PISCATAWAY, NJ 08855-4141 |
Scopus ID | 2-s2.0-85115152238 |
Fulltext Access | |
Citation statistics | |
Document Type | Journal article |
Collection | DEPARTMENT OF COMPUTER AND INFORMATION SCIENCE |
Corresponding Author | Vong, Chi Man |
Affiliation | 1.Shenzhen Univ, Sch Biomed Engn, Hlth Sci Ctr, Shenzhen 518060, Peoples R China 2.Shenzhen Univ, Natl Reg Key Technol Engn Lab Med Ultrasound, Shenzhen 518060, Peoples R China 3.Shenzhen Univ, Marshall Lab Biomed Engn, Shenzhen 518060, Peoples R China 4.Univ Macau, Dept Comp & Informat Sci, Macau 999078, Peoples R China |
Corresponding Author Affilication | University of Macau |
Recommended Citation GB/T 7714 | Du, Jie,Zhou, Yanhong,Liu, Peng,et al. Parameter-Free Loss for Class-Imbalanced Deep Learning in Image Classification[J]. IEEE Transactions on Neural Networks and Learning Systems, 2021, 34(6), 3234 - 3240. |
APA | Du, Jie., Zhou, Yanhong., Liu, Peng., Vong, Chi Man., & Wang, Tianfu (2021). Parameter-Free Loss for Class-Imbalanced Deep Learning in Image Classification. IEEE Transactions on Neural Networks and Learning Systems, 34(6), 3234 - 3240. |
MLA | Du, Jie,et al."Parameter-Free Loss for Class-Imbalanced Deep Learning in Image Classification".IEEE Transactions on Neural Networks and Learning Systems 34.6(2021):3234 - 3240. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment