Residential Collegefalse
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; Wang, Tianfu1,2,3
2021-09-15
Source PublicationIEEE Transactions on Neural Networks and Learning Systems
ISSN2162-237X
Volume34Issue: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.

KeywordClass-imbalanced Deep Learning Dynamic Changes Hyperparameters Tuning Loss Function Parameter-free
DOI10.1109/TNNLS.2021.3110885
URLView the original
Indexed BySCIE
Language英語English
WOS Research AreaComputer Science ; Engineering
WOS SubjectComputer Science, Artificial Intelligence ; Computer Science, Hardware & Architecture ; Computer Science, Theory & Methods ; Engineering, Electrical & Electronic
WOS IDWOS:000732928900001
PublisherIEEE-INST ELECTRICAL ELECTRONICS ENGINEERS INC, 445 HOES LANE, PISCATAWAY, NJ 08855-4141
Scopus ID2-s2.0-85115152238
Fulltext Access
Citation statistics
Document TypeJournal article
CollectionDEPARTMENT OF COMPUTER AND INFORMATION SCIENCE
Corresponding AuthorVong, Chi Man
Affiliation1.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 AffilicationUniversity 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.
Related Services
Recommend this item
Bookmark
Usage statistics
Export to Endnote
Google Scholar
Similar articles in Google Scholar
[Du, Jie]'s Articles
[Zhou, Yanhong]'s Articles
[Liu, Peng]'s Articles
Baidu academic
Similar articles in Baidu academic
[Du, Jie]'s Articles
[Zhou, Yanhong]'s Articles
[Liu, Peng]'s Articles
Bing Scholar
Similar articles in Bing Scholar
[Du, Jie]'s Articles
[Zhou, Yanhong]'s Articles
[Liu, Peng]'s Articles
Terms of Use
No data!
Social Bookmark/Share
All comments (0)
No comment.
 

Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.