Abstract
Precise nodule segmentation in thyroid ultrasound images is important for clinical quantitative analysis and diagnosis. Fully supervised deep learning method can effectively extract representative features from nodules and background. Despite the great success, deep learning–based segmentation methods still face a critical hindrance: the difficulty in acquiring sufficient training data due to high annotation costs. To this end, we propose a weakly supervised framework called uncertainty to fine generative adversarial network (U2F-GAN) for nodule segmentation in thyroid ultrasound images that exploits only a handful of rough bounding box annotations to successfully generate reliable labels from these weak supervisions. Based on feature-matching GAN, the proposed method alternates between generating masks and learning a segmentation network in an adversarial manner. Super-pixel processing mechanism is adopted to reflect low-level image structure features for learning and inferring semantic segmentation, which largely improve the efficiency of training process. In addition, we introduce a similarity comparison module and a distributed loss function with constraints to effectively remove noise in localization annotations and enhance the generalization capability of the network, thus strengthen the overall segmentation performance. Compared to existing weakly supervised approaches, our proposed U2F-GAN yields a significant performance boost. The segmentation results are also comparable to fully supervised methods, but the annotation burden is much lower. Also, the training speed of the network model is much faster than other methods with weak supervisions, which enables the network to be updated in time, thus is beneficial to high-throughput medical image setting.
Similar content being viewed by others
Explore related subjects
Discover the latest articles, news and stories from top researchers in related subjects.References
Blankenship DR, Chin E, Terris DJ. Contemporary management of thyroid cancer. American Journal of Otolaryngology-Head and Neck Medicine and Surgery. 2005;26(4):249–60.
Zhao J, Zheng W, Zhang L, Tian H. Segmentation of ultrasound images of thyroid nodule for assisting fine needle aspiration cytology. Health Inf Sci Syst. 2013;1(1):1–12.
Peng B, Zhang L, Zhang D, Yang J. Image segmentation by iterated region merging with localized graph cuts. Pattern Recognit. 2011;44(10–11):2527–38.
Huang Q, Lee S, Liu L, Lu M, Li A. A robust graph-based segmentation method for breast nodules in ultrasound images. Ultrasonics. 2011;52(2):266–75.
Yap M, Edirisinghe E, Bez H. Processed images in human perception: a case study in ultrasound breast imaging. Eur J Radiol. 2010;73(3):682–7.
Bomeli S, Lebeau S, Ferris R. Evaluation of a thyroid nodule. Otolaryngol Clin North Am. 2010;43(2):229–38.
Suha K, Seunghoon H, Bohyung H. Weakly supervised semantic segmentation using super-pixel pooling network. In: 31st AAAI Conference on Artificial Intelligence. 2017;4111–7.
Dong M, Liu D, Xiong Z, et al. Instance segmentation from volumetric biomedical images without voxel-wise labeling. In: International Conference on Medical Image Computing & Computer-assisted Intervention. Springer, Cham. 2019;83–91.
Arnab A, Torr P. Pixelwise instance segmentation with a dynamically instantiated network. In: 2017 IEEE International Conference on Computer Vision and Pattern Recognition (CVPR). 2017;879–88.
Lin T, Maire M, Belongie S, et al. Microsoft coco: common objects in context. In: 11th European Conference on Computer Vision (ECCV). 2014;740–55.
Yoo I, Yoo D, Paeng K. PseudoEdgeNet: Nuclei segmentation only with point annotations. arXiv: 1906.02924v1. 2019.
Khoreva A, Benenson R, Hosang J, Hein M, Schiele B. Simple does it: weakly supervised instance and semantic segmentation. In: 2017 IEEE International Conference On Computer Vision and Pattern Recognition (CVPR). 2017;1665–74.
Xue H, Liu C, Wan F, Jiao J. DANet: divergent activation for weakly supervised object localization. In: 2019 IEEE/CVF International Conference on Computer Vision (ICCV). 2019;6588–97.
Bouget D, Allan M, Stoyanov D, Jannin P. Vision-based and marker-less surgical tool detection and tracking: a review of the literature. Med Image Anal. 2017;35:633–54.
Zhou Y, Zhu Y, Ye Q, Qiu Q, Jiao J. Weakly supervised instance segmentation using class peak response. In: 2018 IEEE International Conference on Computer Vision and Pattern Recognition (CVPR). 2018;3791–800.
Zhang W, Zhang Q, Cheng J, Bai C, Hao P. End-to-end panoptic segmentation with pixel-level non-overlapping embedding. In: 2019 IEEE International Conference on Multimedia and Expo (ICME). 2019;976–81.
Li Q, Arnab A, Torr P. Weakly- and semi-supervised panoptic segmentation. In: 15th European Conference on Computer Vision (ECCV). 2018;106–24.
Hwang S, Kim H. Self-transfer learning for weakly supervised lesion localization. In: International Conference on Medical Image Computing & Computer-assisted Intervention. Springer, Cham. 2016;239–46.
Wang X, Peng Y, Lu L, Lu Z, Bagheri M, Summers R. Chestx-ray8: hospital-scale chest x-ray database and benchmarks on weakly-supervised classification and localization of common thorax diseases. In: 2017 IEEE International Conference on Computer Vision and Pattern Recognition (CVPR). 2017;2097–106.
Feng X, Yang J, Laine A, Angelini E. Discriminative localization in cnns for weakly-supervised segmentation of pulmonary nodules. In: International Conference on Medical Image Computing & Computer-assisted Intervention. Springer, Cham. 2017;568–76.
Yang X, Wang Z, Liu C, et al. Joint detection and diagnosis of prostate cancer in multi-parametric MRI based on multimodal convolutional neural networks. In: International Conference on Medical Image Computing & Computer-assisted Intervention. Springer, Cham. 2017;426–34.
Demiray, B, Rackerseder, J, Bozhinoski, S, Navab N. Weakly-supervised white and grey matter segmentation in 3D brain ultrasound. arXiv: 1904.05191. 2019.
Carneiro G, Peng T, Bayer C, Navab N. Automatic quantification of tumour hypoxia from multi-modal microscopy images using weakly-supervised learning methods. IEEE Trans Med Imaging. 2017;36(7):1405–17.
Shin SY, Lee S, Yun I, Lee K. Joint weakly and semi-supervised deep learning for localization and classification of masses in breast ultrasound images. IEEE Trans Med Imaging. 2019;38(3):762–74.
Yang L, Zhang Y, Chen J, Zhang S, Chen D. Suggestive annotation: a deep active learning framework for biomedical image segmentation. In: International Conference on Medical Image Computing & Computer-assisted Intervention. Springer, Cham. 2017;399–407.
Zhao Z, Yang L, Zheng H, Guldner I, Zhang S, Chen D. Deep learning based instance segmentation in 3D biomedical images using weak annotation. In: International Conference on Medical Image Computing & Computer-assisted Intervention. Springer, Cham. 2018;352–60.
Khan S, Shahin A, Villafruela J, Shen J, Shao L. Extreme points derived confidence map as a cue for class-agnostic interactive segmentation using deep neural network. In: International Conference on Medical Image Computing & Computer-assisted Intervention. Springer, Cham. 2019;66–73.
Nishimura K, Ker D, Bise, R. Weakly supervised cell instance segmentation by propagating from detection response. arXiv: 1911.13077v1. 2019.
Ganin Y, Ustinova E, Ajakan H, et al. Domain-adversarial training of neural networks. J Mach Learn Res. 2016;17(1):1–35.
Luc P, Couprie C, Chintala S, Verbeek J. Semantic segmentation using adversarial networks. arXiv: 1611.08408. 2016.
Kohl S, Bonekamp D, Schlemmer HP, et al. Adversarial networks for the detection of aggressive prostate cancer. arXiv: 1702.08014. 2017.
Dai W, Doyle J, Liang X, et al. SCAN: structure correcting adversarial network for chest X-rays organ segmentation. arXiv: 1703.08770v1. 2017.
Liu R, Zhou S, Guo Y, Wang Y, Chang C. Nodule localization in thyroid ultrasound images with a joint-training convolutional neural network. J Digit Imaging. 2020;33:1266–79.
Salimans T, Goodfellow I, Zaremba W, Cheung V, Radford A, Chen X. Improved techniques for training GANs. Adv Neural Inf Process Syst. 2016;29:2234–42.
Ronneberger O, Fischer P, Brox T. U-Net: convolutional networks for biomedical image segmentation. In: International Conference on Medical Image Computing & Computer-assisted Intervention. Springer, Cham. 2015;234–41.
Achanta R, Shaji A, Smith K, et al. SLIC Super-pixels compared to state-of-the-art super-pixel methods. IEEE Trans Pattern Anal Mach Intell. 2012;34(11):2274–82.
Milletari F, Navab N, Ahmadi S. (2016) V-net: fully convolutional neural networks for volumetric medical image segmentation. In: 2016 Fourth International Conference on 3D Vision (3DV). 2016;565–71.
Ker J, Wang L, Rao J, Lim T. Deep learning applications in medical image analysis. IEEE Access. 2018;6(3):9375–89.
Lin T, Goyal P, Girshick R, He K, Dollár P. Focal loss for dense object detection. IEEE Trans Pattern Anal Mach Intell. 2020;42(2):318–27.
Ma J, Wu F, Jiang T, Zhu J, Kong D. Cascade convolutional neural networks for automatic detection of thyroid nodules in ultrasound images. Med Phys. 2017;44(5):1678–91.
Kingma D, Ba J. Adam: A method for stochastic optimization. arXiv: 1412.6980. 2015.
Abadi M, Barham P, Chen J, et al. TensorFlow: a system for large-scale machine learning. In: Conference on Operating Systems Design and Implementation. 2016;265–83.
Ma J, Wu F, Jiang T, et al. Ultrasound image-based thyroid nodule automatic segmentation using convolutional neural networks. Int J Comput Assist Radiol Surg. 2017;12:1895–910.
Hu Y, Guo Y, Wang Y, et al. Automatic nodule segmentation in breast ultrasound images using a dilated fully convolutional network combined with an active contour model. Med Phys. 2018;46(1):215–28.
Funding
This work is supported by the National Natural Science Foundation of China (61871135, 81830058, 81627804) and the Science and Technology Commission of Shanghai Municipality (18511102904, 20DZ1100104).
Author information
Authors and Affiliations
Corresponding authors
Ethics declarations
Ethical Approval
This article does not contain any studies with human participants performed by any of the authors.
Informed Consent
Informed consent was obtained from all individual participants included in the study.
Conflict of Interest
The authors declare no competing interests.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Ruoyun Liu and Shichong Zhou contributed equally to this work.
Rights and permissions
About this article
Cite this article
Liu, R., Zhou, S., Guo, Y. et al. U2F-GAN: Weakly Supervised Super-pixel Segmentation in Thyroid Ultrasound Images. Cogn Comput 13, 1099–1113 (2021). https://doi.org/10.1007/s12559-021-09909-7
Received:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s12559-021-09909-7