论文标题

perastai:一个用于创建稀疏神经网络的轻量级库

FasterAI: A Lightweight Library for Creating Sparse Neural Networks

论文作者

Hubens, Nathan

论文摘要

Forperai是一个基于Pytorch的库,旨在促进深度神经网络压缩技术的利用,例如稀疏,修剪,知识蒸馏或正则化。该库的构建是为了实现快速实施和实验。尤其是,压缩技术是利用Fastai和Pytorch Lightning等库的回调系统来带来用户友好和高级API。 Forperai的主要资产是其轻巧但功能强大,使用的简单性。的确,由于它是以非常颗粒的方式开发的,因此用户可以使用不同的参数组合来创建数千个独特的实验。在本文中,我们专注于代表图书馆核心的porastai的稀疏功能。在porterai中对神经网络进行稀疏只需要在传统培训循环中进行单一的代码,但允许执行最先进的技术,例如彩票票证假设实验

FasterAI is a PyTorch-based library, aiming to facilitate the utilization of deep neural networks compression techniques such as sparsification, pruning, knowledge distillation, or regularization. The library is built with the purpose of enabling quick implementation and experimentation. More particularly, compression techniques are leveraging Callback systems of libraries such as fastai and Pytorch Lightning to bring a user-friendly and high-level API. The main asset of FasterAI is its lightweight, yet powerful, simplicity of use. Indeed, because it was developed in a very granular way, users can create thousands of unique experiments by using different combinations of parameters. In this paper, we focus on the sparsifying capabilities of FasterAI, which represents the core of the library. Performing sparsification of a neural network in FasterAI only requires a single additional line of code in the traditional training loop, yet allows to perform state-of-the-art techniques such as Lottery Ticket Hypothesis experiments

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源