Deep Learning Neural Network for Lung Cancer Classification: Enhanced Optimization Function

TL;DR

This paper introduces a convolutional neural network with multispace image pooling and autoencoder enhancements, achieving higher accuracy and faster processing in lung cancer CT image classification.

Contribution

It proposes a novel CNN architecture with multispace image pooling and autoencoder integration, improving accuracy and reducing processing time for lung cancer detection.

Findings

Accuracy improved to 99.5% from 98.9%

Processing time reduced to 12 seconds per frame

Autoencoder and multispace image techniques enhance classification

Abstract

Background and Purpose: Convolutional neural network is widely used for image recognition in the medical area at nowadays. However, overall accuracy in predicting lung tumor is low and the processing time is high as the error occurred while reconstructing the CT image. The aim of this work is to increase the overall prediction accuracy along with reducing processing time by using multispace image in pooling layer of convolution neural network. Methodology: The proposed method has the autoencoder system to improve the overall accuracy, and to predict lung cancer by using multispace image in pooling layer of convolution neural network and Adam Algorithm for optimization. First, the CT images were pre-processed by feeding image to the convolution filter and down sampled by using max pooling. Then, features are extracted using the autoencoder model based on convolutional neural network and multispace image reconstruction technique is used to reduce error while reconstructing the image which then results improved accuracy to predict lung nodule. Finally, the reconstructed images are taken as input for SoftMax classifier to classify the CT images. Results: The state-of-art and proposed solutions were processed in Python Tensor Flow and It provides significant increase in accuracy in classification of lung cancer to 99.5 from 98.9 and decrease in processing time from 10 frames/second to 12 seconds/second. Conclusion: The proposed solution provides high classification accuracy along with less processing time compared to the state of art. For future research, large dataset can be implemented, and low pixel image can be processed to evaluate the classification