arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2312.17615cs.CV

图卷积网络的一次性多速率剪枝

One-Shot Multi-Rate Pruning of Graph Convolutional Networks

Sorbonne University(索邦大学);CNRS(法国国家科学研究中心);LIP6(LIP6实验室)

Hichem Sahbi

AI总结:

提出多速率幅度剪枝(MRMP)方法,联合训练图卷积网络拓扑与权重,实现任意剪枝率的一次性剪枝,在骨架识别任务上高剪枝率下性能显著提升。

摘要:

在本文中,我们设计了一种新颖的轻量级图卷积网络(GCN)结构,称为多速率幅度剪枝(MRMP),该结构联合训练网络拓扑和权重。我们的方法是变分的,通过将学习到的网络的权重分布与先验分布对齐来进行。一方面,这允许实现任何固定的剪枝率,并增强所设计的轻量级GCN的泛化性能。另一方面,MRMP在共享权重的基础上实现了多个GCN的联合训练,以便在任何目标剪枝率下外推准确的网络,而无需重新训练其权重。在具有挑战性的基于骨架的识别任务上进行的大量实验表明,我们的轻量级GCN在极高剪枝率下尤其具有显著优势。

英文摘要:

In this paper, we devise a novel lightweight Graph Convolutional Network (GCN) design dubbed as Multi-Rate Magnitude Pruning (MRMP) that jointly trains network topology and weights. Our method is variational and proceeds by aligning the weight distribution of the learned networks with an a priori distribution. In the one hand, this allows implementing any fixed pruning rate, and also enhancing the generalization performances of the designed lightweight GCNs. In the other hand, MRMP achieves a joint training of multiple GCNs, on top of shared weights, in order to extrapolate accurate networks at any targeted pruning rate without retraining their weights. Extensive experiments conducted on the challenging task of skeleton-based recognition show a substantial gain of our lightweight GCNs particularly at very high pruning regimes.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑