Performance Comparison of Different Loss Functions for Digital Breast Tomosynthesis Classification using 3D Deep Learning Model

摘要

Artificial intelligence (AI) algorithms, especially deep learning methods have proven to be successful in many medical imaging applications. Computerized breast cancer image analysis can improve diagnosis accuracy. Digital Breast Tomosynthesis (DBT) imaging is a new modality and more advantageous compared to classical digital mammography (DM). Therefore, development of new deep learning algorithms compatible with DBT modality are potent to improve DBT imaging reading time efficiency and increase accuracy for breast cancer diagnosis when used as additional tool for radiologists. In this work, we aimed to build a 3D deep learning model to distinguish malignancy and benign breasts using DBT images. We also investigated effects of different loss functions in our deep learning models. We implemented and evaluated our method on a large data set of 546 patients (205 malignancy and 341 benign). Our results showed that different loss functions lead to an influence on the models performance in our classification tasks, and specific loss function may be selected or customized to adjust a specific performance metric for concrete applications.