arXiv:2410.04032cs.CV2024-10被引 17

用测试时训练定位图片篡改区域,零样本下效果显著提升

ForgeryTTT: Zero-Shot Image Manipulation Localization with Test-Time Training

  • 测试时微调模型,针对每张图动态优化定位能力
  • 在五个基准上定位准确率比零样本方法高20.1%
  • 适合应对真实世界中未见过的伪造图像场景

社交媒体日益充斥着逼真的虚假图像,难以信任内容。以往检测方法在新场景中表现不佳,因训练数据有限。为此,我们提出 ForgeryTTT,首个利用测试时训练(TTT)定位图像篡改区域的方法。该方法在训练阶段使用大规模合成数据,通过视觉变压器联合学习分类与定位任务;定位头输出掩码以标记篡改区域,输入令牌据此分为篡改与真实组,分别送入分类头进行区分。测试时,利用定位头预测掩码指导分类头更新图像编码器,实现自适应优化。同时,在每组令牌中采用经典丢弃策略,显著提升性能与效率。在五个标准基准上测试,尽管方法简单,定位准确率相比其他零样本方法提升20.1%,较非零样本技术提高4.3%。代码与数据将在发表后公开。

原文摘要 · Abstract (English)

Social media is increasingly plagued by realistic fake images, making it hard to trust content. Previous algorithms to detect these fakes often fail in new, real-world scenarios because they are trained on specific datasets. To address the problem, we introduce ForgeryTTT, the first method leveraging test-time training (TTT) to identify manipulated regions in images. The proposed approach fine-tunes the model for each individual test sample, improving its performance. ForgeryTTT first employs vision transformers as a shared image encoder to learn both classification and localization tasks simultaneously during the training-time training using a large synthetic dataset. Precisely, the localization head predicts a mask to highlight manipulated areas. Given such a mask, the input tokens can be divided into manipulated and genuine groups, which are then fed into the classification head to distinguish between manipulated and genuine parts. During test-time training, the predicted mask from the localization head is used for the classification head to update the image encoder for better adaptation. Additionally, using the classical dropout strategy in each token group significantly improves performance and efficiency. We test ForgeryTTT on five standard benchmarks. Despite its simplicity, ForgeryTTT achieves a 20.1% improvement in localization accuracy compared to other zero-shot methods and a 4.3% improvement over non-zero-shot techniques. Our code and data will be released upon publication.

图像伪造检测测试时训练零样本学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。