arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.02152cs.CVcs.AI

Group-wise Inhibition based Feature Regularization for Robust Classification

SZU Branch, Shenzhen Institute of Artificial Intelligence and Robotics for Society(深圳市人工智能与机器人研究院深圳大学分中心);National Engineering Laboratory for Big Data System Computing Technology(大数据系统计算技术国家工程实验室);Guangdong Key Laboratory of Intelligent Information Processing(广东省智能信息处理重点实验室)

Haozhe Liu, Haoqian Wu, Weicheng Xie, Feng Liu, Linlin Shen

英文摘要:

The convolutional neural network (CNN) is vulnerable to degraded images with even very small variations (e.g. corrupted and adversarial samples). One of the possible reasons is that CNN pays more attention to the most discriminative regions, but ignores the auxiliary features when learning, leading to the lack of feature diversity for final judgment. In our method, we propose to dynamically suppress significant activation values of CNN by group-wise inhibition, but not fixedly or randomly handle them when training. The feature maps with different activation distribution are then processed separately to take the feature independence into account. CNN is finally guided to learn richer discriminative features hierarchically for robust classification according to the proposed regularization. Our method is comprehensively evaluated under multiple settings, including classification against corruptions, adversarial attacks and low data regime. Extensive experimental results show that the proposed method can achieve significant improvements in terms of both robustness and generalization performances, when compared with the state-of-the-art methods. Code is available at https://github.com/LinusWu/TENET_Training.

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

↑