Large Language Model-Enhanced Algorithm Selection: Towards Comprehensive Algorithm Representation

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wu, Xingyu, Zhong, Yan, Wu, Jibin, Jiang, Bingbing, Tan, Kay Chen
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916248386797568
author Wu, Xingyu
Zhong, Yan
Wu, Jibin
Jiang, Bingbing
Tan, Kay Chen
author_facet Wu, Xingyu
Zhong, Yan
Wu, Jibin
Jiang, Bingbing
Tan, Kay Chen
contents Algorithm selection, a critical process of automated machine learning, aims to identify the most suitable algorithm for solving a specific problem prior to execution. Mainstream algorithm selection techniques heavily rely on problem features, while the role of algorithm features remains largely unexplored. Due to the intrinsic complexity of algorithms, effective methods for universally extracting algorithm information are lacking. This paper takes a significant step towards bridging this gap by introducing Large Language Models (LLMs) into algorithm selection for the first time. By comprehending the code text, LLM not only captures the structural and semantic aspects of the algorithm, but also demonstrates contextual awareness and library function understanding. The high-dimensional algorithm representation extracted by LLM, after undergoing a feature selection module, is combined with the problem representation and passed to the similarity calculation module. The selected algorithm is determined by the matching degree between a given problem and different algorithms. Extensive experiments validate the performance superiority of the proposed model and the efficacy of each key module. Furthermore, we present a theoretical upper bound on model complexity, showcasing the influence of algorithm representation and feature selection modules. This provides valuable theoretical guidance for the practical implementation of our method.
format Preprint
id arxiv_https___arxiv_org_abs_2311_13184
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Large Language Model-Enhanced Algorithm Selection: Towards Comprehensive Algorithm Representation
Wu, Xingyu
Zhong, Yan
Wu, Jibin
Jiang, Bingbing
Tan, Kay Chen
Machine Learning
Computation and Language
Algorithm selection, a critical process of automated machine learning, aims to identify the most suitable algorithm for solving a specific problem prior to execution. Mainstream algorithm selection techniques heavily rely on problem features, while the role of algorithm features remains largely unexplored. Due to the intrinsic complexity of algorithms, effective methods for universally extracting algorithm information are lacking. This paper takes a significant step towards bridging this gap by introducing Large Language Models (LLMs) into algorithm selection for the first time. By comprehending the code text, LLM not only captures the structural and semantic aspects of the algorithm, but also demonstrates contextual awareness and library function understanding. The high-dimensional algorithm representation extracted by LLM, after undergoing a feature selection module, is combined with the problem representation and passed to the similarity calculation module. The selected algorithm is determined by the matching degree between a given problem and different algorithms. Extensive experiments validate the performance superiority of the proposed model and the efficacy of each key module. Furthermore, we present a theoretical upper bound on model complexity, showcasing the influence of algorithm representation and feature selection modules. This provides valuable theoretical guidance for the practical implementation of our method.
title Large Language Model-Enhanced Algorithm Selection: Towards Comprehensive Algorithm Representation
topic Machine Learning
Computation and Language
url https://arxiv.org/abs/2311.13184