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

TL;DR - FARI is a one-step inversion framework for authenticating watermarked diffusion-generated images. It outperforms 50-step DDIM inversion in verification robustness while substantially reducing inference time.

• Exploits the lower curvature of inversion trajectories to compress inversion into one step.
• Prioritizes robustness to external image distortions over minimizing comparatively smaller internal truncation errors.
• Uses lightweight adversarial LoRA fine-tuning of the denoiser to improve watermark extraction.
• Requires approximately 20 minutes of fine-tuning on one NVIDIA RTX A6000 GPU.