arXiv:2412.14598cs.CV2024-12被引 56

用稀疏注意力机制自动提取图像篡改定位特征,无需人工设计特征提取器。

Can We Get Rid of Handcrafted Feature Extractors? SparseViT: Nonsemantics-Centered, Parameter-Efficient Image Manipulation Localization through Spare-Coding Transformer

  • 通过稀疏离散自注意力打破图像语义,强制模型关注篡改敏感的非语义特征。
  • 在多个基准数据集上超越现有方法,参数量减少最高达80%。
  • 适合追求高效、强泛化能力的图像篡改检测研究与应用。

非语义特征(即与图像上下文无关但对篡改敏感的特征)是图像篡改定位(IML)的关键证据。现有方法依赖人工设计的特征提取器,导致模型在未见或复杂场景下泛化能力差。本文提出稀疏视觉变压器(SparseViT),将ViT中的密集全局自注意力重构为稀疏离散形式。该设计打破图像语义结构,迫使模型自适应提取非语义特征。由于非语义特征在图像块间具有一致性(篡改时除外),稀疏交互足以捕捉其特性;而语义差异大的块则需密集交互学习。因此,稀疏自注意力机制显著降低模型复杂度:最大可减少80%的计算量(FLOPs)。大量实验表明,不依赖任何手工特征提取器,SparseViT在多个基准数据集上均实现更优的泛化性能与效率。

原文摘要 · Abstract (English)

Non-semantic features or semantic-agnostic features, which are irrelevant to image context but sensitive to image manipulations, are recognized as evidential to Image Manipulation Localization (IML). Since manual labels are impossible, existing works rely on handcrafted methods to extract non-semantic features. Handcrafted non-semantic features jeopardize IML model's generalization ability in unseen or complex scenarios. Therefore, for IML, the elephant in the room is: How to adaptively extract non-semantic features? Non-semantic features are context-irrelevant and manipulation-sensitive. That is, within an image, they are consistent across patches unless manipulation occurs. Then, spare and discrete interactions among image patches are sufficient for extracting non-semantic features. However, image semantics vary drastically on different patches, requiring dense and continuous interactions among image patches for learning semantic representations. Hence, in this paper, we propose a Sparse Vision Transformer (SparseViT), which reformulates the dense, global self-attention in ViT into a sparse, discrete manner. Such sparse self-attention breaks image semantics and forces SparseViT to adaptively extract non-semantic features for images. Besides, compared with existing IML models, the sparse self-attention mechanism largely reduced the model size (max 80% in FLOPs), achieving stunning parameter efficiency and computation reduction. Extensive experiments demonstrate that, without any handcrafted feature extractors, SparseViT is superior in both generalization and efficiency across benchmark datasets.

图像篡改稀疏注意力视觉Transformer特征提取

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