CVPR 2024 | 和马赛克说拜拜！华为、清华等提出基于认知的万物超分大模型

2024-03-06 07:03

图像超分辨率技术旨在将低分辨率图像转换为高分辨率图像，从而提高图像的清晰度和细节真实性。随着超分技术的发展和手机硬件性能的提升，人们期望拍摄出更加清晰的照片。这项技术在手机影像等领域有着广泛的应用和需求。

然而，现有的超分方法存在一些局限性，如下图所示，主要有以下两个方面：

一是缺乏泛化能力。为了实现更好的超分效果，通常需要针对特定场景使用特定传感器采集到的数据来进行模型训练，这种学习方式拟合了某种低清图像和高清图像间的映射，但在其他场景下表现不佳。此外，逐场景训练的方式计算成本较高，不利于模型的部署和更新。

二是缺乏理解能力。现有的超分方法主要依赖于从大量数据中学习图像的退化分布，忽视了对图像内容的理解，无法利用常识来准确恢复物体的结构和纹理。

▲ 图2. 真实场景超分 SOTA 方法的局限性：（行一）难以处理训练集外的退化分布；（行二）难以利用常识恢复物体结构。

人类在处理信息时，有两种不同的认知反馈系统。诺贝尔奖经济学得主丹尼尔・卡尔曼在《思考，快与慢》中将它们称为系统一和系统二，如图 3 所示。

系统一是快速的、直觉的、基于记忆的反馈，比如，我们可以脱口而出十以内的加减运算。系统二是缓慢的、多步的反馈，比如，28x39 往往需要逐步运算。现有的超分方法更贴近系统一，它们主要依赖于从大量数据中学习图像的退化分布，忽视了对图像内容的理解，无法按照常识来准确恢复物体的结构和纹理，也无法处理域外的退化情况。

▲ 图3. CoSeR 采用类似于人脑中系统二的修复方式

近日，由清华大学、华为诺亚方舟实验室、香港科技大学等机构提出的研究，通过将图像外观和语言理解结合起来生成认知特征，实现了认知超分辨率框架，使 SR 模型能够理解低分辨率图像。

本文认为，真正能有效应用于真实场景的画质大模型应该具备类似系统二的多步修复能力，即基于对图像内容的认知，结合先验知识来实现图像超分（Cognitive Super-Resolution，CoSeR）。

项目主页：

https://coser-main.github.io/

论文地址：

https://arxiv.org/abs/2311.16512

代码地址：

https://github.com/vinhyu/coser

▲ 图1. LR，GR 和 SR 分别为低清图像、基于对低清图像的认知生成的参考图像和超分图像。

CoSeR 模仿了人类专家修复低质量图像自上而下的思维方式，首先建立对图像内容的全面认知，包括识别场景和主要物体的特征，随后将重点转移到对图像细节的检查和还原。本文的主要贡献如下：

1. 提出了一种通用的万物超分画质大模型 CoSeR，它能够从低清图像中提取认知特征，包括场景内容理解和纹理细节信息，从而提高模型的泛化能力和理解能力；

2. 提出了一种基于认知特征的参考图像生成方法，它能够生成与低清图像内容一致的高质量参考图像，用于指导图像的恢复过程，增强图像的保真度和美感度；

3. 提出了一种「All-in-Attention」模块，它能够将低清图像、认知特征、参考图像三个条件注入到模型当中，实现多源信息的融合和增强；

4. 在多个测试集和评价指标上，相较于现有方法，CoSeR 均取得了更好的效果。同时，CoSeR 在真实场景下也展现颇佳。

方法介绍

图 4 展示了 CoSeR 的整体架构。CoSeR 首先使用认知编码器来对低清图像进行解析，将提取到的认知特征传递给 Stable Diffusion 模型，用以激活扩散模型中的图像先验，从而恢复更精细的细节。

此外，CoSeR 利用认知特征来生成与低清图像内容一致的高质量参考图像。这些参考图像作为辅助信息，有助于提升超分辨率效果。最终，CoSeR 使用提出的「All-in-Attention」模块，将低清图像、认知特征、参考图像三个条件注入到模型当中，进一步提升结果的保真度。

▲ 图4. 本文提出的万物超分画质大模型CoSeR

图 5 展示了 CoSeR 参考图像生成的效果。与直接从低清图像中获取描述的方法相比，CoSeR 的认知特征保留了细粒度的图像特征，在生成具有高度相似内容的参考图像时具有优势。在图 5 的第一行，使用 BLIP2 从低清图像生成的描述无法准确识别动物的类别、颜色和纹理。

此外，CoSeR 的认知特征对于低清图像更加鲁棒。例如，在图 5 的第二行，由于输入分布的差异，BLIP2 会生成错误的图像描述，而 CoSeR 生成了内容一致的高质量参考图像。最后，相比于 BLIP2 大模型接近 7B 的参数量，CoSeR 的认知编码器只有其 3% 的参数量，极大提升了推理速度。