Optimization-Inspired Cross-Attention Transformer for Compressive Sensing
This work addresses efficiency and performance issues in compressive sensing for image processing, though it appears incremental as it builds on existing deep unfolding networks.
The paper tackles the problem of high parameter count and feature information loss in deep unfolding networks for compressive sensing by proposing a lightweight OCT-based Unfolding Framework (OCTUF) with a novel Dual Cross Attention module, achieving superior performance compared to state-of-the-art methods with lower complexity.
By integrating certain optimization solvers with deep neural networks, deep unfolding network (DUN) with good interpretability and high performance has attracted growing attention in compressive sensing (CS). However, existing DUNs often improve the visual quality at the price of a large number of parameters and have the problem of feature information loss during iteration. In this paper, we propose an Optimization-inspired Cross-attention Transformer (OCT) module as an iterative process, leading to a lightweight OCT-based Unfolding Framework (OCTUF) for image CS. Specifically, we design a novel Dual Cross Attention (Dual-CA) sub-module, which consists of an Inertia-Supplied Cross Attention (ISCA) block and a Projection-Guided Cross Attention (PGCA) block. ISCA block introduces multi-channel inertia forces and increases the memory effect by a cross attention mechanism between adjacent iterations. And, PGCA block achieves an enhanced information interaction, which introduces the inertia force into the gradient descent step through a cross attention block. Extensive CS experiments manifest that our OCTUF achieves superior performance compared to state-of-the-art methods while training lower complexity. Codes are available at https://github.com/songjiechong/OCTUF.