arXiv:2502.01634cs.LGcs.AI2025-02被引 3

让梯度提升树支持实时增删数据,高效更新模型。

Online Gradient Boosting Decision Tree: In-Place Updates for Efficient Adding/Deleting Data

  • 提出在线学习框架,支持原地增删数据。
  • 可实时更新模型,且保持较高准确率。
  • 适合需要动态数据更新的场景,如安全检测。

梯度提升决策树(GBDT)是广泛应用的机器学习模型。传统训练方式要求所有数据一次性加载,无法在训练后添加或删除数据。本文提出一种高效的在线学习框架,支持增量与减量学习。据我们所知,这是首个在GBDT上实现原地统一增量与减量学习的工作。为降低学习成本,设计了一系列优化策略,可在不重训练的情况下实时增删少量数据。理论分析揭示了优化超参数间的关联,可权衡精度与计算开销。后门攻击实验表明,该框架能成功在已训练模型中注入和移除后门。在公开数据集上的实验证明了该框架的有效性与高效性。

原文摘要 · Abstract (English)

Gradient Boosting Decision Tree (GBDT) is one of the most popular machine learning models in various applications. However, in the traditional settings, all data should be simultaneously accessed in the training procedure: it does not allow to add or delete any data instances after training. In this paper, we propose an efficient online learning framework for GBDT supporting both incremental and decremental learning. To the best of our knowledge, this is the first work that considers an in-place unified incremental and decremental learning on GBDT. To reduce the learning cost, we present a collection of optimizations for our framework, so that it can add or delete a small fraction of data on the fly. We theoretically show the relationship between the hyper-parameters of the proposed optimizations, which enables trading off accuracy and cost on incremental and decremental learning. The backdoor attack results show that our framework can successfully inject and remove backdoor in a well-trained model using incremental and decremental learning, and the empirical results on public datasets confirm the effectiveness and efficiency of our proposed online learning framework and optimizations.

在线学习决策树模型更新

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。