研究视觉变压器在医学图像中对对抗水印的脆弱性及防御效果。
Exploring Adversarial Watermarking in Transformer-Based Models: Transferability and Robustness Against Defense Mechanism for Medical Images
- 用PGD生成对抗水印,测试其跨模型迁移能力。
- 视觉变压器在含水印图像上准确率最低下降至27.6%。
- 对抗训练可将准确率提升至90.0%,增强模型鲁棒性。
深度学习模型在皮肤科图像分析中表现卓越,有望实现皮肤病自动诊断。尽管基于卷积神经网络(CNN)的架构在计算机视觉任务中长期占据主导地位,但近年来基于变换器的模型(如视觉变换器,ViTs)在图像识别、生成和视频分析等任务中逐渐成为主流。ViTs利用自注意力机制,在多个任务上达到领先性能。然而,其依赖全局注意力机制的特点使其容易受到对抗扰动的影响。本文旨在研究ViTs在医学图像上对对抗水印攻击的敏感性——一种通过添加难以察觉的扰动来误导模型的方法。通过使用投影梯度下降(PGD)生成对抗水印,我们考察了此类攻击在不同模型间的迁移能力,并评估了对抗训练作为防御机制的有效性。结果表明,虽然干净图像上的性能未受影响,但ViTs对对抗攻击显著脆弱:准确率最低下降至27.6%。然而,经过对抗训练后,准确率可恢复至90.0%。
原文摘要 · Abstract (English)
Deep learning models have shown remarkable success in dermatological image analysis, offering potential for automated skin disease diagnosis. Previously, convolutional neural network(CNN) based architectures have achieved immense popularity and success in computer vision (CV) based task like skin image recognition, generation and video analysis. But with the emergence of transformer based models, CV tasks are now are nowadays carrying out using these models. Vision Transformers (ViTs) is such a transformer-based models that have shown success in computer vision. It uses self-attention mechanisms to achieve state-of-the-art performance across various tasks. However, their reliance on global attention mechanisms makes them susceptible to adversarial perturbations. This paper aims to investigate the susceptibility of ViTs for medical images to adversarial watermarking-a method that adds so-called imperceptible perturbations in order to fool models. By generating adversarial watermarks through Projected Gradient Descent (PGD), we examine the transferability of such attacks to CNNs and analyze the performance defense mechanism -- adversarial training. Results indicate that while performance is not compromised for clean images, ViTs certainly become much more vulnerable to adversarial attacks: an accuracy drop of as low as 27.6%. Nevertheless, adversarial training raises it up to 90.0%.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。