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

Noise Optimization for Artificial Neural Networks

Institute of Computing Technology, Chinese Academy of Sciences(中国科学院计算技术研究所);Peking University(北京大学);Huazhong University of Science and Technology(华中科技大学)

Li Xiao, Zeliang Zhang, Yijie Peng

英文摘要:

Adding noises to artificial neural network(ANN) has been shown to be able to improve robustness in previous work. In this work, we propose a new technique to compute the pathwise stochastic gradient estimate with respect to the standard deviation of the Gaussian noise added to each neuron of the ANN. By our proposed technique, the gradient estimate with respect to noise levels is a byproduct of the backpropagation algorithm for estimating gradient with respect to synaptic weights in ANN. Thus, the noise level for each neuron can be optimized simultaneously in the processing of training the synaptic weights at nearly no extra computational cost. In numerical experiments, our proposed method can achieve significant performance improvement on robustness of several popular ANN structures under both black box and white box attacks tested in various computer vision datasets.

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

↑