Residual Back Projection With Untrained Neural Networks

Read original: arXiv:2210.14416 - Published 9/19/2024 by Ziyu Shu, Alireza Entezari
Total Score

0

🧠

Sign in to get full access

or

If you already have an account, we'll log you in

Overview

  • Researchers present a framework for iterative reconstruction (IR) in computed tomography (CT) that leverages the hierarchical structure of neural networks, without requiring training.
  • The framework incorporates structural information as a deep image prior (DIP) and uses a novel residual back projection (RBP) connection to form the basis for iterations.
  • Experimental results demonstrate improvements over other state-of-the-art conventional IR methods, as well as pre-trained and untrained models, particularly in few-view, limited-angle, and low-dose imaging configurations.

Plain English Explanation

Computed tomography (CT) is a medical imaging technique that uses X-rays to create detailed images of the body. Researchers have been exploring the use of neural networks, a type of artificial intelligence, to improve the quality of these CT images.

However, there are challenges in applying neural networks to CT imaging, such as the lack of stability, theoretical guarantees for accuracy, and high-quality training data. To address these issues, the researchers have developed a new framework that uses an untrained neural network as a "deep image prior" (DIP) and a novel "residual back projection" (RBP) connection to iteratively improve the reconstruction of CT images.

The key idea is that the hierarchical structure of the neural network can be used to learn the characteristics of the CT images, without the need for extensive training. The RBP connection allows the output of the neural network in one iteration to be used to update the input for the next iteration, helping to refine the image over time.

The researchers found that their RBP-DIP framework performed better than other state-of-the-art methods, particularly in challenging imaging scenarios like few-view, limited-angle, and low-dose CT scans. This is important because these types of scans can be used to reduce radiation exposure or speed up the imaging process, but the image quality is often lower.

Technical Explanation

The researchers propose using an untrained U-net, a type of neural network architecture, in conjunction with a novel residual back projection (RBP) connection to minimize an objective function and achieve high-accuracy CT image reconstruction.

In each iteration of the reconstruction process, the weights of the untrained U-net are optimized, and the output of the U-net in the current iteration is used to update the input of the U-net in the next iteration through the RBP connection. This allows the framework to leverage the hierarchical structure of the neural network as a deep image prior (DIP) without requiring any training data.

Experimental results on both parallel and fan beam X-ray imaging configurations demonstrate that the RBP-DIP framework offers improvements over other state-of-the-art conventional IR methods, as well as pre-trained and untrained models with similar network structures. These improvements are particularly significant in the few-view, limited-angle, and low-dose imaging configurations, which are challenging for many CT reconstruction techniques.

Critical Analysis

The researchers acknowledge that their framework, while showing promising results, still has some limitations. For example, the iterative nature of the reconstruction process can be computationally intensive, and the performance of the framework may be sensitive to the choice of initialization and optimization hyperparameters.

Additionally, the paper does not provide a detailed theoretical analysis of the convergence and stability properties of the RBP-DIP framework, which would be important for understanding its fundamental limitations and potential use cases.

Further research could explore ways to improve the computational efficiency of the framework, as well as to provide stronger theoretical guarantees on its performance. It would also be interesting to see how the RBP-DIP framework compares to other state-of-the-art deep learning-based CT reconstruction methods that leverage large-scale training data.

Conclusion

The RBP-DIP framework presented in this paper offers a novel approach to CT image reconstruction that leverages the hierarchical structure of neural networks without the need for training data. The experimental results demonstrate significant improvements over other state-of-the-art methods, particularly in challenging imaging scenarios.

While the framework has some limitations, it represents an important step towards developing more robust and versatile CT reconstruction algorithms that can adapt to different imaging conditions and requirements. As the field of medical imaging continues to evolve, techniques like the RBP-DIP framework may play an increasingly important role in improving the quality and accessibility of CT scans for various clinical applications.



This summary was produced with help from an AI and may contain inaccuracies - check out the links to read the original source documents!

Follow @aimodelsfyi on 𝕏 →

Related Papers

🧠

Total Score

0

Residual Back Projection With Untrained Neural Networks

Ziyu Shu, Alireza Entezari

Background and Objective: The success of neural networks in a number of image processing tasks has motivated their application in image reconstruction problems in computed tomography (CT). While progress has been made in this area, the lack of stability and theoretical guarantees for accuracy, together with the scarcity of high-quality training data for specific imaging domains pose challenges for many CT applications. In this paper, we present a framework for iterative reconstruction (IR) in CT that leverages the hierarchical structure of neural networks, without the need for training. Our framework incorporates this structural information as a deep image prior (DIP), and uses a novel residual back projection (RBP) connection that forms the basis for our iterations. Methods: We propose using an untrained U-net in conjunction with a novel residual back projection to minimize an objective function and achieve high-accuracy reconstruction. In each iteration, the weights of the untrained U-net are optimized, and the output of the U-net in the current iteration is used to update the input of the U-net in the next iteration through the aforementioned RBP connection. Results: Experimental results demonstrate that the RBP-DIP framework offers improvements over other state-of-the-art conventional IR methods, as well as pre-trained and untrained models with similar network structures under multiple conditions. These improvements are particularly significant in the few-view, limited-angle, and low-dose imaging configurations. Conclusions: Applying to both parallel and fan beam X-ray imaging, our framework shows significant improvement under multiple conditions. Furthermore, the proposed framework requires no training data and can be adjusted on-demand to adapt to different conditions (e.g. noise level, geometry, and imaged object).

Read more

9/19/2024

Res-U2Net: Untrained Deep Learning for Phase Retrieval and Image Reconstruction
Total Score

0

Res-U2Net: Untrained Deep Learning for Phase Retrieval and Image Reconstruction

Carlos Osorio Quero, Daniel Leykam, Irving Rondon Ojeda

Conventional deep learning-based image reconstruction methods require a large amount of training data which can be hard to obtain in practice. Untrained deep learning methods overcome this limitation by training a network to invert a physical model of the image formation process. Here we present a novel untrained Res-U2Net model for phase retrieval. We use the extracted phase information to determine changes in an object's surface and generate a mesh representation of its 3D structure. We compare the performance of Res-U2Net phase retrieval against UNet and U2Net using images from the GDXRAY dataset.

Read more

4/11/2024

Untrained Neural Nets for Snapshot Compressive Imaging: Theory and Algorithms
Total Score

0

Untrained Neural Nets for Snapshot Compressive Imaging: Theory and Algorithms

Mengyu Zhao, Xi Chen, Xin Yuan, Shirin Jalali

Snapshot compressive imaging (SCI) recovers high-dimensional (3D) data cubes from a single 2D measurement, enabling diverse applications like video and hyperspectral imaging to go beyond standard techniques in terms of acquisition speed and efficiency. In this paper, we focus on SCI recovery algorithms that employ untrained neural networks (UNNs), such as deep image prior (DIP), to model source structure. Such UNN-based methods are appealing as they have the potential of avoiding the computationally intensive retraining required for different source models and different measurement scenarios. We first develop a theoretical framework for characterizing the performance of such UNN-based methods. The theoretical framework, on the one hand, enables us to optimize the parameters of data-modulating masks, and on the other hand, provides a fundamental connection between the number of data frames that can be recovered from a single measurement to the parameters of the untrained NN. We also employ the recently proposed bagged-deep-image-prior (bagged-DIP) idea to develop SCI Bagged Deep Video Prior (SCI-BDVP) algorithms that address the common challenges faced by standard UNN solutions. Our experimental results show that in video SCI our proposed solution achieves state-of-the-art among UNN methods, and in the case of noisy measurements, it even outperforms supervised solutions.

Read more

6/7/2024

Low-resolution Prior Equilibrium Network for CT Reconstruction
Total Score

0

Low-resolution Prior Equilibrium Network for CT Reconstruction

Yijie Yang, Qifeng Gao, Yuping Duan

The unrolling method has been investigated for learning variational models in X-ray computed tomography. However, it has been observed that directly unrolling the regularization model through gradient descent does not produce satisfactory results. In this paper, we present a novel deep learning-based CT reconstruction model, where the low-resolution image is introduced to obtain an effective regularization term for improving the network`s robustness. Our approach involves constructing the backbone network architecture by algorithm unrolling that is realized using the deep equilibrium architecture. We theoretically discuss the convergence of the proposed low-resolution prior equilibrium model and provide the conditions to guarantee convergence. Experimental results on both sparse-view and limited-angle reconstruction problems are provided, demonstrating that our end-to-end low-resolution prior equilibrium model outperforms other state-of-the-art methods in terms of noise reduction, contrast-to-noise ratio, and preservation of edge details.

Read more

4/19/2024