Anderson 视角

使用机器学习恢复过度压缩的社交媒体视频

mm
将 Unite.AI 添加到您在 Google 上的首选来源
Main image source: DALL-E 2

中国的一项新研究提出了一种有效且新颖的方法,用于恢复用户上传视频的细节和分辨率,该视频在平台如微信和YouTube上被自动压缩以节省带宽和存储空间。

比较新方法与既有方法准确重建社交媒体平台自动优化过程中丢失细节的能力。来源:https://arxiv.org/pdf/2208.08597.pdf

比较新方法与既有方法准确重建社交媒体平台自动优化过程中丢失细节的能力。来源:https://arxiv.org/pdf/2208.08597.pdf

与之前的方法不同,新方法不使用通用训练数据来升级和采样视频,而是为每一帧压缩视频生成一个退化特征图(DFM),从而有效地概述了帧中由于压缩而受损或退化的区域。

来自新论文的消融研究:右数第二幅是纯退化特征图(DFM)的基准真值;右数第三幅是不使用DFM时的损伤估计;左侧是使用DFM得到的更准确损伤图。

来自新论文的消融研究:右数第二幅是纯退化特征图(DFM)的基准真值;右数第三幅是不使用DFM时的损伤估计;左侧是使用DFM得到的更准确损伤图。

恢复过程利用卷积神经网络(CNN)等技术,根据DFM中的信息进行指导和聚焦,从而使新方法超越之前方法的性能和准确性。

研究人员将高质量视频上传到四个流行的分享平台,再下载压缩结果,从而获得该过程的基准真值。他们开发了能在抽象层面学习压缩伪影和细节损失的计算机视觉处理流程,可将其应用于多个平台,依据完全适合任务的数据,把视频恢复到接近原始的质量。

研究人员的新UVSSM数据集的示例。

研究人员的新UVSSM数据集的示例。

研究材料已被编译成一个名为用户在社交媒体上分享的视频(UVSSM)的HQ/LQ数据集,并已在百度网盘上公开下载(密码:rsqw),供后续研究项目开发新的平台压缩视频恢复方法。

可下载UVSSM数据集中两个对应HQ/LQ样本的比较,源URL见上方链接。即使这个例子也可能受到图像软件、CMS或CDN等多轮压缩的影响,因此请参考原始数据以获得更准确的比较。

可下载UVSSM数据集中两个对应HQ/LQ样本的比较,源URL见上方链接。即使这个例子也可能受到图像软件、CMS或CDN等多轮压缩的影响,因此请参考原始数据以获得更准确的比较。

系统的代码,称为通过自适应退化感知恢复视频(VOTES),也已在GitHub上发布,尽管其实现需要一些依赖项。

该论文题为恢复用户在社交媒体上分享的视频,由深圳大学的三位研究人员和香港理工大学电子和信息工程系的一位研究人员共同撰写。

从伪影到事实

恢复网页抓取视频的质量,同时避免Gigapixel等程序及大多数同类开源软件包凭借通用数据生成有时过度的虚构细节,可能对计算机视觉研究领域产生影响。

计算机视觉研究通常依赖于从YouTube和Twitter等平台获取的视频,但这些平台使用的压缩方法和编解码器是保密的,无法轻易地根据伪影模式或其他视觉指标确定,而且可能定期改变。

大多数使用网页抓取视频的项目并非研究压缩,而是必须适应可用压缩视频质量的限制,因为他们没有访问原始高质量视频的权限。

因此,能够忠实地恢复这些视频的质量和分辨率,而不引入来自其他计算机视觉数据集的下游影响,这将有助于避免计算机视觉项目目前必须为降级视频源做出的常见变通和妥协。

尽管YouTube偶尔会宣布其视频压缩方法的重大变化(如VP9),但没有一个平台会明确透露其压缩用户视频的整个过程或确切的编解码器和设置。

过去约十年间,提高用户上传视频的输出质量已成为一种德鲁伊式神秘技艺,各种大多未经证实的变通办法不断流行又过时。

方法

之前的基于深度学习的视频恢复方法涉及通用特征提取,既可以针对单帧恢复,也可以使用光流的多帧架构,在恢复当前帧时考虑相邻及后续帧。

所有这些方法都必须应对“黑盒”效应,即它们无法检查压缩效果的核心技术,因为不确定这些技术是什么或如何为特定用户上传的视频配置。

VOTES则试图直接从原始和压缩视频中提取显著特征,并确定将泛化到多个平台标准的转换模式。

VOTES的简化概念架构。

VOTES的简化概念架构。

VOTES使用一个专门开发的降级感知模块(DSM)来提取卷积块中的特征。多个帧然后被传递到特征提取和对齐模块(FEAM),然后被传递到降级调制模块(DMM)。最后,重构模块输出恢复的视频。

数据和实验

在新工作中,研究人员专注于恢复上传到微信平台后再下载的视频,但他们希望确保所得到的算法可以适应其他平台。

事实证明,一旦他们为微信视频获得了有效的恢复模型,将其适配到Bilibili、Twitter和YouTube时,每个平台的每个定制模型只需90秒即可完成一个训练轮次(在一台配备4个NVIDIA Tesla P40 GPU和96GB VRAM的机器上)。

将成功的微信模型适配到其他视频分享平台相当容易。这里展示VOTES使用作者的UVSSM数据集和REDS数据集(见下文),在各平台上几乎立即达到相当的性能。

将成功的微信模型适配到其他视频分享平台相当容易。这里展示VOTES使用作者的UVSSM数据集和REDS数据集(见下文),在各平台上几乎立即达到相当的性能。

为了填充UVSSM数据集,研究人员收集了264个视频,时长在5-30秒之间,每个帧率为30fps,来源于移动电话摄像头或互联网。视频的分辨率为1920 x 1080或1280 x 270。

内容(见前图)包括城市景观、风景、人物和动物等,均可通过Creative Commons Attribution许可在公共数据集中使用,允许重用。

作者将214个视频上传到微信,使用五种不同品牌的移动电话,获得微信的默认视频分辨率960×540(除非源视频尺寸已经更小),这是流行平台中对画质损害最严重的转换之一。

左上为带有三个放大区域的原始HQ帧;右上为同一视频经平台压缩退化后的对应帧;左下为压缩帧的计算退化;右下为VOTES关注的工作区域。低质量图像尺寸为HQ图像的一半,此处为便于比较进行了缩放。

左上为带有三个放大区域的原始HQ帧;右上为同一视频经平台压缩退化后的对应帧;左下为压缩帧的计算退化;右下为VOTES关注的工作区域。低质量图像尺寸为HQ图像的一半,此处为便于比较进行了缩放。

对于与其他平台的比较,研究人员上传了50个不在原始214个视频中的视频到Bilibili、YouTube和Twitter。这些视频的原始分辨率为1280×270,下载版本为640×360。

这使得UVSSM数据集包含364对原始(HQ)和共享(LQ)视频,其中214对来自微信,Bilibili、YouTube和Twitter各50对。

实验中,10个随机视频被选为测试集,4个作为验证集,剩余的200个作为核心训练集。实验进行了五次,使用K折交叉验证,结果在这些实例中进行了平均。

在视频恢复测试中,VOTES与时空可变形融合(STDF)比较。在分辨率增强方面,它与增强可变形卷积(EDVR)、RSDN、带时序组注意力的视频超分辨率(VSR_TGA)和BasicVSR比较。Google的单阶段方法COMISR也包括在内,尽管它与其他既有工作的架构类型不同。

方法被测试在UVSS和REDS数据集上,VOTES取得了最高的分数:

作者认为,定性结果也表明VOTES优于之前的系统:

由不同竞争方法恢复的REDS视频帧。指示性分辨率-请参阅论文以获取最终分辨率。

由不同竞争方法恢复的REDS视频帧。指示性分辨率-请参阅论文以获取最终分辨率。

 

首次发表于2022年8月19日。

机器学习撰稿人,专注于人类图像合成的领域专家。曾任 Metaphysic.ai 的研究内容负责人,直至其并入 DNEG 的 Brahma.ai。
网站: martinanderson.ai
联系:martin@martinanderson.ai