Confidence calibration on multiclass classification in medical imaging

Wenjing Yang, Zhantao Cao, Qin Chen, Yuhong Yang, Guowu Yang

Research output: Chapter in Book/Report/Conference proceedingConference contribution

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 languageEnglish (US)
Title of host publicationProceedings - 20th IEEE International Conference on Data Mining, ICDM 2020
EditorsClaudia Plant, Haixun Wang, Alfredo Cuzzocrea, Carlo Zaniolo, Xindong Wu
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages1364-1369
Number of pages6
ISBN (Electronic)9781728183169
DOIs
StatePublished - Nov 2020
Event20th IEEE International Conference on Data Mining, ICDM 2020 - Virtual, Sorrento, Italy
Duration: Nov 17 2020Nov 20 2020

Publication series

NameProceedings - IEEE International Conference on Data Mining, ICDM
Volume2020-November
ISSN (Print)1550-4786

Conference

Conference20th IEEE International Conference on Data Mining, ICDM 2020
Country/TerritoryItaly
CityVirtual, Sorrento
Period11/17/2011/20/20

Bibliographical note

Publisher Copyright:
© 2020 IEEE.

Keywords

  • BI-RADS
  • Classification
  • Confidence calibration
  • Deep learning
  • Medical imaging

Fingerprint

Dive into the research topics of 'Confidence calibration on multiclass classification in medical imaging'. Together they form a unique fingerprint.

Cite this