Deep curriculum learning in task space for multi-class based mammography diagnosis
Abstract
Mammography is used as a standard screening procedure for the potential patients of breast cancer. Over the past decade, it has been shown that deep learning techniques have succeeded in reaching near-human performance in a number of tasks, and its application in mammography is one of the topics that medical researchers most concentrate on. In this work, we propose an end-to-end Curriculum Learning (CL) strategy in task space for classifying the three categories of Full-Field Digital Mammography (FFDM), namely Malignant, Negative, and False recall. Specifically, our method treats this three-class classification as a "harder" task in terms of CL, and creates an "easier" sub-task of classifying False recall against the combined group of Negative and Malignant. We introduce a loss scheduler to dynamically weight the contribution of the losses from the two tasks throughout the entire training process. We conduct experiments on an FFDM dataset of 1,709 images using 5-fold cross validation. The results show that our curriculum learning strategy can boost the performance for classifying the three categories of FFDM compared to the baseline strategies for model training.
- Publication:
-
Medical Imaging 2022: Computer-Aided Diagnosis
- Pub Date:
- April 2022
- DOI:
- 10.1117/12.2612617
- arXiv:
- arXiv:2110.11320
- Bibcode:
- 2022SPIE12033E..0CL
- Keywords:
-
- Computer Science - Computer Vision and Pattern Recognition
- E-Print:
- 4-page abstract. Full paper to appear at SPIE Medical Imaging 2022