Design of lung nodules segmentation and recognition algorithm based on deep learning
Abstract Background Accurate segmentation and recognition algorithm of lung nodules has great important value of reference for early diagnosis of lung cancer. An algorithm is proposed for 3D CT sequence images in this paper based on 3D Res U-Net segmentation network and 3D ResNet50 classification ne...
Guardado en:
Autores principales: | , , , , , |
---|---|
Formato: | article |
Lenguaje: | EN |
Publicado: |
BMC
2021
|
Materias: | |
Acceso en línea: | https://doaj.org/article/ea95d1654c53428fa1eb5fc5cc931d56 |
Etiquetas: |
Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
|
id |
oai:doaj.org-article:ea95d1654c53428fa1eb5fc5cc931d56 |
---|---|
record_format |
dspace |
spelling |
oai:doaj.org-article:ea95d1654c53428fa1eb5fc5cc931d562021-11-14T12:13:03ZDesign of lung nodules segmentation and recognition algorithm based on deep learning10.1186/s12859-021-04234-01471-2105https://doaj.org/article/ea95d1654c53428fa1eb5fc5cc931d562021-11-01T00:00:00Zhttps://doi.org/10.1186/s12859-021-04234-0https://doaj.org/toc/1471-2105Abstract Background Accurate segmentation and recognition algorithm of lung nodules has great important value of reference for early diagnosis of lung cancer. An algorithm is proposed for 3D CT sequence images in this paper based on 3D Res U-Net segmentation network and 3D ResNet50 classification network. The common convolutional layers in encoding and decoding paths of U-Net are replaced by residual units while the loss function is changed to Dice loss after using cross entropy loss to accelerate network convergence. Since the lung nodules are small and rich in 3D information, the ResNet50 is improved by replacing the 2D convolutional layers with 3D convolutional layers and reducing the sizes of some convolution kernels, 3D ResNet50 network is obtained for the diagnosis of benign and malignant lung nodules. Results 3D Res U-Net was trained and tested on 1044 CT subcases in the LIDC-IDRI database. The segmentation result shows that the Dice coefficient of 3D Res U-Net is above 0.8 for the segmentation of lung nodules larger than 10 mm in diameter. 3D ResNet50 was trained and tested on 2960 lung nodules in the LIDC-IDRI database. The classification result shows that the diagnostic accuracy of 3D ResNet50 is 87.3% and AUC is 0.907. Conclusion The 3D Res U-Net module improves segmentation performance significantly with the comparison of 3D U-Net model based on residual learning mechanism. 3D Res U-Net can identify small nodules more effectively and improve its segmentation accuracy for large nodules. Compared with the original network, the classification performance of 3D ResNet50 is significantly improved, especially for small benign nodules.Hui YuJinqiu LiLixin ZhangYuzhen CaoXuyao YuJinglai SunBMCarticleLung noduleConvolutional neural networkU-NetResidual learningImage segmentationImage classificationComputer applications to medicine. Medical informaticsR858-859.7Biology (General)QH301-705.5ENBMC Bioinformatics, Vol 22, Iss S5, Pp 1-21 (2021) |
institution |
DOAJ |
collection |
DOAJ |
language |
EN |
topic |
Lung nodule Convolutional neural network U-Net Residual learning Image segmentation Image classification Computer applications to medicine. Medical informatics R858-859.7 Biology (General) QH301-705.5 |
spellingShingle |
Lung nodule Convolutional neural network U-Net Residual learning Image segmentation Image classification Computer applications to medicine. Medical informatics R858-859.7 Biology (General) QH301-705.5 Hui Yu Jinqiu Li Lixin Zhang Yuzhen Cao Xuyao Yu Jinglai Sun Design of lung nodules segmentation and recognition algorithm based on deep learning |
description |
Abstract Background Accurate segmentation and recognition algorithm of lung nodules has great important value of reference for early diagnosis of lung cancer. An algorithm is proposed for 3D CT sequence images in this paper based on 3D Res U-Net segmentation network and 3D ResNet50 classification network. The common convolutional layers in encoding and decoding paths of U-Net are replaced by residual units while the loss function is changed to Dice loss after using cross entropy loss to accelerate network convergence. Since the lung nodules are small and rich in 3D information, the ResNet50 is improved by replacing the 2D convolutional layers with 3D convolutional layers and reducing the sizes of some convolution kernels, 3D ResNet50 network is obtained for the diagnosis of benign and malignant lung nodules. Results 3D Res U-Net was trained and tested on 1044 CT subcases in the LIDC-IDRI database. The segmentation result shows that the Dice coefficient of 3D Res U-Net is above 0.8 for the segmentation of lung nodules larger than 10 mm in diameter. 3D ResNet50 was trained and tested on 2960 lung nodules in the LIDC-IDRI database. The classification result shows that the diagnostic accuracy of 3D ResNet50 is 87.3% and AUC is 0.907. Conclusion The 3D Res U-Net module improves segmentation performance significantly with the comparison of 3D U-Net model based on residual learning mechanism. 3D Res U-Net can identify small nodules more effectively and improve its segmentation accuracy for large nodules. Compared with the original network, the classification performance of 3D ResNet50 is significantly improved, especially for small benign nodules. |
format |
article |
author |
Hui Yu Jinqiu Li Lixin Zhang Yuzhen Cao Xuyao Yu Jinglai Sun |
author_facet |
Hui Yu Jinqiu Li Lixin Zhang Yuzhen Cao Xuyao Yu Jinglai Sun |
author_sort |
Hui Yu |
title |
Design of lung nodules segmentation and recognition algorithm based on deep learning |
title_short |
Design of lung nodules segmentation and recognition algorithm based on deep learning |
title_full |
Design of lung nodules segmentation and recognition algorithm based on deep learning |
title_fullStr |
Design of lung nodules segmentation and recognition algorithm based on deep learning |
title_full_unstemmed |
Design of lung nodules segmentation and recognition algorithm based on deep learning |
title_sort |
design of lung nodules segmentation and recognition algorithm based on deep learning |
publisher |
BMC |
publishDate |
2021 |
url |
https://doaj.org/article/ea95d1654c53428fa1eb5fc5cc931d56 |
work_keys_str_mv |
AT huiyu designoflungnodulessegmentationandrecognitionalgorithmbasedondeeplearning AT jinqiuli designoflungnodulessegmentationandrecognitionalgorithmbasedondeeplearning AT lixinzhang designoflungnodulessegmentationandrecognitionalgorithmbasedondeeplearning AT yuzhencao designoflungnodulessegmentationandrecognitionalgorithmbasedondeeplearning AT xuyaoyu designoflungnodulessegmentationandrecognitionalgorithmbasedondeeplearning AT jinglaisun designoflungnodulessegmentationandrecognitionalgorithmbasedondeeplearning |
_version_ |
1718429379510403072 |