Lung nodule classification based on deep learning networks and handcraft segmentation

Luis G. Salvador-Torres, Jose A. Almaraz-Damian, Volodymyr I. Ponomaryov, Rogelio Reyes-Reyes, Clara Cruz-Ramos

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

Abstract

This study proposes a hybrid CAD system, where the first stage consists of the handcraft segmentation, following a CNN based on the ResNet-34 architecture. In the segmentation stage, the rib cage (thorax region) is extracted using the K-means algorithm. The extraction of the nodules is performed in two steps, those attached to the pleura are found via a K-means clustering on the rib cage, and the circumscribed and vascular nodules are extracted using morphological operations. The resulting segmentation masks are applied to the test images, decreasing the number of false positives. Finally, the resulting image is splitted of the patches to be classified by the ResNet-34 trained from scratch. Designed CAD system has been implemented on Google Colab platform and a standalone computer with Nvidia®RTX 3090. The experiments with different CAD systems were performed on SPIE and LIDC-IDRI datasets demonstrating better performance of designed technique with reduction of false-positive objects.

Original languageEnglish
Title of host publicationReal-Time Image Processing and Deep Learning 2022
EditorsNasser Kehtarnavaz, Matthias F. Carlsohn
PublisherSPIE
ISBN (Electronic)9781510650800
DOIs
StatePublished - 2022
EventReal-Time Image Processing and Deep Learning 2022 - Virtual, Online
Duration: 6 Jun 202212 Jun 2022

Publication series

NameProceedings of SPIE - The International Society for Optical Engineering
Volume12102
ISSN (Print)0277-786X
ISSN (Electronic)1996-756X

Conference

ConferenceReal-Time Image Processing and Deep Learning 2022
CityVirtual, Online
Period6/06/2212/06/22

Keywords

  • CNN
  • Computer-Aided Detection
  • LIDC-IDRI dataset
  • Lung cancer
  • SPIE dataset
  • computed tomography

Fingerprint

Dive into the research topics of 'Lung nodule classification based on deep learning networks and handcraft segmentation'. Together they form a unique fingerprint.

Cite this