Nest-DGIL: Nesterov-optimized Deep Geometric Incremental Learning for CS Image Reconstruction

TL;DR

Nest-DGIL introduces a Nesterov-optimized deep learning framework for image reconstruction that effectively captures geometric details, reduces artifacts, and converges quickly, outperforming existing methods.

Contribution

The paper proposes a novel end-to-end deep geometric incremental learning framework based on Nesterov proximal gradient optimization, with theoretical guarantees and improved reconstruction quality.

Findings

Outperforms state-of-the-art methods in image reconstruction tasks.

Effectively reconstructs geometric texture details from preliminary linear reconstructions.

Achieves fast convergence and reduces artifacts in reconstructed images.

Abstract

Proximal gradient-based optimization is one of the most common strategies to solve inverse problem of images, and it is easy to implement. However, these techniques often generate heavy artifacts in image reconstruction. One of the most popular refinement methods is to fine-tune the regularization parameter to alleviate such artifacts, but it may not always be sufficient or applicable due to increased computational costs. In this work, we propose a deep geometric incremental learning framework based on the second Nesterov proximal gradient optimization. The proposed end-to-end network not only has the powerful learning ability for high-/low-frequency image features, but also can theoretically guarantee that geometric texture details will be reconstructed from preliminary linear reconstruction. Furthermore, it can avoid the risk of intermediate reconstruction results falling outside the geometric decomposition domains and achieve fast convergence. Our reconstruction framework is decomposed into four modules including general linear reconstruction, cascade geometric incremental restoration, Nesterov acceleration, and post-processing. In the image restoration step, a cascade geometric incremental learning module is designed to compensate for missing texture information from different geometric spectral decomposition domains. Inspired by the overlap-tile strategy, we also develop a post-processing module to remove the block effect in patch-wise-based natural image reconstruction. All parameters in the proposed model are learnable, an adaptive initialization technique of physical parameters is also employed to make model flexibility and ensure converging smoothly. We compare the reconstruction performance of the proposed method with existing state-of-the-art methods to demonstrate its superiority. Our source codes are available at https://github.com/fanxiaohong/Nest-DGIL.