MBA-RAG:一种基于多臂赌博机的自适应检索增强生成方法
📝
内容提要
本文研究了现有检索增强生成(RAG)框架在处理不同复杂度查询时的低效问题,提出了一种基于强化学习的动态检索策略选择框架。通过使用多臂赌博机算法和动态奖励函数,显著提升了检索精度及效率,实现了多个数据集上的最新最优效果,同时降低了检索成本。
➡️
本文研究了现有检索增强生成(RAG)框架在处理不同复杂度查询时的低效问题,提出了一种基于强化学习的动态检索策略选择框架。通过使用多臂赌博机算法和动态奖励函数,显著提升了检索精度及效率,实现了多个数据集上的最新最优效果,同时降低了检索成本。