arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.20114cs.CVcs.AI
收藏

对视觉变换器的去学习进行基准测试

Benchmarking Unlearning for Vision Transformers

University of Warwick, United Kingdom(沃里克大学)

Kairan Zhao, Iurie Luca, Peter Triantafillou

AI总结:

本研究首次对视觉变换器的去学习算法进行基准测试,评估不同数据集、算法和协议的影响,揭示了VTs在记忆训练数据方面的特性及性能基线。

摘要:

机器去学习(MU)的研究已获得强劲进展:MU现在被广泛认为是构建安全和公平AI的关键能力。同时,针对计算机视觉任务的变换器架构研究也取得了巨大成功:越来越多地,视觉变换器(VTs)作为CNNs的有力替代品出现。然而,针对视觉任务的MU研究主要集中在CNNs上,而非VTs。尽管基准测试努力已涉及大语言模型、扩散模型和CNNs,但针对VTs的尚不存在。本工作首次尝试此问题,对不同VT家族(ViT和Swin-T)以及不同容量下的MU算法性能进行基准测试。该工作采用(i)不同数据集,以评估数据集规模和复杂性的影响;(ii)不同MU算法,以代表不同MU方法;以及(iii)单次和持续去学习协议。此外,它专注于评估利用训练数据记忆的MU算法,因为利用记忆已被最近发现显著提高先前SOTA算法的性能。在此过程中,该工作阐述了VTs相对于CNNs在记忆训练数据方面的特性,并评估了不同记忆代理对性能的影响。该基准测试使用统一的评估指标,捕捉了两种互补的遗忘质量概念,以及在未见过(测试)数据和保留数据上的准确性。总体而言,本工作提供了一个基准测试基础,使现有(和未来的)MU算法在VTs上的可重复、公平和全面比较成为可能。此外,首次揭示了现有算法在VT设置中的表现,建立了有前景的参考性能基线。

英文摘要:

Machine unlearning (MU) refers to the post-training capability to remove (the influence of) training examples that are incorrect, biased, or leak sensitive/private information. MU is now widely regarded as critical for building safe and fair AI. In parallel, research into transformer architectures for computer vision has been highly successful: Vision Transformers (VTs) increasingly emerge as strong alternatives to CNNs. Yet, MU research for vision tasks has largely centered on CNNs, not VTs. While MU benchmarks have been developed for LLMs, diffusion models, and CNNs, none currently exist for VTs. This work is the first to attempt this, benchmarking MU algorithm performance across different VT families (ViT, Swin-T, and DINOv2) and at different capacities. The work employs (i) different datasets, selected to assess the impacts of dataset scale and complexity; (ii) different MU algorithms, selected to represent fundamentally different approaches for MU; and (iii) both single-shot and continual unlearning protocols. Additionally, it focuses on benchmarking MU algorithms that leverage training data memorization, since leveraging memorization has been recently discovered to significantly improve the performance of previously SOTA algorithms. En route, the work characterizes how VTs memorize training data relative to CNNs, and assesses the impact of different memorization proxies on performance. The benchmark uses unified evaluation metrics that capture two complementary notions of forget quality along with accuracy on unseen (test) data and on retained data. Overall, this work offers a benchmarking basis, enabling reproducible, fair, and comprehensive comparisons of existing (and future) MU algorithms on VTs. Importantly, for the first time, it sheds light on how well existing algorithms work in VT settings, establishing a promising reference performance baseline.

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

↑