Dirichlet-based Per-Sample Weighting by Transition Matrix for Noisy Label Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Bae, HeeSun, Shin, Seungjae, Na, Byeonghu, Moon, Il-Chul
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910354028625920
author Bae, HeeSun
Shin, Seungjae
Na, Byeonghu
Moon, Il-Chul
author_facet Bae, HeeSun
Shin, Seungjae
Na, Byeonghu
Moon, Il-Chul
contents For learning with noisy labels, the transition matrix, which explicitly models the relation between noisy label distribution and clean label distribution, has been utilized to achieve the statistical consistency of either the classifier or the risk. Previous researches have focused more on how to estimate this transition matrix well, rather than how to utilize it. We propose good utilization of the transition matrix is crucial and suggest a new utilization method based on resampling, coined RENT. Specifically, we first demonstrate current utilizations can have potential limitations for implementation. As an extension to Reweighting, we suggest the Dirichlet distribution-based per-sample Weight Sampling (DWS) framework, and compare reweighting and resampling under DWS framework. With the analyses from DWS, we propose RENT, a REsampling method with Noise Transition matrix. Empirically, RENT consistently outperforms existing transition matrix utilization methods, which includes reweighting, on various benchmark datasets. Our code is available at \url{https://github.com/BaeHeeSun/RENT}.
format Preprint
id arxiv_https___arxiv_org_abs_2403_02690
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Dirichlet-based Per-Sample Weighting by Transition Matrix for Noisy Label Learning
Bae, HeeSun
Shin, Seungjae
Na, Byeonghu
Moon, Il-Chul
Machine Learning
Computer Vision and Pattern Recognition
For learning with noisy labels, the transition matrix, which explicitly models the relation between noisy label distribution and clean label distribution, has been utilized to achieve the statistical consistency of either the classifier or the risk. Previous researches have focused more on how to estimate this transition matrix well, rather than how to utilize it. We propose good utilization of the transition matrix is crucial and suggest a new utilization method based on resampling, coined RENT. Specifically, we first demonstrate current utilizations can have potential limitations for implementation. As an extension to Reweighting, we suggest the Dirichlet distribution-based per-sample Weight Sampling (DWS) framework, and compare reweighting and resampling under DWS framework. With the analyses from DWS, we propose RENT, a REsampling method with Noise Transition matrix. Empirically, RENT consistently outperforms existing transition matrix utilization methods, which includes reweighting, on various benchmark datasets. Our code is available at \url{https://github.com/BaeHeeSun/RENT}.
title Dirichlet-based Per-Sample Weighting by Transition Matrix for Noisy Label Learning
topic Machine Learning
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2403.02690