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

Can GPT-4 Perform Neural Architecture Search?

About

We investigate the potential of GPT-4~\cite{gpt4} to perform Neural Architecture Search (NAS) -- the task of designing effective neural architectures. Our proposed approach, \textbf{G}PT-4 \textbf{E}nhanced \textbf{N}eural arch\textbf{I}tect\textbf{U}re \textbf{S}earch (GENIUS), leverages the generative capabilities of GPT-4 as a black-box optimiser to quickly navigate the architecture search space, pinpoint promising candidates, and iteratively refine these candidates to improve performance. We assess GENIUS across several benchmarks, comparing it with existing state-of-the-art NAS techniques to illustrate its effectiveness. Rather than targeting state-of-the-art performance, our objective is to highlight GPT-4's potential to assist research on a challenging technical problem through a simple prompting scheme that requires relatively limited domain expertise\footnote{Code available at \href{https://github.com/mingkai-zheng/GENIUS}{https://github.com/mingkai-zheng/GENIUS}.}. More broadly, we believe our preliminary results point to future research that harnesses general purpose language models for diverse optimisation tasks. We also highlight important limitations to our study, and note implications for AI safety.

Mingkai Zheng, Xiu Su, Shan You, Fei Wang, Chen Qian, Chang Xu, Samuel Albanie• 2023

Related benchmarks

TaskDatasetResultRank
Image ClassificationCIFAR-10 NAS-Bench-201 (test)
Accuracy93.79
225
Image ClassificationCIFAR-100 NAS-Bench-201 (test)
Accuracy71.3
198
Image ClassificationCIFAR-10 NAS-Bench-201 (val)
Accuracy91.07
169
Image ClassificationImageNet-16-120 NAS-Bench-201 (test)
Accuracy44.96
167
Neural Architecture SearchNAS-Bench-201 ImageNet-16-120 (test)
Accuracy44.96
140
Image ClassificationCIFAR-100 NAS-Bench-201 (val)
Accuracy71.37
139
Image ClassificationImageNet 16-120 NAS-Bench-201 (val)
Accuracy45.29
123
Neural Architecture SearchCIFAR-10 NAS-Bench-201 (val)
Accuracy91.07
111
Neural Architecture SearchNAS-Bench-201 CIFAR-10 (test)
Accuracy93.79
110
Neural Architecture SearchImageNet16-120 NAS-Bench-201 (val)
Accuracy45.29
104
Showing 10 of 12 rows

Other info

Follow for update