AAAI 2022 Oral | 无需人工标注,清华、快手提出基于参考图像的单张生成图像质量评价方法

简介: AAAI 2022 Oral | 无需人工标注,清华、快手提出基于参考图像的单张生成图像质量评价方法
这是一篇来 自清华大学黄高团队和快手 Y-tech 团队合作的论文,该工作探究了如何在基于参考图像的生成任务中实现对于单张生成图像质量的评价。 文中设计的 RISA 模型无需人工标注的训练数据,其评价结果能够与人的主观感受具有高度一致性。 本工作已入选 AAAI 2022 Oral。


引言

现有的生成图像评价工作主要基于生成图像的分布对模型「整体」的生成效果进行评价。然而,一个性能优异的生成模型并不代表其合成的「任何一张」图像都具有高质量的效果。在基于参考图像(reference image)的生成任务中,譬如将用户上传的风景照渲染成某种指定的风格的业务场景中,能够对于「单张」生成图像的质量进行评价,对于提高用户的使用体验是至关重要的。


该研究提出了基于参考图像的单张生成图像质量评价方法 Reference-guided Image Synthesis Assessment(RISA)。

RISA 的贡献和创新点可以总结为以下几个方面:

RISA 的训练图像来自于 GAN 训练过程的中间模型生成的图像,图像的质量标签来自于模型的迭代轮数,无需人工标注,理论上可用于训练的数据无上限。

由于以模型的迭代轮数作为标注不够精细,采用了 pixel-wise interpolation 和 mutiple binary classifiers 的方法来增强训练的稳定性。

引入了无监督的对比学习损失,学习参考图像和生成图像之间的风格相似度。

论文链接:https://arxiv.org/pdf/2112.04163.pdf


实现策略
RISA 的整体框架十分简洁,参考图像和生成图像经过参数共享的风格提取器得到相应的特征向量,接着计算两特征向量的 L1 距离并输入到 mutiple binary classifiers 中得到预测向量,最后预测向量元素取平均得到最终的质量分数。


RISA 的训练数据来自一系列 GAN 训练过程中的中间模型的生成图像,以下图中给出的一性别转换任务为例,可以看到,在 GAN 的训练早期,模型随着训练迭代轮数的增加,生成图像的质量会有显著的提升;而在训练后期,模型的生成图像的质量会趋于稳定。


本文采用一系列中间模型的生成图像作为 RISA 的训练数据,这些图像的样本标签由其对应模型的训练迭代轮数得到。但显然这样的标注形式不太适合训练后期的模型,因为训练后期生成图像质量不会有显著的变化。为了使训练数据更适合 RISA 的训练,文中采用了 pixel-wise interpolation 的技巧,即图像空间的线性插值,用于估计训练后期图像质量变化。

如下图所示,理想情况下,生成图像随着 GAN 的训练轮数的增加单调变好,但实际上对于简单的任务,训练后期生成图像的质量几乎没有变化;对于困难的任务,训练后期生成图像的质量随着训练轮数的增加呈现震荡变好的趋势。因此文中选取了 FID 曲线变化的肘点作为 GAN 的训练前期和后期的分界,对于训练前期直接采样中间模型生成图像,并用迭代轮数作为图像质量标签;对于训练后期,选取开始和最终的两个模型生成具有明显质量差异的图像,再对图像进行线性插值得到一系列中间质量的图像。


插值图像的一些 demo 如下动图所示,图中所示 epsilon 表示两幅图融合时的权重。


除图像空间的插值外,为了保证 RISA 的训练稳定,RISA 的预测使用的是个二值分类器(multiple binary classifers)输出取平均的形式,而没有采用简单的回归器输出拟合值。其中第个二值分类器用于预测当前生成图像质量大于一个特定阈值的概率。实验表明,将质量评估从回归问题转化为分类问题,能够显著地提升 RISA 的性能。

损失函数的设计上考虑了三个方面:1)弱监督损失,用于拟合输入参考图像 & 生成图像对及其对应的质量标签;2)无监督对比学习损失,用于捕捉参考图像和生成图像风格相似度;3)上界损失,用于学到来自真实图像的两个增强图像的风格一致性。

上界损失表达的是和风格信息完全一致,将其输入 RISA 预测结果应当对应于最高的质量分数 1。

在对比学习损失中,文中首先考虑对于参考图像做两次不同的且不破坏图像风格信息的数据增强图像和,即仅包括图像的放缩,裁剪以及翻转。生成图像与构成正样本对,对比学习损失拉近它们的预测输出;同一批输入样本中,与其对应的参考图像,与其不对应的参考图像构成负样本对,对比学习损失拉大它们的预测输出。


实验结果
文中基于四种生成模型,五个数据集上的生成图像分别训练多个 RISA 模型。首先从可视化的角度,下图说明 RISA 能够按照质量从低到高给出对应的质量评价分数。


接着从量化指标的角度,文中进行了广泛的人工评价测试,以说明 RISA 评价结果和人的主观感受具有较高的一致性。具体来说,对于每个任务都选取了上千个三元组样本,包含一张参考图像和两张生成图像。两张生成图像可能来自于同一架构模型的两个不同训练阶段的中间模型,也可能来自于两个充分收敛的不同架构的模型。测试者被要求从二者中选出质量更好的一张。最终对于每个任务,保证了每组样本都有至少三个测试者参与评价,而所有评价均一致的样本被保留,用于评估 RISA 的评价与人的主观感受的一致性。

下表对应于 RISA 的训练数据和测试数据均由相同架构的模型生成的情况。可以看到 RISA 的评价结构能够与人的主观感受具有更高的一致性,且优于现有的主流的有参考和无参考单张图像质量评价方法。


下表对应于 RISA 的训练数据和测试数据均由不同架构的模型生成的情况。表中结果进一步说明 RISA 具有较好的在不同模型之间迁移的能力。


相应地,研究者提供了三元组上的 RISA 和每个数据集下最优的基线方法的可视化对比。可以看到 RISA 能够在考虑生成图像的真实程度的同时,兼具评价生成图像和参考图像的风格相似度水平的能力。


最后研究者进行了两组消融实验,说明了 RISA 引入 multiple binary classifers,pixel-wise interpolation 和其每个损失项的意义。

相关文章
|
安全 关系型数据库 MySQL
Sqlmap2021 -- Cookie注入
Sqlmap2021 -- Cookie注入
919 1
|
存储 SQL 关系型数据库
mysql中 如何设置字段不限长度
在MySQL中,可以使用数据类型TEXT来存储不定长的文本数据。如果不需要限制文本的最大长度,可以使用TEXT数据类型而不指定长度。 例如,下面的SQL语句创建一个名为my_table的表,其中包含一个text_column字段,该字段不限长度:
2114 0
|
5月前
|
消息中间件 缓存 JSON
http 调用优化 的11大绝招 (图解+秒懂+史上最全)
http 调用优化 的11大绝招 (图解+秒懂+史上最全)
|
机器学习/深度学习 算法
扩散模型=进化算法!生物学大佬用数学揭示本质
在机器学习与生物学交叉领域,Tufts和Harvard大学研究人员揭示了扩散模型与进化算法的深刻联系。研究表明,扩散模型本质上是一种进化算法,通过逐步去噪生成数据点,类似于进化中的变异和选择机制。这一发现不仅在理论上具有重要意义,还提出了扩散进化方法,能够高效识别多解、处理高维复杂参数空间,并显著减少计算步骤,为图像生成、视频合成及神经网络优化等应用带来广泛潜力。论文地址:https://arxiv.org/pdf/2410.02543。
412 21
|
机器学习/深度学习 编解码 人工智能
InvSR:开源图像超分辨率生成模型,提升分辨率,修复老旧照片为超清图像
InvSR 是一个创新的图像超分辨率模型,基于扩散模型的逆过程恢复高分辨率图像。它通过深度噪声预测器和灵活的采样机制,能够高效地提升图像分辨率,适用于老旧照片修复、视频监控、医疗成像等多个领域。
2904 9
InvSR:开源图像超分辨率生成模型,提升分辨率,修复老旧照片为超清图像
|
存储 缓存 API
后端的技术方案怎么写?
本文讨论了编写方案的重要性,旨在帮助我们清晰理解需求并确保有效落地。方案应记录实现初衷,方便后续迭代,避免因缺乏文档导致的代码理解困难。通过方案评审,可以考虑通用性和复用性,减少重复工作。文章还强调了在新系统或已有系统上添加功能时需考虑的问题,如系统复杂度、耦合度、API依赖、兼容性、性能和数据处理。此外,方案应涵盖需求背景、系统架构设计(包括业务架构、状态机、存储、服务应用和服务治理)、上线计划和灰度发布策略。最后,文章提到了性能设计、复用性和中台建设的思考,指出设计应以满足实际需求和降低成本为目标。
736 0
|
数据采集 机器学习/深度学习 安全
疫苗效果分析:如何使用疫苗预防率等指标来评估疫苗的预防效果
疫苗效果分析:如何使用疫苗预防率等指标来评估疫苗的预防效果
2738 0
|
Oracle 关系型数据库 MySQL
Centos7下图形化部署单点KFS同步工具并将Oracle增量同步到KES
Centos7下图形化部署单点KFS同步工具并将Oracle增量同步到KES
Centos7下图形化部署单点KFS同步工具并将Oracle增量同步到KES
|
监控 安全 数据处理
淘宝 1688 跨境电商官方接口接入全攻略,跨境卖家必知
本攻略详述了接入1688跨境电商官方接口的全过程,涵盖注册申请、开发调试、数据处理与业务集成、安全合规及上线维护等环节,帮助开发者高效对接1688,拓展跨境业务。
1482 0

热门文章

最新文章