Shading artifact correction in breast CT using an interleaved deep learning segmentation and maximum-likelihood polynomial fitting approach.

Abstract

The purpose of this work was twofold: (a) To provide a robust and accurate method for image segmentation of dedicated breast CT (bCT) volume data sets, and (b) to improve Hounsfield unit (HU) accuracy in bCT by means of a postprocessing method that uses the segmented images to correct for the low-frequency shading artifacts in reconstructed images. A sequential and iterative application of image segmentation and low-order polynomial fitting to bCT volume data sets was used in the interleaved correction (IC) method. Image segmentation was performed through a deep convolutional neural network (CNN) with a modified U-Net architecture. A total of 45621 coronal bCT images from 111 patient volume data sets were segmented (using a previously published segmentation algorithm) and used for neural network training, validation, and testing. All patient data sets were selected from scans performed on four different prototype breast CT systems. The adipose voxels for each patient volume data set, segmented using the proposed CNN, were then fit to a three-dimensional low-order polynomial. The polynomial fit was subsequently used to correct for the shading artifacts introduced by scatter and beam hardening in a method termed "flat fielding." An interleaved utilization of image segmentation and flat fielding was repeated until a convergence criterion was satisfied. Mathematical and physical phantom studies were conducted to evaluate the dependence of the proposed algorithm on breast size and the distribution of fibroglandular tissue. In addition, a subset of patient scans (not used in the CNN training, testing or validation) were used to investigate the accuracy of the IC method across different scanner designs and beam qualities. The IC method resulted in an accurate classification of different tissue types with an average Dice similarity coefficient>95%, precision>97%, recall>95%, and F1-score>96% across all tissue types. The flat fielding correction of bCT images resulted in a significant reduction in either cupping or capping artifacts in both mathematical and physical phantom studies as measured by the integral nonuniformity metric with an average reduction of 71% for cupping and 30% for capping across different phantom sizes, and the Uniformity Index with an average reduction of 53% for cupping and 34% for capping. The validation studies demonstrated that the IC method improves Hounsfield Units (HU) accuracy and effectively corrects for shading artifacts caused by scatter contamination and beam hardening. The postprocessing approach described herein is relevant to the broad scope of bCT devices and does not require any modification in hardware or existing scan protocols. The trained CNN parameters and network architecture are available for interested users.