A New Graph Neural Network-based Join Optimization Algorithm

Yuanfeng Song,Yuqiang Li,Shuhuan Fan,Dongsheng He,Jianming Liao
DOI: https://doi.org/10.1109/admit57209.2022.00012
2022-01-01
Abstract:Improving the query performance of databases has long been a hot research topic in the field of database research. The rapid development of artificial intelligence has introduced new opportunities to database query optimization. To address the problem of the failure to effectively utilize query features and join order features in query optimization, this paper proposes a graph neural network-based database join optimization algorithm, RGOS (Reinforcement learning and GNN-based join Order Selection), which uses a graph convolution algorithm to extract query features and join order features, then employs a reinforcement learning algorithm to learn the database operating environment and continuously optimize the model using feedback to improve the query performance. Finally, the RGOS algorithm is validated on TiDB, an open source distributed relational database. Experimental results show that the RGOS algorithm can improve the average query performance of TiDB by 40%, and the query performance of RGOS algorithm is 25% higher than that of DQ algorithm.
What problem does this paper attempt to address?