Transfer Learning for Lung Nodules Classification with CNN and Random Forest

Abstract

Machine learning and deep neural networks are improving various industries, including healthcare, which improves daily life. Deep neural networks, including Convolutional Neural Networks (CNNs), provide valuable insights and support in improving daily activities. In particular, CNNs enable the recognition and classification of images from CT and MRI scans and other tasks. However, training a CNN requires many datasets to attain optimal accuracy and performance, which is challenging in the medical field due to ethical worries, the lack of descriptive notes from experts and labeled data, and the overall scarcity of disease images. To overcome these challenges, this work proposes a hybrid CNN with transfer learning and a random forest algorithm for classifying lung cancer and non-cancer from CT scan images. This research aims include preprocessing lung nodular data, developing the proposed algorithm, and comparing its effectiveness with other methods. The findings indicate that the proposed hybrid CNN with transfer learning and random forest performs better than standard CNNs without transfer learning. This research demonstrates the potential of using machine learning algorithms in the healthcare industry, especially in disease detection and classification.