English

Robust Watermarking on Gradient Boosting Decision Trees

Artificial Intelligence 2025-11-14 v1

Abstract

Gradient Boosting Decision Trees (GBDTs) are widely used in industry and academia for their high accuracy and efficiency, particularly on structured data. However, watermarking GBDT models remains underexplored 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.

Keywords

Cite

@article{arxiv.2511.09822,
  title  = {Robust Watermarking on Gradient Boosting Decision Trees},
  author = {Jun Woo Chung and Yingjie Lao and Weijie Zhao},
  journal= {arXiv preprint arXiv:2511.09822},
  year   = {2025}
}

Comments

Accepted for publication at the Fortieth AAAI Conference on Artificial Intelligence (AAAI-26)

R2 v1 2026-07-01T07:34:49.776Z