用神经网络进行文件无损压缩:斯坦福大学提出DeepZip

神经网络不仅可以分析、识别特征,提出预测,还可以压缩文件。斯坦福大学的研究者们最近提交的论文中,循环神经网络捕捉长期依赖关系的优势被用于无损压缩任务中,这种被称为DeepZip的技术已在文本和基因组数据文件中得到了实验。研究人员称,其结果颇具潜力。

正在进行的大数据变革让我们收集了大量不同类型的数据,如图像、文本和音频等;新类型的数据如3D VR数据、用于自动驾驶的点云数据、不同类型的基因组数据等,占据着巨量的存储空间。因此,人们对于统计模型和适用于各种数据格式的高效压缩方法有着很大的需求。

近50年来,无损压缩技术已经历了很多重要的发展。在克劳德·香农的一个经典研究中,这位先驱者指出,熵率是给定数据源可能达到的最佳压缩比,同时也给出了一种实现方法(尽管不甚实际)。J. Rissanen提出了算术编码,这是一个实现已知分布熵边界的有效方法。对于未知分布的数据源(如文本和DNA),他还设计了算术编码的自适应变体,它可以通过尝试学习条件k-gram模型的分布来进行压缩。尽管这种过程的复杂度会随k的变化而呈指数级增长,通常上下文会被限制在k=20符号。这会导致压缩比例的显著损失,因为模型无法捕捉长期依赖关系。我们都知道基于循环神经网络(LSTM/GRU)的模型善于捕捉长期依赖关系,同时可以较准确地预测下一个字母/单词。如此一来,能否使用基于RNN的框架来用于压缩任务?在斯坦福大学的一份研究中,研究人员探索了使用基于RNN的语言模型及算术编码来提升无损压缩的性能。

在这一研究的论文中,研究人员首先分析和理解了已知熵情况下,合成数据集上RNN和算术编码方法的表现,其目的是对各种RNN结构的能力和极限进行直观的理解。研究人员也对伪随机数生成序列(PRNG)进行了测试,尽管其熵率为零(因为它们是确定性的),但使用标准技术极难压缩。基于对此前在合成数据集上测试的经验,研究人员使用了文本压缩模型和基因组数据集。

压缩器框架

2.1 概述

首先是用于实验的压缩器模型,其框架可以被分为两个模块:

  1. RNN概率评估器模块:对于数据流S0,S1……SN,RNN概率评估器模块可以基于此前观察到的负号S0,S1……Sk-1来估计Sk的条件概率分布。这一概率估计Pˆ(Sk|S0, S1, . . . , Sk−1)会被递送到算术编码模块;
  2. 算术编码器模块:算法编码器模块可被认为是FSM,它接收下一个符号的概率分布估计并将其编码成一个状态(与解码器的操作相反)。

2.2 RNN概率评估器模块

实际上,RNN评估器模块可以是任何循环神经网络(LSTM/GRU),其中包括最终估算概率的Softmax层。算术编码器模块可以是经典的算术编码FSM,或更快的非对称数字系统(Asymmetric Numeral Systems,ANS)模块。对于模型的运行,有一些重要的限制:

  1. 输入的因果关系:RNN评估器必须是具有因果关系的,它可以视输入为特征,仅仅基于此前的编码符号进行估算(BiLSTM等或许不行)。
  2. 权重更新:权重更新(如执行)应在编码器和解码器中执行。这是必要的,因为我们需要编码器和解码器生成每个符号的分布。

研究人员主要探索了两个模型:符号级别的GRU模型(DeepZip-ChGRU)和基于特征的模型(DeepZip-Feat)。在DeepZip-GRU上,在第k步,GRU模块的输入是Xk-1,而状态k-1是输出的状态,直到k点为止。DeepZip-Feat包含输入作为特征计算因果关系,如过去的20个符号,以及观察到的流内上下文表现记录。此外,研究人员也考虑过基于文字的模型(Attention-RWA模型)。

2.3 算术编码器模块

算术编码器保持在区间[0,1]之间。每个符号流唯一地确定一个范围,这个范围可按顺序计算,并直接基于下一符号的概率评估。它可视为传递至下一迭代的算术编码器的一个状态。最后,该范围被编码,由此形成了压缩数据。在给定概率评估的情况下,解码操作则相反。算术编码操作如图 2 所示。


图 2:i.i.d (0.6, 0.2, 0.1, 0.1)分布式源的序列 (0, 2, 3)的算术编码

2.4 编码器&解码器操作

编码器&解码器操作如下图所示:

  1. 算术编码器模块通常从首个符号S0的自定义概率分布评估开始。完成之后,解码器可以解码首个符号。
  2. 算术编码器和RNN评估器模块都通过迭代传递状态信息。算术编码器的最终状态充当压缩数据。
  3. 如果模型训练超过一个epoch,RNN评估器模块的权重需要被存储,并计算压缩大小(MDL Principle [14])。


图 3:编码器模型架构

接着研究人员讨论了不同模型在上述数据集上的一些有趣实验。模型有:

  1. DeepZip-ChRNN:基于字符级RNN的神经网络模型。
  2. DeepZip-ChGRU:基于字符级GRU的神经网络模型。
  3. DeepZip-Feat:基于GRU的模型,其中包含所有以前观察到的符号的功能,而不仅仅是之前的输入。

3 合成数据集上的实验


图 5:包含128个单元的 DeepZip-ChRNN 模型在Markov-k源上的表现


图 6:包含128个单元的 DeepZip-ChGRU 模型在Markov-k源上的表现


图 7:包含128个单元的DeepZip模型与GZIP[15]、适应性算术编码-CABAC的表现对比


图 8:包含128个单元的DeepZip模型在实际数据集上的表现

论文:DeepZip: Lossless Compression using Recurrent Networks 


论文地址:https://web.stanford.edu/class/cs224n/reports/2761006.pdf

摘要:现今,我们生成的数据量大幅增加。新类型的数据,比如基因组数据[1]、3D-360度 VR 数据、自动驾驶点云数据已经出现。大量努力用在了分析以上数据的统计学信息,以设计好的压缩器。我们由信息论得知,好的压缩器来自好的预测器[2]。我们知道基于循环神经网络(LSTM/GRU)的模型擅长捕捉长期依赖关系 [3],并可以很好地预测下一字符/词。这样RNN可被有效用于压缩吗?我们分析了RNN在数据压缩问题上的应用。

压缩器DeepZip包含两个主要模块:基于RNN的概率评估器和算术编码模块。首先,我们讨论了现有文献和基本的模型架构。接着我们深入到合成及真实文本和基因组数据集的实验结果。最后,我们对发现的结果和未来工作作了讨论。

理论理论论文斯坦福大学
暂无评论
暂无评论~