Perceptual-Distortion Balanced Image Super-Resolution is a Multi-Objective Optimization Problem

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Zhu, Qiwen, Wang, Yanjie, Cai, Shilv, Chen, Liqun, Zhou, Jiahuan, Yan, Luxin, Zhong, Sheng, Zou, Xu
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866913493735702528
author Zhu, Qiwen
Wang, Yanjie
Cai, Shilv
Chen, Liqun
Zhou, Jiahuan
Yan, Luxin
Zhong, Sheng
Zou, Xu
author_facet Zhu, Qiwen
Wang, Yanjie
Cai, Shilv
Chen, Liqun
Zhou, Jiahuan
Yan, Luxin
Zhong, Sheng
Zou, Xu
contents Training Single-Image Super-Resolution (SISR) models using pixel-based regression losses can achieve high distortion metrics scores (e.g., PSNR and SSIM), but often results in blurry images due to insufficient recovery of high-frequency details. Conversely, using GAN or perceptual losses can produce sharp images with high perceptual metric scores (e.g., LPIPS), but may introduce artifacts and incorrect textures. Balancing these two types of losses can help achieve a trade-off between distortion and perception, but the challenge lies in tuning the loss function weights. To address this issue, we propose a novel method that incorporates Multi-Objective Optimization (MOO) into the training process of SISR models to balance perceptual quality and distortion. We conceptualize the relationship between loss weights and image quality assessment (IQA) metrics as black-box objective functions to be optimized within our Multi-Objective Bayesian Optimization Super-Resolution (MOBOSR) framework. This approach automates the hyperparameter tuning process, reduces overall computational cost, and enables the use of numerous loss functions simultaneously. Extensive experiments demonstrate that MOBOSR outperforms state-of-the-art methods in terms of both perceptual quality and distortion, significantly advancing the perception-distortion Pareto frontier. Our work points towards a new direction for future research on balancing perceptual quality and fidelity in nearly all image restoration tasks. The source code and pretrained models are available at: https://github.com/ZhuKeven/MOBOSR.
format Preprint
id arxiv_https___arxiv_org_abs_2409_03179
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Perceptual-Distortion Balanced Image Super-Resolution is a Multi-Objective Optimization Problem
Zhu, Qiwen
Wang, Yanjie
Cai, Shilv
Chen, Liqun
Zhou, Jiahuan
Yan, Luxin
Zhong, Sheng
Zou, Xu
Image and Video Processing
Computer Vision and Pattern Recognition
Training Single-Image Super-Resolution (SISR) models using pixel-based regression losses can achieve high distortion metrics scores (e.g., PSNR and SSIM), but often results in blurry images due to insufficient recovery of high-frequency details. Conversely, using GAN or perceptual losses can produce sharp images with high perceptual metric scores (e.g., LPIPS), but may introduce artifacts and incorrect textures. Balancing these two types of losses can help achieve a trade-off between distortion and perception, but the challenge lies in tuning the loss function weights. To address this issue, we propose a novel method that incorporates Multi-Objective Optimization (MOO) into the training process of SISR models to balance perceptual quality and distortion. We conceptualize the relationship between loss weights and image quality assessment (IQA) metrics as black-box objective functions to be optimized within our Multi-Objective Bayesian Optimization Super-Resolution (MOBOSR) framework. This approach automates the hyperparameter tuning process, reduces overall computational cost, and enables the use of numerous loss functions simultaneously. Extensive experiments demonstrate that MOBOSR outperforms state-of-the-art methods in terms of both perceptual quality and distortion, significantly advancing the perception-distortion Pareto frontier. Our work points towards a new direction for future research on balancing perceptual quality and fidelity in nearly all image restoration tasks. The source code and pretrained models are available at: https://github.com/ZhuKeven/MOBOSR.
title Perceptual-Distortion Balanced Image Super-Resolution is a Multi-Objective Optimization Problem
topic Image and Video Processing
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2409.03179