中文

NeuroGen:基于大语言模型的神经网络参数生成

人工智能 2025-05-26 v2

摘要

神经网络(NN)参数的获取自 NN 提出以来便是机器学习中最重要的问题之一。传统方法,如反向传播和前向仅优化,均通过迭代数据拟合逐步优化参数。本文旨在探索新方向:通过大语言模型生成获取 NN 参数。我们提出 NeuroGen,一种通用且易于实现的两阶段方法,用于根据数据、任务和网络架构描述生成 NN 参数。阶段一为参数参考知识注入,在此基础上,LLM 在 NN 检查点上预训练以构建参数空间的基础理解;阶段二为上下文增强指令调优,使 LLM 通过丰富的、任务感知的提示适应特定任务。实验结果表明,NeuroGen 有效生成可用 NN 参数。我们的发现突显了基于 LLM 的 NN 参数生成的可行性,并指出 LLM 与轻量级 NN 协同共存的有前景的新范式。

关键词

引用

@article{arxiv.2505.12470,
  title  = {NeuroGen: Neural Network Parameter Generation via Large Language Models},
  author = {Jiaqi Wang and Yusen Zhang and Xi Li},
  journal= {arXiv preprint arXiv:2505.12470},
  year   = {2025}
}

备注

The three authors contributed equally to this work. The codes will be public after being accepted