Cross-Validation Is All You Need: A Statistical Approach To Label Noise Estimation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Chen, Jianan, Ramanathan, Vishwesh, Xu, Tony, Martel, Anne L.
Natura: Preprint
Pubblicazione: 2023
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866914876685811712
author Chen, Jianan
Ramanathan, Vishwesh
Xu, Tony
Martel, Anne L.
author_facet Chen, Jianan
Ramanathan, Vishwesh
Xu, Tony
Martel, Anne L.
contents Machine learning models experience deteriorated performance when trained in the presence of noisy labels. This is particularly problematic for medical tasks, such as survival prediction, which typically face high label noise complexity with few clear-cut solutions. Inspired by the large fluctuations across folds in the cross-validation performance of survival analyses, we design Monte-Carlo experiments to show that such fluctuation could be caused by label noise. We propose two novel and straightforward label noise detection algorithms that effectively identify noisy examples by pinpointing the samples that more frequently contribute to inferior cross-validation results. We first introduce Repeated Cross-Validation (ReCoV), a parameter-free label noise detection algorithm that is robust to model choice. We further develop fastReCoV, a less robust but more tractable and efficient variant of ReCoV suitable for deep learning applications. Through extensive experiments, we show that ReCoV and fastReCoV achieve state-of-the-art label noise detection performance in a wide range of modalities, models and tasks, including survival analysis, which has yet to be addressed in the literature. Our code and data are publicly available at https://github.com/GJiananChen/ReCoV.
format Preprint
id arxiv_https___arxiv_org_abs_2306_13990
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Cross-Validation Is All You Need: A Statistical Approach To Label Noise Estimation
Chen, Jianan
Ramanathan, Vishwesh
Xu, Tony
Martel, Anne L.
Machine Learning
Computer Vision and Pattern Recognition
Machine learning models experience deteriorated performance when trained in the presence of noisy labels. This is particularly problematic for medical tasks, such as survival prediction, which typically face high label noise complexity with few clear-cut solutions. Inspired by the large fluctuations across folds in the cross-validation performance of survival analyses, we design Monte-Carlo experiments to show that such fluctuation could be caused by label noise. We propose two novel and straightforward label noise detection algorithms that effectively identify noisy examples by pinpointing the samples that more frequently contribute to inferior cross-validation results. We first introduce Repeated Cross-Validation (ReCoV), a parameter-free label noise detection algorithm that is robust to model choice. We further develop fastReCoV, a less robust but more tractable and efficient variant of ReCoV suitable for deep learning applications. Through extensive experiments, we show that ReCoV and fastReCoV achieve state-of-the-art label noise detection performance in a wide range of modalities, models and tasks, including survival analysis, which has yet to be addressed in the literature. Our code and data are publicly available at https://github.com/GJiananChen/ReCoV.
title Cross-Validation Is All You Need: A Statistical Approach To Label Noise Estimation
topic Machine Learning
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2306.13990