中文

CDS4RAG:面向RAG的循环双序列超参数优化

机器学习 2026-05-12 v1 人工智能 计算与语言 性能 软件工程

摘要

检索增强生成(RAG)对检索器和生成器的众多超参数极为敏感,但仅使用给定查询进行优化具有挑战性,因为其复杂的相互作用和高昂的评估成本。现有算法在收敛速度上无效且缓慢,因其常将RAG视为单一的黑箱或仅优化部分超参数。本文提出CDS4RAG,一种通过新型循环双序列公式优化完整RAG超参数的框架。CDS4RAG的独特之处在于它区分了检索器和生成器的超参数,循环式地对其进行优化。这一范式允许我们设计细粒度的循环内预算分配,并通过跨循环种子加速生成器的优化。CDS4RAG也是一种与算法无关的框架,可与多种通用算法配合使用。通过在四个常见基准测试和两个主干LLM上的实验,我们揭示了CDS4RAG在21/24种情况下均显著提升基础算法的性能,并在所有案例中均显著超过最先进的算法,生成质量提升最高可达1.54倍。

关键词

引用

@article{arxiv.2605.08333,
  title  = {CDS4RAG: Cyclic Dual-Sequential Hyperparameter Optimization for RAG},
  author = {Pengzhou Chen and Tao Chen},
  journal= {arXiv preprint arXiv:2605.08333},
  year   = {2026}
}

备注

Accepted by main track at IJCAI 2026