arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.00148cs.LGcs.CRcs.CV

Rethinking Feature Uncertainty in Stochastic Neural Networks for Adversarial Robustness

National University of Defense Technology(国防科技大学)

Hao Yang, Min Wang, Zhengfei Yu, Yun Zhou

英文摘要:

It is well-known that deep neural networks (DNNs) have shown remarkable success in many fields. However, when adding an imperceptible magnitude perturbation on the model input, the model performance might get rapid decrease. To address this issue, a randomness technique has been proposed recently, named Stochastic Neural Networks (SNNs). Specifically, SNNs inject randomness into the model to defend against unseen attacks and improve the adversarial robustness. However, existed studies on SNNs mainly focus on injecting fixed or learnable noises to model weights/activations. In this paper, we find that the existed SNNs performances are largely bottlenecked by the feature representation ability. Surprisingly, simply maximizing the variance per dimension of the feature distribution leads to a considerable boost beyond all previous methods, which we named maximize feature distribution variance stochastic neural network (MFDV-SNN). Extensive experiments on well-known white- and black-box attacks show that MFDV-SNN achieves a significant improvement over existing methods, which indicates that it is a simple but effective method to improve model robustness.

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

↑