Less is KEN: a Universal and Simple Non-Parametric Pruning Algorithm for Large Language Models

KEN修剪算法:大型语言模型的通用非参数优化方法
本文提出了一种名为KEN的非架构特定、简单且非参数化的修剪算法,用于优化大型语言模型。通过使用核密度估计(KDE),KEN有效地选择并保留关键参数,减少模型内存需求,同时保持甚至提高模型性能。实验结果显示,KEN在多个transformer模型上实现了高达约70%的参数减少,且不影响性能。此外,KENviz工具提供了对修剪过程的可视化解释。

本文是LLM系列文章,针对《Less is KEN: a Universal and Simple Non-Parametric Pruning Algorithm for Large Language Models》的翻译。

Less is KEN:一种通用且简单的大型语言模型非参数修剪算法

摘要

由于神经网络模型的复杂性及其在各个领域的广泛应用,神经网络修剪变得越来越重要。现有的修剪算法经常受到架构特异性、过度复杂和对复杂计算的依赖等限制,这使得它们在现实世界的应用中不切实际。在本文中,我们提出了KEN:一种基于核密度估计(KDE)的简单、通用和非结构化修剪算法。KEN旨在通过选择性地保留最重要的参数,同时将其他参数恢复到其预训练状态,来构建优化的transformer模型。这种方法保持了模型性能,同时只允许存储优化的子网络,从而显著节省了内存。对七个transformer模型的广泛评估表明,KEN实现了与原始模型相同或更好的性能,最小参数减少了25%。与其他修剪和PEFT算法的深入比较证实了KEN的有效性。此外,我们还介绍了KENviz,这是一个可解释的工具,可以可视化优化的模型组成和KEN选择的子网络。

1 引言

2 背景

3 相关工作

4 KEN修剪算法

5 实验

6 结果和讨论

7 结论

在本文中,我们提出了KEN,这是一种新

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

UnknownBody

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值