arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2408.16031cs.LGcs.AI
收藏

EMP: 提升数据剪枝中的记忆能力

EMP: Enhance Memory in Data Pruning

College of Computer(计算机学院);National University of Defense Technology(国防科技大学);School of Computer and Communication Engineering(计算机与通信工程学院);Changsha University of Science and Technology(长沙理工大学);Hunan Provincial Key Laboratory of Intelligent Processing of Big Data on Transp(湖南省大数据智能处理重点实验室)

Jinying Xiao, Ping Li, Jie Nie, Bin Ji, Shasha Li, Xiaodong Liu, Jun Ma, Qingbo Wu, Jie Yu

AI总结:

EMP通过增强模型对数据的记忆能力,在高剪枝率下提升模型性能,首次探讨自监督学习中的记忆问题。

摘要:

最近,大型语言和视觉模型展现出强大的性能,但由于预训练和微调成本高,研究转向通过数据集剪枝实现更快的训练。先前的方法使用样本损失作为评估标准,旨在选择最

英文摘要:

Recently, large language and vision models have shown strong performance, but due to high pre-training and fine-tuning costs, research has shifted towards faster training via dataset pruning. Previous methods used sample loss as an evaluation criterion, aiming to select the most "difficult" samples for training. However, when the pruning rate increases, the number of times each sample is trained becomes more evenly distributed, which causes many critical or general samples to not be effectively fitted. We refer to this as Low-Frequency Learning (LFL). In other words, LFL prevents the model from remembering most samples. In our work, we decompose the scoring function of LFL, provide a theoretical explanation for the inefficiency of LFL, and propose adding a memory term to the scoring function to enhance the model's memory capability, along with an approximation of this memory term. Similarly, we explore memory in Self-Supervised Learning (SSL), marking the first discussion on SSL memory. Using contrastive learning, we derive the memory term both theoretically and experimentally. Finally, we propose Enhance Memory Pruning (EMP), which addresses the issue of insufficient memory under high pruning rates by enhancing the model's memory of data, thereby improving its performance. We evaluated the performance of EMP in tasks such as image classification, natural language understanding, and model pre-training. The results show that EMP can improve model performance under extreme pruning rates. For example, in the CIFAR100-ResNet50 pre-training task, with 70\% pruning, EMP outperforms current methods by 2.2\%.

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

↑