Collaborative Threshold Watermarking
Tameem Bakr, Anish Ambreth, Nils Lukas
Abstract
In federated learning (FL), clients jointly train a model without sharing raw data. Because each participant invests data and compute, clients need mechanisms to later prove the provenance of a jointly trained model. Model watermarking embeds a hidden signal in the weights, but naive approaches either do not scale with many clients as per-client watermarks dilute as grows, or give any individual client the ability to verify and potentially remove the watermark. We introduce -threshold watermarking: clients collaboratively embed a shared watermark during training, while only coalitions of at least clients can reconstruct the watermark key and verify a suspect model. We secret-share the watermark key so that coalitions of fewer than clients cannot reconstruct it, and verification can be performed without revealing in the clear. We instantiate our protocol in the white-box setting and evaluate it on image classification tasks on both IID and non-IID partitions, as well as language models fine-tuning setting. Our watermark remains detectable at scale () with minimal accuracy loss and stays above the detection threshold () under attacks including adaptive fine-tuning using up to 20% of the training data. Code is available at https://github.com/tameemalaa/collaborative-threshold-watermark.
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 c2f69c92-93ec-41f4-98c0-306e818dc6e3Builds on5
- Practical Secure Aggregation for Privacy-Preserving Machine LearningKallista A. Bonawitz, Vladimir Ivanov, Ben Kreuter, Antonio Marcedone et al.CCS 2017 · 3,936 citations
- Turning Your Weakness Into a Strength: Watermarking Deep Neural Networks by BackdooringYossi Adi, Carsten Baum, Moustapha Cissé, Benny Pinkas et al.USENIX Security 2018 · 832 citations
- Entangled Watermarks as a Defense against Model ExtractionHengrui Jia, Christopher A. Choquette-Choo, Varun Chandrasekaran, Nicolas PapernotUSENIX Security 2021 · 287 citations
- RIGA: Covert and Robust White-Box Watermarking of Deep Neural NetworksTianhao Wang, Florian KerschbaumWWW 2021 · 128 citations
- Certified Neural Network Watermarks with Randomized SmoothingArpit Bansal, Ping-Yeh Chiang, Michael J. Curry, Rajiv Jain et al.ICML 2022 · 64 citations
Related papers
- Traceable Black-Box Watermarks For Federated LearningJiahao Xu, Rui Hu, Olivera Kotevska, Zikai ZhangICLR 2026 · 4 citations
- MFL-Owner: Ownership Protection for Multi-modal Federated Learning via Orthogonal Transform WatermarkKeke Gai, Dongjue Wang, Jing Yu, Mohan Wang et al.AAAI 2025 · 6 citations
- FedGMark: Certifiably Robust Watermarking for Federated Graph LearningYuxin Yang, Qiang Li, Yuan Hong, Binghui WangNeurIPS 2024 · 11 citations
- Aion: Robust and Efficient Multi-Round Single-Mask Secure Aggregation Against Malicious ParticipantsYizhong Liu, Zixiao Jia, Xiao Chen, Song Bian et al.USENIX Security 2025
- Defending against Backdoors in Federated Learning with Robust Learning RateMustafa Safa Özdayi, Murat Kantarcioglu, Yulia R. GelAAAI 2021 · 250 citations
