arXiv:2409.05540cs.CVcs.MM2024-09被引 105

利用评分分布信息提升真实场景图像质量评估效果

Exploring Rich Subjective Quality Information for Image Quality Assessment in the Wild

论文配图:Exploring Rich Subjective Quality Information for Image Quality Assessment in the Wild
图 1 · 摘自论文原文
  • 设计三阶段网络模拟人类记忆机制,融合CvT特征提取能力
  • 采用多标签训练,同时学习均值、标准差与评分分布
  • 在多个大规模数据集上超越现有方法,适合高质量评估场景

传统真实场景图像质量评估(IQA)模型通常仅使用平均意见分(MOS)作为标签,忽略了评分中蕴含的丰富主观信息,例如意见分数的标准差(SOS)或分布(DOS)。本文提出一种新方法 RichIQA,旨在挖掘 MOS 之外的主观评分信息以预测图像质量。RichIQA 的核心创新包括:(1)基于卷积视觉变压器(CvT)的三阶段网络,模拟人类短期与长期记忆机制;(2)多标签训练策略,同步利用 MOS、SOS 及 DOS 进行联合训练。该方法可输出图像质量的分布预测,进而获得均值。大量实验表明,三阶段网络能有效建模复杂质量信息,多标签训练显著释放主观评分潜力,提升模型性能与泛化能力。RichIQA 在多个具备丰富主观评分标签的大规模真实场景 IQA 数据库上超越当前最优方法。代码将公开于 GitHub。

原文摘要 · Abstract (English)

Traditional in the wild image quality assessment (IQA) models are generally trained with the quality labels of mean opinion score (MOS), while missing the rich subjective quality information contained in the quality ratings, for example, the standard deviation of opinion scores (SOS) or even distribution of opinion scores (DOS). In this paper, we propose a novel IQA method named RichIQA to explore the rich subjective rating information beyond MOS to predict image quality in the wild. RichIQA is characterized by two key novel designs: (1) a three-stage image quality prediction network which exploits the powerful feature representation capability of the Convolutional vision Transformer (CvT) and mimics the short-term and long-term memory mechanisms of human brain; (2) a multi-label training strategy in which rich subjective quality information like MOS, SOS and DOS are concurrently used to train the quality prediction network. Powered by these two novel designs, RichIQA is able to predict the image quality in terms of a distribution, from which the mean image quality can be subsequently obtained. Extensive experimental results verify that the three-stage network is tailored to predict rich quality information, while the multi-label training strategy can fully exploit the potentials within subjective quality rating and enhance the prediction performance and generalizability of the network. RichIQA outperforms state-of-the-art competitors on multiple large-scale in the wild IQA databases with rich subjective rating labels. The code of RichIQA will be made publicly available on GitHub.

图像质量评估主观评分多标签学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。