AI-NEWS · 2024年 7月 10日

Llama 3超越GPT-4!英伟达推微调新框架RankRAG

材料总结:Llama 3超越GPT-4!英伟达推微调新框架RankRAG

背景与创新
佐治亚理工学院和英伟达的两位华人学者最近提出了一种名为RankRAG的新型微调框架,这一框架简化了传统的RAG(Retrieval-Augmented Generation)流水线,使得同一个大语言模型(LLM)可以同时完成检索、排名和生成任务,显著提升了性能。

技术细节

  1. RAG流程:

    • RAG是一种常用的技术,通过基于文本编码的稠密模型从外部数据库中检索top-k文本段,然后由LLM进行读取和生成。
    • 传统RAG在k值选择上存在局限:k值过大处理速度慢,k值过小则需要高召回率的检索机制。
  2. RankRAG改进:

    • RankRAG通过微调扩展LLM能力,让LLM自己完成检索和排名,提高数据效率和模型性能。
    • 特别在多个通用基准和生物医学知识密集型基准上,RankRAG微调出的Llama38B/70B模型分别超过了ChatQA-1.58B和ChatQA-1.570B模型。

实验结果

  1. 性能表现:

    • RankRAG在九个通用领域数据集上的表现优于当前开源SOTA模型ChatQA-1.5,特别是在长尾QA和多跳QA任务中提高了10%以上的性能。
    • 在生物医学RAG基准Mirage上,即使没有微调,RankRAG也超过了许多专业领域的开源模型。
  2. 用户体验:

    • RankRAG框架具有高度的交互性和可编辑性,用户能实时查看并编辑AI生成的内容,极大提高工作效率。
    • 本次更新让Artifacts不再局限于Claude平台,用户可以轻松分享。

微调策略

  1. 两个阶段的指令微调:
    • 第一阶段:监督微调(SFT),混合多个数据集以提高LLM的指令跟随能力。
    • 第二阶段:微调数据集包含多种QA数据、检索增强的QA数据和上下文排名数据,进一步提升LLM的检索和排名能力。

未来展望
随着RankRAG框架的提出和不断完善,AI与人类协作创作的前景更加光明。独立开发者和研究人员都能利用这一创新框架激发更多创意和可能性,推动技术和应用的发展。

结论
RankRAG展示了其在检索和生成任务中的强大能力,并且在生物医学领域表现出卓越的适应性。该框架有望成为未来AI协作创作的重要工具。

论文链接: arXiv


这份总结概括了RankRAG框架的主要创新点、技术细节、实验结果、用户体验和未来展望,突出了其在性能提升和应用广泛性方面的优势。

Source:https://www.aibase.com/news/10146