arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.17954cs.CVcs.ITcs.LGmath.IT

使用有损图像压缩扩展训练数据

Scaling Training Data with Lossy Image Compression

Granica

Katherine L. Mentzer, Andrea Montanari

AI总结:

提出“存储缩放定律”描述测试误差随样本量和每图像比特数的联合演变,优化有损压缩级别以在给定存储下显著降低模型测试误差。

摘要:

经验确定的缩放定律在预测大型机器学习模型随训练数据和参数数量演变方面取得了广泛成功。因此,它们在优化有限资源(尤其是计算时间)的分配方面非常有用。在某些应用中,存储空间是一个重要的约束条件,因此需要仔细选择数据格式。计算机视觉是一个突出的例子:图像本质上是模拟的,但总是使用有限数量的比特以数字格式存储。给定一个数字图像数据集,可以使用有损数据压缩进一步减少存储每个图像所需的比特数 $L$。然而,这可能会降低在这些图像上训练的模型质量,因为每个样本的分辨率较低。为了捕捉这种权衡并优化训练数据的存储,我们提出了一种“存储缩放定律”,描述了测试误差随样本大小和每张图像比特数的联合演变。我们证明该定律在图像压缩的程式化模型中成立,并在两个计算机视觉任务上进行了经验验证,提取了相关参数。然后,我们展示了该定律可用于优化有损压缩级别。在给定存储量下,在最优压缩图像上训练的模型比在原始数据上训练的模型表现出显著更小的测试误差。最后,我们研究了随机化压缩级别的潜在益处。

英文摘要:

Empirically-determined scaling laws have been broadly successful in predicting the evolution of large machine learning models with training data and number of parameters. As a consequence, they have been useful for optimizing the allocation of limited resources, most notably compute time. In certain applications, storage space is an important constraint, and data format needs to be chosen carefully as a consequence. Computer vision is a prominent example: images are inherently analog, but are always stored in a digital format using a finite number of bits. Given a dataset of digital images, the number of bits $L$ to store each of them can be further reduced using lossy data compression. This, however, can degrade the quality of the model trained on such images, since each example has lower resolution. In order to capture this trade-off and optimize storage of training data, we propose a `storage scaling law' that describes the joint evolution of test error with sample size and number of bits per image. We prove that this law holds within a stylized model for image compression, and verify it empirically on two computer vision tasks, extracting the relevant parameters. We then show that this law can be used to optimize the lossy compression level. At given storage, models trained on optimally compressed images present a significantly smaller test error with respect to models trained on the original data. Finally, we investigate the potential benefits of randomizing the compression level.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑