Self-Relaxed Joint Training: Sample Selection for Severity Estimation with Ordinal Noisy Labels

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Takezaki, Shumpei, Tanaka, Kiyohito, Uchida, Seiichi
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866912093114990592
author Takezaki, Shumpei
Tanaka, Kiyohito
Uchida, Seiichi
author_facet Takezaki, Shumpei
Tanaka, Kiyohito
Uchida, Seiichi
contents Severity level estimation is a crucial task in medical image diagnosis. However, accurately assigning severity class labels to individual images is very costly and challenging. Consequently, the attached labels tend to be noisy. In this paper, we propose a new framework for training with ``ordinal'' noisy labels. Since severity levels have an ordinal relationship, we can leverage this to train a classifier while mitigating the negative effects of noisy labels. Our framework uses two techniques: clean sample selection and dual-network architecture. A technical highlight of our approach is the use of soft labels derived from noisy hard labels. By appropriately using the soft and hard labels in the two techniques, we achieve more accurate sample selection and robust network training. The proposed method outperforms various state-of-the-art methods in experiments using two endoscopic ulcerative colitis (UC) datasets and a retinal Diabetic Retinopathy (DR) dataset. Our codes are available at https://github.com/shumpei-takezaki/Self-Relaxed-Joint-Training.
format Preprint
id arxiv_https___arxiv_org_abs_2410_21885
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Self-Relaxed Joint Training: Sample Selection for Severity Estimation with Ordinal Noisy Labels
Takezaki, Shumpei
Tanaka, Kiyohito
Uchida, Seiichi
Computer Vision and Pattern Recognition
Severity level estimation is a crucial task in medical image diagnosis. However, accurately assigning severity class labels to individual images is very costly and challenging. Consequently, the attached labels tend to be noisy. In this paper, we propose a new framework for training with ``ordinal'' noisy labels. Since severity levels have an ordinal relationship, we can leverage this to train a classifier while mitigating the negative effects of noisy labels. Our framework uses two techniques: clean sample selection and dual-network architecture. A technical highlight of our approach is the use of soft labels derived from noisy hard labels. By appropriately using the soft and hard labels in the two techniques, we achieve more accurate sample selection and robust network training. The proposed method outperforms various state-of-the-art methods in experiments using two endoscopic ulcerative colitis (UC) datasets and a retinal Diabetic Retinopathy (DR) dataset. Our codes are available at https://github.com/shumpei-takezaki/Self-Relaxed-Joint-Training.
title Self-Relaxed Joint Training: Sample Selection for Severity Estimation with Ordinal Noisy Labels
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2410.21885