Comparison between Supervised and Unsupervised Learning in Deep Unfolded Sparse Signal Recovery

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Nagahisa, Koshi, Hayakawa, Ryo, Iiguni, Youji
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917430676160512
author Nagahisa, Koshi
Hayakawa, Ryo
Iiguni, Youji
author_facet Nagahisa, Koshi
Hayakawa, Ryo
Iiguni, Youji
contents This paper investigates the impact of loss function selection in deep unfolding techniques for sparse signal recovery algorithms. Deep unfolding transforms iterative optimization algorithms into trainable lightweight neural networks by unfolding their iterations as network layers, with various loss functions employed for parameter learning depending on application contexts. We focus on deep unfolded versions of the fundamental iterative shrinkage thresholding algorithm (ISTA) and the iterative hard thresholding algorithm (IHT), comparing supervised learning using mean squared error with unsupervised learning using the objective function of the original optimization problem. Our simulation results reveal that the effect of the choice of loss function significantly depends on the convexity of the optimization problem. For convex $\ell_1$-regularized problems, supervised-ISTA achieves better final recovery accuracy but fails to minimize the original objective function, whereas we empirically observe that unsupervised-ISTA converges to a nearly identical solution as conventional ISTA but with accelerated convergence. Conversely, for nonconvex $\ell_0$-regularized problems, both supervised-IHT and unsupervised-IHT converge to better local minima than the original IHT, showing similar performance under the training conditions regardless of the loss function employed. However, when the test conditions differ from the training conditions, unsupervised-IHT generalizes well whereas supervised-IHT tends to suffer from performance degradation, suggesting that unsupervised learning offers better robustness to distribution mismatch. These findings provide valuable insights into the design of effective deep unfolded networks for sparse signal recovery applications.
format Preprint
id arxiv_https___arxiv_org_abs_2509_01331
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Comparison between Supervised and Unsupervised Learning in Deep Unfolded Sparse Signal Recovery
Nagahisa, Koshi
Hayakawa, Ryo
Iiguni, Youji
Signal Processing
This paper investigates the impact of loss function selection in deep unfolding techniques for sparse signal recovery algorithms. Deep unfolding transforms iterative optimization algorithms into trainable lightweight neural networks by unfolding their iterations as network layers, with various loss functions employed for parameter learning depending on application contexts. We focus on deep unfolded versions of the fundamental iterative shrinkage thresholding algorithm (ISTA) and the iterative hard thresholding algorithm (IHT), comparing supervised learning using mean squared error with unsupervised learning using the objective function of the original optimization problem. Our simulation results reveal that the effect of the choice of loss function significantly depends on the convexity of the optimization problem. For convex $\ell_1$-regularized problems, supervised-ISTA achieves better final recovery accuracy but fails to minimize the original objective function, whereas we empirically observe that unsupervised-ISTA converges to a nearly identical solution as conventional ISTA but with accelerated convergence. Conversely, for nonconvex $\ell_0$-regularized problems, both supervised-IHT and unsupervised-IHT converge to better local minima than the original IHT, showing similar performance under the training conditions regardless of the loss function employed. However, when the test conditions differ from the training conditions, unsupervised-IHT generalizes well whereas supervised-IHT tends to suffer from performance degradation, suggesting that unsupervised learning offers better robustness to distribution mismatch. These findings provide valuable insights into the design of effective deep unfolded networks for sparse signal recovery applications.
title Comparison between Supervised and Unsupervised Learning in Deep Unfolded Sparse Signal Recovery
topic Signal Processing
url https://arxiv.org/abs/2509.01331