当前位置:首页 > AI科普

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

xinhack3天前AI科普30

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

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

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

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

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

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

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

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

剪完不是万事大吉

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

和普通人有啥关系

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

常见问题

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

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

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

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

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

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

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

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

(本文由米岸AI整理)

相关文章

训练 AI 的数据从哪来?它到底在学什么

训练 AI 的数据从哪来?它到底在学什么

大模型再强,也是从海量文本里"喂"出来的。本文带你看看训练数据通常是什么、从哪里来,为什么数据的质量和数量同样重要,以及数据过时会对 AI 造成什么影响。…

涌现能力是什么?小模型不会、大模型突然会

涌现能力是什么?小模型不会、大模型突然会

有些能力很奇怪:模型小的时候完全没有,参数涨到一定规模,突然就会了——像攒够了人数才会涌现的“人浪”。这种“量变到质变”叫涌现能力(Emergence)。这篇文章用“凑人数”和“水温烧开”两个比喻,讲…

模型量化是什么?一文看懂手机为什么能跑大模型

模型量化是什么?一文看懂手机为什么能跑大模型

同样一个模型,在云端跑得动,塞进手机却卡死,问题常常出在“精度”。量化(Quantization)就是把模型里成千上万的小数“四舍五入”成更小的数,体积和算力需求大幅降下来,效果却几乎不掉。这篇文章用…

AI 是怎么看懂一张照片的?一文看懂卷积神经网络

AI 是怎么看懂一张照片的?一文看懂卷积神经网络

你上传一张猫的照片,AI 立刻认出是猫;拍一张街景,软件分出车道和行人。可计算机最初只看到一堆像素点。卷积神经网络(CNN)就是让 AI 学会“看”的关键结构,它像用放大镜在图上滑动、提取边缘和纹理,…

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

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

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

生成对抗网络是什么?造假者与鉴别者的猫鼠游戏

生成对抗网络是什么?造假者与鉴别者的猫鼠游戏

大多数模型在「判断」,生成对抗网络(GAN)却另辟蹊径地「创造」——它能凭空画出人脸、生成逼真照片。诀窍是让两个模型互相较劲:一个拼命造假,一个拼命打假。本文用假币与验钞的比喻,讲清这场猫鼠游戏怎么把…

自编码器是什么?AI 怎么把复杂信息压缩再还原

自编码器是什么?AI 怎么把复杂信息压缩再还原

你手机传照片会先压缩,AI 也有一套类似的「打包术」叫自编码器:它先把数据压成一小团精华,再试着还原回去,逼自己只留住最要紧的特征。本文用搬家断舍离和缩略图的比喻,讲清它怎么顺手干成了降噪、找异常这些…

上下文窗口到底是什么?AI 一次到底能「读」进去多少字

上下文窗口到底是什么?AI 一次到底能「读」进去多少字

总听人说某模型「上下文 12 万 token」,这数到底管啥用?本文讲清上下文窗口是模型的「短期记忆容量」,为啥超了就忘事、长文档咋喂才不浪费,再聊聊这跟你能不能一口气聊完一整本书到底有关系,以及咱们…