用扩散语言模型提升文本无损压缩速度与效率
Diffuse to Compress: Leveraging Diffusion LMs for Lossless Compression
- 以扩散模型替代自回归模型,实现并行生成符号,突破逐字生成瓶颈
- 在enwik8数据集上超越zstd等通用压缩器,达到新压缩性能纪录
- 适合需要高速压缩的场景,如大规模日志、代码库存储
我们研究无损文本压缩问题,针对数字文本数据(包括纯文本、源代码和XML等结构化格式)快速增长带来的存储挑战,以及神经语言模型压缩技术的进展。当前基于大语言模型的方法虽显著优于zstd、gzip等通用压缩器,但因每次仅生成一个符号,存在严重吞吐瓶颈。本文首次在无损神经文本压缩中引入扩散语言模型(DLM),作为自回归模型的替代推理范式。通过允许每轮前向传播独立决定编码符号的数量与位置,解决传统方法的逐步生成限制。设计了高效的算法策略应对扩散模型在无损压缩中的新挑战,并在enwik8基准上评估性能。实验表明,新框架实现了无损文本压缩的最新技术水平。由于DLM仍处于早期阶段,未来模型能力与效率的提升将带来更大改进空间。
原文摘要 · Abstract (English)
We study the problem of lossless text compression, motivated by the rapid growth in the collection and storage of digital textual data - including plain text, source code, and structured formats such as XML - and by recent advances in neural language model-based compression. In particular, recent LLM-based approaches, whether built on symbol-ranking pipelines or paired with a statistical compressor, have demonstrated compression ratios significantly superior to general-purpose compressors such as zstd, gzip, or bzip on text and code. However, these neural approaches suffer from severe throughput limitations, making them not yet practically usable. For the first time in the context of lossless neural text compression, we introduce Diffusion Language Models (DLMs) as an alternative inference paradigm to autoregressive LLM-based approaches. We argue that replacing autoregressive LLMs with DLMs within the same compression framework could overcome the throughput bottleneck caused by their one-symbol-per-step limitation. However, achieving these improvements requires addressing algorithmic challenges introduced by applying DLMs to lossless compression, where the architecture allows the number and positions of symbols encoded at each forward pass to be decided independently. We design efficient and effective strategies to solve these challenges and evaluate them experimentally against LLM-based and general-purpose compressors on enwik8, a well-established textual benchmark. Our results show that the newly proposed DLM-based framework advances the state of the art in lossless text compression. Moreover, as DLMs are still a relatively young paradigm, recent advances toward increasingly capable and efficient models suggest substantial room for further improvements.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。