Author
Listed:
- Qian Zhang
(School of Information Technology, Jiangsu Open University, Nanjing 210036, China
These authors contributed equally to this work.)
- De Yu
(School of Information Technology, Jiangsu Open University, Nanjing 210036, China
These authors contributed equally to this work.)
- Xinru Zhou
(School of Information Technology, Jiangsu Open University, Nanjing 210036, China)
- Hanmeng Gong
(School of Information Technology, Jiangsu Open University, Nanjing 210036, China)
- Zheng Li
(School of Information Technology, Jiangsu Open University, Nanjing 210036, China)
- Yiming Liu
(School of Information Technology, Jiangsu Open University, Nanjing 210036, China)
- Ruirui Shao
(School of Information Technology, Jiangsu Open University, Nanjing 210036, China)
Abstract
Deep neural networks (DNNs) have achieved breakthrough progress in various fields, largely owing to the support of large-scale datasets with manually annotated labels. However, obtaining such datasets is costly and time-consuming, making high-quality annotation a challenging task. In this work, we propose an improved noisy sample selection method, termed “sample selection framework”, based on a mixup loss and recalibration strategy (SMR). This framework enhances the robustness and generalization abilities of models. First, we introduce a robust mixup loss function to pre-train two models with identical structures separately. This approach avoids additional hyperparameter adjustments and reduces the need for prior knowledge of noise types. Additionally, we use a Gaussian Mixture Model (GMM) to divide the entire training set into labeled and unlabeled subsets, followed by robust training using semi-supervised learning (SSL) techniques. Furthermore, we propose a recalibration strategy based on cross-entropy (CE) loss to prevent the models from converging to local optima during the SSL process, thus further improving performance. Ablation experiments on CIFAR-10 with 50% symmetric noise and 40% asymmetric noise demonstrate that the two modules introduced in this paper improve the accuracy of the baseline (i.e., DivideMix) by 1.5% and 0.5%, respectively. Moreover, the experimental results on multiple benchmark datasets demonstrate that our proposed method effectively mitigates the impact of noisy labels and significantly enhances the performance of DNNs on noisy datasets. For instance, on the WebVision dataset, our method improves the top-1 accuracy by 0.7% and 2.4% compared to the baseline method.
Suggested Citation
Qian Zhang & De Yu & Xinru Zhou & Hanmeng Gong & Zheng Li & Yiming Liu & Ruirui Shao, 2024.
"A Noisy Sample Selection Framework Based on a Mixup Loss and Recalibration Strategy,"
Mathematics, MDPI, vol. 12(15), pages 1-22, July.
Handle:
RePEc:gam:jmathe:v:12:y:2024:i:15:p:2389-:d:1447195
Download full text from publisher
Corrections
All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:gam:jmathe:v:12:y:2024:i:15:p:2389-:d:1447195. See general information about how to correct material in RePEc.
If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.
We have no bibliographic references for this item. You can help adding them by using this form .
If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.
For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: MDPI Indexing Manager (email available below). General contact details of provider: https://www.mdpi.com .
Please note that corrections may take a couple of weeks to filter through
the various RePEc services.