论文标题
迈向可靠的钙化检测:冠状动脉光相干断层扫描图像中不确定性的校准
Towards reliable calcification detection: calibration of uncertainty in coronary optical coherence tomography images
论文作者
论文摘要
光学相干断层扫描(OCT)对于有助于治疗冠状动脉疾病(CAD)变得越来越重要。在CAD处理过程中,广泛使用了图像指导的溶液,例如经皮冠状动脉介入(PCI)。但是,狭窄的动脉内未鉴定的钙化区域可能会损害PCI的结果。在治疗之前,对象检测对于自动在动脉内钙化的位置和厚度上自动读取至关重要。在各种应用中探索了基于深度学习的对象检测方法。对象检测预测的质量可能导致不确定的结果,这在安全至关重要的情况下是不可取的。在这项工作中,我们在冠状OCT图像中的钙化检测框架上实现了一个对象检测模型,即您的仅外观-Once V5(YOLO)。我们根据预期的校准误差评估预测的不确定性,从而评估了检测结果的确定性水平。为了校准预测的置信度得分,我们使用每个检测结果的置信度和中心坐标实现依赖的逻辑校准。随着每个预测的校准评分,我们降低了钙化检测预测的不确定性。我们的结果表明,与其他对象检测模型相比,Yolo可以达到更高的精度和回忆,同时产生更可靠的结果。预测的校准置信度导致置信度误差约为0.13,这表明钙化检测的置信度校准可以提供更可信赖的结果,这表明在成像引导过程中处理CAD的临床评估具有很大的潜力。
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.