arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2305.13232cs.CV

重新审视模型压缩中的数据增强:一项实证与全面研究

Revisiting Data Augmentation in Model Compression: An Empirical and Comprehensive Study

Tsinghua University(清华大学)

Muzhou Yu, Linfeng Zhang, Kaisheng Ma

AI总结:

本文全面研究了模型压缩中数据增强的作用,发现不同大小的模型偏好不同强度的数据增强,并提出利用预训练大模型预测结果作为设计数据增强策略的标准。

摘要:

深度神经网络的优异性能通常伴随着大量的参数和计算量,这限制了它们在资源受限的边缘设备上的使用。为了解决这个问题,人们提出了大量方法,如剪枝、量化和知识蒸馏,以压缩神经网络并取得了重大突破。然而,这些压缩方法大多关注神经网络的架构或训练方法,却忽略了数据增强的影响。在本文中,我们重新审视了数据增强在模型压缩中的使用,并对模型大小与其最优数据增强策略之间的关系进行了全面研究。总的来说,我们主要有以下三个观察结果:(A) 不同大小的模型偏好不同强度的数据增强。因此,在迭代剪枝中,采用不同强度的数据增强比采用一致强度的数据增强能带来更好的性能。(B) 高强度的数据增强可能显著提升大模型的性能,但会损害小模型的性能。幸运的是,小模型仍然可以通过首先使用“附加参数”学习强数据增强,然后在推理期间丢弃这些“附加参数”来从中受益。(C) 预训练大模型的预测结果可用于衡量数据增强的难度。因此,它可以被用作设计更好数据增强策略的标准。我们希望本文能促进更多关于模型压缩中数据增强使用的研究。

英文摘要:

The excellent performance of deep neural networks is usually accompanied by a large number of parameters and computations, which have limited their usage on the resource-limited edge devices. To address this issue, abundant methods such as pruning, quantization and knowledge distillation have been proposed to compress neural networks and achieved significant breakthroughs. However, most of these compression methods focus on the architecture or the training method of neural networks but ignore the influence from data augmentation. In this paper, we revisit the usage of data augmentation in model compression and give a comprehensive study on the relation between model sizes and their optimal data augmentation policy. To sum up, we mainly have the following three observations: (A) Models in different sizes prefer data augmentation with different magnitudes. Hence, in iterative pruning, data augmentation with varying magnitudes leads to better performance than data augmentation with a consistent magnitude. (B) Data augmentation with a high magnitude may significantly improve the performance of large models but harm the performance of small models. Fortunately, small models can still benefit from strong data augmentations by firstly learning them with "additional parameters" and then discard these "additional parameters" during inference. (C) The prediction of a pre-trained large model can be utilized to measure the difficulty of data augmentation. Thus it can be utilized as a criterion to design better data augmentation policies. We hope this paper may promote more research on the usage of data augmentation in model compression.

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

↑