首页/问答百科/模型剪枝(Pruning)
🔧 技术与方法

模型剪枝(Pruning)

模型剪枝权重剪枝结构化剪枝非结构化剪枝Neural Network Pruning

模型剪枝是一种模型压缩技术,通过移除神经网络中不重要的权重、神经元或通道来减小模型体积和加速推理,是模型部署的关键优化手段。


模型剪枝是什么

模型剪枝(Model Pruning)是神经网络模型压缩的核心技术之一,通过剔除冗余或重要性低的参数来缩小模型规模,同时尽量保持模型精度。

分类方式

按粒度,剪枝分为非结构化剪枝(移除单个权重,产生稀疏矩阵)和结构化剪枝(移除整行、整列、整个通道或注意力头)。结构化剪枝更受工业界青睐,因其直接减少计算量且对硬件更友好。按时机,分为训练后剪枝剪枝即训练

常见方法

主要方法包括:幅度剪枝(Magnitude Pruning),以权重绝对值为重要性指标,简单高效;梯度感知剪枝(如 Movement Pruning),结合梯度信息动态评估参数重要性;结构化剪枝(如 Channel Pruning),按通道或注意力头裁剪,直接加速推理。

与量化的关系

剪枝通常与模型量化配合使用,是模型优化的两大支柱。剪枝移除冗余参数,量化减少参数位宽,二者结合可达到数倍到数十倍的压缩率。


相关术语

分享: