Latent Posterior-Mean Rectified Flow for Higher-Fidelity Perceptual Face Restoration

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Luo, Xin, Zhang, Menglin, Lan, Yunwei, Zhang, Tianyu, Li, Rui, Liu, Chang, Liu, Dong
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866915366887751680
author Luo, Xin
Zhang, Menglin
Lan, Yunwei
Zhang, Tianyu
Li, Rui
Liu, Chang
Liu, Dong
author_facet Luo, Xin
Zhang, Menglin
Lan, Yunwei
Zhang, Tianyu
Li, Rui
Liu, Chang
Liu, Dong
contents The Perception-Distortion tradeoff (PD-tradeoff) theory suggests that face restoration algorithms must balance perceptual quality and fidelity. To achieve minimal distortion while maintaining perfect perceptual quality, Posterior-Mean Rectified Flow (PMRF) proposes a flow based approach where source distribution is minimum distortion estimations. Although PMRF is shown to be effective, its pixel-space modeling approach limits its ability to align with human perception, where human perception is defined as how humans distinguish between two image distributions. In this work, we propose Latent-PMRF, which reformulates PMRF in the latent space of a variational autoencoder (VAE), facilitating better alignment with human perception during optimization. By defining the source distribution on latent representations of minimum distortion estimation, we bound the minimum distortion by the VAE's reconstruction error. Moreover, we reveal the design of VAE is crucial, and our proposed VAE significantly outperforms existing VAEs in both reconstruction and restoration. Extensive experiments on blind face restoration demonstrate the superiority of Latent-PMRF, offering an improved PD-tradeoff compared to existing methods, along with remarkable convergence efficiency, achieving a 5.79X speedup over PMRF in terms of FID. Our code will be available as open-source.
format Preprint
id arxiv_https___arxiv_org_abs_2507_00447
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Latent Posterior-Mean Rectified Flow for Higher-Fidelity Perceptual Face Restoration
Luo, Xin
Zhang, Menglin
Lan, Yunwei
Zhang, Tianyu
Li, Rui
Liu, Chang
Liu, Dong
Computer Vision and Pattern Recognition
Image and Video Processing
The Perception-Distortion tradeoff (PD-tradeoff) theory suggests that face restoration algorithms must balance perceptual quality and fidelity. To achieve minimal distortion while maintaining perfect perceptual quality, Posterior-Mean Rectified Flow (PMRF) proposes a flow based approach where source distribution is minimum distortion estimations. Although PMRF is shown to be effective, its pixel-space modeling approach limits its ability to align with human perception, where human perception is defined as how humans distinguish between two image distributions. In this work, we propose Latent-PMRF, which reformulates PMRF in the latent space of a variational autoencoder (VAE), facilitating better alignment with human perception during optimization. By defining the source distribution on latent representations of minimum distortion estimation, we bound the minimum distortion by the VAE's reconstruction error. Moreover, we reveal the design of VAE is crucial, and our proposed VAE significantly outperforms existing VAEs in both reconstruction and restoration. Extensive experiments on blind face restoration demonstrate the superiority of Latent-PMRF, offering an improved PD-tradeoff compared to existing methods, along with remarkable convergence efficiency, achieving a 5.79X speedup over PMRF in terms of FID. Our code will be available as open-source.
title Latent Posterior-Mean Rectified Flow for Higher-Fidelity Perceptual Face Restoration
topic Computer Vision and Pattern Recognition
Image and Video Processing
url https://arxiv.org/abs/2507.00447