TriggerBN ++

motivation

用两个BN(一个用于干净样本, 一个用于对抗样本), 结果当使用 B N n a t \mathrm{BN}_{nat} BNnat的时候, 精度能够上升, 而使用 B N a d v \mathrm{BN}_{adv} BNadv的时候, 也有相当的鲁棒性. 原文采用的是
α L ( f ( x ) , y ) + ( 1 − α ) L ( f ( x + δ ) , y ) , \alpha \mathcal{L}(f(x), y) + (1-\alpha) \mathcal{L}(f(x+\delta), y), αL(f(x),y)+(1α)L(f(x+δ),y),
来训练(这里 f ( x ) f(x) f(x)输出的是概率向量而非logits), 试试看别的组合方式, 比如
L ( α f ( x n a t ) + ( 1 − α ) f ( x a d v ) , y ) . \mathcal{L}(\alpha f(x_{nat}) + (1-\alpha)f(x_{adv}) ,y). L(αf(xnat)+(1α)f(xadv),y).

settings

AttributeValue
attackpgd-linf
batch_size128
beta10.9
beta20.999
datasetcifar10
descriptionAT=0.5=default-sgd-0.1=pgd-linf-0.0314-0.25-10=128=default
epochs100
epsilon0.03137254901960784
learning_policy[50, 75] x 0.1
leverage0.5
losscross_entropy
lr0.1
modelresnet32
momentum0.9
optimizersgd
progressFalse
resumeFalse
seed1
stats_logFalse
steps10
stepsize0.25
transformdefault
weight_decay0.0005

results

x轴为 α \alpha α 0 0 0变化到 1 1 1.

AccuracyRobustness
0.5 L n a t + 0.5 L a d v 0.5 \mathcal{L}_{nat} + 0.5\mathcal{L}_{adv} 0.5Lnat+0.5Ladvimage-20210606124444152image-20210606124508889
L ( 0.5 p n a t + 0.5 p a d v , y ) \mathcal{L}(0.5 p_{nat} + 0.5p_{adv}, y) L(0.5pnat+0.5padv,y)image-20210610175902568image-20210610175919179
0.1 L n a t + 0.9 L a d v 0.1 \mathcal{L}_{nat} + 0.9\mathcal{L}_{adv} 0.1Lnat+0.9Ladv 48.350image-20210614083553167image-20210614083606765
L ( 0.1 p n a t + 0.9 p a d v , y ) \mathcal{L}(0.1 p_{nat} + 0.9p_{adv}, y) L(0.1pnat+0.9padv,y) 48.270image-20210613093454435image-20210613093517965
0.2 L n a t + 0.8 L a d v 0.2 \mathcal{L}_{nat} + 0.8\mathcal{L}_{adv} 0.2Lnat+0.8Ladv 48.310image-20210614093535960image-20210614093550959
L ( 0.2 p n a t + 0.8 p a d v , y ) \mathcal{L}(0.2 p_{nat} + 0.8p_{adv}, y) L(0.2pnat+0.8padv,y) 47.960image-20210613110736970image-20210613110749156

似乎原来的形式情况更好一点.

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值