journal article·2021-01-06Open access

Using Generative Adversarial Networks and Parameter Optimization of Convolutional Neural Networks for Lung Tumor Classification

Chun-Hui Lin, Cheng-Jian Lin, Yu-Chi Li, Shyh-Hau Wang

Applied Sciences· Volume 11· Issue 2

Abstract

Cancer is the leading cause of death worldwide. Lung cancer, especially, caused the most death in 2018 according to the World Health Organization. Early diagnosis and treatment can considerably reduce mortality. To provide an efficient diagnosis, deep learning is overtaking conventional machine learning techniques and is increasingly being used in computer-aided design systems. However, a sparse medical data set and network parameter tuning process cause network training difficulty and cost longer experimental time. In the present study, the generative adversarial network was proposed to generate computed tomography images of lung tumors for alleviating the problem of sparse data. Furthermore, a parameter optimization method was proposed not only to improve the accuracy of lung tumor classification, but also reduce the experimental time. The experimental results revealed that the average accuracy can reach 99.86% after image augmentation and parameter optimization.
Source: crossref · Quality: baseline

Research context

OpenAlex topics: AI in cancer detection · COVID-19 diagnosis using AI · Radiomics and Machine Learning in Medical Imaging

SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.

Affiliations

Department of Computer Science and Information Engineering, National Cheng Kung University, Tainan 701, Taiwan

Department of Computer Science and Information Engineering, National Chin-Yi University of Technology, Taichung 411, Taiwan

College of Intelligence, National Taichung University of Science and Technology, Taichung 404, Taiwan

Intelligent Manufacturing Research Center, National Cheng Kung University, Tainan 701, Taiwan

Funding

the Ministry of Science and Technology of the Republic of China

References (29)

Show source-provided references
1. World Health Organization (2018, September 12). A Report About Cancer. Available online: https://www.who.int/news-room/fact-sheets/detail/cancer.
2. Cancer incidence and mortality worldwide: Sources, methods and major patterns in GLOBOCAN 201210.1002/ijc.29210 ↗
3. The Danish Randomized Lung Cancer CT Screening Trial—Overall Design and Results of the Prevalence Round10.1097/JTO.0b013e3181a0d98f ↗
4. PET/CT versus MRI for diagnosis, staging, and follow-up of lung cancer10.1002/jmri.24776 ↗
5. Ding, C., and Tao, D. (2017). Trunk-Branch Ensemble Convolutional Neural Networks for Video-based Face Recognition. IEEE Trans. Pattern Anal. Mach. Intell.10.1109/TPAMI.2017.2700390 ↗
6. Rasti, P., Uiboupin, T., Escalera, S., and Anbarjafari, G. (2016, January 13–15). Convolutional Neural Network Super Resolution for Face Recognition in Surveillance Monitoring. Proceedings of the International Conference on Articulated Motion and Deformable Objects, Palma de Mallorca, Spain.10.1007/978-3-319-41778-3_18 ↗
7. Levi, G., and Hassncer, T. (2015, January 7–12). Age and gender classification using convolutional neural networks. Proceedings of the 2015 IEEE Conference on Computer Vision and Pattern Recognition Workshops, Boston, MA, USA.10.1109/CVPRW.2015.7301352 ↗
8. Gradient-based learning applied to document recognition10.1109/5.726791 ↗
9. ImageNet classification with deep convolutional neural networks
10. Szegedy, C., Liu, W., Jia, Y., Sermanet, P., Reed, S., Anguelov, D., Erhan, D., Vanhoucke, V., and Rabinovich, A. (2015, January 7–12). Going deeper with convolutions. Proceedings of the 2015 IEEE Conference on Computer Vision and Pattern Recognition, Boston, MA, USA.10.1109/CVPR.2015.7298594 ↗
11. Simonyan, K., and Zisserman, A. (2014). Very Deep Convolutional Networks for Large-Scale Image Recognition. arXiv.
12. He, K., Zhang, X., Ren, S., and Sun, J. (2016, January 27–30). Deep Residual Learning for Image Recognition. Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition, Las Vegas, NV, USA.10.1109/CVPR.2016.90 ↗
13. A survey on deep learning in medical image analysis10.1016/j.media.2017.07.005 ↗
14. Guest Editorial Deep Learning in Medical Imaging: Overview and Future Promise of an Exciting New Technique10.1109/TMI.2016.2553401 ↗
15. Convolutional Neural Networks for Medical Image Analysis: Full Training or Fine Tuning?10.1109/TMI.2016.2535302 ↗
16. Stacked deep polynomial network based representation learning for tumor classification with small ultrasound image dataset10.1016/j.neucom.2016.01.074 ↗
17. Brosch, T., and Tam, R. (2013, January 22–26). Manifold Learning of Brain MRIs by Deep Learning. Proceedings of the Medical Image Computing and Computer-Assisted Intervention-MICCAI, Nagoya, Japan.
18. A survey on Image Data Augmentation for Deep Learning10.1186/s40537-019-0197-0 ↗
19. Gupta, A., Vedaldi, A., and Zisserman, A. (2016, January 27–30). Synthetic Data for Text Localisation in Natural Images. Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, NV, USA.10.1109/CVPR.2016.254 ↗
20. Generative Adversarial Networks
21. Radford, A., Metz, L., and Chintala, S. (2016). Unsupervised Representation Learning with Deep Convolutional Generative Adversarial Networks. arXiv.
22. Stoyanov, D. (2018). Data Augmentation for Skin Lesion Analysis. OR 2.0 Context-Aware Operating Theaters, Computer Assisted Robotic Endoscopy, Clinical Image-Based Procedures, and Skin Image Analysis, Springer. Lecture Notes in Computer Science.
23. Madani, A., Moradi, M., Karargyris, A., and Syeda-Mahmood, T. (2018, January 2). Chest x-ray generation and data augmentation for cardiovascular abnormality classification. Proceedings of the SPIE 10574, Medical Imaging 2018: Image Processing, 105741M, Houston, TX, USA.10.1117/12.2293971 ↗
24. Real, E., Moore, S., Selle, A., Saxena, S., Suematsu, Y.L., Tan, J., Le, Q., and Kurakin, A. (2017, January 6–11). Large-Scale Evolution of Image Classifiers. Proceedings of the 34th International Conference on Machine Learning, Sydney, Australia.
25. Autonomous deep learning: A genetic DCNN designer for image classification10.1016/j.neucom.2019.10.007 ↗
26. Design optimization of cutting parameters for turning operations based on the Taguchi method10.1016/S0924-0136(98)00079-X ↗
27. Reduce optimisation time and effort: Taguchi experimental design methods10.1016/j.fsigss.2007.10.050 ↗
28. Parameters Optimization for Sustainable Machining by using Taguchi Method10.1016/j.matpr.2019.07.380 ↗
29. Armato, S.G., Hadjiiski, L., Tourassi, G.D., Drukker, K., Giger, M.L., Li, F., Redmond, G., Farahani, K., Kirby, J.S., and Clarke, L.P. (2015). SPIE-AAPM-NCI Lung Nodule Classification Challenge Dataset. Cancer Imaging Arch.