Learning step sizes for unfolded sparse coding
Not on the map. Why?
Abstract (arXiv)
Sparse coding is typically solved by iterative optimization techniques, such as the Iterative Shrinkage-Thresholding Algorithm (ISTA). Unfolding and learning weights of ISTA using neural networks is a practical way to accelerate estimation. In this paper, we study the selection of adapted step sizes for ISTA. We show that a simple step size strategy can improve the convergence rate of ISTA by leveraging the sparsity of the iterates. However, it is impractical in most large-scale applications. Therefore, we propose a network architecture where only the step sizes of ISTA are learned. We demonstrate that for a large class of unfolded algorithms, if the algorithm converges to the solution of the Lasso, its last layers correspond to ISTA with learned step sizes. Experiments show that our method is competitive with state-of-the-art networks when the solutions are sparse enough.
Key facts. Cited by 29 papers in the map (data through 7 Oct 2026). First posted on arXiv 27 May 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 11; 2022: 4; 2023: 7; 2024: 3; 2025: 3; 2026: 1.
29 papers.
Sort:
- 2021Learning to Optimize: A Primer and A Benchmarkcited by 206
- 2025From Flat to Hierarchical: Extracting Sparse Representations with Matching Pursuitcited by 45
- 2021Learning to Learn Graph Topologiescited by 26
- 2021Hyperparameter Tuning is All You Need for LISTAcited by 22
- 2022Theoretical Perspectives on Deep Learning Methods in Inverse Problemscited by 22
- 2023Machine Learning for Large-Scale Optimization in 6G Wireless Networkscited by 18
- 2021Stable and Interpretable Unrolled Dictionary Learningcited by 15
- 2023Towards Constituting Mathematical Structures for Learning to Optimizecited by 13
- 2021Understanding approximate and unrolled dictionary learning for pattern recoverycited by 11
- 2021Algorithm Unrolling for Massive Access via Deep Neural Network with Theoretical Guaranteecited by 5
- 2022Optimization-Derived Learning with Essential Convergence Analysis of Training and Hyper-trainingcited by 5
- 2021Unrolling PALM for sparse semi-blind source separationcited by 3
17 more
- 2022Learning Variational Models with Unrolling and Bilevel Optimizationcited by 3
- 2023Hierarchical Optimization-Derived Learningcited by 3
- 2025Mechanistic Interpretability with Sparse Autoencoder Neural Operatorscited by 3
- 2021Learned ISTA with Error-based Thresholding for Adaptive Sparse Codingcited by 2
- 2023Uncertainty quantification for learned ISTAcited by 2
- 2023Learning Interpretable Deep Disentangled Neural Networks for Hyperspectral Unmixingcited by 2
- 2024From Learning to Optimize to Learning Optimization Algorithmscited by 2
- 2024Accelerating Multi-Block Constrained Optimization Through Learning to Optimizecited by 2
- 2021A Design Space Study for LISTA and Beyondcited by 1
- 2022Gabor is Enough: Interpretable Deep Denoising with a Gabor Synthesis Dictionary Priorcited by 1
- 2023Probabilistic Unrolling: Scalable, Inverse-Free Maximum Likelihood Estimation for Latent Gaussian Modelscited by 1
- 2025Evaluating Sparse Autoencoders: From Shallow Design to Matching Pursuitcited by 1
- 2021Deep Learning Techniques for Compressive Sensing-Based Reconstruction and Inference -- A Ubiquitous Systems Perspectivecited by 0
- 2021Learned Interpretable Residual Extragradient ISTA for Sparse Codingcited by 0
- 2023Active Inference in Hebbian Learning Networkscited by 0
- 2024Analysis and Synthesis Denoisers for Forward-Backward Plug-and-Play Algorithmscited by 0
- 2026Accelerating Hierarchical Sparse Predictive Coding with Hybrid Amortized Inferencecited by 0
- 2026Accelerating Hierarchical Sparse Predictive Coding with Hybrid Amortized Inference26 Jun 2026
- 2025Mechanistic Interpretability with Sparse Autoencoder Neural Operators3 Sep 2025
- 2025Evaluating Sparse Autoencoders: From Shallow Design to Matching Pursuit5 Jun 2025
- 2025From Flat to Hierarchical: Extracting Sparse Representations with Matching Pursuit3 Jun 2025
- 2024Analysis and Synthesis Denoisers for Forward-Backward Plug-and-Play Algorithms20 Nov 2024
- 2024Accelerating Multi-Block Constrained Optimization Through Learning to Optimize25 Sep 2024
- 2024From Learning to Optimize to Learning Optimization Algorithms28 May 2024
- 2023Learning Interpretable Deep Disentangled Neural Networks for Hyperspectral Unmixing3 Oct 2023
- 2023Uncertainty quantification for learned ISTA14 Sep 2023
- 2023Active Inference in Hebbian Learning Networks8 Jun 2023
- 2023Probabilistic Unrolling: Scalable, Inverse-Free Maximum Likelihood Estimation for Latent Gaussian Models5 Jun 2023
- 2023Towards Constituting Mathematical Structures for Learning to Optimize29 May 2023
17 more
- 2023Hierarchical Optimization-Derived Learning11 Feb 2023
- 2023Machine Learning for Large-Scale Optimization in 6G Wireless Networks3 Jan 2023
- 2022Learning Variational Models with Unrolling and Bilevel Optimization26 Sep 2022
- 2022Theoretical Perspectives on Deep Learning Methods in Inverse Problems29 Jun 2022
- 2022Optimization-Derived Learning with Essential Convergence Analysis of Training and Hyper-training16 Jun 2022
- 2022Gabor is Enough: Interpretable Deep Denoising with a Gabor Synthesis Dictionary Prior23 Apr 2022
- 2021Learned ISTA with Error-based Thresholding for Adaptive Sparse Coding21 Dec 2021
- 2021Unrolling PALM for sparse semi-blind source separation10 Dec 2021
- 2021Hyperparameter Tuning is All You Need for LISTA29 Oct 2021
- 2021Learning to Learn Graph Topologies19 Oct 2021
- 2021Learned Interpretable Residual Extragradient ISTA for Sparse Coding22 Jun 2021
- 2021Algorithm Unrolling for Massive Access via Deep Neural Network with Theoretical Guarantee19 Jun 2021
- 2021Understanding approximate and unrolled dictionary learning for pattern recovery11 Jun 2021
- 2021Stable and Interpretable Unrolled Dictionary Learning31 May 2021
- 2021Deep Learning Techniques for Compressive Sensing-Based Reconstruction and Inference -- A Ubiquitous Systems Perspective26 May 2021
- 2021A Design Space Study for LISTA and Beyond8 Apr 2021
- 2021Learning to Optimize: A Primer and A Benchmark23 Mar 2021
- 2021Learning to Optimize: A Primer and A Benchmarkcited by 206
- 2025From Flat to Hierarchical: Extracting Sparse Representations with Matching Pursuitcited by 45
- 2021Learning to Learn Graph Topologiescited by 26
- 2021Hyperparameter Tuning is All You Need for LISTAcited by 22
- 2022Theoretical Perspectives on Deep Learning Methods in Inverse Problemscited by 22
- 2023Machine Learning for Large-Scale Optimization in 6G Wireless Networkscited by 18
- 2021Stable and Interpretable Unrolled Dictionary Learningcited by 15
- 2023Towards Constituting Mathematical Structures for Learning to Optimizecited by 13
- 2021Understanding approximate and unrolled dictionary learning for pattern recoverycited by 11
- 2021Algorithm Unrolling for Massive Access via Deep Neural Network with Theoretical Guaranteecited by 5
- 2022Optimization-Derived Learning with Essential Convergence Analysis of Training and Hyper-trainingcited by 5
- 2021Unrolling PALM for sparse semi-blind source separationcited by 3
12 most cited of 29 All 29 (JSON)
- 2026Accelerating Hierarchical Sparse Predictive Coding with Hybrid Amortized Inference26 Jun 2026
- 2025Mechanistic Interpretability with Sparse Autoencoder Neural Operators3 Sep 2025
- 2025Evaluating Sparse Autoencoders: From Shallow Design to Matching Pursuit5 Jun 2025
- 2025From Flat to Hierarchical: Extracting Sparse Representations with Matching Pursuit3 Jun 2025
- 2024Analysis and Synthesis Denoisers for Forward-Backward Plug-and-Play Algorithms20 Nov 2024
- 2024Accelerating Multi-Block Constrained Optimization Through Learning to Optimize25 Sep 2024
- 2024From Learning to Optimize to Learning Optimization Algorithms28 May 2024
- 2023Learning Interpretable Deep Disentangled Neural Networks for Hyperspectral Unmixing3 Oct 2023
- 2023Uncertainty quantification for learned ISTA14 Sep 2023
- 2023Active Inference in Hebbian Learning Networks8 Jun 2023
- 2023Probabilistic Unrolling: Scalable, Inverse-Free Maximum Likelihood Estimation for Latent Gaussian Models5 Jun 2023
- 2023Towards Constituting Mathematical Structures for Learning to Optimize29 May 2023
12 newest of 29 All 29 (JSON)