Classification of Breast Cancer Histology Images Through Transfer Learning Using a Pre-trained Inception Resnet V2
Breast cancer is one of the leading causes of female death worldwide. The histological analysis of breast tissue allows for the differentiation of the tissue suspected to be abnormal into four classes: normal tissue, benign tumor, in situ carcinoma and invasive carcinoma. Automatic diagnostic systems can help in that task. In this sense, this work propose a deep neural network approach using transfer learning to classify breast cancer histology images. First, the added top layers are trained and a second fine-tunning is done on some feature extraction layers that are frozen previously. The used network is an Inception Resnet V2. In order to overcome the lack of data, data augmentation is performed too. This work is a suggested solution for the ICIAR 2018 BACH-Challenge and the accuracy is 0.76 in the blind test set.
KeywordsBreast cancer diagnosis Breast histology images classification Convolutional neural network Inception resnet v2 Transfer learning Data augmentation Fine-tuning
This work is financed by the North Portugal Regional Operational Programme (NORTE 2020), under the PORTUGAL 2020 Partnership Agreement, and the European Regional Development Fund (ERDF), within the project “NanoSTIMA: Macro-to-Nano Human Sensing: Towards Integrated Multimodal Health Monitoring and Analytics/NORTE-01-0145-FEDER-000016”.
- 1.Kasper, D., Fauci, A., Hauser, S., Longo, D., Jameson, J.: Harrison’s Principles of Internal Medicine. McGraw-Hill Education, New York (2015)Google Scholar
- 2.National Breast Cancer Foundation Inc: Breast Cancer Facts. http://www.nationalbreastcancer.org/breast-cancer-facts
- 3.American Cancer Society: Breast Cancer Facts&Figures 2017–2018. American Cancer Society, Inc., Atlanta (2017)Google Scholar
- 9.Belsare, A., Mushrif, M., Pangarkar, M., Meshram, M.: Classification of breast cancer histopathology images using texture feature analysis. In: TENCON 2015–2015 IEEE Region 10 Conference, Macau (2015)Google Scholar
- 10.Zhang, B.: Breast cancer diagnosis from biopsy images by serial fusion of Random Subspace ensembles. In: 4th International Conference on Biomedical Engineering and Informatics, Shanghai (2011)Google Scholar
- 11.Cruz-Roa, A., Basavanhally, A., González, F., Gilmore, H., Feldman, M., Ganesan, S., Shih, N., Tomaszewsk, J., Madabhushi, A.: Automatic detection of invasive ductal carcinoma in whole slide images with convolutional neural networks. In: Proceedings of the Medical Imaging 2014: Digital Pathology, San Diego, vol. 9041 (2014)Google Scholar
- 12.Szegedy, C., Ioffe, S., Vanhoucke, V.: Inception-v4, inception-resnet and the impact of residual connections on learning (2016). arXiv:1602.07261