Structure-preserving quality improvement of cone beam CT images using contrastive learning

Abstract

Cone-beam CT (CBCT) is widely used in dental clinics but exhibits limitations in assessing soft tissue pathology because of its lack of contrast resolution and low Hounsfield Units (HU) quantification accuracy. We aimed to increase the image quality and HU accuracy of CBCTs while preserving anatomical structures. We generated CT-like images from CBCT images using a patchwise contrastive learning-based GAN model. Our model was trained on unpaired CT and CBCT datasets with the novel combination of losses and the feature extractor pretrained on our training dataset. We evaluated the quality of the images generated by our model in terms of Frechet inception distance (FID), peak signal-to-noise ratio (PSNR), mean absolute error (MAE), and root mean square error (RMSE). Additionally, the structure preservation performance was assessed by the structure score. As a result, the generated CT-like images by our model were significantly superior to those generated by various baseline models in terms of FID, PSNR, MAE, RMSE, and structure score. Therefore, we demonstrated that our model provided the complementary benefits of preserving the anatomical structures of the input CBCT images and improving the image quality to be similar to those of CT images.