arXiv:2507.18988cs.CVcs.CR2025-07AAAI被引 6

无需训练,通过双重建比提升生成图像溯源准确率

AEDR: Training-Free AI-Generated Image Attribution via Autoencoder Double-Reconstruction

  • 用连续自编码器进行两次重建,取损失比作为溯源信号
  • 在8个顶级扩散模型上准确率提升25.5%,计算量仅为1%
  • 自动校准图像复杂度偏差,适合高效率溯源场景

图像生成技术的快速发展使任何人均可使用生成模型创作逼真图像,引发重大安全问题。为防止滥用,追踪图像来源至关重要。基于重构的溯源方法虽有前景,但在先进生成模型上常面临准确率下降与计算成本高的问题。为此,我们提出AEDR(AutoEncoder Double-Reconstruction),一种面向连续自编码器生成模型的无训练溯源方法。不同于依赖单次重构损失的现有方法,AEDR利用模型自编码器执行两次连续重构,并以两次重构损失的比值作为溯源信号。该信号进一步通过图像同质性指标校准,有效消除由图像复杂度带来的绝对偏差;同时,基于自编码器的重构机制确保了卓越的计算效率。在八个顶尖潜在扩散模型上的实验表明,AEDR相较现有重构方法实现25.5%的准确率提升,且仅需1%的计算时间。

原文摘要 · Abstract (English)

The rapid advancement of image-generation technologies has made it possible for anyone to create photorealistic images using generative models, raising significant security concerns. To mitigate malicious use, tracing the origin of such images is essential. Reconstruction-based attribution methods offer a promising solution, but they often suffer from reduced accuracy and high computational costs when applied to state-of-the-art (SOTA) models. To address these challenges, we propose AEDR (AutoEncoder Double-Reconstruction), a novel training-free attribution method designed for generative models with continuous autoencoders. Unlike existing reconstruction-based approaches that rely on the value of a single reconstruction loss, AEDR performs two consecutive reconstructions using the model's autoencoder, and adopts the ratio of these two reconstruction losses as the attribution signal. This signal is further calibrated using the image homogeneity metric to improve accuracy, which inherently cancels out absolute biases caused by image complexity, with autoencoder-based reconstruction ensuring superior computational efficiency. Experiments on eight top latent diffusion models show that AEDR achieves 25.5% higher attribution accuracy than existing reconstruction-based methods, while requiring only 1% of the computational time.

图像溯源自编码器生成模型无训练

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