Learning a Convex Patch-Based Synthesis Model via Deep Equilibrium

S. Ducotterd, S. Neumayer, M. Unser

Proceedings of the Forty-Ninth IEEE International Conference on Acoustics, Speech, and Signal Processing (ICASSP'24), Seoul Special City, Republic of Korea, April 14-19, 2024, pp. 2031–2035.

Download PDFDownload PSDownload Global BibTeX

We investigate the learning of a convex patch-based synthesis model for the reconstruction of images. In essence, we propose to learn a dictionary via bilevel optimization for denoising. Using implicit differentiation, we find a closed-form formula of the derivative of the minimizer of an objective with respect to the atoms of the dictionary. We also propose a novel way to handle the mean of each patch of the predictions, which improves our model when it is applied to other inverse problems. For minimizing the objective involving the learnt dictionary, we propose an early stopping criterion to further improve the performance of the model for denoising. Finally, we assess our model in a compressed sensing MRI inverse problem and show that, despite being trained on denoising only, our model yields good reconstruction performances.

@INPROCEEDINGS(http://bigwww.epfl.ch/publications/ducotterd2402.html,
AUTHOR="Ducotterd, S. and Neumayer, S. and Unser, M.",
TITLE="Learning a Convex Patch-Based Synthesis Model via Deep
	Equilibrium",
BOOKTITLE="Proceedings of the Forty-Ninth IEEE International Conference
	on Acoustics, Speech, and Signal Processing ({ICASSP'24})",
YEAR="2024",
editor="",
volume="",
series="",
pages="2031--2035",
address="Seoul Special City, Republic of Korea",
month="April 14-19,",
organization="",
publisher="",
note="")

DOI: 10.1109/ICASSP48485.2024.10446773