arXiv:2512.12281cs.CV2025-12被引 1

用大模型自动设计检测模型,省去人工调参

Cognitive-YOLO: LLM-Driven Architecture Synthesis from First Principles of Data for Object Detection

  • 大模型驱动智能组件拼装,根据数据特征自动生成网络结构
  • 轻量模型参数仅190万,检测精度超越YOLOv11n
  • 适合垂直领域快速部署,无需专家经验

通用目标检测算法在垂直场景部署时面临域偏移瓶颈,手动定制网络架构依赖专家经验且试错成本高。为此,我们提出基于大语言模型(LLM)与自主代理协同的自适应目标检测模型合成系统Cognitive-YOLO。该系统构建了全自动的‘分析-合成-编译’流程:首先,数据画像模块定量提取数据集特征,引导采用ReAct范式的代理从精选组件库中检索匹配模块;其次,LLM基于数据特征进行逻辑推理与拓扑组装,生成符合规范的神经网络描述;最后,后端混合编译器动态实例化模型,完成端到端训练与评估调度。在轨面缺陷和水稻病害等数据集上的交叉验证表明,Cognitive-YOLO生成的轻量模型参数量压缩至190万,同时在[email protected]:0.95上达到或超过主流基线如YOLOv11n的水平,有效平衡了模型紧凑性与特征表达能力。

原文摘要 · Abstract (English)

General-purpose object detection algorithms face a domain-shift bottleneck when deployed in vertical scenarios, and manually customizing a network architecture relies heavily on expert experience with high trial-and-error cost. To address these problems, we present \textbf{Cognitive-YOLO}, an adaptive object-detection model synthesis system driven by a large language model (LLM) collaborating with an autonomous agent. Cognitive-YOLO builds a fully automated \emph{Analyze--Synthesize--Compile} pipeline. First, a data-profiling module quantitatively extracts dataset features, guiding a ReAct-paradigm agent to autonomously retrieve matching network modules from a curated component library. Second, the LLM performs logical reasoning and topological assembly conditioned on the data features, producing a specification-compliant neural architecture description. Finally, a backend hybrid compiler dynamically instantiates the model, closing the loop with end-to-end training and evaluation scheduling. Cross-validation on datasets including rail-surface defects and rice diseases shows that the lightweight models synthesized by Cognitive-YOLO markedly compress the parameter count (down to 1.9M) while matching or surpassing mainstream baselines such as YOLOv11n in [email protected]:0.95, effectively balancing model compactness and feature-representation capability.

目标检测大模型自动化设计

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