Breast Cancer Detection Model Training Strategy Based on Continual Learning

Conference: CAIBDA 2022 - 2nd International Conference on Artificial Intelligence, Big Data and Algorithms
06/17/2022 - 06/19/2022 at Nanjing, China

Proceedings: CAIBDA 2022

Pages: 5Language: englishTyp: PDF

Authors:
Chen, Shufeng (UM-SJTU Joint Institute, Shanghai Jiao Tong University, Shanghai, China)
Tang, Fenghui (Software Institute, Nanjing University, Nanjing, China)

Abstract:
Breast cancer is the most diagnosed cancer but also the most curable cancer if detected early. Convolutional neural network (CNN) is widely adopted in breast cancer diagnosis for its efficiency and accuracy. However, due to its structure and features, catastrophic forgetting and inaccuracy are inevitable in the diagnostic process when the original CNN is used to deal with future tasks based on the previous learning experience. To make things worse, the available dataset of breast tumour samples is very limited, bringing challenges to CNN classification models’ accuracy when encountering new samples. In our research, we included three types of strategies for processing datasets and embedded continual learning method EWC in several popular CNN structures. By comparing various parameters in different strategies and structures, we tested the most accurate learning model we introduced. Surprisingly, our approach turns out to be more accurate and adaptable to the classification of microscopic new images of breast tumour tissue into benign and malignant ones. The experimental outcome shed light on a new research direction in improving state-of-the-art CNN meant for diagnostic function.