姓名:黄世民 学号:20021210922
【嵌牛导读】:随着深度学习的蓬勃发展,人们注意到了其在视频压缩方面的潜力,传统视频压缩方法依赖于手工设计以及极高的专业知识,而使用深度学习可以让更多人加入到视频压缩的工作中。
【嵌牛鼻子】:深度学习,视频压缩
【嵌牛提问】:传统视频压缩和端到端深度学习视频压缩的关系和区别?
【嵌牛正文】:
转自:https://www.jqr.com/article/000289
2018年6月14日,图鸭与论智联合举办了一场线上公开课深度学习之视频图像压缩。讲师为周雷博士,图鸭科技深度学习算法研究员。以下为论智整理的听课笔记。因水平有限,难免有错漏不当之处,仅供参考。
首先,我们简单回顾下传统图像与视频压缩技术。
JPEG
以下为JPEG编解码流程示意图。
从上图我们可以看到,图像数据编码过程如下:
离散余弦变换。简单来说,离散余弦变换是一种矩阵运算。
经过离散余弦变换后,高频数据和低频数据分流了,矩阵左上方是高频数据(较大的数值),右下方是低频数据(较小的数值)。这样我们就可以对其进行量化了,在JPEG中是除以量化步长再取整。
量化之后对其进行熵编码,得到压缩表示。
解码的过程与编码过程相逆,经过反熵编码、反量化、反离散余弦变换重建图像。
JPEG2000
JPEG2000和JPEG的最大差别是使用了离散小波变换。此外还加上了一些预处理步骤。
WebP
WebP是来源于VP8的图片压缩格式。主要特色是基于块预测。
BPG
和WebP类似,BPG同样源于视频编码技术(HEVC)。BGP的主要特点如下:
HEVC
HEVC的编码示意图如下:
上图中,Ref.表示参考样本。T & Q表示转换、量化过程,Q−1Q−1 & T−1T−1为其逆过程。Deblk.为Deblocking(去区块)的缩写。HEVC会将影像分为区块再进行编码,因此重建时会在区块边缘出现不连续的现象,称为区块效应。去区块过程可以减轻区块效应。SAO为Sample Adaptive Offset(样本自适应偏移量)的缩写,通过分析去区块后的数据与原始数据的差异,补偿量化过程造成的损失,使其尽可能接近原始数据。
CABAC为自适应二进制算术编码。算术编码利用符号出现的概率将符号序列编码为一个数字。
相应的解码过程:
从以上的编解码过程中,我们可以看到,先验概率估计的精确程度对编码的效率影响很大。HEVC使用动态更新的概率模型实现自适应二进制算术编码。
除了帧内估计、预测(参见前面提到的BPG)之外,HEVC视频编码还需考虑运动估计等帧间的关系。
深度学习图像压缩框架
下为深度学习图片压缩的典型框架示意图:
上图中每个模块的具体作用,可以参考如何设计基于深度学习的图像压缩算法中的解释。
图像压缩数据集
设计好网络模型后,需要使用图像进行训练。由于图像压缩属于无监督学习,无需人工标注,因此数据集是比较容易搜集的。无论是从网上爬取,还是自行使用相机拍摄,都不难得到大量高清图片。
常用的测试集有:
Kodak PhotoCD数据集,图像分辨率768x512,约40万像素;
Tecnick数据集,约一百四十万像素。
CVPR 2818 CLIC数据集,图像类别广泛,分辨率不等(512至2048),文件尺寸不等(几百K到几M)。
深度学习视频压缩框架
深度学习视频压缩与图像压缩的主要差别在于增加了帧间预测/差值。
帧间预测能极大得减少帧间冗余。以1个参考帧,预测N-1帧为例,帧间预测的约束为参考帧和预测码字远小于每帧单独压缩的码字:
深度学习图像视频压缩进展介绍
1. 深度学习图像压缩的主要发展方向:
RNN
CNN
GAN
这部分内容可以参考公开课ppt以及概览CVPR 2018神经网络图像压缩领域进展一文。
2. 视频压缩方面,近年来的研究热点是将CNN与现有的视频编码器相结合。
编码单元选择
Liu Z、Yu X、Chen S等在2016年发表了CNN oriented fast HEVC intra CU mode decision,使用CNN学习预测编码单元模式的分类(2N x 2N或N x N)。
下采样编码
Jiahao Li等在2018年发表的Fully Connected Network-Based Intra Prediction for Image Coding对视频帧进行了分块处理,对适合进行下采样的块执行下采样操作,而对不适合进行下采样的块不执行下采样操作。之后,对下采样的块根据情况分别使用CNN或DCTIF进行上采样,以重建图像。为了达到更好的效果,亮度通道和色度通道使用了不同的网络架构。
视频帧环路滤波和后处理
Park W S和Kim M在2016年发表的CNN-based in-loop filtering for coding efficiency improvement中,使用CNN提升了HEVC的环路滤波(包括去区块滤波和SAO滤波)的效果。
类似地,Yuanying Dai等在2016年发表的A Convolutional Neural Network Approach for Post-Processing in HEVC Intra Coding,使用CNN网络改进了HEVC的后处理过程。
使用深度学习进行单纯的图像压缩,应用场景有一定局限性。深度学习在视频压缩领域潜力更大。深度学习在视频压缩领域的主要优势在于:
能够实现更好的变换学习,从而取得更好的效果。
端到端的深度学习模型能够自行学习,而传统的视频压缩工作需要手工设计很多东西。
传统的视频压缩方法通常通过一些启发式的方法进行帧间预测,从而减少帧间冗余。而深度学习能够基于光流等进行预测。
劣势:
另一方面,基于深度学习进行视频压缩也会遇到很多挑战。比如控制实现帧间预测占用的比特。
CVPR 2018 CLIC
在CVPR 2018学习图像压缩挑战上,图鸭团队为三个赢家之一,MOS、MS-SSIM两项指标均为第一。
TucodecTNGcnn4p基于端到端的深度学习算法,其中使用了层次特征融合的网络结构,以及新的量化方式、码字估计技术。网络使用了卷积模块和残差模块,损失函数纳入了MS-SSIM。
基于深度学习超分辨率重建图像
在这一领域,图鸭科技重点关注低码率下的超分辨率重建。因为低码率下图像难免有比较多的失真,应用超分辨率重建技术能缓解这些图像上的瑕疵,取得更好的显示效果。而高码率图像保留了原图更多的细节,相对而言不是非常适合应用超分辨率技术。
基于深度学习的视频压缩
如前所述,图鸭科技认为相对图像压缩,深度学习在视频压缩领域潜力更大。目前图鸭科技在基于深度学习的视频压缩方面,已经能够取得与x265媲美的效果。
基于深度学习的结构化存储
相比传统方法,深度学习编码图像的算力负担较重。然而,另一方面,图像的压缩特征不仅可以用于重建图像,还可以为语义分割、图像分类提供帮助。
基于GAN进行图像压缩
GAN主要用于图像生成领域。但在进行图像压缩时,GAN会遇到一个问题,就是它会改变一些细节(生成一些新的细节)。因此,GAN这一的技术方向的选择常常取决于项目需求。例如,对于人脸图像来说,如果感兴趣区域是人脸,那么对感兴趣区域以外的区域可以使用非常低的码率压缩,重建图像时利用GAN生成细节。
量化方法的选择
建议大家参考相关论文自行选择。因为量化方法的选择往往还和网络中的其他模块相关。例如,如果编码器部分选用的激活输出的是二值(0、1),那么量化其实就不是那么重要了。
压缩时间
一般而言,基于深度学习的压缩算法,和传统算法相比,在CPU上压缩时间处于劣势。不过也有例外。比如,在图鸭科技的测试中,在CPU上,基于CNN的算法实际上比H266要快。H266虽然属于传统算法,但是复杂度其实很高。
未来随着GPU、专用深度学习芯片的算力提升,压缩时间不会成为应用深度学习压缩算法的最大障碍