arXiv 机器学习· Durmus Karatay, Richard Newman·· 7 小时前AI 评分39
TreeWalker:面向成组树集成模型推理的部分求值
TreeWalker: Partial Evaluation for Grouped Tree-Ensemble Inference
AI 导读
TreeWalker 将部分求值引入树集成模型的成组推理,无需重新训练即可直接支持标准 LightGBM 和 XGBoost 模型。它通过每组仅遍历一次树并跳过空子树,在 Intel 参考配置下较单行独立遍历提速 2.5-3.2x,在 G=128 的生存数据集上提速 6.8-7.8x,在 Arm 架构上提升更显著。
来源:arXiv 机器学习 · arxiv.org