mlsanityData to 7 Oct 2026

A Unified Transferable Model for ML-Enhanced DBMS

arXiv 2105.02418 ↗First posted 6 May 2021cs.DB cs.AIcited by 10

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

  1. 2017Attention Is All You Needcited by 59,932
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
  3. 2014Adam: A Method for Stochastic Optimizationcited by 36,960
  4. 2020Language Models are Few-Shot Learnerscited by 32,199
  5. 2015Sequence Level Training with Recurrent Neural Networkscited by 568
  6. 2020TAPAS: Weakly Supervised Table Parsing via Pre-trainingcited by 432
  7. 2015MADE: Masked Autoencoder for Distribution Estimationcited by 306
  8. 2015Federated Optimization:Distributed Optimization Beyond the Datacentercited by 193
  9. 2017The Case for Learned Index Structurescited by 188
  10. 2019Neo: A Learned Query Optimizercited by 80
  11. 2018Learned Cardinalities: Estimating Correlated Joins with Deep Learningcited by 64
  12. 2019ALEX: An Updatable Adaptive Learned Indexcited by 43
14 more
  1. 2019An End-to-End Learning-based Cost Estimatorcited by 43
  2. 2019Deep Unsupervised Cardinality Estimationcited by 36
  3. 2019Learning Multi-dimensional Indexescited by 30
  4. 2018Deep Reinforcement Learning for Join Order Enumerationcited by 25
  5. 2020Tsunami: A Learned Multi-dimensional Index for Correlated Data and Skewed Workloadscited by 25
  6. 2019Plan-Structured Deep Neural Network Models for Query Performance Predictioncited by 24
  7. 2020FLAT: Fast, Lightweight and Accurate Method for Cardinality Estimationcited by 22
  8. 2021Flow-Loss: Learning Cardinality Estimates That Mattercited by 22
  9. 2020Are We Ready For Learned Cardinality Estimation?cited by 19
  10. 2020Cost Models for Big Data Query Processing: Learning, Retrofitting, and Our Findingscited by 16
  11. 2018Learning State Representations for Query Optimization with Deep Reinforcement Learningcited by 14
  12. 2020BayesCard: Revitilizing Bayesian Frameworks for Cardinality Estimationcited by 10
  13. 2020FSPN: A New Class of Probabilistic Graphical Modelcited by 6
  14. 2016WiSeDB: A Learning-based Workload Management Advisor for Cloud Databasescited by 4

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2021: 3; 2022: 3; 2023: 0; 2024: 2; 2025: 1; 2026: 1.

10 papers.