LATuner: An LLM-Enhanced Database Tuning System Based on Adaptive Surrogate Model

  • Chongjiong Fan
  • , Zhicheng Pan
  • , Wenwen Sun
  • , Chengcheng Yang
  • , Wei Neng Chen*
  • *Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

3 Scopus citations

Abstract

Database Management Systems (DBMSs) offer a plethora of configurable parameters—termed “knobs”—that control the system behavior. Identifying the optimal configuration for these knobs, i.e., Knob Tuning (KT) is acknowledged as a critical way to enhance the DBMS performance. However, the increasing number of adjustable knobs and the complexity inherent in KTs have rendered manual tuning an antiquated and impractical approach. Recently, automatic KTs based on Machine Learning (ML) techniques have demonstrated significant potentials. Despite the advancements, they are also hindered by notable drawbacks such as the lack of domain knowledge and low tuning efficiency. Meanwhile, Large Language Model (LLM), which is pre-trained on diverse corpora including web content, database manuals, could offer a novel, training-free approach to significantly mitigate the aforementioned issues. In light of this, we propose an LLM-enhanced dAtabase Tuner, called LATuner. Firstly, since KT often suffers from the cold-start problem, we harness the extensive domain knowledge of LLMs to identify critical knobs and to warm start the tuning process, thus obtaining high-quality training samples. Secondly, as KT requires multiple rounds of sampling during the training process, we leverage LLMs to guide the sampling procedure, accelerating the convergence of the model training. Finally, to balance the tuning cost and efficiency between LLM-based KT and traditional ML-based KT, we design an adaptive surrogate strategy based on multi-armed bandit, achieving cost-effective tuning performance. Extensive experiments performed on well-established benchmarks have proven the efficacy and superiority of our proposal.

Original languageEnglish
Title of host publicationMachine Learning and Knowledge Discovery in Databases. Research Track - European Conference, ECML PKDD 2024, Proceedings
EditorsAlbert Bifet, Jesse Davis, Tomas Krilavicius, Meelis Kull, Eirini Ntoutsi, Indre Žliobaite
PublisherSpringer Science and Business Media Deutschland GmbH
Pages372-388
Number of pages17
ISBN (Print)9783031703614
DOIs
StatePublished - 2024
EventEuropean Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Databases, ECML PKDD 2024 - Vilnius, Lithuania
Duration: 9 Sep 202413 Sep 2024

Publication series

NameLecture Notes in Computer Science
Volume14945 LNAI
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

ConferenceEuropean Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Databases, ECML PKDD 2024
Country/TerritoryLithuania
CityVilnius
Period9/09/2413/09/24

Keywords

  • Bayesian Optimization
  • Database Tuning
  • Large Language Model
  • Multi-armed Bandit

Fingerprint

Dive into the research topics of 'LATuner: An LLM-Enhanced Database Tuning System Based on Adaptive Surrogate Model'. Together they form a unique fingerprint.

Cite this