Natural Language Query to Configuration for Retrieval Agents

TL;DR

BRANE方法通过LLM提取查询特征,实现MuSiQue等数据集89%成本节省的动态检索配置优化。

cs.AI 🔴 高级 2026-05-27 64 次浏览
Melissa Z. Pan Negar Arabzadeh Mathew Jacob Fiodar Kazhamiaka Esha Choukse Matei Zaharia
自然语言处理 信息检索 大规模语言模型 系统优化 机器学习

核心发现

方法论

本文提出BRANE框架,利用大型语言模型(LLM)从自然语言查询中提取工作负载特定的二元特征,作为查询与检索管线配置之间的桥梁。针对预定义的配置空间中每个配置,训练轻量级二分类预测器,估计该配置对特定查询的正确回答概率。推理阶段,BRANE基于预测的准确率与成本加权得分,动态选择最优配置,实现成本与准确率的可调权衡。该方法通过离线配置性能剖析、特征提取、预测器训练及基于拉格朗日松弛的推理选择三阶段完成,支持多种检索器、LLM、文档数量和合成策略组合,显著超越静态配置和现有基于LLM路由的基线。

关键结果

  • 在MuSiQue、BrowseComp-Plus和FinanceBench三个公开基准上,BRANE在保持与最佳静态配置相同准确率的前提下,实现了最高达89.4%的成本节省,平均节省约59.7%。
  • BRANE在成本-准确率的帕累托前沿上持续领先,优于包括Carrot(LLM路由)、METIS(规则路由)、Adaptive-RAG(检索策略选择)及端到端微调的BERT和Qwen3-4B模型。
  • 消融实验表明,LLM自动生成的工作负载特定二元特征优于通用语义嵌入作为预测器输入,且BRANE对特征数量和特征提取LLM的选择表现稳健。

研究意义

本研究首次系统性地将自然语言查询映射至完整检索管线配置,实现了按查询动态优化配置,突破了传统静态工作负载调优的局限。该方法显著降低了推理成本,同时保持甚至提升了答案质量,解决了多参数组合配置空间庞大、交互复杂、难以穷尽搜索的难题。对工业界大规模知识检索系统具有重要指导意义,推动了智能问答和知识搜索系统的高效部署与个性化服务。

技术贡献

BRANE提出了基于LLM自动发现工作负载特定查询特征的创新机制,作为查询到配置映射的中间层,有效捕获了查询与配置间复杂非线性关系。通过为每个配置训练独立轻量级预测器,避免了联合建模的高昂成本和复杂性。采用拉格朗日松弛实现推理时的成本-准确率权衡,支持灵活调整。此外,提出模糊帕累托剪枝策略,显著减少训练和推理负担,提升系统可扩展性。

新颖性

BRANE是首个将自然语言查询转化为完整检索管线配置的端到端框架,区别于仅选择LLM模型的路由方法,覆盖了检索器、文档数、多跳次数及合成策略等多维配置空间。其创新在于利用LLM自动生成工作负载特定的二元查询特征,替代传统固定或通用特征,从而实现更细粒度的查询区分和配置选择,显著提升了配置的个性化和效率。

局限性

  • BRANE依赖于离线配置性能剖析,初期采样和评估成本较高,尤其在配置空间极大时,可能限制快速部署。
  • 当前方法假设查询特征与配置准确率之间的关系稳定,面对动态变化的工作负载或语料库时,模型需重新训练以保持性能。
  • 尽管BRANE在多个基准表现优异,但对极端复杂查询或跨领域迁移的适应性尚未充分验证。

未来方向

未来工作可聚焦于减少离线剖析成本,探索在线学习和自适应更新机制以应对动态工作负载。此外,结合更丰富的查询语义理解和多模态信息,提升查询特征表达能力。扩展至跨领域和多语言环境,增强系统的通用性和鲁棒性。

AI 总览摘要

现代知识检索系统因其复杂的多组件配置空间而面临成本与准确率之间的权衡难题。传统方法通常对整个工作负载采用固定配置,忽视了不同查询间的多样性和个性化需求,导致资源浪费和性能瓶颈。本文针对这一挑战,提出了BRANE框架,实现了基于自然语言查询的动态配置选择。

BRANE利用大型语言模型自动从查询中提取工作负载特定的二元特征,作为查询与检索管线配置之间的桥梁。针对每个预定义配置,训练轻量级预测器估计其对特定查询的正确回答概率。推理阶段,BRANE基于预测准确率与成本的加权得分,动态选择最优配置,支持用户灵活调节成本与质量的权衡。

该方法通过离线性能剖析、特征提取和预测器训练三阶段完成,避免了传统联合建模的高昂计算开销。实验在MuSiQue、BrowseComp-Plus和FinanceBench三个公开基准上验证,BRANE在保持最佳静态配置准确率的同时,最高节省89.4%推理成本,平均节省近60%,显著优于现有基线方法。

BRANE的创新在于工作负载特定的查询特征自动发现机制,细粒度捕捉查询差异,提升配置选择的个性化和效率。模糊帕累托剪枝策略有效减少训练和推理负担,增强系统扩展性。该研究为知识检索系统的动态优化提供了实用且高效的解决方案,推动了智能问答和知识搜索技术的产业应用。

尽管BRANE表现优异,但依赖于离线剖析数据,初期成本较高,且对动态变化的工作负载适应性有待提升。未来工作将探索在线自适应机制,丰富查询特征表达,扩展跨领域和多语言应用,进一步提升系统的鲁棒性和通用性。

深度分析

研究背景

随着大规模语言模型(LLM)和信息检索技术的发展,现代知识搜索系统日益复杂,广泛应用于客户支持、科研辅助和商业问答等场景。代表性系统如Perplexity、ChatGPT、Gemini和Claude均结合了多步检索与LLM生成,提升答案质量。然而,这类系统涉及多维配置空间,包括LLM型号、检索器选择、检索文档数量、多跳次数及答案合成策略等,配置不当将导致成本激增或准确率下降。现有方法多为针对固定工作负载手动调优,忽视了查询间的个性化差异,难以充分利用配置空间的潜力。近期LLM路由方法虽能动态选择模型,但仅限于模型选择,未覆盖整个管线配置,且训练成本高昂。如何基于自然语言查询动态选择最优配置,兼顾成本与准确率,成为亟待解决的核心问题。

核心问题

本文聚焦于知识搜索系统中,如何针对单个自然语言查询,在预定义的多维配置空间内选择最优管线配置,以满足给定的准确率或成本预算目标。该问题面临三大挑战:一是配置空间庞大且组合复杂,LLM、检索器、文档数、多跳次数和合成策略等参数交互非线性且无封闭解;二是查询本身语义丰富且噪声多,难以直接映射至配置选择;三是离线性能剖析成本高,难以覆盖所有配置和查询。传统静态配置忽视查询间差异,导致资源浪费和性能不均。如何高效、精准地从自然语言查询中提取有效特征,预测各配置的准确率,并动态选择配置,是本文的核心问题。

核心创新

本文提出三大核心创新:


1. 工作负载特定查询特征自动发现:利用强大LLM(gpt-4o)从少量示例查询中自动生成一组二元特征,捕捉查询的细粒度语义和结构差异,替代传统固定或通用特征,提升配置选择的区分度。


2. 每配置独立轻量级预测器:针对预定义配置空间中通过模糊帕累托剪枝筛选的配置,训练单独的二分类器预测该配置对查询的正确率,避免联合建模的高复杂度和数据需求,支持灵活扩展和快速训练。


3. 拉格朗日松弛推理策略:在推理阶段基于预测准确率与成本的加权得分动态选择配置,用户可通过调节权重λ实现成本-准确率的灵活权衡,形成完整的端到端Query2Conf框架。

方法详解

  • �� 离线配置性能剖析:采集工作负载中的N个查询,针对每个查询执行预定义的C个配置,记录准确率(正确/错误)和成本,构建准确率矩阵和成本矩阵。

  • �� 工作负载特定特征生成:使用强力LLM(gpt-4o)对少量示例查询进行提示,自动生成d个二元特征(如requires_multi_hop, involves_regional_cuisine),每个特征均可由查询文本判定是/否。

  • �� 查询特征标注:使用较小、廉价的LLM对所有查询进行特征标注,生成二元特征向量Fq。

  • �� 预测器训练:对通过模糊帕累托剪枝筛选的配置,分别训练轻量级二分类器(包括逻辑回归、决策树、随机森林、XGBoost、LightGBM等),输入为Fq,输出为该配置回答正确的概率ˆc(Fq)。

  • �� 推理时配置选择:给定查询q,计算Fq,评估所有配置的ˆc(Fq),结合配置平均成本cost(c),通过拉格朗日松弛公式πλ(q)=argmax_c [ˆc(Fq) - λ·cost(c)]选择配置,λ控制成本与准确率权衡。

  • �� 模糊帕累托剪枝:为减少训练和推理负担,仅对成本-准确率帕累托前沿及其近邻配置训练预测器,利用容忍度τacc和τcost避免因采样噪声误删配置。

实验设计

实验在MuSiQue、BrowseComp-Plus和FinanceBench三个公开知识搜索基准上进行。每个基准包含150至600个查询,配置空间涵盖多种LLM(如gpt-4o、gpt-4o、GPT-5.4)、检索器、检索文档数、多跳次数及合成策略(LLM-only、分块摘要、迭代检索)。基线包括静态最优配置、LLM路由方法Carrot、规则路由METIS、Adaptive-RAG检索策略选择、以及端到端微调的BERT和Qwen3-4B模型。评估指标为准确率(与gpt-4o判定的参考答案一致率)和推理成本(基于生成token计价)。采用5折交叉验证,使用Optuna调优预测器超参数,消融研究特征数量、特征提取LLM及预测器类型对性能影响。

结果分析

BRANE在三个基准上均显著优于所有基线,达到与最佳静态配置相同准确率的同时,成本节省最高达89.4%,平均节省约59.7%。在成本-准确率帕累托前沿上,BRANE持续领先,覆盖广泛成本范围。LLM自动生成的工作负载特定二元特征显著优于通用语义嵌入,提升预测器性能。模糊帕累托剪枝有效减少训练配置数,保持性能同时降低计算开销。BRANE是唯一在所有基准上均能达到严格准确率目标且实现成本节省的方法,展示了其强大的泛化能力和实用价值。

应用场景

BRANE适用于需要高效知识检索的智能问答系统、客户支持机器人、科研文献检索助手及金融信息分析平台。通过动态配置选择,系统可根据查询复杂度和预算灵活调整资源分配,提升响应速度和准确率,降低云计算成本。该方法支持多种LLM和检索器组合,便于集成现有工业级检索架构,推动个性化和按需服务的发展。

局限与展望

BRANE依赖于离线性能剖析,初期采样和评估成本较高,限制了快速迭代和部署。模型假设查询特征与配置性能关系稳定,面对动态变化的语料库或查询分布时,需重新训练以维持效果。当前实验主要覆盖英文公开基准,跨语言和跨领域的适应性尚待验证。此外,极端复杂查询的配置选择策略仍有提升空间。

通俗解读 非专业人士也能看懂

想象你在一家大型餐厅工作,厨师们有很多不同的烹饪工具和配料组合,每道菜的做法和用料都不同。传统上,餐厅为所有顾客准备固定菜单,不管顾客点什么菜,厨师都用同样的配方和工具,结果有些菜做得太复杂浪费材料,有些菜又做得太简单味道不佳。BRANE就像一个聪明的点菜助手,它能根据顾客点的菜(自然语言查询),快速判断这道菜需要哪些工具和配料组合(检索管线配置),既保证菜的味道(答案准确率),又节省成本和时间。它通过学习大量历史点菜和做菜的经验,提炼出每道菜的关键特征(工作负载特定查询特征),然后用轻量级模型预测不同做法的成功率,最后挑选出最合适的做法。这样,餐厅既能满足不同顾客的口味需求,又能高效利用资源,避免浪费。

简单解释 像给14岁少年讲一样

嘿,想象你在玩一个超级复杂的游戏,每个任务都有很多不同的装备和技能组合。以前你总是用同一套装备去完成所有任务,结果有些任务太难,有些又浪费资源。BRANE就像你的游戏助手,它能根据每个任务的特点,帮你挑选最合适的装备和技能组合,既能保证你赢得任务,又能节省游戏币!它先学会识别任务的关键特征,然后预测不同装备组合的成功率,最后帮你选出最棒的搭配。这样你玩游戏更轻松,花的钱也少多了!

术语表

大型语言模型 (Large Language Model, LLM)

基于深度学习的自然语言处理模型,能理解和生成自然语言文本。

BRANE使用LLM提取查询特征和评估答案准确率。

检索管线配置 (Retrieval Pipeline Configuration)

包括选择LLM、检索器、文档数量、多跳次数和合成策略等参数的组合。

BRANE动态选择最优配置以平衡成本和准确率。

工作负载特定查询特征 (Workload-Specific Query Characteristics)

针对特定任务集合自动生成的二元查询特征,用于区分不同查询类型。

BRANE通过LLM自动发现这些特征作为预测器输入。

拉格朗日松弛 (Lagrangian Relaxation)

一种优化技术,通过引入拉格朗日乘子将约束问题转化为无约束问题。

BRANE用其实现成本与准确率的权衡选择。

模糊帕累托剪枝 (Fuzzy Pareto Pruning)

基于成本和准确率的容忍度,筛选接近最优的配置以减少训练负担。

BRANE用此策略减少预测器数量。

准确率 (Accuracy)

模型输出正确答案的比例。

BRANE预测每个配置对查询的准确率。

成本 (Cost)

执行查询所需的计算资源和费用,通常以美元计。

BRANE在选择配置时考虑成本。

多跳推理 (Multi-hop Reasoning)

需要跨多个信息片段综合推理以回答查询的能力。

BRANE的查询特征中包括是否需要多跳推理。

知识搜索基准 (Knowledge Search Benchmark)

用于评估知识检索系统性能的标准数据集,如MuSiQue、BrowseComp-Plus、FinanceBench。

BRANE在这些基准上进行实验验证。

预测器 (Predictor)

用于估计特定配置对查询回答正确概率的机器学习模型。

BRANE为每个配置训练独立预测器。

开放问题 这项研究留下的未解疑问

  • 1 如何在动态变化的工作负载和语料库环境下,实现BRANE模型的在线自适应更新,保持性能稳定?
  • 2 如何进一步降低离线性能剖析的成本,使BRANE适用于配置空间极其庞大的工业系统?
  • 3 跨语言和跨领域的查询特征自动发现机制如何设计,以提升BRANE的通用性?
  • 4 在极端复杂或模糊查询场景下,BRANE的配置预测准确率和成本效益如何保障?
  • 5 如何结合多模态信息(如图像、视频)扩展BRANE框架,实现更丰富的查询理解和配置优化?

应用场景

近期应用

智能客服系统

通过BRANE动态选择检索配置,提升客户问题回答的准确率和响应速度,降低云计算成本。

科研文献检索助手

针对不同科研查询自动优化检索策略,提高文献检索的相关性和效率。

金融信息分析平台

根据查询复杂度动态调整检索深度和模型,提升金融数据分析的准确性和实时性。

远期愿景

个性化知识搜索引擎

结合用户历史和偏好,实现按查询动态定制检索配置,提供个性化高效服务。

跨领域多语言智能问答系统

扩展BRANE至多语言和多领域环境,支持全球用户多样化查询需求。

原文摘要

Modern retrieval agents expose many configuration choices -- LLM, retriever, number of documents, number of hops, and synthesis strategy -- each shaping both answer quality and serving cost. Today, these pipelines are typically hand-tuned once per workload, leaving substantial per-query optimization untapped. We formulate the problem: given a natural-language query and either an accuracy or a budget target, select from a predefined pipeline catalog the configuration that minimizes cost or maximizes accuracy at inference time. We propose **BRANE**, which uses an LLM to convert each query into workload-specific characteristics, then trains a lightweight per-configuration predictor that estimates whether the pipeline will answer the query correctly. At inference time, **BRANE** selects the configuration that maximizes predicted correctness penalized by cost, exposing a tunable cost-quality tradeoff without retraining. Across MuSiQue, BrowseComp-Plus, and FinanceBench, **BRANE** consistently pushes the cost-quality Pareto frontier, matches the best fixed configuration's accuracy at up to 89% lower cost, and outperforms LLM-routing, rule-based, and fine-tuned Qwen3-4B baselines. These results show that per-query configuration of the full retrieval pipeline is a practical alternative to static workload-level tuning.

cs.AI eess.SY

参考文献 (19)

Adaptive-RAG: Learning to Adapt Retrieval-Augmented Large Language Models through Question Complexity

Soyeong Jeong, Jinheon Baek, Sukmin Cho 等

2024 480 引用 ⭐ 高影响力 查看解读 →

CARROT: A Cost Aware Rate Optimal Router

Seamus Somerstep, Felipe Maia Polo, Allysson Flavio Melo de Oliveira 等

2025 21 引用 ⭐ 高影响力 查看解读 →

Murakkab: Resource-Efficient Agentic Workflow Orchestration in Cloud Platforms

G. Chaudhry, Esha Choukse, Haoran Qiu 等

2025 12 引用 ⭐ 高影响力 查看解读 →

METIS: Fast Quality-Aware RAG Systems with Configuration Adaptation

Siddhant Ray, Rui Pan, Zhuohan Gu 等

2024 14 引用 ⭐ 高影响力 查看解读 →

FinanceBench: A New Benchmark for Financial Question Answering

Pranab Islam, Anand Kannappan, Douwe Kiela 等

2023 217 引用 ⭐ 高影响力 查看解读 →

♫ MuSiQue: Multihop Questions via Single-hop Question Composition

H. Trivedi, Niranjan Balasubramanian, Tushar Khot 等

2021 848 引用 查看解读 →

OmniRouter: Budget and Performance Controllable Multi-LLM Routing

K. Mei, Wujiang Xu, Shuhang Lin 等

2025 28 引用 查看解读 →

RAG over Thinking Traces Can Improve Reasoning Tasks

Negar Arabzadeh, Wenjie Ma, Sewon Min 等

2026 1 引用 查看解读 →

RouterBench: A Benchmark for Multi-LLM Routing System

Qi Hu, J. Bieker, Xiuyu Li 等

2024 170 引用 查看解读 →

BrowseComp-Plus: A More Fair and Transparent Evaluation Benchmark of Deep-Research Agent

Zijian Chen, Xueguang Ma, Shengyao Zhuang 等

2025 98 引用 查看解读 →

The Distracting Effect: Understanding Irrelevant Passages in RAG

Chen Amiraz, Florin Cuconasu, Simone Filice 等

2025 31 引用 查看解读 →

RouteLLM: Learning to Route LLMs with Preference Data

Isaac Ong, Amjad Almahairi, Vincent Wu 等

2024 416 引用 查看解读 →

Enterprise Deep Research: Steerable Multi-Agent Deep Research for Enterprise Analytics

Akshara Prabhakar, Roshan Ram, Zixiang Chen 等

2025 11 引用 查看解读 →

R2-Router: A New Paradigm for LLM Routing with Reasoning

Jiaqi Xue, Qian Lou, Jiarong Xing 等

2026 6 引用 查看解读 →

The Power of Noise: Redefining Retrieval for RAG Systems

Florin Cuconasu, Giovanni Trappolini, F. Siciliano 等

2024 395 引用 查看解读 →

LightGBM: A Highly Efficient Gradient Boosting Decision Tree

Guolin Ke, Qi Meng, Thomas Finley 等

2017 14591 引用

DeepResearcher: Scaling Deep Research via Reinforcement Learning in Real-world Environments

Yuxiang Zheng, Dayuan Fu, Xiangkun Hu 等

2025 211 引用 查看解读 →

XGBoost: A Scalable Tree Boosting System

Tianqi Chen, Carlos Guestrin

2016 54283 引用 查看解读 →

HedraRAG: Coordinating LLM Generation and Database Retrieval in Heterogeneous RAG Serving

Zhengding Hu, Vibha Murthy, Zaifeng Pan 等

2025 4 引用 查看解读 →