Skip to content

[ACM MM 2024] QPT V2: An MIM-based pretraining framework for IQA, VQA, and IAA.

Notifications You must be signed in to change notification settings

KeiChiTse/QPT-V2

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 

Repository files navigation

QPT V2: Masked Image Modeling Advances Visual Scoring

Arxiv   

Qizhi Xie1,2 | Kun Yuan2 | Yunpeng Qu1,2 | Mingda Wu2 | Ming Sun2 | Chao Zhou2 | Jihong Zhu1

1Tsinghua University, 2Kuaishou Technology.

Overview

Quality assessment and aesthetics assessment aim to evaluate the perceived quality and aesthetics of visual content. Current learning-based methods suffer greatly from the scarcity of labeled data and usually perform sub-optimally in terms of generalization. Although masked image modeling (MIM) has achieved noteworthy advancements across various high-level tasks (e.g., classification, detection etc.). In this work, we take on a novel perspective to investigate its capabilities in terms of quality- and aesthetics-awareness. To this end, we propose Quality- and aesthetics-aware PreTraining (QPT V2), the first pretraining framework based on MIM that offers a unified solution to quality and aesthetics assessment. To perceive the high-level semantics and fine-grained details, pretraining data is curated. To comprehensively encompass quality- and aesthetics-related factors, degradation is introduced. To capture multi-scale quality and aesthetic information, model structure is modified. Extensive experimental results on 11 downstream benchmarks clearly show the superior performance of QPT V2 in comparison with current state-of-the-art approaches and other pretraining paradigms.

Updates

Code and model weights will be available soon!

About

[ACM MM 2024] QPT V2: An MIM-based pretraining framework for IQA, VQA, and IAA.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published