HyperAIHyperAI

Command Palette

Search for a command to run...

免费的对抗训练!

Ali Shafahi Mahyar Najibi Amin Ghiasi Zheng Xu John Dickerson Christoph Studer Larry S. Davis Gavin Taylor Tom Goldstein

摘要

对抗训练是一种通过在对抗样本上训练网络来防御对抗攻击的方法,是少数能够抵御强攻击的有效手段之一。不幸的是,生成强对抗样本的高成本使得标准的对抗训练在大规模问题(如ImageNet)上难以实施。我们提出了一种算法,通过循环利用更新模型参数时计算的梯度信息,消除了生成对抗样本的额外开销。我们的“免费”对抗训练算法在CIFAR-10和CIFAR-100数据集上实现了与PGD(Projected Gradient Descent)对抗训练相当的鲁棒性,其附加成本几乎可以忽略不计,并且比其他强对抗训练方法快7到30倍。使用配备4个P100 GPU的工作站运行2天,我们可以在大规模ImageNet分类任务中训练出一个鲁棒模型,在面对PGD攻击时仍能保持40%的准确率。代码已发布在https://github.com/ashafahi/free_adv_train


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供