阿里云 Elasticsearch 使用 RRF 混排优化语义查询结果对比

本文涉及的产品
检索分析服务 Elasticsearch 版,2核4GB开发者规格 1个月
简介: Elasticsearch 从8.8版本开始,新增 RRF,支持对多种不同方式召回的多个结果集进行综合再排序,返回最终的排序结果。之前 Elasticsearch 已经分别支持基于 BM25 的相关性排序和向量相似度的召回排序,通过 RRF 可以对这两者的结果进行综合排序,可以提升排序的准确性。

背景介绍

Elasticsearch 从8.8版本开始,新增 RRF,支持对多种不同方式召回的多个结果集进行综合再排序,返回最终的排序结果。之前Elasticsearch已经分别支持基于BM25的相关性排序和向量相似度的召回排序,通过RRF可以对这两者的结果进行综合排序,可以提升排序的准确性。本文通过一个具体例子来介绍这个过程。


前置条件

需要准备一个8.8+版本Elasticsearch集群,阿里云Elasticsearch已经上线售卖8.9.1的集群可以购买


选择版本8.9


选择节点配置


随后选择网络配置,点击购买即可。


RRF试验

RRF介绍

算法公式:

k是一个常数值,默认是60,R包含每个查询结果的doc排序集合。r(d)表示doc d在某个查询条件中排序号,排序次序从1开始。


如下是一个排序的示例:


数据和模型准备

按照ESRE 系列(一)文章中介绍的方法,通过 Eland 导入 text_embedding 的模型和启动模型部署,通过 kibana 上传初始数据,配置 text-embeddings pipeline,通过索引重建,生成带向量的索引数据。


查询效果评估

同样,我们从 TREC 2019 Deep Learning Track 对“段落排名任务”的查询中选取了1个查询,分别使用文本查询,向量召回,和RRF混排,获得前 10 个结果并参考 TREC 判断


我们以"hydrogen is a liquid below what temperature" 这个查询条件为例,分别构造三个查询

// RRF混排查询GETcollection-with-embeddings/_search{
"size": 10,
"query": {
"query_string": {
"query": "hydrogen is a liquid below what temperature"    }
  },
"knn": [
    {
"field": "text_embedding.predicted_value",
"k": 10,
"num_candidates": 100,
"query_vector_builder": {
"text_embedding": {
"model_id": "sentence-transformers__msmarco-minilm-l-12-v3",
"model_text": "hydrogen is a liquid below what temperature"        }
      }
    }
  ],
"_source": [
"id"  ],
"rank": {
"rrf": {
"window_size": 10,
"rank_constant": 1    }
  }
}
//向量查询GETcollection-with-embeddings/_search{
"size": 10,
"knn": [
    {
"field": "text_embedding.predicted_value",
"k": 10,
"num_candidates": 100,
"query_vector_builder": {
"text_embedding": {
"model_id": "sentence-transformers__msmarco-minilm-l-12-v3",
"model_text": "hydrogen is a liquid below what temperature"        }
      }
    }
  ],
"_source": [
"id"  ] 
}
//文本查询GETcollection-with-embeddings/_search{
"size": 10,
"query": {
"query_string": {
"query": "how are some sharks warm blooded"    }
  },
"_source": [
"id"  ] 
}

对于查询"hydrogen is a liquid below what temperature",三种查询方式返回结果和准确性如下。准确性分为四个等级:不相关 (0)、相关(段落切题但没有回答问题)(1)、高度相关 (2) 和完全相关 (3)。准确性分数越高越相关


RRF混排

向量召回

文本查询

段落ID 

准确性

段落ID  

准确性

段落ID  

准确性

8588222

0

8588222

0

7911557

3

8588219

3

8588219

3

8588219

3

7911557

3

6080460

3

8588222

0

128984

3

128984

3

2697752

2

6080460

3

4254815

1

128984

3

2697752

2

6343521

1

1721142

0

4254815

1

1020793

0

8588227

0

1721142

0

4254811

3

302210

1

6343521

1

1959030

0

2697746

2

8588227

0

4254813

1

7350325

0


可以看出RRF的排序结果,在向量查询结果的基础上,结合了文本查询的结果,将没有在向量召回出现的结果"7911557" 排到了前面,提高了召回的准确性。


与此同时,RRF的排序结果,也将没有在文本查询中出现的结果"6080460" 排到了前面,提高了召回的准确性。




Elasticsearch技术社区2群

image.png

相关实践学习
使用阿里云Elasticsearch体验信息检索加速
通过创建登录阿里云Elasticsearch集群,使用DataWorks将MySQL数据同步至Elasticsearch,体验多条件检索效果,简单展示数据同步和信息检索加速的过程和操作。
ElasticSearch 入门精讲
ElasticSearch是一个开源的、基于Lucene的、分布式、高扩展、高实时的搜索与数据分析引擎。根据DB-Engines的排名显示,Elasticsearch是最受欢迎的企业搜索引擎,其次是Apache Solr(也是基于Lucene)。 ElasticSearch的实现原理主要分为以下几个步骤: 用户将数据提交到Elastic Search 数据库中 通过分词控制器去将对应的语句分词,将其权重和分词结果一并存入数据 当用户搜索数据时候,再根据权重将结果排名、打分 将返回结果呈现给用户 Elasticsearch可以用于搜索各种文档。它提供可扩展的搜索,具有接近实时的搜索,并支持多租户。
相关文章
|
14天前
|
存储 自然语言处理 Java
Elasticsearch写入优化
【10月更文挑战第3天】Elasticsearch:从写入原理谈写入优化
55 2
|
12天前
|
存储 JSON 监控
大数据-167 ELK Elasticsearch 详细介绍 特点 分片 查询
大数据-167 ELK Elasticsearch 详细介绍 特点 分片 查询
28 4
|
17天前
|
自然语言处理 搜索推荐 Java
SpringBoot 搜索引擎 海量数据 Elasticsearch-7 es上手指南 毫秒级查询 包括 版本选型、操作内容、结果截图(一)
SpringBoot 搜索引擎 海量数据 Elasticsearch-7 es上手指南 毫秒级查询 包括 版本选型、操作内容、结果截图
37 0
|
17天前
|
存储 自然语言处理 搜索推荐
SpringBoot 搜索引擎 海量数据 Elasticsearch-7 es上手指南 毫秒级查询 包括 版本选型、操作内容、结果截图(二)
SpringBoot 搜索引擎 海量数据 Elasticsearch-7 es上手指南 毫秒级查询 包括 版本选型、操作内容、结果截图(二)
22 0
|
1月前
|
JSON 自然语言处理 算法
ElasticSearch基础2——DSL查询文档,黑马旅游项目查询功能
DSL查询文档、RestClient查询文档、全文检索查询、精准查询、复合查询、地理坐标查询、分页、排序、高亮、黑马旅游案例
ElasticSearch基础2——DSL查询文档,黑马旅游项目查询功能
|
2月前
|
自然语言处理 Java 关系型数据库
ElasticSearch 实现分词全文检索 - 聚合查询 cardinality
ElasticSearch 实现分词全文检索 - 聚合查询 cardinality
47 1
|
2月前
|
存储 自然语言处理 Java
ElasticSearch 实现分词全文检索 - 经纬度定位商家距离查询
ElasticSearch 实现分词全文检索 - 经纬度定位商家距离查询
20 0
|
2月前
|
自然语言处理 Java
ElasticSearch 实现分词全文检索 - 高亮查询
ElasticSearch 实现分词全文检索 - 高亮查询
55 0
|
2月前
|
缓存 自然语言处理 Java
ElasticSearch 实现分词全文检索 - filter查询
ElasticSearch 实现分词全文检索 - filter查询
40 0
|
2月前
|
自然语言处理 Java
ElasticSearch 实现分词全文检索 - 复合查询
ElasticSearch 实现分词全文检索 - 复合查询
53 0

相关产品

  • 检索分析服务 Elasticsearch版