Abstract
Current deep learning methods developed to address classification problems related to medical imaging for disease detection and diagnosis are primarily based on binary labels and also with limited focus on confidence calibration. Confidence estimates are closely related to classification accuracy. While existing neural networks have the capability of extending binary labels to multiclass labels, the confidence calibration procedure is generally overlooked. To address the issue, we propose a method called knowledge discriminator risk network (KDR) and a confidence calibration voting algorithm (KDR-CCV) that together enhance classification accuracy, with an emphasis on confidence calibration. Comparative studies on multiclass classification based on the Breast Imaging Reporting and Data Systems (BI-RADS) assessment categories with a dataset containing only binary labels of ultrasound images are conducted. Experimental results show KDR-CCV achieves the overall best classification performance in comparison to other methods that conform to the BI-RADS criterion in addition to the effective improvement on classification accuracy. The proposed method incorporates BI-RADS assessment and artificial intelligence from an application-based broad practice, and can be extended to other medical imaging problems.
Original language | English (US) |
---|---|
Title of host publication | Proceedings - 20th IEEE International Conference on Data Mining, ICDM 2020 |
Editors | Claudia Plant, Haixun Wang, Alfredo Cuzzocrea, Carlo Zaniolo, Xindong Wu |
Publisher | Institute of Electrical and Electronics Engineers Inc. |
Pages | 1364-1369 |
Number of pages | 6 |
ISBN (Electronic) | 9781728183169 |
DOIs | |
State | Published - Nov 2020 |
Event | 20th IEEE International Conference on Data Mining, ICDM 2020 - Virtual, Sorrento, Italy Duration: Nov 17 2020 → Nov 20 2020 |
Publication series
Name | Proceedings - IEEE International Conference on Data Mining, ICDM |
---|---|
Volume | 2020-November |
ISSN (Print) | 1550-4786 |
Conference
Conference | 20th IEEE International Conference on Data Mining, ICDM 2020 |
---|---|
Country/Territory | Italy |
City | Virtual, Sorrento |
Period | 11/17/20 → 11/20/20 |
Bibliographical note
Publisher Copyright:© 2020 IEEE.
Keywords
- BI-RADS
- Classification
- Confidence calibration
- Deep learning
- Medical imaging