skip to main content
10.1145/3546607.3546615acmotherconferencesArticle/Chapter ViewAbstractPublication PagesicvarsConference Proceedingsconference-collections
research-article

Uncertainty Analysis Based Attention Network for Lung Nodule Segmentation from CT Images

Published: 25 August 2022 Publication History

Abstract

In recent years, with the development of computer science and technology, computer aided diagnosis (CAD) systems have played important role in clinical practice. Using deep learning technology, existing CAD systems can predict whether CT images contain lung nodule automatically. However, for the precise segmentation of lung nodule and nodule boundary, further diagnosis by doctors is required. The current widely used segmentation networks still have segmentation uncertainty in the lung nodules boundary, which will interfere the accuracy of segmentation results. To solve this problem, this paper propose a UAA-UNet (Uncertainty Analysis Based Attention UNet) based on the uncertainty analysis of edge regions. The network structure is divided into two stages. In the first stage, the initial segmentation map of the lung nodule is obtained, and the second stage focuses on the uncertainty region of the initial segmentation map. By learning the features of the uncertainty region, the uncertainty is reduced and the segmentation accuracy is improved. The second stage includes two modules, the uncertainty attention module and the uncertainty elimination module. In uncertainty attention module, the entropy map of the initial segmentation map of the lung nodule is input into the network as attention information to improve the network's ability to understand uncertainty. In uncertainty elimination module, by using EWCE (entropy map weighted cross entropy loss function), the entropy map of the prediction result is fed back to the network as a weight factor to further improve the network's learning ability of the uncertain region. We selected lung nodule slices from 1012 patients in the Lung Image Database Consortium (LIDC) to validate the feasibility and effectiveness of the proposed method. The experiment result shows that, in the lung nodule segmentation task, by leveraging uncertainty analysis, the network achieves significant improvements over the baseline network.

References

[1]
Center M, Siegel R, Jemal A. Global cancer facts & figures. Atlanta: American Cancer Society, 2011, 3: 1-52.
[2]
Wanqing Chen PhD, MD, Rongshou Zheng MPH, Peter D. Baade PhD, Siwei Zhang BMedSc, Hongmei Zeng PhD, MD, Freddie Bray PhD, Ahmedin Jemal DVM, PhD, Xue Qin Yu PhD, MPH, Jie He MD. Cancer statistics in China, 2015. CA: a cancer journal for clinicians, 2016, 66(2): 115-132.
[3]
David R.Baldwin. Prediction of risk of lung cancer in populations and in pulmonary nodules: significant progress to drive changes in paradigms. Lung Cancer, 2015, 89(1): 1-3.
[4]
Gabriel I. Barbash, M.D., M.P.H., and Sherry A. Glied, Ph.D. New technology and health care costs–the case of robot-assisted surgery. The New England journal of medicine, 2010, 363(8): 701.
[5]
Joao Rodrigo Ferreirada Silva Sousa, Aristofanes Correa Silva, Anselmo Cardosode Paiva, Rodolfo Acatauassu Nunes. Methodology for automatic detection of lung nodules in computerized tomography images. Computer methods and programs in biomedicine, 2010, 98(1): 1-14.
[6]
Shanhui Sun, Christian Bauer, Reinhard Beichel. Automated 3-D segmentation of lungs with lung cancer in CT data using a novel robust active shape model approach. IEEE transactions on medical imaging, 2011, 31(2): 449-460.
[7]
Jinsa Kuruvilla, K.Gunavathi. "Lung cancer classification using neural networks for CT images." Computer methods and programs in biomedicine 113.1 (2014): 202-209.
[8]
Messay, T., Hardie, R. C., & Tuinstra, T. R. (2015). Segmentation of pulmonary nodules in computed tomography using a regression neural network approach and its application to the lung image database consortium and image database resource initiative dataset. Medical image analysis, 22(1), 48-62.
[9]
Mohsen Keshani, Zohreh Azimifar, Farshad Tajeripour, Reza Boostani. Lung nodule segmentation and recognition using SVM classifier and active contour modeling: A complete intelligent system. Computers in biology and medicine, 2013, 43(4): 287-300.
[10]
Shelhamer Evan, Long Jonathan, Darrell Trevor. Fully convolutional networks for semantic segmentation. IEEE transactions on pattern analysis and machine intelligence, 2016, 39(4): 640-651.
[11]
LiangChieh Chen, George Papandreou, Iasonas Kokkinos, Kevin Murphy, Alan L. Yuille. Deeplab: Semantic image segmentation with deep convolutional nets, atrous convolution, and fully connected crfs. IEEE transactions on pattern analysis and machine intelligence, 2017, 40(4): 834-848.
[12]
Shuo Wang, Mu Zhou, Zaiyi Liu, Zhenyu Liu, Dongsheng Gu, Yali Zang, Di Dong, Olivier Gevaert, Jie Tian. Central focused convolutional neural networks: Developing a data-driven model for lung nodule segmentation. Medical image analysis, 2017, 40: 172-183.
[13]
Sihang Chen, Yifan Wang. Pulmonary nodule segmentation in computed tomography with an encoder-decoder architecture. In 2019 10th International Conference on Information Technology in Medicine and Education (ITME) (pp. 157-162). IEEE.
[14]
M Liu, J Dong, X Dong, H Yu, L Qi. Segmentation of lung nodule in CT images based on mask R-CNN//2018 9th International Conference on Awareness Science and Technology (iCAST). IEEE, 2018: 1-6.
[15]
[online] Available:https://wiki.cancerimagingarchive.net/display/Public/LIDC-IDRI.
[16]
Rafael Wiemker, Martin Bergtholdt, Ekta Dharaiya, Sven Kabus, Michael C. Lee. Agreement of CAD features with expert observer ratings for characterization of pulmonary nodules in CT using the LIDC-IDRI database//Medical Imaging 2009: Computer-Aided Diagnosis. International Society for Optics and Photonics, 2009, 7260: 72600H.
[17]
Wook-Jin Choi, Tae-Sun Choi. Automated pulmonary nodule detection system in computed tomography images: A hierarchical block classification approach. Entropy, 2013, 15(2): 507-523.
[18]
Qiang Li,Shusuke Sone,Kunio Doi. Selective enhancement filters for nodules, vessels, and airway walls in two‐and three‐dimensional CT scans. Medical physics, 2003, 30(8): 2040-2051.
[19]
Tomokazu Oda, Mitsuru Kubo, Yoshiki Kawata, Noboru Niki, Kenji Eguchi, Hironobu Ohmatsu, Ryutaro Kakinuma, Masahiro Kaneko, Masahiko Kusumoto, Noriyuki Moriyama, Kiyoshi Mori, Hiroyuki Nishiyama. Detection algorithm of lung cancer candidate nodules on multislice CT images//Medical Imaging 2002: Image Processing. International Society for Optics and Photonics, 2002, 4684: 1354-1361.
[20]
Ezhil E.Nithila, S.S.Kumar. Automatic detection of solitary pulmonary nodules using swarm intelligence optimized neural networks on CT images. Engineering science and technology, an international journal, 2017, 20(3): 1192-1202.
[21]
O Ronneberger, P Fischer, T Brox. (2015, October). U-net: Convolutional networks for biomedical image segmentation. In International Conference on Medical image computing and computer-assisted intervention (pp. 234-241). Springer, Cham.
[22]
O Oktay, J Schlemper, L L Folgoc, M Lee, M Heinrich, K Misawa, K Mori, S McDonagh, N Y Hammerla, B Kainz, B Glocker, D Rueckert. (2018). Attention u-net: Learning where to look for the pancreas. arXiv preprint arXiv:1804.03999.
[23]
Xiao Xiao, Shen Lian, Zhiming Luo, Shaozi Li. Weighted res-unet for high-quality retina vessel segmentation//2018 9th international conference on information technology in medicine and education (ITME). IEEE, 2018: 327-331.
[24]
Md Zahangir Alom, Mahmudul Hasan, Chris Yakopcic, Tarek M. Taha, Vijayan K. Asari. Recurrent residual convolutional neural network based on u-net (r2u-net) for medical image segmentation. arXiv preprint arXiv:1802.06955, 2018.
[25]
Sijing Cai, Yunxian Tian, Harvey Lui, Haishan Zeng, Yi Wu, Guannan Chen. Dense-UNet: a novel multiphoton in vivo cellular image segmentation model based on a convolutional neural network. Quantitative imaging in medicine and surgery, 2020, 10(6): 1275.

Cited By

View all
  • (2023)Lung Cancer Classification Using Modified U-Net Based Lobe Segmentation and Nodule DetectionIEEE Access10.1109/ACCESS.2023.328582111(60279-60291)Online publication date: 2023

Index Terms

  1. Uncertainty Analysis Based Attention Network for Lung Nodule Segmentation from CT Images

    Recommendations

    Comments

    Information & Contributors

    Information

    Published In

    cover image ACM Other conferences
    ICVARS '22: Proceedings of the 2022 6th International Conference on Virtual and Augmented Reality Simulations
    March 2022
    119 pages
    ISBN:9781450387330
    DOI:10.1145/3546607
    Permission to make digital or hard copies of all or part of this work for personal or classroom use is granted without fee provided that copies are not made or distributed for profit or commercial advantage and that copies bear this notice and the full citation on the first page. Copyrights for components of this work owned by others than ACM must be honored. Abstracting with credit is permitted. To copy otherwise, or republish, to post on servers or to redistribute to lists, requires prior specific permission and/or a fee. Request permissions from [email protected]

    Publisher

    Association for Computing Machinery

    New York, NY, United States

    Publication History

    Published: 25 August 2022

    Permissions

    Request permissions for this article.

    Check for updates

    Author Tags

    1. Attention Mechanism
    2. Entropy Map
    3. Lung Nodule Segmentation
    4. UNet
    5. Uncertainty Analysis

    Qualifiers

    • Research-article
    • Research
    • Refereed limited

    Conference

    ICVARS 2022

    Contributors

    Other Metrics

    Bibliometrics & Citations

    Bibliometrics

    Article Metrics

    • Downloads (Last 12 months)16
    • Downloads (Last 6 weeks)0
    Reflects downloads up to 25 Jan 2025

    Other Metrics

    Citations

    Cited By

    View all
    • (2023)Lung Cancer Classification Using Modified U-Net Based Lobe Segmentation and Nodule DetectionIEEE Access10.1109/ACCESS.2023.328582111(60279-60291)Online publication date: 2023

    View Options

    Login options

    View options

    PDF

    View or Download as a PDF file.

    PDF

    eReader

    View online with eReader.

    eReader

    HTML Format

    View this article in HTML Format.

    HTML Format

    Figures

    Tables

    Media

    Share

    Share

    Share this Publication link

    Share on social media