7 天前

范数-范数损失在图像质量评估中的应用:更快的收敛速度与更优的性能

Dingquan Li, Tingting Jiang, Ming Jiang
范数-范数损失在图像质量评估中的应用:更快的收敛速度与更优的性能
摘要

目前,大多数图像质量评估(IQA)模型均采用MAE或MSE损失函数进行监督训练,但这类方法通常存在收敛速度较慢的问题。众所周知,归一化有助于加速收敛过程。因此,本文探索在IQA损失函数设计中引入归一化机制。具体而言,我们首先对预测的质量得分及其对应的主观质量得分进行归一化处理,随后基于归一化后数值之间的差异范数定义损失函数。由此提出的“归一化之归一化”(Norm-in-Norm)损失函数,能够促使IQA模型对主观质量得分做出线性预测。训练完成后,通过最小二乘回归确定从预测质量到主观质量的线性映射关系。研究表明,该新型损失函数与两种常见的IQA性能评估指标(PLCC和RMSE)具有紧密关联。通过理论分析进一步证明,嵌入的归一化机制可使损失函数的梯度更加稳定且更具可预测性,从而显著促进IQA模型的快速收敛。为进一步验证所提损失函数的有效性,我们将其应用于一个具有挑战性的实际问题:真实场景图像(in-the-wild)的质量评估。在两个相关数据集(KonIQ-10k和CLIVE)上的实验结果表明,相较于传统的MAE或MSE损失,新损失函数可使IQA模型的收敛速度提升约10倍,且最终模型的性能更优。此外,所提出的模型在该挑战性任务上达到了当前最先进的预测性能。为支持可复现的科学研究,相关代码已公开发布于:https://github.com/lidq92/LinearityIQA。

范数-范数损失在图像质量评估中的应用:更快的收敛速度与更优的性能 | 最新论文 | HyperAI超神经