DiffStereo: High-Frequency Aware Diffusion Model for Stereo Image Restoration

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Cao, Huiyun, Shi, Yuan, Xia, Bin, Jin, Xiaoyu, Yang, Wenming
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913654959505408
author Cao, Huiyun
Shi, Yuan
Xia, Bin
Jin, Xiaoyu
Yang, Wenming
author_facet Cao, Huiyun
Shi, Yuan
Xia, Bin
Jin, Xiaoyu
Yang, Wenming
contents Diffusion models (DMs) have achieved promising performance in image restoration but haven't been explored for stereo images. The application of DM in stereo image restoration is confronted with a series of challenges. The need to reconstruct two images exacerbates DM's computational cost. Additionally, existing latent DMs usually focus on semantic information and remove high-frequency details as redundancy during latent compression, which is precisely what matters for image restoration. To address the above problems, we propose a high-frequency aware diffusion model, DiffStereo for stereo image restoration as the first attempt at DM in this domain. Specifically, DiffStereo first learns latent high-frequency representations (LHFR) of HQ images. DM is then trained in the learned space to estimate LHFR for stereo images, which are fused into a transformer-based stereo image restoration network providing beneficial high-frequency information of corresponding HQ images. The resolution of LHFR is kept the same as input images, which preserves the inherent texture from distortion. And the compression in channels alleviates the computational burden of DM. Furthermore, we devise a position encoding scheme when integrating the LHFR into the restoration network, enabling distinctive guidance in different depths of the restoration network. Comprehensive experiments verify that by combining generative DM and transformer, DiffStereo achieves both higher reconstruction accuracy and better perceptual quality on stereo super-resolution, deblurring, and low-light enhancement compared with state-of-the-art methods.
format Preprint
id arxiv_https___arxiv_org_abs_2501_10325
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle DiffStereo: High-Frequency Aware Diffusion Model for Stereo Image Restoration
Cao, Huiyun
Shi, Yuan
Xia, Bin
Jin, Xiaoyu
Yang, Wenming
Computer Vision and Pattern Recognition
Diffusion models (DMs) have achieved promising performance in image restoration but haven't been explored for stereo images. The application of DM in stereo image restoration is confronted with a series of challenges. The need to reconstruct two images exacerbates DM's computational cost. Additionally, existing latent DMs usually focus on semantic information and remove high-frequency details as redundancy during latent compression, which is precisely what matters for image restoration. To address the above problems, we propose a high-frequency aware diffusion model, DiffStereo for stereo image restoration as the first attempt at DM in this domain. Specifically, DiffStereo first learns latent high-frequency representations (LHFR) of HQ images. DM is then trained in the learned space to estimate LHFR for stereo images, which are fused into a transformer-based stereo image restoration network providing beneficial high-frequency information of corresponding HQ images. The resolution of LHFR is kept the same as input images, which preserves the inherent texture from distortion. And the compression in channels alleviates the computational burden of DM. Furthermore, we devise a position encoding scheme when integrating the LHFR into the restoration network, enabling distinctive guidance in different depths of the restoration network. Comprehensive experiments verify that by combining generative DM and transformer, DiffStereo achieves both higher reconstruction accuracy and better perceptual quality on stereo super-resolution, deblurring, and low-light enhancement compared with state-of-the-art methods.
title DiffStereo: High-Frequency Aware Diffusion Model for Stereo Image Restoration
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2501.10325