Adaptive Convolutional Dictionary Network for CT Metal Artifact Reduction

TL;DR

This paper introduces an adaptive convolutional dictionary network that combines model-based priors with deep learning to effectively reduce metal artifacts in CT images, offering interpretability and adaptability.

Contribution

It proposes a novel network that explicitly embeds prior knowledge of artifact structures into a deep learning framework, enhancing interpretability and performance in metal artifact reduction.

Findings

Outperforms existing methods on synthetic and clinical datasets

Demonstrates strong generalization ability across different data types

Provides interpretable model structure aligned with artifact characteristics

Abstract

Inspired by the great success of deep neural networks, learning-based methods have gained promising performances for metal artifact reduction (MAR) in computed tomography (CT) images. However, most of the existing approaches put less emphasis on modelling and embedding the intrinsic prior knowledge underlying this specific MAR task into their network designs. Against this issue, we propose an adaptive convolutional dictionary network (ACDNet), which leverages both model-based and learning-based methods. Specifically, we explore the prior structures of metal artifacts, e.g., non-local repetitive streaking patterns, and encode them as an explicit weighted convolutional dictionary model. Then, a simple-yet-effective algorithm is carefully designed to solve the model. By unfolding every iterative substep of the proposed algorithm into a network module, we explicitly embed the prior structure into a deep network, \emph{i.e.,} a clear interpretability for the MAR task. Furthermore, our ACDNet can automatically learn the prior for artifact-free CT images via training data and adaptively adjust the representation kernels for each input CT image based on its content. Hence, our method inherits the clear interpretability of model-based methods and maintains the powerful representation ability of learning-based methods. Comprehensive experiments executed on synthetic and clinical datasets show the superiority of our ACDNet in terms of effectiveness and model generalization. {\color{blue}{{\textit{Code is available at {\url{https://github.com/hongwang01/ACDNet}}.}}}}