论文标题
基于快速变压器的通用无损压缩机
A Fast Transformer-based General-Purpose Lossless Compressor
论文作者
论文摘要
由于压缩比的改善,基于深度学习的压缩机最近获得了兴趣。但是,现代方法遭受了长时间的执行时间。为了缓解这个问题,本文的目标是减少基于学习的压缩机的执行时间。顺序建立历史依赖性(例如,复发性神经网络)负责长期推断潜伏期。取而代之的是,我们将变压器引入深度学习压缩机中,以平行构建历史依赖性。但是,现有的变压器在计算方面太重,并且不兼容压缩任务。 本文通过设计基于单层变压器的压缩友好结构,提出了快速通用无损压缩机Trace。我们首先设计一个新的指标,以建议压缩模型结构的选择部分。进一步提出了字节组和共享-FFN方案,以充分利用单层变压器的容量。这些功能允许痕迹达到竞争性压缩率和更快的速度。此外,我们通过设计控制器来减少更新开销的参数来进一步加速压缩过程。实验表明,Trace达到了总体$ \ sim $ 3倍的加速,同时保持与最先进的压缩机相当的压缩比。跟踪的源代码和数据集的链接可在https://github.com/mynotwo/a-fast-transformer基于general-purpose-losslessCompressor上获得。
Deep-learning-based compressor has received interests recently due to much improved compression ratio. However, modern approaches suffer from long execution time. To ease this problem, this paper targets on cutting down the execution time of deep-learning-based compressors. Building history-dependencies sequentially (e.g., recurrent neural networks) is responsible for long inference latency. Instead, we introduce transformer into deep learning compressors to build history-dependencies in parallel. However, existing transformer is too heavy in computation and incompatible to compression tasks. This paper proposes a fast general-purpose lossless compressor, TRACE, by designing a compression-friendly structure based on a single-layer transformer. We first design a new metric to advise the selection part of compression model structures. Byte-grouping and Shared-ffn schemes are further proposed to fully utilize the capacity of the single-layer transformer. These features allow TRACE to achieve competitive compression ratio and a much faster speed. In addition, we further accelerate the compression procedure by designing a controller to reduce the parameter updating overhead. Experiments show that TRACE achieves an overall $\sim$3x speedup while keeps a comparable compression ratio to the state-of-the-art compressors. The source code for TRACE and links to the datasets are available at https://github.com/mynotwo/A-Fast-Transformer-based-General-Purpose-LosslessCompressor.