Deep greedy unfolding: Sorting out argsorting in greedy sparse recovery algorithms
This addresses a bottleneck for integrating greedy sparse recovery into deep learning, enabling structured sparse recovery applications, though it is incremental as it adapts existing methods.
The paper tackled the non-differentiability of greedy sparse recovery algorithms like OMP and IHT in neural networks by proposing Soft-OMP and Soft-IHT using soft permutation matrices, enabling fully trainable architectures that approximate the original algorithms with controllable accuracy.
Gradient-based learning imposes (deep) neural networks to be differentiable at all steps. This includes model-based architectures constructed by unrolling iterations of an iterative algorithm onto layers of a neural network, known as algorithm unrolling. However, greedy sparse recovery algorithms depend on the non-differentiable argsort operator, which hinders their integration into neural networks. In this paper, we address this challenge in Orthogonal Matching Pursuit (OMP) and Iterative Hard Thresholding (IHT), two popular representative algorithms in this class. We propose permutation-based variants of these algorithms and approximate permutation matrices using "soft" permutation matrices derived from softsort, a continuous relaxation of argsort. We demonstrate -- both theoretically and numerically -- that Soft-OMP and Soft-IHT, as differentiable counterparts of OMP and IHT and fully compatible with neural network training, effectively approximate these algorithms with a controllable degree of accuracy. This leads to the development of OMP- and IHT-Net, fully trainable network architectures based on Soft-OMP and Soft-IHT, respectively. Finally, by choosing weights as "structure-aware" trainable parameters, we connect our approach to structured sparse recovery and demonstrate its ability to extract latent sparsity patterns from data.