《计算机应用》唯一官方网站

• •    下一篇

基于浅层网络预测的元标签校正方法

黄雨鑫1,黄贻望2,黄辉3   

  1. 1. 福建理工大学计算机科学与数学学院
    2. 铜仁学院
    3. 福建农业职业技术学院现代农业技术学院
  • 收稿日期:2023-11-22 修回日期:2024-02-22 接受日期:2024-03-08 发布日期:2024-03-12 出版日期:2024-03-12
  • 通讯作者: 黄贻望
  • 基金资助:
    国家自然科学基金资助项目(62066040);铜仁市科技局资助项目(铜仁市科研[2022]5 号)。

Meta-Label Correction Method Based on Shallow Network Predictions

  • Received:2023-11-22 Revised:2024-02-22 Accepted:2024-03-08 Online:2024-03-12 Published:2024-03-12
  • Supported by:
    This work is partially supported by National Natural Science Foundation of China (62066040), Tongren Science and Technology Bureau (Tongren City Scientific Research [2022]5).

摘要: 针对深度神经模型对含有噪声标签的图像数据具有记忆行为的问题,提出了一种基于浅层网络预测的元标签校正技术。该方法采用弱监督训练方式,通过设置标签重加权网络对噪声数据进行加权操作,利用元学习方法使模型动态的学习噪声数据,并将模型中深层与浅层网络的预测输出作为新的标签训练模型,同时利用知识蒸馏算法使深层网络指导浅层网络训练,从而有效缓解模型的记忆行为并提升了模型泛化性。在CIFAR10、CIFAR100、Clothing1M数据集上进行实验,对比于其他噪声标签学习方法,该方法在准确率方面有明显的提升效果。

关键词: 噪声标签, 元学习, 标签校正, 标签重加权, 知识蒸馏

Abstract: Abstract: A meta-label correction technique based on shallow network predictions is proposed to address overfitting and memorization behaviors exhibited by deep neural models when dealing with noisy labeled data. This approach employs weakly supervised training, where a label re-weighting network is introduced to re-weight noisy data, and utilizes meta-learning methods to enable the model to dynamically adapt to noisy data. The model is trained using the predictions from both deep and shallow networks as new labels, and knowledge distillation techniques are applied to have the deep network guide the training of the shallow network, so as to effectively alleviate the model's memorization behavior and enhance the model's generalizability. In experiments conducted on the CIFAR10, CIFAR100, and Clothing1M datasets, compared to other noise label learning methods, this approach demonstrates a significant improvement in accuracy.

Key words: noise label, meta learning, label correction, label reweighting, knowledge distillation

中图分类号: