Skip to content

Latest commit

 

History

History
59 lines (38 loc) · 1.82 KB

README.md

File metadata and controls

59 lines (38 loc) · 1.82 KB
Raw

RingID 🔍

RingID: Rethinking Tree-Ring Watermarking for Enhanced Multi-Key Identification

Hai Ci*  Pei Yang*  Yiren Song*  Mike Zheng Shou

National University of Singapore

Project Page | Arxiv

RingID presents a robust diffusion image watermarking approach to imprint multiple keys. It bases on the training-free approach Tree-Ring[1], but significantly enhances in both watermark verification and multi-key identification.

Method

RingID identifies the limitations in Tree-Ring's design and suggests a series of approaches for enhanced distinguishability and robustness.

Dependencies

  • PyTorch == 1.13.0
  • transformers == 4.23.1
  • diffusers == 0.11.1
  • datasets

Note: higher diffusers version may not be compatible with the DDIM inversion code.

Usage

Verification

python verify.py --run_name verification --online

Identification

python identify.py --run_name identification --online

Citation

@article{ci2024ringid,
  title={RingID: Rethinking Tree-Ring Watermarking for Enhanced Multi-Key Identification},
  author={Ci, Hai and Yang, Pei and Song, Yiren and Shou, Mike Zheng},
  journal={arXiv preprint arXiv:2404.14055},
  year={2024}
}

Acknowledgement

[1] Tree-ring watermarks: Fingerprints for diffusion images that are invisible and robust