arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2005.01699cs.LGcs.ITmath.ITstat.ML

Depth-2 Neural Networks Under a Data-Poisoning Attack

University of Florida(佛罗里达大学);The University of Manchester(曼彻斯特大学)

Sayar Karmakar, Anirbit Mukherjee, Theodore Papamarkou

英文摘要:

In this work, we study the possibility of defending against data-poisoning attacks while training a shallow neural network in a regression setup. We focus on doing supervised learning for a class of depth-2 finite-width neural networks, which includes single-filter convolutional networks. In this class of networks, we attempt to learn the network weights in the presence of a malicious oracle doing stochastic, bounded and additive adversarial distortions on the true output during training. For the non-gradient stochastic algorithm that we construct, we prove worst-case near-optimal trade-offs among the magnitude of the adversarial attack, the weight approximation accuracy, and the confidence achieved by the proposed algorithm. As our algorithm uses mini-batching, we analyze how the mini-batch size affects convergence. We also show how to utilize the scaling of the outer layer weights to counter output-poisoning attacks depending on the probability of attack. Lastly, we give experimental evidence demonstrating how our algorithm outperforms stochastic gradient descent under different input data distributions, including instances of heavy-tailed distributions.

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

↑