Towards reliable calcification detection: calibration of uncertainty in coronary optical coherence tomography images

TL;DR

This paper enhances calcification detection in coronary OCT images by applying YOLO and calibrating prediction confidence to improve reliability and support clinical decision-making in CAD treatment.

Contribution

It introduces a calibration method for uncertainty in deep learning-based calcification detection, improving confidence reliability in coronary OCT imaging.

Findings

YOLO outperforms other models in precision and recall

Calibrated confidence scores reduce prediction uncertainty

Confidence calibration achieves a confidence error of approximately 0.13

Abstract

Optical coherence tomography (OCT) has become increasingly essential in assisting the treatment of coronary artery disease (CAD). Image-guided solutions such as Percutaneous Coronary Intervention (PCI) are extensively used during the treatment of CAD. However, unidentified calcified regions within a narrowed artery could impair the outcome of the PCI. Prior to treatments, object detection is paramount to automatically procure accurate readings on the location and thickness of calcifications within the artery. Deep learning-based object detection methods have been explored in a variety of applications. The quality of object detection predictions could lead to uncertain results, which are not desirable in safety-critical scenarios. In this work, we implement an object detection model, You-Only-Look-Once v5 (YOLO), on a calcification detection framework within coronary OCT images. We evaluate the uncertainty of predictions based on the expected calibration errors, thus assessing the certainty level of detection results. To calibrate confidence scores of predictions, we implement dependent logistic calibration using each detection result's confidence and center coordinates. With the calibrated confidence score of each prediction, we lower the uncertainty of predictions in calcification detection. Our results show that the YOLO achieves higher precision and recall in comparison with the other object detection model, meanwhile producing more reliable results. The calibrated confidence of prediction results in a confidence error of approximately 0.13, suggesting that the confidence calibration on calcification detection could provide a more trustworthy result, indicating a great potential to assist clinical evaluation of treating the CAD during the imaging-guided procedure.