arXiv:2506.20249cs.AIcs.CL2025-06NeurIPS被引 8

用大模型自动生成新语言模型架构,效果媲美人工设计。

Language Modeling by Language Models

  • 多智能体系统模拟科研全流程,从构想到验证
  • 生成1162个新架构,6/9基准上超越GPT2等经典模型
  • 基于遗传编程提升设计成功率,适合自动化模型研发

能否利用大语言模型(LLM)来发现新型语言模型架构?受真实研究流程启发,我们提出一种多智能体LLM方法,模拟从构思、文献调研(提案阶段)到设计实现(代码生成)、生成式预训练及下游评估(验证)的完整研究流程。基于缩放定律,系统Genesys采用分层缩放策略:在14M~350M参数范围内,逐步扩大模型规模,同时缩减每轮可训练模型数量。为提升发现效率与可分解性,Genesys引入新颖的遗传编程主干,相比常见的直接提示生成流程,成功生成设计的比例提升约86个百分点(关键瓶颈)。实验共发现1162个新架构(其中1062个经完整预训练验证),最优设计在6/9常见基准上优于GPT2、Mamba2等已知架构。结合系统级消融实验与形式化分析,揭示了高效自主发现系统的构建规律。

原文摘要 · Abstract (English)

Can we leverage LLMs to model the process of discovering novel language model (LM) architectures? Inspired by real research, we propose a multi-agent LLM approach that simulates the conventional stages of research, from ideation and literature search (proposal stage) to design implementation (code generation), generative pre-training, and downstream evaluation (verification). Using ideas from scaling laws, our system, Genesys, employs a Ladder of Scales approach; new designs are proposed, adversarially reviewed, implemented, and selectively verified at increasingly larger model scales (14M$\sim$350M parameters) with a narrowing budget (the number of models we can train at each scale). To help make discovery efficient and factorizable, Genesys uses a novel genetic programming backbone, which we show has empirical advantages over commonly used direct prompt generation workflows (e.g., $\sim$86\% percentage point improvement in successful design generation, a key bottleneck). We report experiments involving 1,162 newly discovered designs (1,062 fully verified through pre-training) and find the best designs to be highly competitive with known architectures (e.g., outperform GPT2, Mamba2, etc., on 6/9 common benchmarks). We couple these results with comprehensive system-level ablations and formal results, which give broader insights into the design of effective autonomous discovery systems.

模型发现多智能体遗传编程自动化

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