Agent Tuning & Optimization 相关度: 5/10

Bonsai: A Framework for Convolutional Neural Network Acceleration Using Criterion-Based Pruning

Joseph Bingham, Sam Helmich
arXiv: 2602.17145v1 发布: 2026-02-19 更新: 2026-02-19

AI 摘要

Bonsai框架提出了一种基于准则的CNN剪枝方法,旨在加速和压缩模型。

主要贡献

  • 提出了Combine剪枝框架
  • 比较了不同准则对模型的影响
  • 提出了一些新的准则函数

方法论

迭代地移除CNN中不重要的权重(filters),基于不同的准则函数来决定移除哪些权重。

原文摘要

As the need for more accurate and powerful Convolutional Neural Networks (CNNs) increases, so too does the size, execution time, memory footprint, and power consumption. To overcome this, solutions such as pruning have been proposed with their own metrics and methodologies, or criteria, for how weights should be removed. These solutions do not share a common implementation and are difficult to implement and compare. In this work, we introduce Combine, a criterion- based pruning solution and demonstrate that it is fast and effective framework for iterative pruning, demonstrate that criterion have differing effects on different models, create a standard language for comparing criterion functions, and propose a few novel criterion functions. We show the capacity of these criterion functions and the framework on VGG inspired models, pruning up to 79\% of filters while retaining or improving accuracy, and reducing the computations needed by the network by up to 68\%.

标签

CNN Pruning Acceleration Model Compression

arXiv 分类

cs.AI