Zeyu Ren1, Yudong Zhang1, Shuihua Wang1. 1. School of Computing and Mathematical Sciences, 4488University of Leicester, Leicester LE1 7RH, UK.
Abstract
Objective: The only possible solution to increase the patients' fatality rate is lung cancer early-stage detection. Recently, deep learning techniques became the most promising methods in medical image analysis compared with other numerous computer-aided diagnostic techniques. However, deep learning models always get lower performance when the model is overfitting. Methods: We present a Lung Cancer Data Augmented Ensemble (LCDAE) framework to solve the overfitting and lower performance problems in the lung cancer classification tasks. The LCDAE has 3 parts: The Lung Cancer Deep Convolutional GAN, which can synthesize images of lung cancer; A Data Augmented Ensemble model (DA-ENM), which ensembled 6 fine-tuned transfer learning models for training, testing, and validating on a lung cancer dataset; The third part is a Hybrid Data Augmentation (HDA) which combines all the data augmentation techniques in the LCDAE. Results: By comparing with existing state-of-the-art methods, the LCDAE obtains the best accuracy of 99.99%, the precision of 99.99%, and the F1-score of 99.99%. Conclusion: Our proposed LCDAE can overcome the overfitting issue for the lung cancer classification tasks by applying different data augmentation techniques, our method also has the best performance compared to state-of-the-art approaches.
Objective: The only possible solution to increase the patients' fatality rate is lung cancer early-stage detection. Recently, deep learning techniques became the most promising methods in medical image analysis compared with other numerous computer-aided diagnostic techniques. However, deep learning models always get lower performance when the model is overfitting. Methods: We present a Lung Cancer Data Augmented Ensemble (LCDAE) framework to solve the overfitting and lower performance problems in the lung cancer classification tasks. The LCDAE has 3 parts: The Lung Cancer Deep Convolutional GAN, which can synthesize images of lung cancer; A Data Augmented Ensemble model (DA-ENM), which ensembled 6 fine-tuned transfer learning models for training, testing, and validating on a lung cancer dataset; The third part is a Hybrid Data Augmentation (HDA) which combines all the data augmentation techniques in the LCDAE. Results: By comparing with existing state-of-the-art methods, the LCDAE obtains the best accuracy of 99.99%, the precision of 99.99%, and the F1-score of 99.99%. Conclusion: Our proposed LCDAE can overcome the overfitting issue for the lung cancer classification tasks by applying different data augmentation techniques, our method also has the best performance compared to state-of-the-art approaches.
Entities:
Keywords:
ensemble; generative adversarial networks; machine learning; medical image analysis
Authors: Andrew A Borkowski; Catherine P Wilson; Steven A Borkowski; L Brannon Thomas; Lauren A Deland; Stefanie J Grewe; Stephen M Mastorides Journal: Fed Pract Date: 2019-10
Authors: Sajid A Khan; Muhammad Nazir; Muhammad A Khan; Tanzila Saba; Kashif Javed; Amjad Rehman; Tallha Akram; Muhammad Awais Journal: Microsc Res Tech Date: 2019-04-11 Impact factor: 2.769
Authors: Miles N Wernick; Yongyi Yang; Jovan G Brankov; Grigori Yourganov; Stephen C Strother Journal: IEEE Signal Process Mag Date: 2010-07 Impact factor: 12.551