Large Language Model Assisted Adversarial Robustness Neural Architecture Search
Large Language Models (LLMs) have shown signif-icant promise as evolutionary optimizers. This paper introduces a novel LLM-based Optimizer (LLMO) to address Neural Archi-tecture Search Considering Adversarial Robustness (ARNAS), a classic combinatorial optimization problem. Using the standard CRISPE framework, we design the prompt and employ Gemini to iteratively refine solutions based on its responses. In our numerical experiments, we investigate the performance of LLMO on NAS-Bench-201-based ARNAS tasks with CIFAR-IO and CIFAR-IOO datasets. The results, compared with six well-known metaheuristic algorithms (MHAs), highlight the superiority and competitiveness of using LLMs as combinatorial optimizers. The source code is available at https://github.com/RuiZhong961230/LLMO.
Authors
- Masaharu Munetomo (ORCID: https://orcid.org/0000-0002-5750-9217)
- Jun Yu (ORCID: https://orcid.org/0000-0001-5029-0294)
- Rui Zhong (ORCID: https://orcid.org/0000-0003-0126-7543)
- Yang Cao (ORCID: https://orcid.org/0009-0001-0918-9378)
Institutions
- Hokkaido University (JP)
- Niigata Institute of Technology (JP)
- Niigata University (JP)
Publication Details
- Published
- 2024-08-16
- DOI
- https://doi.org/10.1109/docs63458.2024.10704419
- Citations
- 5
- Primary Topic
- Adversarial Robustness in Machine Learning
- Type
- article
- Field-Weighted Citation Impact
- 1.74