arXiv:2410.05078cs.LGcs.AI2024-10ICML被引 16

小模型也能高效压缩多模态数据,性能优于传统算法。

Compression via Pre-trained Transformers: A Study on Byte-Level Multimodal Data

  • 用百万级参数的Transformer在字节级多模态数据上训练
  • 对跨模态测试数据压缩比最低达0.49,优于FLAC等主流算法
  • 适合需要轻量级、跨模态压缩方案的研究与应用

基础模型具备强大的数据压缩能力,但因其参数量大,整体压缩比反而不如标准压缩算法。简单减少参数量会损害预测性能,进而影响压缩效果。本文通过大规模实证研究,寻找预训练Transformer在压缩性能上的最佳平衡点。我们在165GB原始字节序列数据(包括文本、图像、音频及组合)上训练模型,并对每种模态的1GB分布外(OOD)数据进行压缩。结果发现,仅数百万参数的小模型即可超越通用压缩算法(gzip、LZMA2)甚至领域专用压缩器(PNG、JPEG-XL、FLAC),即使计入参数大小。例如,在跨模态音频数据上达到最低压缩比0.49(对比FLAC的0.54)。我们进行了广泛的消融实验与超参数调优,分析模型规模、数据规模的影响,并比较单模态与多模态训练的效果。结果显示,小模型可在多模态上表现良好,但相比大规模基础模型,其对未见模态的迁移能力普遍较弱。

原文摘要 · Abstract (English)

Foundation models are strong data compressors, but when accounting for their parameter size, their compression ratios are inferior to standard compression algorithms. Naively reducing the parameter count does not necessarily help as it deteriorates predictions and, accordingly, compression. We conduct a large-scale empirical study to find a sweet spot where pre-trained vanilla transformers can achieve competitive compression ratios. To this end, we train models on 165GB of raw byte sequences of either text, image, or audio data (and all possible combinations of the three) and then compress 1GB of out-of-distribution (OOD) data from each modality. We find that relatively small models (millions of parameters) can outperform standard general-purpose compression algorithms (gzip, LZMA2) and even domain-specific compressors (PNG, JPEG-XL, FLAC) $\unicode{x2013}$ even when accounting for parameter size. We achieve, e.g., the lowest compression ratio of 0.49 on OOD audio data (vs. 0.54 for FLAC). We conduct extensive ablations and hyperparameter sweeps to study the impact of model- and dataset scale, and we investigate the effect of unimodal versus multimodal training. We find that even small models can be trained to perform well on multiple modalities, but unlike large-scale foundation models, transfer to unseen modalities is generally weak.

数据压缩Transformer多模态轻量化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。