只有加法也能做深度学习的 AdderNet 开源了
source link: https://www.infoq.cn/article/JiAHH0phDlAgmSAYiHRr
Go to the source link to view the article. You can view the picture content, updated content and better typesetting reading experience. If the link is broken, please click the button below to view the snapshot at that time.
2 月 25 日,华为诺亚方舟实验室将 AdderNet 正式开源!
一个月以前,一篇名为《我们真的需要在深度学习中使用乘法吗?》的论文刷爆朋友圈,这篇来自华为诺亚方舟实验室与北大合作的论文提出了一种新的方法,名为: AdderNet。
根据论文介绍:与加法运算相比,乘法运算具有更高的计算复杂度。在深度神经网络中被广泛使用的卷积计算,正是衡量输入特征和卷积滤波器之间相似度的交叉相关计算,这涉及到浮点值之间的大量乘法。而 AdderNet 可以将深度神经网络中,特别是卷积神经网络中的乘法,转换为更简单的加法运算,以便减少计算成本。
在 AdderNet 中,研究者采用了 L1 正则距离,用于计算滤波器和输入特征之间的距离,分析了这种新的相似性度量方法对神经网络优化的影响。为了获得更好的性能,研究人员还开发了一种特殊的反向传播方法来研究 AdderNet 的全精度梯度。然后,根据每个神经元梯度的大小,提出一种自适应学习速率策略来增强 AdderNet 的训练过程。
因此,在不使用卷积乘法的情况下,AdderNet 在 ResNet-50 上对 ImageNet 数据集进行训练后,能够取得 74.9% 的 top-1 精确度和 91.7% 的 top-5 精确度,如下图所示:
开源后,用户可以在自己的设备上对 AdderNet 进行配置,需要的基础环境为:
- python 3
- pytorch >= 1.1.0
- torchvision
用户可以按照 pytorch / examples 准备 ImageNet 数据。此外,官方表示预训练的模型将很快发布。
目前,开发者可以通过运行如下代码来对 ImageNet val 集进行评估:
复制代码
pythontest.py--data_dir'path/to/imagenet_root/'
在 ResNet-50 上对 ImageNet 数据集训练,用户将达到 74.9%的最高精度和 91.7%的 Top-5 精度。开发者可尝试运行如下代码,在 CIFAR-10 上进行求值运算:
复制代码
pythontest.py--datasetcifar10--model_dirmodels/ResNet20-AdderNet.pth--data_dir'path/to/cifar10_root/'
此外,在 ResNet-20 上对 CIFAR-10 数据集进行训练,用户将达到 91.8%的精度。
由于加法器过滤器是由 torch.cdist 实现的,所以 AdderNet 的推理很慢,但是用户可以编写 cuda 以获得更高的推理速度。
虽然 AdderNet 简化了运算成本,但是有一些反对的声音认为:类似的工作是可以通过硬件的优化(如 AI 芯片)来解决的,因此加法的优势其实并不算明显。如果有读者对 AdderNet 有更多见解,欢迎在留言区发表您的观点。
GitHub 开源链接:
https://github.com/huawei-noah/AdderNet
论文地址:
Recommend
-
30
-
43
风格迁移是VGG网络架构的拿手好戏。难道其他架构的CNN,比如ResNet就不可以吗?可以,但是直接用它来做风格迁移,原来的风格仍然非常明显,几乎没有迁移。 就像这样:
-
9
点击 蓝字 关注我们 扫码关注我们 公众号 : 计算机视觉战队 扫码回复: 加法网络 ,获取源码及论文链接 CVPR2020收录的结果已...
-
17
让你眼里的人,只有心中的那一个——用人工智能做单人CUT同济大学 计算机科学与技术硕士在读无聊是第一生产力,尤其是在吃饱了撑的的情况下;而懒又是另一个促进科技...
-
14
加法与减法,高频打低频:拼多多的取舍场长阿尔法工场创始人只关...
-
8
《自然》论文:人们更倾向于做加法而不是做减法解决问题 来源:中国新闻网202...
-
7
先做加法,再做减法,看看一个行动派是如何在知乎上赚钱的?非著名程序员公众号「非著名程序员」主理人,程序员/复业者/生涯规划师...
-
13
poj 1503 高精度加法 2013-09-07 分类:未分类 阅读(4401) 评论(0) 把输入...
-
1
只有“疯子”才能做动画观娱象限54分钟前比动画更热血的,是中国的动画人。从月初的《俑之城》,到即将播出的《白蛇2:青蛇劫起》《冲出地球》,尽管前者...
-
2
“断卡行动”下的银行人:有误伤,我们能做的只有解释36氪的朋友们·2022-08-14 03:00如何保证客户体验? 近日,有网友反映...
About Joyk
Aggregate valuable and interesting links.
Joyk means Joy of geeK