Robust Watermarking on Gradient Boosting Decision Trees
Jun Woo Chung, Yingjie Lao, Weijie Zhao
Abstract
Gradient Boosting Decision Trees (GBDTs) are widely used in industry and academia for their high accuracy and efficiency, particularly on structured data. However, the subject of watermarking GBDT models remains underexplored, especially compared to neural networks. In this work, we present the first robust watermarking framework tailored to GBDT models, utilizing in-place fine-tuning to embed imperceptible and resilient watermarks. We propose four embedding strategies, each designed to minimize impact on model accuracy while ensuring watermark robustness. Through experiments across diverse datasets, we demonstrate that our methods achieve high watermark embedding rates, low accuracy degradation, and strong resistance to post-deployment fine-tuning.
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.
Builds on7
- 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
- Domain Watermark: Effective and Harmless Dataset Copyright Protection is Closed at HandJunfeng Guo, Yiming Li, Lixu Wang, Shu-Tao Xia et al.NeurIPS 2023 · 93 citations
- SketchBoost: Fast Gradient Boosted Decision Tree for Multioutput ProblemsLeonid Iosipoi, Anton VakhrushevNeurIPS 2022 · 20 citations
- Integrity Authentication in Tree ModelsWeijie Zhao, Yingjie Lao, Ping LiKDD 2022 · 3 citations
- Rethinking White-Box Watermarks on Deep Learning Models under Neural Structural ObfuscationYifan Yan, Xudong Pan, Mi Zhang, Min YangUSENIX Security 2023
Related papers
- DRGW: Learning Disentangled Representations for Robust Graph WatermarkingJiasen Li, Yanwei Liu, Zhuoyi Shang, Xiaoyan Gu et al.WWW 2026 · 3 citations
- Certified Neural Network Watermarks with Randomized SmoothingArpit Bansal, Ping-Yeh Chiang, Michael J. Curry, Rajiv Jain et al.ICML 2022 · 64 citations
- Achieving Resolution-Agnostic DNN-based Image Watermarking: A Novel Perspective of Implicit Neural RepresentationYuchen Wang, Xingyu Zhu, Guanhui Ye, Shiyao Zhang et al.ACM MM 2024 · 6 citations
- PTW: Pivotal Tuning Watermarking for Pre-Trained Image GeneratorsNils Lukas, Florian KerschbaumUSENIX Security 2023
- KGMark: A Diffusion Watermark for Knowledge GraphsHongrui Peng, Haolang Lu, Yuanlong Yu, Weiye Fu et al.ICML 2025
