SV-Learner: Support-Vector Contrastive Learning for Robust Learning With Noisy Labels

被引:1
作者
Liang, Xin [1 ]
Ji, Yanli [1 ,2 ]
Zheng, Wei-Shi [3 ]
Zuo, Wangmeng [4 ]
Zhu, Xiaofeng [1 ,5 ]
机构
[1] UESTC, Shenzhen Inst Adv Study, Shenzhen 518110, Peoples R China
[2] UESTC, Sch Comp Sci & Engn, Chengdu 610056, Peoples R China
[3] Sun Yat Sen Univ, Sch Comp Sci & Engn, Guangzhou 510275, Peoples R China
[4] Harbin Inst Technol, Sch Comp Sci & Technol, Harbin 150001, Peoples R China
[5] UESTC, Sch Comp Sci & Engn, Chengdu 611731, Peoples R China
关键词
Noise measurement; Self-supervised learning; Vectors; Noise; Reliability; Training; Support vector machines; Learning with noisy labels; semi-supervised learning; support - vector contrastive learning (SVCL);
D O I
10.1109/TKDE.2024.3386829
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Noisy-label data inevitably gives rise to confusion in various perception applications. In this work, we revisit the theory of support vector machines (SVMs) which mines support vectors to build the maximum-margin hyperplane for robust classification, and propose a robust-to-noise deep learning framework, SV-Learner, including the Support Vector Contrastive Learning (SVCL) and Support Vector-based Noise Screening (SVNS). The SV-Learner mines support vectors to solve the learning problem with noisy labels (LNL) reliably. SVCL adopts support vectors as positive and negative samples, driving robust contrastive learning to enlarge the feature distribution margin for learning convergent feature distributions. SVNS uses support vectors with valid labels to assist in screening noisy ones from confusable samples for reliable clean-noisy sample screening. Finally, Semi-Supervised classification is performed to realize the recognition of noisy samples. Extensive experiments are evaluated on CIFAR-10, CIFAR-100, Clothing1M, and Webvision datasets, and results demonstrate the effectiveness of our proposed approach.
引用
收藏
页码:5409 / 5422
页数:14
相关论文
共 66 条
  • [1] Arazo E, 2019, PR MACH LEARN RES, V97
  • [2] Arpit D, 2017, PR MACH LEARN RES, V70
  • [3] Bekker AJ, 2016, INT CONF ACOUST SPEE, P2682, DOI 10.1109/ICASSP.2016.7472164
  • [4] Caron M., 2020, PROC INT C NEURAL IN, P1
  • [5] Chen Pengfei, 2019, P MACHINE LEARNING R, V97
  • [6] Chen T., 2020, Advances in Neural Information Processing Systems, V33, P22243
  • [7] Chen T., 2020, PMLR, P1597, DOI DOI 10.48550/ARXIV.2002.05709
  • [8] Sample Prior Guided Robust Model Learning to Suppress Noisy Labels
    Chen, Wenkai
    Zhu, Chuang
    Li, Mengting
    [J]. MACHINE LEARNING AND KNOWLEDGE DISCOVERY IN DATABASES: RESEARCH TRACK, ECML PKDD 2023, PT II, 2023, 14170 : 3 - 19
  • [9] Exploring Simple Siamese Representation Learning
    Chen, Xinlei
    He, Kaiming
    [J]. 2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, : 15745 - 15753
  • [10] Cordeiro F. R., 2021, PROC BRIT MACH VIS C, P1