Abstract
Deep convolutional neural networks have driven substantial advancements in the automatic understanding of images. Requiring a large collection of images and their associated annotations is one of the main bottlenecks limiting the adoption of deep networks. In the task of medical image segmentation, requiring pixel-level semantic annotations performed by human experts exacerbate this difficulty. This paper proposes a new framework to train a fully convolutional segmentation network from a large set of cheap unreliable annotations and a small set of expert-level clean annotations. We propose a spatially adaptive reweighting approach to treat clean and noisy pixel-level annotations commensurately in the loss function. We deploy a meta-learning approach to assign higher importance to pixels whose loss gradient direction is closer to those of clean data. Our experiments on training the network using segmentation ground truth corrupted with different levels of annotation noise show how spatial reweighting improves the robustness of deep networks to noisy annotations.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Codella, et al.: Skin lesion analysis toward melanoma detection: a challenge at the 2017 ISBI. arXiv:1710.05006 (2017)
Goldberger, J., Ben-Reuven, E.: Training deep neural-networks using a noise adaptation layer. In: ICLR (2017)
Ibtehaz, N., Rahman, M. S.: Multiresunet: rethinking the U-Net architecture for multimodal biomedical image segmentation. arXiv preprint arXiv:1902.04049 (2019)
Jiang, et al.: Mentornet: regularizing very deep neural networks on corrupted labels. arXiv preprint arXiv:1712.05055 (2017)
Long, et al.: Fully convolutional networks for semantic segmentation. In: IEEE CVPR, pp. 3431–3440 (2015)
Mirikharaji, Z., Hamarneh, G.: Star shape prior in fully convolutional networks for skin lesion segmentation. In: Frangi, A.F., Schnabel, J.A., Davatzikos, C., Alberola-López, C., Fichtinger, G. (eds.) MICCAI 2018. LNCS, vol. 11073, pp. 737–745. Springer, Cham (2018). https://doi.org/10.1007/978-3-030-00937-3_84
Oktay, et al.: Attention U-Net: learning where to look for the pancreas. In: MIDL (2018)
Ren, et al.: Learning to reweight examples for robust deep learning. In: ICML (2018)
Ronneberger, et al.: U-Net: convolutional networks for biomedical image segmentation. In: Navab, N., Hornegger, J., Wells, W.M., Frangi, A.F. (eds.) MICCAI 2015. LNCS, vol. 9351, pp. 234–241. Springer, Cham (2015). https://doi.org/10.1007/978-3-319-24574-4_28
Siegel, et al.: Cancer statistics. CA. Cancer J. Clin. 67(1), 7–30 (2017)
Sukhbaatar, et al.: Learning from noisy labels with deep neural networks. arXiv preprint arXiv:1406.2080 (2014)
Vahdat, A.: Toward robustness against label noise in training deep discriminative neural networks. In: NIPS, pp. 5596–5605 (2017)
Veit, et al.: Learning from noisy large-scale datasets with minimal supervision. In: IEEE CVPR, pp. 839–847 (2017)
Venkatesh, et al.: A deep residual architecture for skin lesion segmentation. In: OR 2.0 Context-Aware Operating Theaters, Computer Assisted Robotic Endoscopy, Clinical Image-Based Procedures, and Skin Image Analysis, pp. 277–284 (2018)
Wang, et al.: Iterative learning with open-set noisy labels. In: IEEE CVPR, pp. 8688–8696 (2018)
Xiao, et al.: Learning from massive noisy labeled data for image classification. In: IEEE CVPR, pp. 2691–2699 (2015)
Yuan, et al.: Automatic skin lesion segmentation using deep fully convolutional networks with jaccard distance. IEEE TMI 36(9), 1876–1886 (2017)
Zhang, et al.: Understanding deep learning requires rethinking generalization. In: ICLR (2017)
Acknowledgments
We thank NVIDIA Corporation for the donation of Titan X GPUs used in this research and Compute Canada for computational resources.
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2019 Springer Nature Switzerland AG
About this paper
Cite this paper
Mirikharaji, Z., Yan, Y., Hamarneh, G. (2019). Learning to Segment Skin Lesions from Noisy Annotations. In: Wang, Q., et al. Domain Adaptation and Representation Transfer and Medical Image Learning with Less Labels and Imperfect Data. DART MIL3ID 2019 2019. Lecture Notes in Computer Science(), vol 11795. Springer, Cham. https://doi.org/10.1007/978-3-030-33391-1_24
Download citation
DOI: https://doi.org/10.1007/978-3-030-33391-1_24
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-030-33390-4
Online ISBN: 978-3-030-33391-1
eBook Packages: Computer ScienceComputer Science (R0)