arrow
返回

Sparsity-control ternary weight networks

delete2022-01-01
delete4
delete
OA
AI
X
Xiang Deng *
Z
Zhongfei Zhang
DOI:10.1016/j.neunet.2021.10.018delete
delete原文链接
delete分享
delete收藏
查看原文
摘要

摘要

En 中文
Deep neural networks (DNNs) have been widely and successfully applied to various applications, but they require large amounts of memory and computational power. This severely restricts their deployment on resource-limited devices. To address this issue, many efforts have been made on training low-bit weight DNNs. In this paper, we focus on training ternary weight {-1, 0, +1} networks which can avoid multiplications and dramatically reduce the memory and computation requirements. A ternary weight network can be considered as a sparser version of the binary weight counterpart by replacing some -1s or 1s in the binary weights with 0s, thus leading to more efficient inference but more memory cost. However, the existing approaches to train ternary weight networks cannot control the sparsity (i.e., percentage of 0s) of the ternary weights, which undermines the advantage of ternary weights. In this paper, we propose to our best knowledge the first sparsity-control approach (SCA) to train ternary weight networks, which is simply achieved by a weight discretization regularizer (WDR). SCA is different from all the existing regularizer-based approaches in that it can control the sparsity of the ternary weights through a controller alpha and does not rely on gradient estimators. We theoretically and empirically show that the sparsity of the trained ternary weights is positively related to alpha. SCA is extremely simple, easy-to-implement, and is shown to consistently outperform the state-of-the-art approaches significantly over several benchmark datasets and even matches the performances of the full-precision weight counterparts. (c) 2021 Elsevier Ltd. All rights reserved.
Keyword:
Ternary weight networks
Sparsity control
Model compression
Image classification
AI总结

AI总结

对已上传原文的论文进行重点信息的提取,主要内容包括:简要概述、研究摘要、背景介绍、关键亮点、图文解析、展望与总结。

期刊

Neural Networks 封面图
Neural Networks
IF:
6.3
论文数:
8.2K
被引数:
3.0W

机构

S
state university of new york (suny) system
学者数:
6.5W
论文数: 5.8W
被引数: 65
引用论文

引用论文

An epidemiologic study of insect allergy in children. I. Characteristics of the disease
err1982-04-01
err0
PREAI
errKenneth C. Schuberth; Lawrence M. Lichtenstein; Anne Kagey-Sobotka; Moyses Szklo; Kathleen A. Kwiterovich; Martin D. Valentine
err分享
err收藏
err分享
err收藏
err分享
err收藏
学者 查看更多内容