BLOCKSET (Block-Aligned Serialized Trees): Reducing Inference Latency for Tree ensemble Deployment
Meghana Madhyastha, Kunal Lillaney, James Browne, Joshua T. Vogelstein, Randal C. Burns
Abstract
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.
Ask about this paper
Ask your agent about it.
Lune has read the top-tier papers around this one, so every answer names the papers it rests on.
Your agent calls
Lunesearch_papers
Free to start. No credit card required.
Terminal
Install the CLIlune papers get f05cacd5-09b7-48ba-a513-729fbc95accbRelated papers
- T-Rex (Tree-Rectangles): Reformulating Decision Tree Traversal as Hyperrectangle EnclosureMeghana Madhyastha, Tamas Budavari, Vladimir Braverman, Joshua T. Vogelstein et al.ICDE 2024
- ScalaGBM: Memory Efficient GBDT Training for High-Dimensional Data on GPUBorui Xu, Zeyi Wen, Yao Chen, Weiguo Liu et al.KDD 2025
- Serving Deep Learning Models with Deduplication from Relational DatabasesLixi Zhou, Jiaqing Chen, Amitabh Das, Hong Min et al.VLDB 2022 · 30 citations
- Trees with Attention for Set Prediction TasksRoy Hirsch, Ran Gilad-BachrachICML 2021 · 7 citations
- DeltaBoost: Gradient Boosting Decision Trees with Efficient Machine UnlearningZhaomin Wu, Junhui Zhu, Qinbin Li, Bingsheng HeSIGMOD 2023 · 18 citations
