当前位置:首页 > AI科普

模型剪枝是啥?AI 大模型是怎么给自己“减肥”的

xinhack3天前AI科普29

模型剪枝是啥?AI 大模型是怎么给自己“减肥”的 AI科普

模型剪枝(Pruning)说白了就是给 AI “减肥”。一个训练好的大模型,里头成千上万亿个参数,真管用的可能就一小半,剩下不少是平时基本不吭声的“闲人”。模型剪枝干的事,就是把这批闲人请走,模型体积缩了、跑起来快了,效果还不怎么掉。这活儿在手机、摄像头这些带不动大模型的设备上,简直是救命稻草。

模型里咋就有这么多“闲人”

你琢磨琢磨,训练的时候模型为了不漏掉任何特征,恨不得把每个神经元都留着。结果就是一大盘棋里,好多连接的权重小到几乎为零,等于摆设。好比一个班组二十号人,真天天干活的就五六个,剩下那些你裁了,活儿照样转。模型剪枝要的就是这个效果——把接近零的那些连接咔嚓掉。

剪法分两种,粗暴的和讲理的

剪枝也分门派。一种叫“非结构化”,挨个神经元看,谁没用砍谁,结果是模型里满是零散的洞,省了算力但硬件不一定买账,因为零零散散不好加速。另一种叫“结构化”,直接成片砍——整根通道、整层不要,虽然狠了点,但剩下的规整,芯片跑得动。要我说,落地还是后者更实在。

老哥见过“剪秃了”的惨案

这事儿我当年真见过翻车的。有个项目图省事,一把剪掉七成参数,指标当场塌了,模型连训练集都认不全。为啥?剪太狠,把真管用的也误伤了。后来才懂,剪完得重新“回炉”微调一阵,让剩下的神经元重新适应,这叫剪后重训。减肥可以,但不能把骨头也减了。

剪完不是万事大吉

以为咔嚓完就完事?没那么简单。剪出来的稀疏模型,得有配套的推理框架认它、加速它,不然省了参数却没省时间,等于白剪。好在现在不少工具链都支持,端侧跑大模型才成了可能。米岸AI 之前聊过的端侧 AI,背后常就有模型剪枝一份功。

和普通人有啥关系

关系大了。你手机相册能本地识别人脸、输入法能本地预测下一个词,很多时候就是模型被剪小了才塞得进那点内存。没这招,这些功能要么卡成幻灯片,要么非得把你的数据传到云上,又慢又让人犯嘀咕。所以模型剪枝这手艺,是藏在咱兜里的。

常见问题

问:模型剪枝和模型量化是一回事吗?

答:不是一家,但常搭伙干。量化是把数字存得“粗糙”点(比如用更少的位数),模型剪枝是直接删参数。一个让数变小,一个让数变少,目标都是瘦身,经常一起上。

问:剪完模型会不会变傻?

答:剪得讲究,基本看不出来;剪狠了真会掉链子。关键在剪完重训和拿捏比例,不是越狠越好。老话讲,过犹不及,模型也一样。

问:我的手机 APP 用到模型剪枝了吗?

答:但凡有“本地就能跑、不联网也成”的 AI 功能,背后多半动过模型剪枝或量化的手脚。不然那点内存,装不下原版的大块头。

问:普通用户能自己给模型剪枝吗?

答:得会点代码和框架,不是点几下的事。但网上开源工具不少,想折腾的能玩;图省事的话,直接用厂商剪好的现成模型最稳妥。

(本文由米岸AI整理)

相关文章

什么是对齐?怎么让人工智能学会该做与不该做

什么是对齐?怎么让人工智能学会该做与不该做

一个模型再聪明,如果分不清“能不能做”和“该不该做”,也靠不住。对齐(Alignment)就是让 AI 的目标和人类的价值一致——不仅要答得对,还要答得合适、安全、守边界。这篇文章用“教孩子规矩”的比…

生成式 AI 和判别式 AI 有什么区别?一个会造一个会判

生成式 AI 和判别式 AI 有什么区别?一个会造一个会判

同样是 AI,有的能写出一篇文章、画出一幅画,有的只能回答“这是猫还是狗”。前者叫生成式 AI,后者叫判别式 AI,它们干的活天差地别。理解这对区别,你就能看懂为什么聊天机器人能创作、而传统模型只能分…

激活函数是什么?为什么 AI 需要「非线性」这道弯

激活函数是什么?为什么 AI 需要「非线性」这道弯

神经网络靠一层层加权求和做判断,但如果只做加减乘除,它永远只能画直线。激活函数就是给网络加上的'非线性开关',让 AI 能拟合曲线、分辨复杂图案。本文用开关和闸口的比喻,讲清它为什么不可或缺,以及选它…

反向传播是什么?神经网络的错误怎么一层层传回去

反向传播是什么?神经网络的错误怎么一层层传回去

模型做错题,错的不只是最后一层。反向传播就是把「总错误」沿着网络一层层往回分,算出每个权重该为这次失误负多少责。本文用传话出错追责和工厂返工的比喻,讲清这个让深度学习得以训练的核心算法,以及它为什么能…

预训练是什么?大模型上岗前先读的「万卷书」

预训练是什么?大模型上岗前先读的「万卷书」

我们常说大模型「见多识广」,它的见识来自一个叫预训练的阶段:在正式干活前,先用海量文本自己琢磨语言规律。本文用通识教育和入职培训的比喻,讲清预训练到底是什么,以及它和微调、推理各司何职。读完你会明白大…

少样本学习是什么?大模型为什么看几个例子就会

少样本学习是什么?大模型为什么看几个例子就会

老式 AI 要学一样东西得喂成千上万条样本,今天的大模型却看两三个例子就能上手。这种「少样本学习」靠的是预训练积累的通识,而不是从零学起。本文用看样例学做事的比喻,讲清它为什么管用、又在哪里会翻车,以…

聚类是什么?AI 怎么把相似的东西自动「归成一堆」

聚类是什么?AI 怎么把相似的东西自动「归成一堆」

前面聊的很多模型都要先有标签才能学,聚类却反着来:不给任何答案,只丢一堆杂乱数据,让 AI 自己把相似的分到一组。本文用聚会自动分桌和整理衣柜的比喻,讲清 K-Means 怎么运作、簇数怎么定,以及它…

残差连接是什么?神经网络为什么能堆到上百层还不崩

残差连接是什么?神经网络为什么能堆到上百层还不崩

把神经网络层数堆深,本想学得更精细,可早年一加深就反而变差——不是算力不够,是信息传不下去。残差连接给了个极简解法:给数据开一条「直通车」,让它跳过几层直接往后传。本文用走楼梯和坐电梯的比喻,讲清这招…