arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.13652cs.LGcs.AI
收藏

OATS:通过稀疏与低秩分解实现的异常值感知剪枝

OATS: Outlier-Aware Pruning Through Sparse and Low Rank Decomposition

University of Toronto(多伦多大学)

Stephen Zhang, Vardan Papyan

AI总结:

针对大模型高内存高算力成本及现有剪枝方法压缩时性能下降的问题,提出基于输入嵌入二阶矩的稀疏低秩分解剪枝方法OATS,无需重训练即可在60%压缩率下实现SOTA性能与最高1.37倍CPU加速。

摘要:

近期向大规模基础模型的范式转变为深度学习开启了新纪元,尽管该范式在实践中取得了巨大成功,但也因高内存消耗与高计算量带来了高昂得令人却步的成本。为缓解这些问题,学界围绕无需昂贵重训练的事后神经网络剪枝技术展开了协同研究。尽管已取得显著进展,但现有方法通常会随着压缩率提升出现模型性能的持续下降。本文提出一种压缩大型Transformer的新方法,命名为OATS(异常值感知剪枝),该方法利用输入嵌入中的二阶矩信息,将模型权重分解为稀疏矩阵与低秩矩阵之和。在无需任何重训练的情况下,OATS在对Llama-3、Phi-3等大语言模型以及ViT、DINOv2等视觉Transformer进行最高60%比例的压缩时,取得了当前最优性能,同时相较于同等压缩程度的剪枝模型,CPU加速比最高可达1.37倍。

英文摘要:

The recent paradigm shift to large-scale foundation models has brought about a new era for deep learning that, while has found great success in practice, has also been plagued by prohibitively expensive costs in terms of high memory consumption and compute. To mitigate these issues, there has been a concerted effort in post-hoc neural network pruning techniques that do not require costly retraining. Despite the considerable progress being made, existing methods often exhibit a steady drop in model performance as the compression increases. In this paper, we present a novel approach to compressing large transformers, coined OATS, that utilizes the second moment information in the input embeddings to decompose the model weights into a sum of sparse and low-rank matrices. Without any retraining, OATS achieves state-of-the-art performance when compressing models by up to $60\%$ on large language models such as Llama-3 and Phi-3 and vision transformers such as ViT and DINOv2 while delivering up to $1.37\times$ the CPU acceleration versus a model that was comparably pruned.

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

↑