TRUST -- Transformer-Driven U-Net for Sparse Target Recovery

TL;DR

TRUST is a novel neural network architecture combining Transformers and U-Net for improved sparse signal recovery in inverse problems with unknown sensing operators, outperforming traditional methods.

Contribution

The paper introduces TRUST, a hybrid Transformer-U-Net model that learns sensing operators and reconstructs sparse signals from limited data, advancing inverse problem solutions.

Findings

Outperforms traditional sparse recovery methods in SSIM and PSNR.

Effectively suppresses hallucination artifacts in reconstructions.

Achieves robust and accurate recovery with limited observation data.

Abstract

In the context of inverse problems $\bf y = Ax$, sparse recovery offers a powerful paradigm shift by enabling the stable solution of ill-posed or underdetermined systems through the exploitation of structure, particularly sparsity. Sparse regularization techniques via $\ell_0$- or $\ell_1$-norm minimization encourage solutions $\bf x$ that are both consistent with observations $\bf y$ and parsimonious in representation, often yielding physically meaningful interpretations. In this work, we address the classical inverse problem under the challenging condition where the sensing operator $\bf A$ is unknown and only a limited set of observation-target pairs $\{ \bf x,\bf y \}$ is available. We propose a novel neural architecture, TRUST, that integrates the attention mechanism of Transformers with the decoder pathway of a UNet to simultaneously learn the sensing operator and reconstruct the sparse signal. The TRUST model incorporates a Transformer-based encoding branch to capture long-range dependencies and estimate sparse support, which then guides a U-Net-style decoder to refine reconstruction through multiscale feature integration. The skip connections between the transformer stages and the decoder not only enhance image quality but also enable the decoder to access image features at different levels of abstraction. This hybrid architecture enables more accurate and robust recovery by combining global context with local details. Experimental results demonstrate that TRUST significantly outperforms traditional sparse recovery methods and standalone U-Net models, achieving superior performance in SSIM and PSNR metrics while effectively suppressing hallucination artifacts that commonly plague deep learning-based inverse solvers.