Database Workload Characterization with Query Plan Encoders
Debjyoti Paul, Jie Cao, Feifei Li, Vivek Srikumar
Abstract
Smart databases are adopting artificial intelligence (AI) technologies to achieve instance optimality , and in the future, databases will come with prepackaged AI models within their core components. The reason is that every database runs on different workloads, demands specific resources, and settings to achieve optimal performance. It prompts the necessity to understand workloads running in the system along with their features comprehensively, which we dub as workload characterization.
To address this workload characterization problem, we propose our query plan encoders that learn essential features and their correlations from query plans. Our pretrained encoders captures the structural and the computational performance of queries independently. We show that our pretrained encoders are adaptable to workloads that expedites the transfer learning process. We performed independent assessments of structural encoder and performance encoders with multiple downstream tasks. For the overall evaluation of our query plan encoders, we architect two downstream tasks (i) query latency prediction and (ii) query classification. These tasks show the importance of feature-based workload characterization. We also performed extensive experiments on individual encoders to verify the effectiveness of representation learning, and domain adaptability.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext f839aa7f-2972-47ff-aa9d-a1bbba5c5db4Cited by top-tier papers9
- A Comparative Study and Component Analysis of Query Plan Representation Techniques in ML4DB StudiesYue Zhao, Zhaodonghui Li, Gao CongVLDB 2024 · 19 citations
- The Holon Approach for Simultaneously Tuning Multiple Components in a Self-Driving Database Management System with Machine Learning via Synthesized Proto-ActionsWilliam Zhang, Wan Shen Lim, Matthew Butrovich, Andrew PavloVLDB 2024 · 13 citations
- Wred: Workload Reduction for Scalable Index TuningMatteo Brucato, Tarique Siddiqui, Wentao Wu, Vivek R. Narasayya et al.SIGMOD 2024 · 7 citations
- Wii: Dynamic Budget Reallocation In Index TuningXiaoying Wang, Wentao Wu, Chi Wang, Vivek R. Narasayya et al.SIGMOD 2024 · 6 citations
- Esc: An Early-Stopping Checker for Budget-aware Index TuningXiaoying Wang, Wentao Wu, Vivek R. Narasayya, Surajit ChaudhuriVLDB 2025 · 4 citations
Related papers
- Lequa: A Learning-Based Query-Aware Framework for Selective Query OptimizationGuoneng Li, Pengfei Zheng, Ling Xu, Yan Li et al.ICDE 2026
- APQO: An Adaptive Framework for Parametric Query OptimizationSijia Li, Peng Cai, Zhifan Zhang, Huiqi Hu et al.SIGMOD 2026 · 2 citations
- Learned Offline Query Planning via Bayesian OptimizationJeffrey Tao, Natalie Maus, Haydn Thomas Jones, Yimeng Zeng et al.SIGMOD 2025 · 5 citations
- QCFE: An Efficient Feature Engineering for Query Cost EstimationYu Yan, Hongzhi Wang, Junfang Huang, Dake Zhong et al.ICDE 2024 · 3 citations
- QueryFormer: A Tree Transformer Model for Query Plan RepresentationYue Zhao, Gao Cong, Jiachen Shi, Chunyan MiaoVLDB 2022 · 117 citations
