论文标题

Mirst-DM:多稳定RST带有滴度最大层用于良好的乳腺癌分类

MIRST-DM: Multi-Instance RST with Drop-Max Layer for Robust Classification of Breast Cancer

论文作者

Sun, Shoukun, Xian, Min, Vakanski, Aleksandar, Ghanem, Hossny

论文摘要

强大的自我训练(RST)可以增强图像分类模型的对抗鲁棒性,而无需显着牺牲模型的通用性。但是,RST和其他最先进的防御方法未能保留普遍性,并在小型医学图像集上重现其良好的对抗性鲁棒性。在这项工作中,我们提出了具有滴度最大层的多命名RST,即Mirst-DM,它涉及训练期间的一系列迭代生成的对抗实例,以学习小型数据集上的更轻松的决策边界。提出的滴度最大层消除了不稳定的功能,并有助于学习图像扰动的强大表示形式。使用带有1,190张图像的小型乳房超声数据集对所提出的方法进行了验证。结果表明,所提出的方法实现了对三种普遍攻击的最先进的对抗性鲁棒性。

Robust self-training (RST) can augment the adversarial robustness of image classification models without significantly sacrificing models' generalizability. However, RST and other state-of-the-art defense approaches failed to preserve the generalizability and reproduce their good adversarial robustness on small medical image sets. In this work, we propose the Multi-instance RST with a drop-max layer, namely MIRST-DM, which involves a sequence of iteratively generated adversarial instances during training to learn smoother decision boundaries on small datasets. The proposed drop-max layer eliminates unstable features and helps learn representations that are robust to image perturbations. The proposed approach was validated using a small breast ultrasound dataset with 1,190 images. The results demonstrate that the proposed approach achieves state-of-the-art adversarial robustness against three prevalent attacks.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源