LITS: An Optimized Learned Index for Strings (An Extended Version)
Yifan Yang, Shimin Chen
TL;DR
This work addresses the difficulty of applying learned indexes to variable-length string keys by introducing LITS, a string-optimized learned index that combines a Hash-enhanced Prefix Table (HPT) for accurate CDF estimation with per-node local linear models, compact leaf nodes, and subtries for hybrid structure selection via PMSS. The approach eliminates costly last-mile searches and leverages trie strengths, achieving up to $2.43\times$ and $2.27\times$ improvements over HOT and ART for point queries while delivering comparable scan performance across diverse datasets. Key contributions include the HPT-based CDF model, the compact leaf node design, and the PMSS-driven hybridization with subtries, validated on 11 string datasets with substantial performance gains. The results suggest practical significance for real-world databases requiring fast string-key indexing, particularly for workloads with heavy point queries and mixed operations.
Abstract
Index is an important component in database systems. Learned indexes have been shown to outperform traditional tree-based index structures for fixed-sized integer or floating point keys. However, the application of the learned solution to variable-length string keys is under-researched. Our experiments show that existing learned indexes for strings fail to outperform traditional string indexes, such as HOT and ART. String keys are long and variable sized, and often contain skewed prefixes, which make the last-mile search expensive, and adversely impact the capability of learned models to capture the skewed distribution of string keys. In this paper, we propose a novel learned index for string keys, LITS (Learned Index with Hash-enhanced Prefix Table and Sub-tries). We propose an optimized learned model, combining a global Hash-enhanced Prefix Table (HPT) and a per-node local linear model to better distinguish string keys. Moreover, LITS exploits compact leaf nodes and hybrid structures with a PMSS model for efficient point and range operations. Our experimental results using eleven string data sets show that LITS achieves up to 2.43x and 2.27x improvement over HOT and ART for point operations, and attains comparable scan performance.
