中文

分裂分层变分压缩

图像与视频处理 2022-04-06 v1 计算机视觉与模式识别 机器学习 机器学习

摘要

变分自编码器(VAEs)在图像数据集压缩方面取得了巨大成功。这一成功由 bits-back 编码框架所实现,在许多基准上产生了具有竞争力的压缩性能。然而尽管如此,VAE 架构目前受限于编码实用性与压缩比的综合制约。亦即,不仅最先进的方法(如归一化流)常表现出更优性能,而且编码所需的初始比特使得单图与并行图像压缩颇具挑战。为弥补此缺陷,我们引入 Split Hierarchical Variational Compression (SHVC,分裂分层变分压缩)。SHVC 带来两点新颖之处。首先,我们提出一种高效的自回归先验,即自回归亚像素卷积,其允许在逐像素自回归与完全因子化概率模型之间作推广。其次,我们定义了我们的编码框架,即自回归初始比特,其灵活支持并行编码,并首次避免了通常与 bits-back 编码相关的诸多实用性难题。在我们的实验中,我们证明 SHVC 能够在全分辨率无损图像压缩任务上实现最先进的压缩性能,且模型参数量较竞争的 VAE 方法最多少 100 倍。

关键词

引用

@article{arxiv.2204.02071,
  title  = {Split Hierarchical Variational Compression},
  author = {Tom Ryder and Chen Zhang and Ning Kang and Shifeng Zhang},
  journal= {arXiv preprint arXiv:2204.02071},
  year   = {2022}
}