Advertisement

Prediction of Skin Cancer Based on Convolutional Neural Network

  • Jianfeng He
  • Qingqing Dong
  • Sanli Yi
Conference paper
Part of the Advances in Intelligent Systems and Computing book series (AISC, volume 856)

Abstract

Skin cancer is one of the most lethal cancers in the world, which is a common type of cancer. And the early diagnosis of it is quite important. Aiming to improve the classification of skin cancer, a modified convolutional neural net- work model is proposed. Firstly, a convolutional layer and a pooling layer is added on the original network, constructed an 8-layer convolutional neural network. And then the network output value and the data label by the backward propagation adjust the network parameters. Lastly, by constantly debugging the network, the best model for the recognition network is selected. The proposed network applied to the dataset consisting of 900 cases of melanoma with the malignancy status pathologically confirmed. The results of the revised network prediction accuracy got 91.92% on training set and 89.5% on test set. The results obtained demonstrate that the proposed method is performant and can be used to predict whether the skin cancer is benign or malignant. With the development of deep learning and the clinical application of medical images, this method is expected to become a diagnostic tool for doctors to diagnose skin cancer.

Keywords

Deep learning Melanoma Image classification Convolutional neural network 

References

  1. 1.
    Zhang, L., Liu, X., Wang, J., et al.: A method for predicting benign and malignant lung neoplasms based on quantitative image histology. Acta Autom. Sin. 43(12), 2109–2114 (2017)Google Scholar
  2. 2.
    Luo, J., Wu, J.: Review of fine grained image classification based on deep convolution. Acta Autom. Sin. 43(08), 1306–1318 (2017)zbMATHGoogle Scholar
  3. 3.
    Wan, L., Zeiler, M., Zhang, S., Le Cun, Y., Fergus, R.: Regularization of neural networks using dropconnect. In: Proceedings of the International Conference on Machine Learning, Atlanta, USA, pp. 1058–1066 (2013)Google Scholar
  4. 4.
    Esteva, A., Kuprel, B., Novoa, R.A., et al.: Dermatologist-level classification of skin cancer with deep neural networks. Nature 542(7639), 115–118 (2017)CrossRefGoogle Scholar
  5. 5.
    Goodfellow, I.J., Warde-Farley, D., Mirza, M., Courville, A.C., Bengio, Y.: Maxout networks. In: Proceedings of the International Conference on Machine Learning, Atlanta, USA, pp. 1319–1327 (2013)Google Scholar
  6. 6.
    Vasconcelos, C.N., Vasconcelos, B.N.: Experiments using deep learning for dermoscopy image analysis. Pattern Recognit. Lett. (2017). ISSN: 0167-8655Google Scholar
  7. 7.
    Chougrad, H., Zouaki, H., Alheyane, O.: Deep Convolutional Neural Networks for breast cancer screening. Comput. Methods Programs Biomed. 157, 19–30 (2018)CrossRefGoogle Scholar
  8. 8.
    Yang, T., Huang, S.: Examples of application of improved convolutional neural networks in classification and recommendation. Appl. Comput. Res. (04) 2018Google Scholar
  9. 9.
    Jia, Y., Shelhamer, E., Donahue, J., Karayev, S., Long, J., Girshick, R., Guadarrama, S., Darrell, T.: Caffe: convolutional architecture for fast feature embedding. In: ACM International Conference on Multimedia, pp. 675–678 (2014)Google Scholar
  10. 10.
    Kingma, D.P., Ba, J.: Adam: a method for stochastic optimization. Comput. Sci. (2014)Google Scholar

Copyright information

© Springer Nature Switzerland AG 2019

Authors and Affiliations

  1. 1.School of Information Engineering and AutomationKunming University of Science and TechnologyKunmingChina

Personalised recommendations