Noise-Free Optimization in Early Training Steps for Image Super-resolution

Authors

  • MinKyu Lee Sungkyunkwan University
  • Jae-Pil Heo Sungkyunkwan University

DOI:

https://doi.org/10.1609/aaai.v38i4.28073

Keywords:

CV: Low Level & Physics-based Vision, CV: Applications, CV: Computational Photography, Image & Video Synthesis, CV: Learning & Optimization for CV

Abstract

Recent deep-learning-based single image super-resolution (SISR) methods have shown impressive performance whereas typical methods train their networks by minimizing the pixel-wise distance with respect to a given high-resolution (HR) image. However, despite the basic training scheme being the predominant choice, its use in the context of ill-posed inverse problems has not been thoroughly investigated. In this work, we aim to provide a better comprehension of the underlying constituent by decomposing target HR images into two subcomponents: (1) the optimal centroid which is the expectation over multiple potential HR images, and (2) the inherent noise defined as the residual between the HR image and the centroid. Our findings show that the current training scheme cannot capture the ill-posed nature of SISR and becomes vulnerable to the inherent noise term, especially during early training steps. To tackle this issue, we propose a novel optimization method that can effectively remove the inherent noise term in the early steps of vanilla training by estimating the optimal centroid and directly optimizing toward the estimation. Experimental results show that the proposed method can effectively enhance the stability of vanilla training, leading to overall performance gain. Codes are available at github.com/2minkyulee/ECO.

Published

2024-03-24

How to Cite

Lee, M., & Heo, J.-P. (2024). Noise-Free Optimization in Early Training Steps for Image Super-resolution. Proceedings of the AAAI Conference on Artificial Intelligence, 38(4), 2920-2928. https://doi.org/10.1609/aaai.v38i4.28073

Issue

Section

AAAI Technical Track on Computer Vision III