English

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

Image and Video Processing 2025-06-03 v1

Abstract

In the context of inverse problems y=Ax\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 0\ell_0- or 1\ell_1-norm minimization encourage solutions x\bf x that are both consistent with observations y\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 A\bf A is unknown and only a limited set of observation-target pairs {x,y}\{ \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.

Keywords

Cite

@article{arxiv.2506.01112,
  title  = {TRUST -- Transformer-Driven U-Net for Sparse Target Recovery},
  author = {Di An and Dylan Poppert and Jiayue Li and Mark Foster and Trac D. Tran},
  journal= {arXiv preprint arXiv:2506.01112},
  year   = {2025}
}

Comments

10 pages, 5 figures. Under review at NeurIPS 2025

R2 v1 2026-07-01T02:53:21.845Z