A New Paradigm in Tuning Learned Indexes: A Reinforcement Learning Enhanced Approach

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wang, Taiyi, Liang, Liang, Yang, Guang, Heinis, Thomas, Yoneki, Eiko
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916618814095360
author Wang, Taiyi
Liang, Liang
Yang, Guang
Heinis, Thomas
Yoneki, Eiko
author_facet Wang, Taiyi
Liang, Liang
Yang, Guang
Heinis, Thomas
Yoneki, Eiko
contents Learned Index Structures (LIS) have significantly advanced data management by leveraging machine learning models to optimize data indexing. However, designing these structures often involves critical trade-offs, making it challenging for both designers and end-users to find an optimal balance tailored to specific workloads and scenarios. While some indexes offer adjustable parameters that demand intensive manual tuning, others rely on fixed configurations based on heuristic auto-tuners or expert knowledge, which may not consistently deliver optimal performance. This paper introduces LITune, a novel framework for end-to-end automatic tuning of Learned Index Structures. LITune employs an adaptive training pipeline equipped with a tailor-made Deep Reinforcement Learning (DRL) approach to ensure stable and efficient tuning. To accommodate long-term dynamics arising from online tuning, we further enhance LITune with an on-the-fly updating mechanism termed the O2 system. These innovations allow LITune to effectively capture state transitions in online tuning scenarios and dynamically adjust to changing data distributions and workloads, marking a significant improvement over other tuning methods. Our experimental results demonstrate that LITune achieves up to a 98% reduction in runtime and a 17-fold increase in throughput compared to default parameter settings given a selected Learned Index instance. These findings highlight LITune's effectiveness and its potential to facilitate broader adoption of LIS in real-world applications.
format Preprint
id arxiv_https___arxiv_org_abs_2502_05001
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle A New Paradigm in Tuning Learned Indexes: A Reinforcement Learning Enhanced Approach
Wang, Taiyi
Liang, Liang
Yang, Guang
Heinis, Thomas
Yoneki, Eiko
Databases
Artificial Intelligence
Systems and Control
Learned Index Structures (LIS) have significantly advanced data management by leveraging machine learning models to optimize data indexing. However, designing these structures often involves critical trade-offs, making it challenging for both designers and end-users to find an optimal balance tailored to specific workloads and scenarios. While some indexes offer adjustable parameters that demand intensive manual tuning, others rely on fixed configurations based on heuristic auto-tuners or expert knowledge, which may not consistently deliver optimal performance. This paper introduces LITune, a novel framework for end-to-end automatic tuning of Learned Index Structures. LITune employs an adaptive training pipeline equipped with a tailor-made Deep Reinforcement Learning (DRL) approach to ensure stable and efficient tuning. To accommodate long-term dynamics arising from online tuning, we further enhance LITune with an on-the-fly updating mechanism termed the O2 system. These innovations allow LITune to effectively capture state transitions in online tuning scenarios and dynamically adjust to changing data distributions and workloads, marking a significant improvement over other tuning methods. Our experimental results demonstrate that LITune achieves up to a 98% reduction in runtime and a 17-fold increase in throughput compared to default parameter settings given a selected Learned Index instance. These findings highlight LITune's effectiveness and its potential to facilitate broader adoption of LIS in real-world applications.
title A New Paradigm in Tuning Learned Indexes: A Reinforcement Learning Enhanced Approach
topic Databases
Artificial Intelligence
Systems and Control
url https://arxiv.org/abs/2502.05001