Learning to Segment from Noisy Annotations: A Spatial Correction Approach

Jiachen Yao, Yikai Zhang, Songzhu Zheng, Mayank Goswami, Prateek Prasanna +1 more
7/21/2023
eess.IVcs.CVcs.LG

Official code: michaelofsbu/SpatialCorrection (20 stars, last commit 2023-07-21).

Abstract

Noisy labels can significantly affect the performance of deep neural networks (DNNs). In medical image segmentation tasks, annotations are error-prone due to the high demand in annotation time and in the annotators' expertise. Existing methods mostly assume noisy labels in different pixels are \textit{i.i.d}. However, segmentation label noise usually has strong spatial correlation and has prominent bias in distribution. In this paper, we propose a novel Markov model for segmentation noisy annotations that encodes both spatial correlation and bias. Further, to mitigate such label noise, we propose a label correction method to recover true label progressively. We provide theoretical guarantees of the correctness of the proposed method. Experiments show that our approach outperforms current state-of-the-art methods on both synthetic and real-world noisy annotations.

View on arXivView PDF

Code Implementations(1)

Learning to Segment from Noisy Annotations: A Spatial Correction Approach

202Shell, PythonFeb 21, 20233 years ago

Similar Papers

Cite this paper

@article{yao2023learning,
  title  = {Learning to Segment from Noisy Annotations: A Spatial Correction Approach},
  author = {Jiachen Yao and Yikai Zhang and Songzhu Zheng and Mayank Goswami and Prateek Prasanna and Chao Chen},
  year   = {2023},
  eprint = {2308.02498},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2308.02498v1},
  journal = {ICLR 2023 2023}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2308.02498.svg)](https://codeofpaper.com/paper/2308.02498)

Discussion