Breast Cancer Detection using Residual Convolutional Neural Network and Weighted Loss

Abstract

This research presents a breast cancer detection system using deep learning method. Breast cancer detection in a large slide of biopsy image is a hard task because it needs manual observation by a pathologist to find the malignant region. The deep learning model used in this research is made up of multiple layers of the residual convolutional neural network, and instead of using another type of classifier, a multilayer neural network was used as the classifier and stacked together and trained using end-to-end training approach. The system is trained using invasive ductal carcinoma dataset from the Hospital of the University of Pennsylvania and The Cancer Institute of New Jersey. From this dataset, 80% and 20% were randomly sampled and used as training and testing data respectively. Training a neural network on an imbalanced dataset is quite challenging. Weighted loss function was used as the objective function to tackle this problem. We achieve 78.26% and 78.03% for Recall and F1-Score metrics, respectively which are an improvement compared to the previous approach.