Designing a Three-layer Back Propagation Artificial Neural Network for Breast Thermogram Classification

Abstract

Breast cancer is largely responsible for female mortality across globe. Infrared imaging helps to screen early abnormal signs based on the difference in contralateral temperatures of the breasts and can be used to improve the patient survival rate. Image data is huge to process as it is. In this work, 15 biostatistical features are extracted from the breast region. Using feature selection to achieve high performance prediction, the designed three-layer back propagation artificial neural network (ANN) employs 9 significant features to classify the thermograms as malignant or benign. For this research work, thermal images from the public Visual Lab dataset have been used. The best performance evaluation metrics, viz., accuracy, sensitivity and specificity obtained are 93.8%, 90% and 95.5%, respectively for the model with 10 neurons in the hidden layer. The outcome is promising with value of overall Area Under the Curve greater than 0.9 for both classes. The design of ANN with gradient descent algorithm used in this study outperforms the other neural network models in the literature indicating that a well-designed neural network can boost the capability of thermography to predict breast abnormalities.

KEYWORDS:

• Breast cancer
• Breast thermography
• Infrared imaging
• Thermal imaging
• Artificial neural network
• CAD system

Additional information

Notes on contributors

Aayesha Hakim

Aayesha Hakim is a research scholar at Veermata Jijabai Technological Institute, Mumbai. Her areas of interest are biomedical image processing, machine learning, pattern recognition and its applications.

R. N. Awale

R N Awale is a professor at Veermata Jijabai Technological Institute, Mumbai. His active research areas are cross layer design for QoS, adhoc wireless networks, image processing, antenna design and data security. Email: [email protected]