← 最新论文
🤖 AI

FARI: Robust One-Step Inversion for Watermarking in Diffusion Models

FARI 是一个鲁棒的一步反向框架,它利用反向轨迹的低曲率和对抗性 LoRA 微调,使得在扩散模型中实现比传统多步反向方法更快、更鲁棒的水印验证。

原作者: Jindong Yang, Han Fang, Weiming Zhang, Nenghai Yu, Kejiang Chen

发布于 2026-07-30
📖 1 分钟阅读☕ 轻松阅读

原作者: Jindong Yang, Han Fang, Weiming Zhang, Nenghai Yu, Kejiang Chen

原始论文采用 CC BY 4.0 许可(http://creativecommons.org/licenses/by/4.0/)。 这是对下方论文的AI生成解释。它不是由作者撰写或认可的。如需技术准确性,请参阅原始论文。 阅读完整免责声明

想象一下这样一个世界:计算机只需阅读你输入的一句话,就能画出画作、谱写歌曲、设计电影。这就是“扩散模型”(diffusion models)的魔力,这是一种人工智能技术,它从一团混乱的静态噪声开始,通过一步步地清理,最终显现出一幅清晰的图像。你可以把它想象成一位雕塑家在从大理石块中凿去多余的部分,但过程恰好相反:人工智能从一团随机噪声开始,通过凿去“噪声”来揭示出一座完美的雕像。由于这些机器如此擅长创作艺术,人们产生了一种日益增长的担忧:我们如何知道一张图片是由人类还是机器人制作的?为了解决这个问题,科学家们开发了一种数字“水印”。这就像是在雕塑家开始凿刻之前,就在第一粒沙子(初始噪声)中隐藏了一个秘密代码。如果你想证明这座雕像是你的,你就必须逆转雕刻的过程,将完成的雕像还原回最初的那粒沙子,从而读取代码。

问题在于,逆转这个过程极其困难且缓慢。这就像试图“反向烘焙”一个蛋糕或“反向混合”一杯奶昔;你逆转的步骤越多,就越容易出错或洒掉原料。如果蛋糕被挤压在袋子里(被 JPEG 压缩或裁剪所扭曲),那么尝试“反向烘焙”就会变成一场噩梦。现有的方法试图做到极其精确,通过数百个微小的步骤来逆转过程,但这既耗时又在图像受损时失效。本文介绍了一种新的“反向烘焙”方法,它不仅速度极快,而且具有惊人的抗破坏能力。

这项工作的研究人员——杨金东及其团队,注意到了一种关于 AI 路径的奇特现象。当 AI 创作图像时,它就像一个在浓雾森林中穿行的徒步旅行者,不断变换方向以避开障碍物;路径是弯曲且不可预测的。然而,当你试图逆转过程以寻找原始噪声时,路径却变得非常笔直,就像徒步旅行者走在一条铺设良好的高速公路上。因为这条“逆向路径”如此笔直,你不需要数百个步骤来跟随它;你可以跨出一大步,直接落在你需要的地方。

该团队将他们的新方法称为 FARI(快速非对称鲁棒逆转)。传统方法在逆转图像时需要进行缓慢且谨慎的 5 对 50 步,而 FARI 仅需仅仅一步即可完成。这就像意识到你不需要通过房屋的每一个房间才能回到前门,你可以直接跳出窗户,落在门廊上。但巧妙之处在于:因为这一跳非常迅速,计算机可以更好地“学习”如何处理杂乱、受损的图像。他们使用一张强大的显卡(NVIDIA RTX A6000)对系统进行了约 20 分钟的训练,使其即使在图像被挤压、模糊或裁剪的情况下,也能熟练掌握这一单步跳跃。

结果令人印象深刻。在测试中,FARI 在处理受损图像中的隐藏水印方面,成功率超过了旧有的、缓慢的 50 步法。例如,在检查一种被称为“Tree-Ring”的特定类型水印时,即使在图像受到严重扭曲的情况下,FARI 仍能几乎 100% 地正确识别隐藏代码,而旧方法则显得力不从心。论文指出,通过跳过不必要的步骤,系统反而变得更加鲁棒,而非削弱。这有点像一位武术家意识到,比起缓慢复杂的招式舞动,一次快速、直接的重拳往往更加有效。

作者还指出,虽然其他方法试图在处理纯净图像时达到完美,但在面对杂乱图像时却会失败。FARI 接受了单步跳跃在处理完美图像时可能并不完美的事实,但在图像经历过磨难时,它的表现要优越得多。他们使用了一种名为 LoRA(低秩自适应)的技术来教会 AI 这个新技巧。你可以把 LoRA 想象成 AI 大脑中一个微小的、可拆卸的插件。当 AI 正在绘画时,这个插件是关闭的,以确保艺术品保持美观;但当有人尝试检查水印时,插件会被开启,以帮助 AI 快速且准确地跳回起点。

简而言之,本文认为我们并不需要通过缓慢和谨慎来换取准确性。通过理解“回到起点”的路径比“走向终点”的路径更简单,我们可以构建出一个既快速又强韧的系统。该团队发现,这种方法在验证水印(尤其是图像经过修改时)方面,比目前的尖端方法更有效。这提醒我们,有时解决问题的最快方式,就是停止过度思考步骤,而是完成一次大胆且训练有素的飞跃。

您所在领域的论文太多了?

获取与您研究关键词匹配的最新论文每日摘要——附技术摘要,使用您的语言。

试用 Digest →