arXiv:2412.19206cs.CV2024-12CVPR被引 7

用多智能体协作设计神经网络,突破传统搜索空间限制。

NADER: Neural Architecture Design via Multi-Agent Collaboration

  • 多智能体协同迭代优化架构,动态学习经验避免重复错误。
  • 在多个基准任务上超越现有方法,发现未预设的高性能结构。
  • 采用图结构表示,专注设计细节,无需处理代码干扰。

神经架构设计在深度学习中面临巨大挑战。尽管神经架构搜索(NAS)能自动寻找最优架构,但现有方法常受限于预定义的搜索空间,可能遗漏关键架构。本文提出NADER(基于大模型多智能体协作的神经架构设计),将神经架构设计建模为大语言模型驱动的多智能体协作问题。NADER通过一组专业智能体对基础架构进行迭代修改。现有基于LLM的架构设计方法通常独立运行,无法从过往经验中学习,导致重复错误和低效探索。为此,我们引入反射器(Reflector),有效结合即时反馈与长期经验。此外,不同于以往使用代码表示架构的方法,我们采用图结构表示,使智能体专注于设计层面,不受编码细节干扰。在多个基准任务上的大量实验表明,NADER能发现超出预设搜索空间的高性能架构,显著优于当前最先进方法。代码即将开源。

原文摘要 · Abstract (English)

Designing effective neural architectures poses a significant challenge in deep learning. While Neural Architecture Search (NAS) automates the search for optimal architectures, existing methods are often constrained by predetermined search spaces and may miss critical neural architectures. In this paper, we introduce NADER (Neural Architecture Design via multi-agEnt collaboRation), a novel framework that formulates neural architecture design (NAD) as a LLM-based multi-agent collaboration problem. NADER employs a team of specialized agents to enhance a base architecture through iterative modification. Current LLM-based NAD methods typically operate independently, lacking the ability to learn from past experiences, which results in repeated mistakes and inefficient exploration. To address this issue, we propose the Reflector, which effectively learns from immediate feedback and long-term experiences. Additionally, unlike previous LLM-based methods that use code to represent neural architectures, we utilize a graph-based representation. This approach allows agents to focus on design aspects without being distracted by coding. We demonstrate the effectiveness of NADER in discovering high-performing architectures beyond predetermined search spaces through extensive experiments on benchmark tasks, showcasing its advantages over state-of-the-art methods. The codes will be released soon.

神经架构设计多智能体大模型应用

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。