VARIATIONAL IMAGE COMPRESSION WITH A SCALE HYPERPRIOR

发布时间 2023-11-24 17:59:02作者: 浪矢-CL

abstruct

描述了一个变分自编码器的端到端图像压缩模型。这个模型结合了超先验来捕获潜在表示的空间依赖性,这种超先验涉及到了边信息。并且改模型与底层的自编码器结合联合训练一个复杂的先验。

introduction

熵编码依赖于编码器和解码器都知道的量化表示的先验概率模型(熵模型)。这种压缩潜在表示的熵模型通常表示为联合分布,甚至全因子分布\(p_{\hat y}(\hat y)\)。这里要注意区分潜在表示的实际分布\(m(\hat y)\)和熵模型\(p_{\hat y}(\hat y)\)

  • 熵模型通常被假定具有某种参数形式,参数和数据匹配
  • 边际分布则是一种未知的分布,它既来自于图像的分布,也来自于用来推断替代表示y的方法。
    一对编码器-解码器所能达到的最小平均编码长度由两个分布之间的香农交叉熵给出:

\[R = E_{\hat y}∼m[− log_2 p_{\hat y}(\hat y)] \]

在这里,作者认为,边信息可以被看做是熵模型参数的先验,使他们成为潜表示的超先验。
该文章在Ballé et al. (2017)的基础上,加入了超先验,以捕捉潜在表示的空间相邻元素在尺度上趋于共同变化这一事实

量化会带来误差,这在有损压缩中是可以容忍的,从而产生了速率-失真优化问题。速率是压缩表示法的预期编码长度(比特率):假设熵编码技术能有效运行,这又可以写成交叉熵:
在这种情况下,潜在表示的边际分布来自(未知)图像分布 px 和分析变换的属性。失真度是指重建 x^ 与原始图像 x 之间的预期差异,以规范或感知度量。量化的粗糙度,或者说分析和合成变换所隐含的表示的扭曲,会影响速率和失真度,从而导致权衡,即速率越高,失真度越低

模型

image
如图3所示,为了捕获潜在表示的空间依赖关系,引入了额外的随机变量z。
每个元素\(\tilde y_i\)被建模为具有自身标准差的零均值高斯(标准差是通过对\(\tilde z\)应用参数变换\(h_s\)来预测的)
我们只需要在y的基础上堆叠另一个参数变换\(h_a\)即可以建立一个单联合因子分解变分后验