Deep learning based projection domain metal segmentation for metal artifact reduction in cone beam computed tomography

TL;DR

This paper introduces a deep learning approach for metal segmentation in CBCT projections using synthetic training data from X-ray simulations, significantly improving metal artifact reduction in challenging clinical scenarios.

Contribution

It presents a novel synthetic data generation method for training deep learning models to segment metals in CBCT projections, enhancing artifact reduction performance.

Findings

Simulations with fewer photons are effective for segmentation.

Training with combined full size and cropped projections improves robustness.

Significant image quality improvements in severe artifact cases.

Abstract

Metal artifact correction is a challenging problem in cone beam computed tomography (CBCT) scanning. Metal implants inserted into the anatomy cause severe artifacts in reconstructed images. Widely used inpainting-based metal artifact reduction (MAR) methods require segmentation of metal traces in the projections as a first step, which is a challenging task. One approach is to use a deep learning method to segment metals in the projections. However, the success of deep learning methods is limited by the availability of realistic training data. It is laborious and time consuming to get reliable ground truth annotations due to unclear implant boundaries and large numbers of projections. We propose to use X-ray simulations to generate synthetic metal segmentation training dataset from clinical CBCT scans. We compare the effect of simulations with different numbers of photons and also compare several training strategies to augment the available data. We compare our model's performance on real clinical scans with conventional region growing threshold-based MAR, moving metal artifact reduction method, and a recent deep learning method. We show that simulations with relatively small number of photons are suitable for the metal segmentation task and that training the deep learning model with full size and cropped projections together improves the robustness of the model. We show substantial improvement in the image quality affected by severe motion, voxel size under-sampling, and out-of-FOV metals. Our method can be easily integrated into the existing projection-based MAR pipeline to get improved image quality. This method can provide a novel paradigm to accurately segment metals in CBCT projections.