计算机科学 ›› 2026, Vol. 53 ›› Issue (8): 298-306.doi: 10.11896/jsjkx.250600179

• 人工智能 • 上一篇    下一篇

LSQ-RAG:基于LLM增强排序器的检索增强生成框架

张忠林, 夏航   

  1. 兰州交通大学电子与信息工程学院 兰州 730070
  • 收稿日期:2025-06-24 修回日期:2025-09-30 发布日期:2026-08-17
  • 通讯作者: 张忠林(zhangzl@mail.lzjtu.cn)
  • 基金资助:
    国家自然科学基金面上项目(62062049);甘肃省哲学社会科学规划项目阶段性研究成果(20YB056)

LSQ-RAG:Retrieval-enhanced Generation Framework Based on LLM-enhanced Ranker

ZHANG Zhonglin, XIA Hang   

  1. School of Electronic and Information Engineering, Lanzhou Jiaotong University, Lanzhou 730070, China
  • Received:2025-06-24 Revised:2025-09-30 Online:2026-08-17
  • About author:ZHANG Zhonglin,born in 1965,Ph.D,professor.His main research interests include NLP and QA.

摘要: 大语言模型(Large Language Model,LLM)在对话、推理和知识保留能力方面展现了卓越的生成和推理能力,但仍存在许多局限性,例如生成包含幻觉的答案,依赖过时的参数化知识,以及模型解释差等。检索增强生成(Retrieval Augmented Generation,RAG)通过集成非参数数据存储来解决这些问题,但是直接集成信息检索或端到端训练这些组件通常会导致次优结果或计算效率低下。为此,提出了LSQ-RAG框架,其通过LLM监督的排序器增强了LLM的上下文理解能力,并提高了所提供段落的质量和准确性。首先,利用LSQ-RAG对LLM进行微调,使其遵循指令并有区别地使用提供的信息;随后,利用微调后的LLM生成排名分数,并将其作为训练排序器的监督信号。通过充分发挥LLM的强大功能,所提方法消除了排序器训练中对人工标注的依赖,同时实现了更高的性能。实验结果表明,LSQ-RAG在开放域QA和事实核查任务上的表现优于现有的检索增强LLM,同时在应用于不同的LLM时均表现出持续的性能提升,凸显了其多功能性和有效性。

关键词: LSQ-RAG框架, 检索增强生成, 指令微调, 排序器, 大语言模型

Abstract: Large language model(LLM) has demonstrated remarkable generation and reasoning capabilities in terms of conversation,reasoning,and knowledge retention,but they still suffer from numerous limitations,such as generating answers containing hallucinations,relying on outdated parameterized knowledge,and poor model interpretation.RAG(Retrieval Augmented Generation) addresses these issues by integrating non-parametric data stores.However,directly integrating information retrieval or training these components end-to-end often leads to suboptimal results or computational inefficiencies.To this end,this paper proposes the LSQ-RAG framework,which enhances the contextual understanding capabilities of the LLM with an LLM-supervised ranker and improves the quality and accuracy of the provided passages.LSQ-RAG fine-tunes the LLM to follow instructions and discriminatively use the provided information.Subsequently,it leverages the fine-tuned LLM to generate ranking scores,which serve as a supervisory signal for training the ranker.By leveraging the power of the LLM,the proposed approach eliminates the reliance on manual annotations in ranker training while achieving higher performance.Experimental results demonstrate that LSQ-RAG outperforms existing retrieval-augmented LLMs on open-domain QA and fact-checking tasks,while exhibiting consis-tent performance improvements when applied to different LLMs,highlighting its versatility and effectiveness.

Key words: LSQ-RAG framework, Retrieval-enhanced generation, Instruction fine-tuning, Ranker, Large language model

中图分类号: 

  • TP391
[1] BROWN T,MANN B,RYDER N,et al.Language models are few-shot learners[J].Advances in Neural Information Processing Systems,2020,33:1877-1901.
[2] LEWIS P,PEREZ E,PIKTUS A,et al.Retrieval-augmentedgen-eration for knowledge-intensive nlp tasks[J].Advances in Neural Information Processing Systems,2020,33:9459-9474.
[3] MALLEN A,ASAI A,ZHONG V,et al.When not to trust lan-guage models:Investigating effectiveness of parametric and non-parametric memories[J].arXiv:2212.10511,2022.
[4] IZACARD G,GRAVE E.Leveraging passage retrieval with gen-erative models for open domain question answering[J].arXiv:2007.01282,2020.
[5] ARSLAN M,GHANEM H,MUNAWAR S,et al.A Survey on RAG with LLMs[J].Procedia Computer Science,2024,246:3781-3790.
[6] SHI W,MIN S,YASUNAGA M,et al.Replug:Retriev-al-augmented black-box language models[J].arXiv:2301.12652,2023.
[7] YU W H,ITER D,WANG S H,et al.Generate rather thanretrieve:large language models are strong context genera-tors[J].arXiv:2209.10063,2022.
[8] FENG Z Y,FENG X C,ZHAO D Z,et al.Retriev-al-generation synergy augmented large language mod-els[C]//Proceedings of the 2024 IEEE International Conference on Acoustics,Speech and Signal Processing.Piscataway:IEEE,2024:11661-11665.
[9] SHAO Z H,GONG Y Y,SHEN Y L,et al.Enhancing retrieval-augmented large language models with iterative retrieval-generation synergy[J].arXiv:2305.15294,2023.
[10] LV K,YANG Y,LIU T,et al.Full parameter fine-tuning for large language models with limited resources[J].arXiv:2306.09782,2023.
[11] HAN Z,GAO C,LIU J,et al.Parameter-efficient fine-tuning for large models:A comprehensive survey[J].arXiv:2403.14608,2024.
[12] WEI J,TAY Y,BOMMASANI R,et al.Emergent abilities of large language models[J].arXiv:2206.07682,2022.
[13] LV K,YAN H,GUO Q,et al.Adalomo:Low-memory optimization with adaptive learning rate[J].arXiv:2310.10195,2023.
[14] CHUNG H W,HOU L,LONGPRE S,et al.Scaling instruc-tion-finetuned language models[J].Journal of Machine Learning Research,2024,25(70):1-53.
[15] WANG Y,KORDI Y,MISHRA S,et al.Self-instruct:Aligning language models with self-generated instructions[C]//Procee-dings of the 61st Annual Meeting of the Association for Computational Linguistics.2023.
[16] DETTMERS T,PAGNONI A,HOLTZMAN A,et al.Qlora:Efficient finetuning of quantized llms[J].Advances in Neural Information Processing Systems,2023,36:10088-10115.
[17] HAN Z,GAO C,LIU J,et al.Parameter-efficient fine-tuning for large models:A comprehensive survey[J].arXiv:2403.14608,2024.
[18] NOGUEIRA R,CHO K.Passage Re-ranking with BERT[J].arXiv:1901.04085,2019.
[19] ZHANG T,GOLDSTEIN A,LEVIN M.Classical sorting algorithms as a model of morphogenesis:Self-sorting arrays reveal unexpected competencies in a minimal model of basal intelligence[J].Adaptive Behavior,2025,33(1):25-54.
[20] NOGUEIRA R,YANG W,CHO K,et al.Multi-stage document ranking with BERT[J].arXiv:1910.14424,2019.
[21] GOU J,YU B,MAYBANKS J,et al.Knowledge distillation:A survey[J].International Journal of Computer Vision,2021,129(6):1789-1819.
[22] ZHOU W,ZHANG S,POON H,et al.Context-faithful prompting for large language models[J].arXiv:2303.11315,2023.
[23] FANG W,CHUANG Y S,GLASS J.Joint Inference of Retrieval and Generation for Passage Re-ranking[C]//Findings of the Association for Computational Linguistics:EACL 2024.2024:2289-2298.
[24] IVISON H,WANG Y,PYATKIN V,et al.Camels in a changing climate:Enhancing lm adaptation with tulu 2[J].arXiv:2311.10702,2023.
[25] KARPUKHIN V,OČUZ B,MIN S,et al.Dense Passage Retrieval for Open-Domain Question Answering[J].arXiv:2004.04906,2020.
[26] IZACARD G,LEWIS P,LOMELI M,et al.Atlas:Few-shotlearning with retrieval augmented language models[J].Journal of Machine Learning Research,2023,24(251):1-43.
[27] KARPUKHIN V,OGUZ B,MIN S,et al.Dense Passage Retrieval for Open-Domain Question Answering[C]//EMNLP.2020:6769-6781.
[28] DOUZE M,GUZHVA A,DENG C,et al.The faiss library[J].arXiv:2401.08281,2024.
[29] REIMERS N,GUREVYCH I.Sentence-bert:Sentence embed-dings using siamese bert-networks[J].arXiv:1908.10084,2019.
[30] BAJAJ P,CAMPOS D,CRASWELL N,et al.Ms marco:A human generated machine reading comprehension dataset[J].arXiv:1611.09268,2016.
[31] PETRONI F,PIKTUS A,FAN A,et al.KILT:a benchmark for knowledge intensive language tasks[J].arXiv:2009.02252,2020.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!