OmniMark: Efficient and Scalable Latent Diffusion Model Fingerprinting
Jianwei Fei, Yunshu Dai, Zhihua Xia, Fangjun Huang, Jiantao Zhou
Abstract
We introduce OmniMark, a novel and efficient fingerprinting method for Latent Diffusion Models (LDM). OmniMark can encode user-specific fingerprints across diverse dimensions of the weights of the LDM, including kernels, filters, channels, and spatial domains. The LDM is fine-tuned to encode the invisible fingerprint into generated images, which can be decoded by a decoder. By altering fingerprints and re-encoding the weights, OmniMark supports efficient and scalable adhoc generation (<100 ms) of numerous models with unique fingerprints that enable user accountability and model attribution. Extensive experiments demonstrate that OmniMark can be applied to various image generation and editing tasks and achieve highly accurate fingerprint detection without compromising image quality. Furthermore, OmniMark demonstrates good robustness against both white-box model attacks and image attacks, including fine-tuning and JPEG compression.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 1a1d8993-2c4b-4b30-b9b4-aec1057671e2Cited by top-tier papers3
- LiteGuard: Efficient Task-Agnostic Model Fingerprinting with Enhanced GeneralizationGuang Yang, Ziye Geng, Yihang Chen, Changqing LuoICLR 2026 · 2 citations
- Scalable Dual Fingerprinting for Hierarchical Attribution of Text-to-Image ModelsJianwei Fei, Yunshu Dai, Peipeng Yu, Zhe Kong et al.ICCV 2025 · 1 citation
- MSAT-LDM: Toward Transferable High-Fidelity Watermarking for Latent Diffusion Model via Modular Self-Augmented TrainingLu Zhang, Liang ZengAAAI 2026
Builds on11
- Sharpness-aware Minimization for Efficiently Improving GeneralizationPierre Foret, Ariel Kleiner, Hossein Mobahi, Behnam NeyshaburICLR 2021 · 1,861 citations
- Attributing Fake Images to GANs: Learning and Analyzing GAN FingerprintsNing Yu, Larry Davis, Mario FritzICCV 2019 · 533 citations
- Tree-Rings Watermarks: Invisible Fingerprints for Diffusion ImagesYuxin Wen, John Kirchenbauer, Jonas Geiping, Tom GoldsteinNeurIPS 2023 · 253 citations
- Responsible Disclosure of Generative Models Using Scalable FingerprintingNing Yu, Vladislav Skripniuk, Dingfan Chen, Larry S. Davis et al.ICLR 2022 · 118 citations
- Towards Discovery and Attribution of Open-world GAN Generated ImagesSharath Girish, Saksham Suri, Sai Saketh Rambhatla, Abhinav ShrivastavaICCV 2021 · 87 citations
Related papers
- WOUAF: Weight Modulation for User Attribution and Fingerprinting in Text-to-Image Diffusion ModelsChanghoon Kim, Kyle Min, Maitreya Patel, Sheng Cheng et al.CVPR 2024
- Attributing Image Generative Models using Latent FingerprintsGuangyu Nie, Changhoon Kim, Yezhou Yang, Yi RenICML 2023 · 23 citations
- OptMark: Robust Multi-bit Diffusion Watermarking via Inference Time OptimizationJiazheng Xing, Hai Ci, Hongbin Xu, Hangjie Yuan et al.AAAI 2026 · 2 citations
- How to Trace Latent Generative Model Generated Images without Artificial Watermark?Zhenting Wang, Vikash Sehwag, Chen Chen, Lingjuan Lyu et al.ICML 2024 · 24 citations
- Your Text Encoder Can Be an Object-Level Watermarking ControllerNaresh Kumar Devulapally, Mingzhen Huang, Vishal Asnani, Shruti Agarwal et al.ICCV 2025 · 1 citation
