A Unified Transferable Model for ML-Enhanced DBMS
Not on the map. Why?
Abstract (arXiv)
Recently, the database management system (DBMS) community has witnessed the power of machine learning (ML) solutions for DBMS tasks. Despite their promising performance, these existing solutions can hardly be considered satisfactory. First, these ML-based methods in DBMS are not effective enough because they are optimized on each specific task, and cannot explore or understand the intrinsic connections between tasks. Second, the training process has serious limitations that hinder their practicality, because they need to retrain the entire model from scratch for a new DB. Moreover, for each retraining, they require an excessive amount of training data, which is very expensive to acquire and unavailable for a new DB. We propose to explore the transferabilities of the ML methods both across tasks and across DBs to tackle these fundamental drawbacks. In this paper, we propose a unified model MTMLF that uses a multi-task training procedure to capture the transferable knowledge across tasks and a pre-train fine-tune procedure to distill the transferable meta knowledge across DBs. We believe this paradigm is more suitable for cloud DB service, and has the potential to revolutionize the way how ML is used in DBMS. Furthermore, to demonstrate the predicting power and viability of MTMLF, we provide a concrete and very promising case study on query optimization tasks. Last but not least, we discuss several concrete research opportunities along this line of work.
Key facts. Cited by 10 papers in the map (data through 7 Oct 2026). Builds on 26 of its 49 references that are in the map. First posted on arXiv 6 May 2021.
Builds on
26 of 49 references.
References · 26
- 2017Attention Is All You Needcited by 59,932
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2015Sequence Level Training with Recurrent Neural Networkscited by 568
- 2020TAPAS: Weakly Supervised Table Parsing via Pre-trainingcited by 432
- 2015MADE: Masked Autoencoder for Distribution Estimationcited by 306
- 2015Federated Optimization:Distributed Optimization Beyond the Datacentercited by 193
- 2017The Case for Learned Index Structurescited by 188
- 2019Neo: A Learned Query Optimizercited by 80
- 2018Learned Cardinalities: Estimating Correlated Joins with Deep Learningcited by 64
- 2019ALEX: An Updatable Adaptive Learned Indexcited by 43
14 more
- 2019An End-to-End Learning-based Cost Estimatorcited by 43
- 2019Deep Unsupervised Cardinality Estimationcited by 36
- 2019Learning Multi-dimensional Indexescited by 30
- 2018Deep Reinforcement Learning for Join Order Enumerationcited by 25
- 2020Tsunami: A Learned Multi-dimensional Index for Correlated Data and Skewed Workloadscited by 25
- 2019Plan-Structured Deep Neural Network Models for Query Performance Predictioncited by 24
- 2020FLAT: Fast, Lightweight and Accurate Method for Cardinality Estimationcited by 22
- 2021Flow-Loss: Learning Cardinality Estimates That Mattercited by 22
- 2020Are We Ready For Learned Cardinality Estimation?cited by 19
- 2020Cost Models for Big Data Query Processing: Learning, Retrofitting, and Our Findingscited by 16
- 2018Learning State Representations for Query Optimization with Deep Reinforcement Learningcited by 14
- 2020BayesCard: Revitilizing Bayesian Frameworks for Cardinality Estimationcited by 10
- 2020FSPN: A New Class of Probabilistic Graphical Modelcited by 6
- 2016WiSeDB: A Learning-based Workload Management Advisor for Cloud Databasescited by 4
Built on by
Citing papers per year: 2021: 3; 2022: 3; 2023: 0; 2024: 2; 2025: 1; 2026: 1.
10 papers.
- 2021Cardinality Estimation in DBMS: A Comprehensive Benchmark Evaluationcited by 30
- 2022FactorJoin: A New Cardinality Estimation Framework for Join Queriescited by 13
- 2024PRICE: A Pretrained Model for Cross-Database Cardinality Estimationcited by 8
- 2024UniDM: A Unified Framework for Data Manipulation with Large Language Modelscited by 7
- 2022Detect, Distill and Update: Learned DB Systems Facing Out of Distribution Datacited by 6
- 2022Unveiling the Black Box of PLMs with Semantic Anchors: Towards Interpretable Neural Semantic Parsingcited by 5
- 2025In-Context Adaptation to Concept Drift for Learned Database Operationscited by 3
- 2021Glue: Adaptively Merging Single Table Cardinality to Estimate Join Query Sizecited by 2
- 2021Baihe: SysML Framework for AI-driven Databasescited by 0
- 2026MICRO: A Lightweight Middleware for Optimizing Cross-store Cross-model Graph-Relation Joins [Technical Report]cited by 0
- 2026MICRO: A Lightweight Middleware for Optimizing Cross-store Cross-model Graph-Relation Joins [Technical Report]14 Mar 2026
- 2025In-Context Adaptation to Concept Drift for Learned Database Operations7 May 2025
- 2024PRICE: A Pretrained Model for Cross-Database Cardinality Estimation3 Jun 2024
- 2024UniDM: A Unified Framework for Data Manipulation with Large Language Models10 May 2024
- 2022FactorJoin: A New Cardinality Estimation Framework for Join Queries11 Dec 2022
- 2022Detect, Distill and Update: Learned DB Systems Facing Out of Distribution Data11 Oct 2022
- 2022Unveiling the Black Box of PLMs with Semantic Anchors: Towards Interpretable Neural Semantic Parsing4 Oct 2022
- 2021Baihe: SysML Framework for AI-driven Databases29 Dec 2021
- 2021Glue: Adaptively Merging Single Table Cardinality to Estimate Join Query Size7 Dec 2021
- 2021Cardinality Estimation in DBMS: A Comprehensive Benchmark Evaluation13 Sep 2021