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

THAT: Two Head Adversarial Training for Improving Robustness at Scale

Fudan University(复旦大学);Facebook AI(Facebook人工智能研究院);University of Maryland(马里兰大学)

Zuxuan Wu, Tom Goldstein, Larry S. Davis, Ser-Nam Lim

英文摘要:

Many variants of adversarial training have been proposed, with most research focusing on problems with relatively few classes. In this paper, we propose Two Head Adversarial Training (THAT), a two-stream adversarial learning network that is designed to handle the large-scale many-class ImageNet dataset. The proposed method trains a network with two heads and two loss functions; one to minimize feature-space domain shift between natural and adversarial images, and one to promote high classification accuracy. This combination delivers a hardened network that achieves state of the art robust accuracy while maintaining high natural accuracy on ImageNet. Through extensive experiments, we demonstrate that the proposed framework outperforms alternative methods under both standard and "free" adversarial training settings.

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

↑