JBFnet -- Low Dose CT Denoising by Trainable Joint Bilateral Filtering
This addresses the problem of interpretability and accountability in deep learning for medical imaging denoising, though it is incremental in combining known filtering techniques with neural networks.
The paper tackles low-dose CT denoising by introducing JBFnet, a neural network that uses iterative joint bilateral filtering with few parameters, and it outperforms state-of-the-art methods like CPCE3D, GAN, and deep GFnet in noise removal while preserving structures on the AAPM dataset.
Deep neural networks have shown great success in low dose CT denoising. However, most of these deep neural networks have several hundred thousand trainable parameters. This, combined with the inherent non-linearity of the neural network, makes the deep neural network diffcult to understand with low accountability. In this study we introduce JBFnet, a neural network for low dose CT denoising. The architecture of JBFnet implements iterative bilateral filtering. The filter functions of the Joint Bilateral Filter (JBF) are learned via shallow convolutional networks. The guidance image is estimated by a deep neural network. JBFnet is split into four filtering blocks, each of which performs Joint Bilateral Filtering. Each JBF block consists of 112 trainable parameters, making the noise removal process comprehendable. The Noise Map (NM) is added after filtering to preserve high level features. We train JBFnet with the data from the body scans of 10 patients, and test it on the AAPM low dose CT Grand Challenge dataset. We compare JBFnet with state-of-the-art deep learning networks. JBFnet outperforms CPCE3D, GAN and deep GFnet on the test dataset in terms of noise removal while preserving structures. We conduct several ablation studies to test the performance of our network architecture and training method. Our current setup achieves the best performance, while still maintaining behavioural accountability.