Skip to main content
QUICK REVIEW

[论文解读] Semi-Supervised Deep Ensembles for Blind Image Quality Assessment

Zhihua Wang, Dingquan Li|arXiv (Cornell University)|Jun 26, 2021
Image and Video Quality Assessment参考文献 30被引用 7
一句话总结

该论文提出了一种用于盲图像质量评估(BIQA)的半监督深度集成方法,通过保真度损失联合优化在有标签数据上的集成准确率和在无标签数据上的多样性。该方法在真实场景中实现了优越的泛化能力,并能有效识别模型失败情况,其在泛化能力和主动学习效率方面优于SOTA模型UNIQUE。

ABSTRACT

Ensemble methods are generally regarded to be better than a single model if the base learners are deemed to be "accurate" and "diverse." Here we investigate a semi-supervised ensemble learning strategy to produce generalizable blind image quality assessment models. We train a multi-head convolutional network for quality prediction by maximizing the accuracy of the ensemble (as well as the base learners) on labeled data, and the disagreement (i.e., diversity) among them on unlabeled data, both implemented by the fidelity loss. We conduct extensive experiments to demonstrate the advantages of employing unlabeled data for BIQA, especially in model generalization and failure identification.

研究动机与目标

  • 为解决由于标注数据有限和模型容量过大导致的深度BIQA模型泛化能力差的问题。
  • 利用丰富的无标签图像数据提升盲图像质量评估中模型的鲁棒性与多样性。
  • 开发一种半监督集成框架,同时最大化预测准确率与学习器间的分歧度。
  • 实现内置的失败识别功能,以支持后续主动学习,提升模型优化效率。
  • 证明通过无标签数据促进多样性可超越标准相关性度量,进一步提升泛化能力。

提出的方法

  • 采用多头卷积网络,其中每个头作为基学习器,共享早期特征但具有独立的后期层用于质量预测。
  • 集成输出为各基学习器预测结果的平均值,通过特征归一化与尺度对齐技术确保输出尺度的一致性。
  • 采用保真度损失联合优化两个目标:在有标签成对排序数据上的准确率与在无标签数据上的多样性。
  • 通过无标签样本上基学习器之间的分歧损失实现多样性目标,促进结构多样性而不损失准确率。
  • 使用权衡参数γ平衡准确率与多样性,消融实验表明在低γ值时性能最优。
  • 消融研究探讨了基学习器数量、特征共享的分割点以及权衡参数γ的影响。

实验结果

研究问题

  • RQ1通过半监督学习引入无标签数据是否能提升深度BIQA模型的泛化能力?
  • RQ2在集成中提升基学习器间的多样性是否能带来更好的失败检测与模型鲁棒性?
  • RQ3集成准确率与多样性之间的权衡如何影响在标准与分布外IQA基准上的性能表现?
  • RQ4集成成员间的分歧是否能有效识别失败案例以支持主动学习?
  • RQ5半监督BIQA集成中基学习器、特征共享与超参数的最优配置是什么?

主要发现

  • 该半监督集成在标准数据集(KonIQ-10k、SPAQ、FLIVE)上的SRCC与PLCC与UNIQUE相当或略低,但在gMAD竞赛中显著优于UNIQUE,表明其具备更优的泛化能力。
  • 在FLIVE数据集上,该集成在所有样本规模下均实现了最低的失败检测相关性(0.427 SRCC),证明其在识别失败案例方面具有更高的效率。
  • 在不同基学习器数量(M ∈ {2,4,6,8,12})下性能稳定,且随着M增加略有提升,表明对这一超参数具有鲁棒性。
  • 最优权衡参数γ值较小(如0.04),因为较大的γ值会因过度优先考虑多样性而损害准确率。
  • 从早期层(如首个卷积层或残差块后)共享计算可实现稳定性能,从而提升推理速度并降低内存消耗。
  • 集成成员间分歧最大的样本在主动学习中极为有效,表明其在半监督设置中具有强大的主动微调潜力。

更好的研究,从现在开始

从阅读论文到最终审阅,大幅缩短您的研究时间。

无需绑定信用卡

本解读由 AI 生成,并经人工编辑审核。