arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2001.03311cs.LGstat.ML

Guess First to Enable Better Compression and Adversarial Robustness

University of Virginia(弗吉尼亚大学);State University of New York at Buffalo(纽约州立大学布法罗分校);University of Electronic Science and Technology of China(电子科技大学)

Sicheng Zhu, Bang An, Shiyu Niu

英文摘要:

Machine learning models are generally vulnerable to adversarial examples, which is in contrast to the robustness of humans. In this paper, we try to leverage one of the mechanisms in human recognition and propose a bio-inspired classification framework in which model inference is conditioned on label hypothesis. We provide a class of training objectives for this framework and an information bottleneck regularizer which utilizes the advantage that label information can be discarded during inference. This framework enables better compression of the mutual information between inputs and latent representations without loss of learning capacity, at the cost of tractable inference complexity. Better compression and elimination of label information further bring better adversarial robustness without loss of natural accuracy, which is demonstrated in the experiment.

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

↑