Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

LLMs as In-Context Meta-Learners for Model and Hyperparameter Selection

About

Model and hyperparameter selection are critical but challenging in machine learning, typically requiring expert intuition or expensive automated search. We investigate whether large language models (LLMs) can act as in-context meta-learners for this task. By converting each dataset into interpretable metadata, we prompt an LLM to recommend both model families and hyperparameters. We study two prompting strategies: (1) a zero-shot mode relying solely on pretrained knowledge, and (2) a meta-informed mode augmented with examples of models and their performance on past tasks. Across synthetic and real-world benchmarks, we show that LLMs can exploit dataset metadata to recommend competitive models and hyperparameters without search, and that improvements from meta-informed prompting demonstrate their capacity for in-context meta-learning. These results highlight a promising new role for LLMs as lightweight, general-purpose assistants for model selection and hyperparameter optimization.

Youssef Attia El Hili, Albert Thomas, Malik Tiomoko, Abdelhakim Benechehab, Corentin L\'eger, Corinne Ancourt, Bal\'azs K\'egl• 2025

Related benchmarks

TaskDatasetResultRank
Model and Hyperparameter SelectionKaggle Allstate Private (test)
p-rank74.68
12
Binary ClassificationKaggle Attrition 2023
P-Rank61.12
6
Binary ClassificationKaggle Horses 2023
p-rank82.78
6
Binary ClassificationKaggle Mental Health 2024
P-Rank92.99
6
Binary ClassificationKaggle Molecules 2012
p-rank97.52
6
Model and Hyperparameter SelectionKaggle attrition private (test)
p-rank61.12
6
Model and Hyperparameter SelectionKaggle blueberry private (test)
p-rank81.16
6
Model and Hyperparameter SelectionKaggle Covertype private (test)
P-Rank67.78
6
Model and Hyperparameter SelectionKaggle crab age private (test)
p-rank68.87
6
Model and Hyperparameter SelectionKaggle heat flux fi private (test)
p-rank93.4
6
Showing 10 of 44 rows

Other info

Follow for update