Pinot与向量索引技术探秘 GenAI时代的实时数据分析 Apache
发布时间:2024-11-15 05:34:13点击:
1.数据价值随时间的变化
如上图所示,可以看出,对于单个事件来说,数据价值会随着时间逐渐降低,比如,用户对小红书上某个帖子或淘宝上的产品发生了点击行为,之后就会有相关的产品或广告的推荐,以及进行步更多的链接、视频等各式各类的推荐,推荐的效果在点击发生当时最有用,之后效果会递减。同时,由于有很多人有类似的数据,我们可以在这些数据上做一些聚合分析,这类聚合分析的价值会随着时间和数据量的增加而增加。之前我们会把大部分的时间和精力都聚焦在这类批式的后处理的分析上,而对实时数据分析的关注较少。
随着实时机器学习以及数据分析应用的发展,实时数据的价值越来越多地被挖掘出来。
过去,数据分析更多是面向公司内部的分析,比如运营数据分析等;未来,数据分析将更多用来服务最终用户。比如,推荐算法,本身就是用户自己的数据来训练模型,通过把用户的数据变为产品又给用户使用,完成整个数据的生命周期,并且能给用户更好的体验。
2.实时数据分析边界拓展
实时数据分析的边界也在不停地拓展。比如:
3.典型的数据分析技术栈
上图是一个典型的数据分析技术栈。左边是数据的生成端,在各种设备上,生成各种原始数据;中间是基于用户或产品特性的数据,比如,电商的订单数据,广告的事件数据,这些数据存到 OLTP,如 MySQL 数据库中,或 Kafka 的消息队列;接下来,OLAP 分析端,离线分析端一般会将 ETL 来的数据,存入数仓或数据湖里,实时分析端一般从 Kafka 直接消费进入实时数据库中。这里的目标是低数据查询延时、高数据查询精度,并支持更大的查询量。
Apache Pinot 是一个支持高可用、高并发、低延时的快速实时分布式 OLAP。为了方便处理互联网上的用户数据,Apache Pinot 是采用列式存储的 OLAP,并且可以与 Kafka 集成起来,进行实时数据处理。
Apache Pinot 已在一些大公司里被使用。在 LinkedIn 支持每秒上百万事件,20万 QPS 的毫秒级查询;在 stripe 上,支持单表 1 千亿行,数据量超 PB,2 万 QPS,25 毫秒查询延时;Uber 也是对实时性要求很高的应用,用 Pinot 来实现地理信息的查询。
Pinot 是 Apache 基金会的项目,所有内容都完全开源,很多公司都在使用,如上图所示。大家如果有兴趣,也欢迎试用,并在社区中交流。
Pinot 采用 Lambda 架构,架构图所示,包括:
为了支持低延时、高并发的查询,Pinot 的工程优化目标是尽量减少单次查询需要扫描的数据量。
为此 Pinot 实现了很多索引来帮助缩小查询范围,包括常见的反向索引、排序索引,以及对于特殊数据格式和类型的索引,如区间范围的索引、半结构化 Json 数据的索引、地理信息的索引等。
同时,对于聚合查询,相比于对于全维度的预聚合,Pinot 内部可以根据维度的,动态的聚合结果,使得查询效率更高。
Pinot 采用可插拔架构,查询层、索引层、存储层都支持 SPI 接口,用户可以方便地在 Pinot 上进行二次扩展开发。
上图展示了 Pinot 的可插拔架构。基于 Pinot SPI 包,开发者可以配置开发自己的数据存储、输入格式、流式引擎、索引、查询等。开发者只需要写一些简单的实现,就可以构建、打包然后应用,非常方便。比如,作者仅花一周左右时间,就实现了向量索引功能。
下面将介绍在 AI 时代,Pinot 如何支持 AI 应用。
Pinot 可以作为一个向量数据库为 AI 提供 Embedding 数据存储以及向量检索。
Embedding 可以看成一个浮点数的数组,存到数据库中。有了 Embedding 向量后,当一个向量进来,需要查询和这个向量最相似的 K 个结果。
这类应用场景有很多,如常见的搜索场景和推荐场景。
例如二手车搜索场景。用户可能会去二手车网站,搜索想要的车。在搜索过程中,用户会有很多偏好,比如喜好品牌、年份区间、价格区间、颜色、天窗等。直接使用筛选的话,结果可能过多或过少。我们可以把用户搜索的偏好描述,作为一个 Embedding,存入数据库。然后根据用户偏好描述,返回一个与描述相似的 topK 排序,展示给用户。
用户推荐餐厅场景。可以根据用户历史喜欢吃的东西,同时结合用户当前的描述,生成一个 Embedding,存储起来。然后根据 Embedding,给用户推荐和他偏好相近的产品,这可以得到更好的用户产品体验。
为什么需要实时呢?比如问 ChatGPT,“在足球界,药厂是什么梗?”,回答“勒沃库森”“但至今未能获得联赛冠军”,而熟悉足球的小伙伴会知道,“勒沃库森”在 2024 年第一次获得德甲联赛冠军。
因此,可以看出 LLM 模型和数据都是有时效性的,如果没有最准确最实时的数据,可能获取不到准确的结果。通过 RAG 和实时向量数据库的帮助,模型可以提供更实时和准确的结果。
为了实现实时 RAG,在向量数据方面,我们需要通过将原始数据变成 Embedding,存到向量数据库中,从而提供给用户查询。
上图是一个一般的流程,用来实现用户回复中对于实时数据的要求。
将原始文本文件、音频或照片根据 Embedding 模型,得到他们的 Embedding 表示,写入 Kafka 并存入 Pinot,这样实时的数据流就变成向量进入了向量库。
对于一个用户,将他的问题,也转换成对应的 Embedding,然后通过向量相似性搜索,就可以找到一堆类似的数据;这些数据称为加强上下文(Enhance Context),也就是把相关的文本和用户本身的问题合在一起,交给 Prompt,再通过 LLM,生成给用户的回答。
通过这样的数据处理,可以将实时数据流与 RAG 结合起来,达到更好的准确性。
Pinot 提供了很多向量计算的支持,比如距离计算函数,支持 L1、L2、余弦、点积等距离。可以根据不同的用户场景,选择不同的距离计算方法,达到不同的效果。
Pinot 通过向量索引可以提高查询效率。采用 HNSW 算法,来建立向量索引,然后通过多层迭代的方式,帮助用户输入的向量,找到离它最近的 n 个向量。这样相对于全表扫描对每个向量都计算一个距离的方法,计算复杂度大幅降低。
而对于开发者来说,这个查询也很简单。如上图右边代码所示,定义了算法使用的距离,以及向量相似度查询的语句,返回 10 条最近的结果。
算法在不同数据集上的性能表现如上图所示。我们使用了 Lucene Benchmark,采用 Lucene 的 HNSW 算法的实现。大家也可以点击链接去看看不同算法的实现和性能,因 Lucene 也是开源的,我们会跟社区一起进行性能优化。
通过混合负载,可以将 OLAP 和向量数据库结合起来,以实现维度数据的过滤,并结合其它数据,提高 prompt 的准确度。
比如,用户可以有产品评价数据的 Embedding,也有产品本身的特征,比如 SKU_ID,类别等。在做向量查询时,需要根据类别等维度特征过滤,再做相似性查询,查完的结果合起来,输出给 Prompt,进行后面的反馈生成。
最后,对本次分享的主要内容进行一下总结。