Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

DP-IQA: Utilizing Diffusion Prior for Blind Image Quality Assessment in the Wild

About

Blind image quality assessment (IQA) in the wild, which assesses the quality of images with complex authentic distortions and no reference images, presents significant challenges. Given the difficulty in collecting large-scale training data, leveraging limited data to develop a model with strong generalization remains an open problem. Motivated by the robust image perception capabilities of pre-trained text-to-image (T2I) diffusion models, we propose a novel IQA method, diffusion priors-based IQA (DP-IQA), to utilize the T2I model's prior for improved performance and generalization ability. Specifically, we utilize pre-trained Stable Diffusion as the backbone, extracting multi-level features from the denoising U-Net guided by prompt embeddings through a tunable text adapter. Simultaneously, an image adapter compensates for information loss introduced by the lossy pre-trained encoder. Unlike T2I models that require full image distribution modeling, our approach targets image quality assessment, which inherently requires fewer parameters. To improve applicability, we distill the knowledge into a lightweight CNN-based student model, significantly reducing parameters while maintaining or even enhancing generalization performance. Experimental results demonstrate that DP-IQA achieves state-of-the-art performance on various in-the-wild datasets, highlighting the superior generalization capability of T2I priors in blind IQA tasks. To our knowledge, DP-IQA is the first method to apply pre-trained diffusion priors in blind IQA. Codes and checkpoints are available at https://github.com/RomGai/DP-IQA.

Honghao Fu, Yufei Wang, Wenhan Yang, Alex C. Kot, Bihan Wen• 2024

Related benchmarks

TaskDatasetResultRank
Image Quality AssessmentSPAQ
SRCC0.923
250
Image Quality AssessmentKonIQ
SRCC0.942
119
Blind Image Quality AssessmentFLIVE
SRCC0.579
115
No-Reference Image Quality AssessmentKonIQ-10k
SROCC0.942
111
No-Reference Image Quality AssessmentSPAQ
SROCC0.923
92
Image Quality AssessmentCLIVE
SRCC0.893
54
Image Quality AssessmentKonIQ (test)
SROCC0.781
53
No-Reference Image Quality AssessmentKonIQ-10k (test)
SRCC0.781
35
No-Reference Image Quality AssessmentCLIVE (test)
SROCC0.833
34
No-Reference Image Quality AssessmentCLIVE
SRCC0.893
21
Showing 10 of 13 rows

Other info

Follow for update