小语言模型推理能力路由研究:模型容量与提示词优化对下游质量和效率的影响
Small Language Models for Inference-Capability Routing: Effects of Model Capacity and Prompt Optimization on Downstream Quality and Efficiency
AI 导读
一项 arXiv 研究用提示词驱动的小语言模型(Qwen2.5,1.5B 至 14B 参数)对推理、检索、网络搜索和工具调用四种能力做独立路由决策。
正文 · 原文
Abstract:Optional reasoning, retrieval, web search, and tool use can improve large language model responses but incur unnecessary computation when activated indiscriminately. This study investigates whether prompt optimization can compensate for limited router capacity and whether classification gains translate into downstream efficiency. The proposed framework uses prompt-driven small language models to make four independent capability decisions, allowing multiple capabilities to be selected per query. Qwen2.5 routers ranging from 1.5B to 14B parameters are compared across five prompting strategies, followed by an evaluation of capability-specific optimized configurations using a fixed GPT-OSS-120B downstream model on 127 questions. Under minimal prompting, increasing router capacity does not consistently improve classification performance. Prompt optimization narrows the capacity gap for reasoning and calculator routing, whereas RAG and web search retain larger benefits from higher capacity. Similar classification F1 scores do not necessarily imply equivalent downstream behavior: false-positive activations increase unnecessary computation, while missed capabilities can reduce answer quality. The 14B router achieves a mean BERTScore F1 of 0.8408, compared with 0.8418 when all capabilities are enabled, while reducing average downstream token consumption by 68.7\% (8,837 to 2,764 tokens) and median downstream latency by 80.5\% (9.79 to 1.91 seconds). These savings exclude router overhead, which is measured separately in a different inference environment and increases with router capacity. The findings demonstrate that the relationship between router capacity, prompt design, and downstream efficiency is capability-dependent. Consequently, router selection must consider error profiles and routing overhead alongside classification performance.
| Comments: | 15 pages, 8 figures |
| Subjects: | Artificial Intelligence (cs.AI) |
| Cite as: | arXiv:2611.16414 [cs.AI] |
| (or arXiv:2611.16414v1 [cs.AI] for this version) | |
| https://doi.org/10.48550/arXiv.2611.16414 arXiv-issued DOI via DataCite (pending registration) |
Submission history
From: Sailendra Kolluru [view email]
[v1]
Thu, 12 Nov 2026 07:25:46 UTC (4,083 KB)
来源:arXiv cs.AI · arxiv.org