Automatic classification of breast lesions using Transfer Learning

Authors

  • Rafael Pedro da Silva Neto Federal University of Piaui
  • Antônio Oseas de Carvalho Filho

Abstract

Breast lesions are one of the most common types of lesions among women in Brazil and worldwide, accounting for about 28% of new cases each year. These lesions may have Benign or Malignant behaviors. In this work, a computational method for image classification was developed to differentiate malignant and benign breast lesions, aiming at a low computational cost and good efficiency. In our approach, different Convolutional Neural Networks architectures and several classifiers were tested. Transfer Learning was employed to deal with the limitation of the small number of images in the database, reaching an accuracy of 81.73%, a sensitivity of 85.66%, a specificity of 78.40%, Kappa of 0.63 and ROC curve of 0.82. Finally, we believe our method can integrate a CAD tool by acting as patient screening or by providing a second opinion to the specialist.

Downloads

Download data is not yet available.

Author Biography

Antônio Oseas de Carvalho Filho

Received a Ph.D. in Electrical Engineering at Federal University of Maranhão -Brazil in 2016. Currently, he is a professor at the Federal University of Piaui (UFPI).

Published

2020-02-16

How to Cite

Pedro da Silva Neto, R., & Oseas de Carvalho Filho, A. (2020). Automatic classification of breast lesions using Transfer Learning. IEEE Latin America Transactions, 17(12), 1964–1969. Retrieved from https://latamt.ieeer9.org/index.php/transactions/article/view/2837

Issue

Section

Special Isssue on Deep Learning