arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2202.09284cs.LG
收藏

顺从稀疏网络调查器

Amenable Sparse Network Investigator

University of Maryland, Baltimore County(马里兰大学巴尔的摩县分校)

Saeed Damadi, Erfan Nouri, Hamed Pirsiavash

摘要:

我们提出了“顺从稀疏网络调查员”(ASNI)算法,该算法利用了一种新的修剪策略,该策略基于一个S形函数,该函数在一轮训练过程中全局诱导稀疏水平。ASNI算法实现了当前最先进的策略只能完成其中一个任务的两个任务。ASNI算法有两个子算法:1)ASNI-I,2)ASNI-II。ASNI-I仅在一轮训练中就能学习到精确的稀疏现成网络。ASNI-II学习一个稀疏网络和一个初始化,该初始化是量化的,压缩的,并且稀疏网络是可训练的。学习的初始化被量化,因为对于每个层L中的非零参数的初始化仅学习两个数字。因此,用于整个网络的初始化的量化级别是2L。此外,学习的初始化被压缩,因为它是由2L个数字组成的集合。可以从这种量化和压缩的初始化训练的特殊稀疏网络被称为顺从的。据我们所知,没有其他算法可以学习量化和压缩的初始化,网络仍然是可训练的,并且能够解决两个修剪任务。我们的数值实验表明,有一个量化和压缩的初始化,学习稀疏网络可以训练,并达到与密集版本的精度。我们的实验表明,这2L级的量化是在ASNI-I学习稀疏网络的每一层中的参数的集中点。为了证实上述观点,我们在ImageNet、CIFAR 10和MNIST数据集上利用ResNets、VGG风格、小卷积和全连接网络等网络进行了一系列实验。

英文摘要:

We present "Amenable Sparse Network Investigator" (ASNI) algorithm that utilizes a novel pruning strategy based on a sigmoid function that induces sparsity level globally over the course of one single round of training. The ASNI algorithm fulfills both tasks that current state-of-the-art strategies can only do one of them. The ASNI algorithm has two subalgorithms: 1) ASNI-I, 2) ASNI-II. ASNI-I learns an accurate sparse off-the-shelf network only in one single round of training. ASNI-II learns a sparse network and an initialization that is quantized, compressed, and from which the sparse network is trainable. The learned initialization is quantized since only two numbers are learned for initialization of nonzero parameters in each layer L. Thus, quantization levels for the initialization of the entire network is 2L. Also, the learned initialization is compressed because it is a set consisting of 2L numbers. The special sparse network that can be trained from such a quantized and compressed initialization is called amenable. To the best of our knowledge, there is no other algorithm that can learn a quantized and compressed initialization from which the network is still trainable and is able to solve both pruning tasks. Our numerical experiments show that there is a quantized and compressed initialization from which the learned sparse network can be trained and reach to an accuracy on a par with the dense version. We experimentally show that these 2L levels of quantization are concentration points of parameters in each layer of the learned sparse network by ASNI-I. To corroborate the above, we have performed a series of experiments utilizing networks such as ResNets, VGG-style, small convolutional, and fully connected ones on ImageNet, CIFAR10, and MNIST datasets.

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

↑