Advertisement

Normal Appearance Autoencoder for Lung Cancer Detection and Segmentation

  • Mehdi AstarakiEmail author
  • Iuliana Toma-Dasu
  • Örjan Smedby
  • Chunliang Wang
Conference paper
Part of the Lecture Notes in Computer Science book series (LNCS, volume 11769)

Abstract

One of the major differences between medical doctor training and machine learning is that doctors are trained to recognize normal/healthy anatomy first. Knowing the healthy appearance of anatomy structures helps doctors to make better judgement when some abnormality shows up in an image. In this study, we propose a normal appearance autoencoder (NAA), that removes abnormalities from a diseased image. This autoencoder is semi-automatically trained using another partial convolutional in-paint network that is trained using healthy subjects only. The output of the autoencoder is then fed to a segmentation net in addition to the original input image, i.e. the latter gets both the diseased image and a simulated healthy image where the lesion is artificially removed. By getting access to knowledge of how the abnormal region is supposed to look, we hypothesized that the segmentation network could perform better than just being shown the original slice. We tested the proposed network on the LIDC-IDRI dataset for lung cancer detection and segmentation. The preliminary results show the NAA approach improved segmentation accuracy substantially in comparison with the conventional U-Net architecture.

Keywords

Lung nodule segmentation Anomaly detection Convolutional variational autoencoder 

Notes

Acknowledgment

This study was supported by the Swedish Childhood Cancer Foundation (grant no. MT2016-0016) and the Swedish innovation agency Vinnova (grant no. 2017-01247).

References

  1. 1.
    Zhang, J., Xia, Y., Cui, H., Zhang, Y.: Pulmonary nodule detection in medical images: a survey. Biomed. Signal Process. Control 43, 138–147 (2018).  https://doi.org/10.1016/j.bspc.2018.01.011CrossRefGoogle Scholar
  2. 2.
    Setio, A.A.A., et al.: Pulmonary nodule detection in CT images: false positive reduction using multi-view convolutional networks. IEEE Trans. Med. Imaging 35, 1160–1169 (2016).  https://doi.org/10.1109/TMI.2016.2536809CrossRefGoogle Scholar
  3. 3.
    Baur, C., Wiestler, B., Albarqouni, S., Navab, N.: Deep autoencoding models for unsupervised anomaly segmentation in brain MR images. In: Crimi, A., Bakas, S., Kuijf, H., Keyvan, F., Reyes, M., van Walsum, T. (eds.) BrainLes 2018. LNCS, vol. 11383, pp. 161–169. Springer, Cham (2019).  https://doi.org/10.1007/978-3-030-11723-8_16CrossRefGoogle Scholar
  4. 4.
    Chen, X., Konukoglu, E.: Unsupervised detection of lesions in brain MRI using constrained adversarial auto-encoders (2018)Google Scholar
  5. 5.
    Schlegl, T., Seeböck, P., Waldstein, S.M., Schmidt-Erfurth, U., Langs, G.: Unsupervised anomaly detection with generative adversarial networks to guide marker discovery. In: Niethammer, M., et al. (eds.) IPMI 2017. LNCS, vol. 10265, pp. 146–157. Springer, Cham (2017).  https://doi.org/10.1007/978-3-319-59050-9_12CrossRefGoogle Scholar
  6. 6.
    Liu, G., Reda, F.A., Shih, K.J., Wang, T.-C., Tao, A., Catanzaro, B.: Image inpainting for irregular holes using partial convolutions (2018)Google Scholar
  7. 7.
    Alex, V., Mohammed Safwan, K.P., Chennamsetty, S.S., Krishnamurthi, G.: Generative adversarial networks for brain lesion detection. Presented at the 24 February (2017)Google Scholar
  8. 8.
    Wang, C., Smedby, Ö.: Automatic whole heart segmentation using deep learning and shape context. In: Pop, M., et al. (eds.) STACOM 2017. LNCS, vol. 10663, pp. 242–249. Springer, Cham (2018).  https://doi.org/10.1007/978-3-319-75541-0_26CrossRefGoogle Scholar
  9. 9.
    Wang, S., et al.: Central focused convolutional neural networks: developing a data-driven model for lung nodule segmentation. Med. Image Anal. 40, 172–183 (2017).  https://doi.org/10.1016/j.media.2017.06.014CrossRefGoogle Scholar
  10. 10.
    Shakibapour, E., Cunha, A., Aresta, G., Mendonça, A.M., Campilho, A.: An unsupervised metaheuristic search approach for segmentation and volume measurement of pulmonary nodules in lung CT scans. Expert Syst. Appl. 119, 415–428 (2019).  https://doi.org/10.1016/j.eswa.2018.11.010CrossRefGoogle Scholar
  11. 11.
    Wang, Z., Yuan, H., Ji, S.: Spatial variational auto-encoding via matrix-variate normal distributions (2017)Google Scholar

Copyright information

© Springer Nature Switzerland AG 2019

Authors and Affiliations

  • Mehdi Astaraki
    • 1
    • 2
    Email author
  • Iuliana Toma-Dasu
    • 2
  • Örjan Smedby
    • 1
  • Chunliang Wang
    • 1
  1. 1.Department of Biomedical Engineering and Health SystemsKTH Royal Institute of TechnologyHuddingeSweden
  2. 2.Department of Oncology-Pathology, Karolinska InstitutetKarolinska UniversitetssjukhusetStockholmSweden

Personalised recommendations