BLOCKSET (Block-Aligned Serialized Trees): Reducing Inference Latency for Tree ensemble Deployment
Meghana Madhyastha, Kunal Lillaney, James Browne, Joshua T. Vogelstein, Randal C. Burns
摘要
We present methods to serialize and deserialize gradient-boosted trees and random forests that optimize inference latency when models are not loaded into memory. This arises when models are larger than memory, but also systematically when models are deployed on low-resource devices in the Internet of Things or run as cloud microservices where resources are allocated on demand. Block-Aligned Serialized Trees (BLOCKSET) introduce the concept of selective access for random forests and gradient boosted trees in which only the parts of the model needed for inference are deserialized and loaded into memory. %BLOCKSET combines concepts from external memory algorithms and data-parallel %layouts of random forests that maximize I/O-density for in-memory models. Using principles from external memory algorithms, we block-align the serialization format in order to minimize the number of I/Os. For gradient boosted trees, this results in a more than five time reduction in inference latency over layouts that do not perform selective access and a 2 times latency reduction over techniques that are selective, but do not encode I/O block boundaries in the layout.
问问这篇 Paper
问问你的智能体。
Lune 读过与它相关的顶会 Paper,每个回答都会注明依据哪几篇。
相关 Paper
- T-Rex (Tree-Rectangles): Reformulating Decision Tree Traversal as Hyperrectangle EnclosureMeghana Madhyastha, Tamas Budavari, Vladimir Braverman, Joshua T. Vogelstein 等ICDE 2024
- ScalaGBM: Memory Efficient GBDT Training for High-Dimensional Data on GPUBorui Xu, Zeyi Wen, Yao Chen, Weiguo Liu 等KDD 2025
- Serving Deep Learning Models with Deduplication from Relational DatabasesLixi Zhou, Jiaqing Chen, Amitabh Das, Hong Min 等VLDB 2022 · 被引用 30 次
- Trees with Attention for Set Prediction TasksRoy Hirsch, Ran Gilad-BachrachICML 2021 · 被引用 7 次
- DeltaBoost: Gradient Boosting Decision Trees with Efficient Machine UnlearningZhaomin Wu, Junhui Zhu, Qinbin Li, Bingsheng HeSIGMOD 2023 · 被引用 18 次
