TorchRadon: Fast Differentiable Routines for Computed Tomography

Abstract

This work presents TorchRadon -- an open source CUDA library which contains a set of differentiable routines for solving computed tomography (CT) reconstruction problems. The library is designed to help researchers working on CT problems to combine deep learning and model-based approaches. The package is developed as a PyTorch extension and can be seamlessly integrated into existing deep learning training code. Compared to the existing Astra Toolbox, TorchRadon is up to 125 faster. The operators implemented by TorchRadon allow the computation of gradients using PyTorch backward(), and can therefore be easily inserted inside existing neural networks architectures. Because of its speed and GPU support, TorchRadon can also be effectively used as a fast backend for the implementation of iterative algorithms. This paper presents the main functionalities of the library, compares results with existing libraries and provides examples of usage.

References in corpus (1)

• Tomographic inversion using ${\ell }_1$-norm regularization of wavelet coefficients

Cited by in corpus (7)

• Known Operator Learning and Hybrid Machine Learning in Medical Imaging -- A Review of the Past, the Present, and the Future
• Quad-Net: Quad-domain Network for CT Metal Artifact Reduction
• Nest-DGIL: Nesterov-optimized Deep Geometric Incremental Learning for CS Image Reconstruction
• 2-Step Sparse-View CT Reconstruction with a Domain-Specific Perceptual Network
• TFPnP: Tuning-free Plug-and-Play Proximal Algorithm with Applications to Inverse Imaging Problems
• Generative Tomography Reconstruction
• Missing Cone Artifacts Removal in ODT using Unsupervised Deep Learning in Projection Domain