链载Ai
标题: RAG检索增强之ReRank(重新排序)模型 [打印本页]
作者: 链载Ai 时间: 5 小时前
标题: RAG检索增强之ReRank(重新排序)模型
ingFang SC", "Hiragino Sans GB", "Microsoft YaHei", "WenQuanYi Micro Hei", "Helvetica Neue", Arial, sans-serif;font-style: normal;font-variant-ligatures: normal;font-variant-caps: normal;letter-spacing: normal;orphans: 2;text-align: justify;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;white-space: normal;background-color: rgb(255, 255, 255);text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">了解ReRank模型前先回顾一下RAG的流程ingFang SC", "Hiragino Sans GB", "Microsoft YaHei", "WenQuanYi Micro Hei", "Helvetica Neue", Arial, sans-serif;font-style: normal;font-variant-ligatures: normal;font-variant-caps: normal;letter-spacing: normal;orphans: 2;text-align: justify;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;white-space: normal;background-color: rgb(255, 255, 255);text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">
ingFang SC", "Hiragino Sans GB", "Microsoft YaHei", "WenQuanYi Micro Hei", "Helvetica Neue", Arial, sans-serif;font-style: normal;font-variant-ligatures: normal;font-variant-caps: normal;letter-spacing: normal;orphans: 2;text-align: justify;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;white-space: normal;background-color: rgb(255, 255, 255);text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">什么是Rerank模型?ingFang SC", "Hiragino Sans GB", "Microsoft YaHei", "WenQuanYi Micro Hei", "Helvetica Neue", Arial, sans-serif;font-size: 18px;font-style: normal;font-variant-ligatures: normal;font-variant-caps: normal;font-weight: 400;letter-spacing: normal;orphans: 2;text-align: justify;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;white-space: normal;background-color: rgb(255, 255, 255);text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;"> Rerank模型是一种用于优化信息检索结果排序的机器学习模型,通过精细化评估文档与查询的相关性,提升最终结果的准确性和语义匹配度。以下是其核心要点: 属于重排序算法,作用于初步检索(如关键词匹配或向量相似度检索)之后,对候选文档进行二次筛选和排序。在RAG(检索增强生成)流程中,与Embedding模型配合使用,形成“粗筛+精排”的协同机制。- 解决初步检索的局限性:弥补传统检索方法(如倒排索引或Embedding相似度计算)在语义理解深度上的不足。
- 提升结果质量:通过多维度评估(如语义一致性、上下文关联性)对文档重新打分,确保高相关性的内容优先展示。
- 监督学习训练:基于大量正确与不正确的查询-文档对,模型学习最大化正确对的分数、最小化错误对的分数。
- 相关性评分:输入查询和文档,直接输出两者的匹配分数,并依此排序。
- RAG系统:优化检索文档的排序,提升大模型生成回答的准确性。
- 搜索引擎/推荐系统:精细化调整结果顺序,增强用户满意度。
ingFang SC", "Hiragino Sans GB", "Microsoft YaHei", "WenQuanYi Micro Hei", "Helvetica Neue", Arial, sans-serif;font-style: normal;font-variant-ligatures: normal;font-variant-caps: normal;letter-spacing: normal;orphans: 2;text-align: justify;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;white-space: normal;background-color: rgb(255, 255, 255);text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">排序的关键维度ingFang SC", "Hiragino Sans GB", "Microsoft YaHei", "WenQuanYi Micro Hei", "Helvetica Neue", Arial, sans-serif;font-style: normal;font-variant-ligatures: normal;font-variant-caps: normal;letter-spacing: normal;orphans: 2;text-align: justify;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;white-space: normal;background-color: rgb(255, 255, 255);text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">语义相关性#用户问题:"糖尿病患者的饮食禁忌有哪些?"
候选文档1:详细列举12种糖尿病饮食禁忌(相关度高)
候选文档2:讲解胰岛素注射方法(相关度低)
ingFang SC", "Hiragino Sans GB", "Microsoft YaHei", "WenQuanYi Micro Hei", "Helvetica Neue", Arial, sans-serif;font-style: normal;font-variant-ligatures: normal;font-variant-caps: normal;letter-spacing: normal;orphans: 2;text-align: justify;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;white-space: normal;background-color: rgb(255, 255, 255);text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">时效性权重文档A:2023年《中国糖尿病防治指南》(权重+20%)
文档B:2010年某医院内部资料(权重-30%)
ingFang SC", "Hiragino Sans GB", "Microsoft YaHei", "WenQuanYi Micro Hei", "Helvetica Neue", Arial, sans-serif;font-style: normal;font-variant-ligatures: normal;font-variant-caps: normal;letter-spacing: normal;orphans: 2;text-align: justify;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;white-space: normal;background-color: rgb(255, 255, 255);text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">多样性控制避免返回3篇都讲"糖分控制"的文章
保留1篇"运动管理"的补充内容
ingFang SC", "Hiragino Sans GB", "Microsoft YaHei", "WenQuanYi Micro Hei", "Helvetica Neue", Arial, sans-serif;font-style: normal;font-variant-ligatures: normal;font-variant-caps: normal;letter-spacing: normal;orphans: 2;text-align: justify;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;white-space: normal;background-color: rgb(255, 255, 255);text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">排序面临的挑战长尾问题
用户提问:“如何训练导盲犬AI机器人?”检索结果:前10篇:通用机器人训练方法(未命中)仅1篇:《基于多模态感知的导盲犬机器人训练指南》(命中但排名靠后)
解决策略:
数据增强:合成"导盲犬+机器人"的伪数据微调模型混合检索:结合关键词(“导盲犬”+“AI”)与语义检索主动学习:标注低置信度结果,迭代优化模型
ingFang SC", "Hiragino Sans GB", "Microsoft YaHei", "WenQuanYi Micro Hei", "Helvetica Neue", Arial, sans-serif;font-style: normal;font-variant-ligatures: normal;font-variant-caps: normal;letter-spacing: normal;orphans: 2;text-align: justify;text-indent: 0px;text-transform: none;widows: 2;word-spacing: 0px;-webkit-text-stroke-width: 0px;white-space: normal;background-color: rgb(255, 255, 255);text-decoration-thickness: initial;text-decoration-style: initial;text-decoration-color: initial;">语义鸿沟用户问:“手机发热严重怎么降温?检索结果:《移动设备SoC功耗管理与散热优化方案》(语义相关)《智能手机电池保养技巧》(字面相关但非核心)
解决策略:
1.查询扩展:用LLM生成同义表述(如"发热"→"散热"、“降温"→"温度控制”)2.上下文增强:提取文档中"发热"相关段落提升权重3.用户反馈:记录用户最终点击的文档,反向优化模型
多语言混合
中文提问:“量子纠缠的实际应用有哪些?检索结果:中文文档:《里子通信技术白皮书》(匹配度一般)英文论文:《QuantumEntanglementinCommercialSystems》(Nature2023,高相关)
解决策略:
1.实时翻译对齐:将英文论文摘要翻译后参与排序2.跨语言模型:使用mBERT等模型直接计算中英文相似度3.多语言标签:为文档添加语言/领域元数据辅助过滤
计算效率
数据规模:100万篇医疗文献库查询需求:实时返回"阿尔茨海默症新药研发进展"Top5结果(<500ms)
解决策略:
1.两阶段排序:第一阶段:BM25快速筛选1000篇(耗时50ms)第二阶段:Reranker精排Top100(耗时400ms)2.模型蒸馏:将BERT-large蒸馏为Tiny版,速度提升5倍3.硬件加速:使用TensorRT部署模型,GPU推理吞吐量提升10倍
主流模型选型
模型 | 特点 | 性能优势 |
BGE ReRanker | 支持多语言 | 多语言场景、高精度需求 |
Jina Reranker | 8k上下文支持 | 长文本排序、低延迟场景 |
BCE-Reranker | 网易有道开源,中英跨语言优化 | 中英混合场景、高召回率需求 |
完整排序流程示例
用户问题:"美联储加息对A股的影响"
返回50篇文档:
- 10篇关于美国货币政策
- 15篇A股市场分析
- 20篇历史加息案例
- 5篇无关内容
for文档in50篇:
计算语义相关性(BERT模型)→得分0.6-0.95
叠加时效性权重(2023年文档×1.2)
扣除低权威惩罚(自媒体文章×0.7)
最终得分=语义分×时效权重×权威系数
排序后Top3:
1.《2023年美联储政策与新兴市场联动分析》(0.94)
2.《跨境资本流动对A股的影响机制》(0.91)
3.《历史六轮加息周期中的板块表现》(0.89)
Reranker本质
Reranker是知识库的"智能质检员",假设你在图书馆找书,先用关键词检索到100本书,但需要找出最相关的3本才行。
图书管理员(Reranker)会综合评估书籍内容、出版时间、作者权威性等维度进行二次筛选。
将检索到的候选文档(如100条)按照与问题的相关度重新排序,把最匹配的结果提升到Top位置。
Reranker模型是RAG系统中的智能排序引擎, 在知识库检索流程中承担着关键的优化角色。它通过多维度智能分析,对初步检索结果进行精细化处理,将最符合用户真实需求的信息精准呈现在前列。
不同于初检阶段的粗粒度筛选,Reranker会综合评估语义相关度(如问题与内容的深层匹配)、时效性(优先最新资料)、权威性(区分专家论述与普通观点)以及内容完整性(覆盖关键要素的程度)等多个核心维度,通过算法加权计算出每个结果的最终排序得分。
在企业级应用中, 这种智能排序机制有效解决了传统检索中面临的长尾问题、语义鸿沟等挑战,大幅提升了知识库的可用性和准确性,是确保专业用户获得高价值信息的关键技术保障。
| 欢迎光临 链载Ai (https://www.lianzai.com/) |
Powered by Discuz! X3.5 |