Multi-Instance Classification of Histopathological Breast Cancer Images with Visual Explanation

Abstract

Breast cancer is a big concern for women due to its higher mortality compared to other cancers. Objective and accurate early diagnosis is primordial for the treatment and survival improvement of patients. Histopathological image classification is considered the gold standard and is usually the last and most dependent diagnosis approach for doctors to make patient treatment proposals. In particular, recent deep learning-based methods provide remarkable classification results. However, these methods ignore rationale or logical explanation that is important for diagnosis reliability and human-level understanding. This paper proposes a multi-instance classification network (MICNet) with the mechanism of visual explanation to achieve the explainable classification of histopathological breast cancer images. The method uses a simple two-dimensional convolution kernel to generate explanation maps (i.e., visual explanation) through features coming from the end of the feature extractor in the VGG11 model pre-trained by ImageNet. Multiple instance learning (MIL) based on mirror padding and overlap cropping is adopted to improve the network’s classification performance. We also design a weighted average pooling method to encourage the network to learn more accurate visual explanation. Experiments on BreakHis and Camelyon16 patch-based datasets demonstrate that our MICNet outperforms other CNN models in classification and is able to provide a logical visual explanation that supports the network’s prediction.