
RAG系统的技术基石是embedding模型,基于embedding模型可以将私域数据矢量化处理并存储。如果数据和文档的量不大,那一些都很美好,但是像所有软件系统一样,随着数据规模的增长,RAG系统的可扩展性(scaling)就成了一个问题. 一方面,存储(尤其是RAM)成本陡增;另一方面,搜索过程中的计算成本也变得更复杂。 今天就给大家扒一扒,如何对embedding矢量数据“
瘦身”,让RAG变得轻盈高效!
核心问题
- 如何充分降低embedding存储成本的同时,不让损害搜索质量?
- 如何选择embedding存储的优化策略:量化(quantization),降维(dimensionality reduction)? 可否将量化和降维联合使用?效果如何?
- 如何在内存成本和性能之间做平衡?
Embedding”瘦身“方法
大家都知道,embedding模型会把token进行高纬矢量化(上千个维度),每个维度由一个高精度的数值表示。所以“瘦身”的方法也自然有两种:
- 量化(Quantization): 降低每个维度数值的精度,从原有的32位浮点数(float32)减低为8位浮点数(float8),8位整数(int8), 甚至是二进制(binary). 类似对embedding适量数据进行“轻断食减重”, 降低身体的 “能量密度”(数值精度)
- 降维(dimensionality reduction): 降低每个适量的维度。常见的算法有:PCA(Principal Component Analysis)等。 好像对embedding适量数据“局部塑形抽脂”,直接抽掉无用的脂肪(冗余维度)
关键结论
- Float8量化综合效果最佳:
- 存储成本线性降低4倍, 性能只损失0.3%
- Int8有同样的成本优势,但是量化处理的逻辑要更简单。因为Int8需要一个额外的标定过程。
- PCA是降维的最佳算法:
- PCA能够在降低维度的同时, 保持搜索的质量, 而且算法本身也更简单。
- PCA降维和Float8量化效果更佳:
- 先用PCA降维50%,再用 float8量化处理,总体能降低8倍的成本。这个“组合拳”比单独用Int8量化效果的搜索质量还好。
- Binary量化过于激进:
- 存储成本确实降低了32倍,但是搜索性能也受损严重。
- 更高维的embedding更经“折腾”:
- 原本维度更高的embedding数据,”瘦身“后的效果要更稳定一点,本身维度就不高的,就不建议这么处理了。
实战建议
-
在实战中如何选型呢?可以把各种策略的组合实际尝试下,然后绘制一条成本压缩和性能保持直接的关系曲线。再根据你实际的内存限制,找到最合理的策略!
-
适合端侧设备或者浏览器,因为这些场景下内存都非常有限。
-
相比重新训练模型,“瘦身”要容易得多。
这里给大家精心整理了一份
全面的AI大模型学习资源,包括:AI大模型全套学习路线图(从入门到实战)、精品AI大模型学习书籍手册、视频教程、实战学习、面试题等,资料免费分享!
👇👇扫码免费领取全部内容👇👇

1. 成长路线图&学习规划
要学习一门新的技术,作为新手一定要先学习成长路线图,方向不对,努力白费。
这里,我们为新手和想要进一步提升的专业人士准备了一份详细的学习成长路线图和规划。可以说是最科学最系统的学习成长路线。

2. 大模型经典PDF书籍
书籍和学习文档资料是学习大模型过程中必不可少的,我们精选了一系列深入探讨大模型技术的书籍和学习文档,它们由领域内的顶尖专家撰写,内容全面、深入、详尽,为你学习大模型提供坚实的理论基础。(书籍含电子版PDF)

3. 大模型视频教程
对于很多自学或者没有基础的同学来说,书籍这些纯文字类的学习教材会觉得比较晦涩难以理解,因此,我们提供了丰富的大模型视频教程,以动态、形象的方式展示技术概念,帮助你更快、更轻松地掌握核心知识。

4. 2026行业报告
行业分析主要包括对不同行业的现状、趋势、问题、机会等进行系统地调研和评估,以了解哪些行业更适合引入大模型的技术和应用,以及在哪些方面可以发挥大模型的优势。

5. 大模型项目实战
学以致用 ,当你的理论知识积累到一定程度,就需要通过项目实战,在实际操作中检验和巩固你所学到的知识,同时为你找工作和职业发展打下坚实的基础。

6. 大模型面试题
面试不仅是技术的较量,更需要充分的准备。
在你已经掌握了大模型技术之后,就需要开始准备面试,我们将提供精心整理的大模型面试题库,涵盖当前面试中可能遇到的各种技术问题,让你在面试中游刃有余。

7. 资料领取:全套内容免费抱走,学 AI 不用再找第二份
不管你是 0 基础想入门 AI 大模型,还是有基础想冲刺大厂、了解行业趋势,这份资料都能满足你!
现在只需按照提示操作,就能免费领取:
👇👇扫码免费领取全部内容👇👇

464



被折叠的 条评论
为什么被折叠?



