官方网站-首页官方网站-首页

首页 新闻资讯 如何在基于Arm Neoverse平台的Google Axion处理器上构建RAG应用
如何在基于Arm Neoverse平台的Google Axion处理器上构建RAG应用
发布日期:2025-05-11 19:31:04

【导(dǎo)语(yǔ)】在(zài)人(rén)工(gōng)智(zhì)能(néng)领(lǐng)域,确(què)保(bǎo)聊(liáo)天(tiān)机(jī)器(qì)人(rén)提(tí)供(gōng)准(zhǔn)确(què)且(qiě)及(jí)时(shí)的(de)回(huí)答(dá)至(zhì)关重(zhòng)要(yào)。检(jiǎn)索(suǒ)增(zēng)强(qiáng)生(shēng)成(chéng)(RAG)技(jì)术(shù)为(wèi)此(cǐ)提(tí)供(gōng)了(le)一(yī)种(zhǒng)高(gāo)效(xiào)解(jiě)决(jué)方(fāng)案(àn)。本(běn)文由(yóu)Arm基(jī)础(chǔ)设(shè)施(shī)事(shì)业(yè)部(bù)的(de)AI解(jiě)决(jué)方(fāng)案(àn)架(jià)构(gòu)师(shī)Na Li等(děng)人(rén)撰(zhuàn)写(xiě),将(jiāng)深(shēn)入(rù)探(tàn)讨(tǎo)RAG的(de)性(xìng)能(néng)优(yōu)势(shì),并(bìng)分(fēn)享(xiǎng)如(rú)何(hé)在(zài)基(jī)于(yú)Arm Neoverse平(píng)台(tái)的(de)Google Axion处(chù)理(lǐ)器(qì)上(shàng)构(gòu)建(jiàn)RAG应(yīng)用(yòng),以(yǐ)优(yōu)化(huà)AI工(gōng)作(zuò)负(fù)载(zài)。测(cè)试(shì)显(xiǎn)示(shì),相(xiāng)较(jiào)于(yú)x86架(jià)构(gòu)处(chù)理(lǐ)器(qì),Google Axion处(chù)理(lǐ)器(qì)在(zài)性(xìng)能(néng)和(hé)成(chéng)本(běn)效(xiào)益(yì)方(fāng)面(miàn)均(jūn)表(biǎo)现(xiàn)出(chū)色(sè),为(wèi)实(shí)时(shí)、动(dòng)态(tài)AI应(yīng)用(yòng)提(tí)供(gōng)了(le)有(yǒu)力(lì)支(zhī)持(chí)。无(wú)论(lùn)你(nǐ)是(shì)AI开(kāi)发(fā)者(zhě)还(hái)是(shì)对(duì)企(qǐ)业(yè)AI战(zhàn)略(è)感(gǎn)兴(xìng)趣(qù),本(běn)文都(dōu)将(jiāng)为(wèi)你(nǐ)提(tí)供(gōng)宝(bǎo)贵(guì)的(de)见(jiàn)解(jiě)和(hé)实(shí)践(jiàn)指(zhǐ)导(dǎo)。

作(zuò)者(zhě):Arm 基(jī)础(chǔ)设(shè)施(shī)事(shì)业(yè)部(bù) AI 解(jiě)决(jué)方(fāng)案(àn)架(jià)构(gòu)师(shī) Na Li 等(děng)

你(nǐ)是(shì)否(fǒu)好(hǎo)奇(qí)如(rú)何(hé)防(fáng)止(zhǐ)人(rén)工(gōng)智(zhì)能(néng) (AI) 聊(liáo)天(tiān)机(jī)器(qì)人(rén)给(gěi)出(chū)过(guò)时(shí)或(huò)不(bù)准(zhǔn)确(què)的(de)答(dá)案(àn)?检(jiǎn)索(suǒ)增(zēng)强(qiáng)生(shēng)成(chéng) (Retrieval-Augmented Generation, RAG) 技(jì)术(shù)提(tí)供(gōng)了(le)一(yī)种(zhǒng)强(qiáng)大(dà)的(de)解(jiě)决(jué)方(fāng)案(àn),能(néng)够(gòu)显(xiǎn)著(zhe)提(tí)升(shēng)答(dá)案(àn)的(de)准(zhǔn)确(què)性(xìng)和(hé)相(xiāng)关性(xìng)。

本(běn)文将(jiāng)探(tàn)讨(tǎo) RAG 的(de)性(xìng)能(néng)优(yōu)势(shì),并(bìng)分(fēn)享(xiǎng)如(rú)何(hé)在(zài)基(jī)于(yú) Arm Neoverse 平(píng)台(tái)的(de) Google Axion 处(chù)理(lǐ)器(qì)上(shàng)构(gòu)建(jiàn) RAG 应(yīng)用(yòng),以(yǐ)优(yōu)化(huà) AI 工(gōng)作(zuò)负(fù)载(zài)。在(zài)本(běn)文的(de)测(cè)试(shì)中(zhōng),Google Axion 处(chù)理(lǐ)器(qì)相(xiāng)较(jiào)于(yú) x86 架(jià)构(gòu)处(chù)理(lǐ)器(qì),性(xìng)能(néng)提(tí)升(shēng)了(le) 2.5 倍(bèi),并(bìng)节(jié)省(shěng)了(le) 64% 的(de)成(chéng)本(běn)。Google Axion 处(chù)理(lǐ)器(qì)通(tōng)过(guò)更(gèng)好(hǎo)的(de) RAG 性(xìng)能(néng)加(jiā)速(sù)推(tuī)理(lǐ)过(guò)程(chéng),从(cóng)而(ér)实(shí)现(xiàn)更(gèng)快(kuài)的(de)知(zhī)识(shi)检(jiǎn)索(suǒ)、更(gèng)低(dī)的(de)响(xiǎng)应(yīng)延(yán)迟(chí)和(hé)更(gèng)高(gāo)效(xiào)的(de) AI 推(tuī)理(lǐ),这(zhè)对(duì)于(yú)实(shí)时(shí)、动(dòng)态(tài) AI 应(yīng)用(yòng)至(zhì)关重(zhòng)要(yào)。

了(le)解(jiě) RAG:高(gāo)效(xiào)的(de) AI 文本(běn)生(shēng)成(chéng)方(fāng)法(fǎ)

RAG 是(shì)一(yī)款(kuǎn)主流(liú) AI 框(kuāng)架(jià),能(néng)够(gòu)实(shí)时(shí)检(jiǎn)索(suǒ)相(xiāng)关外(wài)部(bù)知(zhī)识(shi),从(cóng)而(ér)提(tí)升(shēng)大(dà)语(yǔ)言(yán)模(mó)型(xíng) (LLM) 生(shēng)成(chéng)文本(běn)的(de)质(zhì)量(liàng)和(hé)相(xiāng)关性(xìng)。与(yǔ)仅(jǐn)依(yī)赖(lài)静(jìng)态(tài)预(yù)训(xun)练(liàn)数(shù)据(jù)集的方法不同,RAG 动态集成了最新外部资源信息,能够生成更精确且贴近上下文的输出结果。这使得 RAG 在实际应用场景中表现出色,例如客服聊天机器人、智能体工具和动态内容生成等场景。

何时选择 RAG 而非微调或重新训练?

基础 LLM 通过类似人类的文本生成功能彻底改变了 AI 领域,但其有效性取决于模型是否拥有企业所需的最新信息。对经过预训练的 LLM 模型进行重新训练和微调是集成额外知识的两种常用方法。重新训练 LLM 是一个资源密集型的复杂过程;而微调则能够使用特定数据集对 LLM 进行训练,调整模型的权重,以更好地完成目标任务。不过,模型仍然需要定期重新部署,以保持与时俱进。

通常,在将 LLM 纳入 AI 战略时,必须评估 LLM 的能力和局限性。主要考虑因素包括:

训练数据集的局限性:对于训练数据集未包含的主题,LLM 可能难以提供准确或最新的信息。

资源需求高:重新训练这些大模型需要大量的算力和工程资源,使得频繁更新难以实施。

对内部知识的访问受到限制:由于企业的主要业务数据受到防火墙的保护,因此 LLM 无法通过定期重新训练纳入专有信息,这可能会限制 LLM 在企业内部使用时的相关性。

RAG 的优势

RAG 无需修改 LLM,只需利用外部数据源更新知识库,将动态信息检索与语言模型的生成能力相结合。如果你所在的领域知识经常变化,那么 RAG 是保持准确性和相关性,并减少 LLM 幻觉的理想解决方案。

RAG 的实际应用:对比分析

在以下所举的例子中,比较了通用 LLM(左)和经过 RAG(右)增强的聊天机器人。左图中,由于信息过时或缺乏特定领域的知识,聊天机器人难以准确回答用户的询问;而 RAG 增强型聊天机器人能够从上传的文件中检索最新信息,提供准确且相关的回复。

8f36d394-217d-11f0-9310-92fbcf53809c.jpg

图 1:通过 LLM 实现的聊天机器人(左)

和经过 RAG 增强的聊天机器人(右)

为何选择 Axion 来实现 RAG 解决方案

基于 Arm Neoverse 平台的 Google Axion 处理器为运行 LLM 的 AI 推理功能提供了理想平台,该处理器能够以高性能和高效率支持 RAG 应用的运行。

优化 AI 加速:基于 Neoverse 平台的 CPU 具有高吞吐量向量处理和矩阵乘法功能,这对于高效处理 RAG 至关重要。

云计算的效率和可扩展性:基于 Neoverse 平台的 CPU 可最大限度地提高每瓦性能,在高速处理和能效之间取得平衡。因此,特别适用于需要在云端快速推理并兼顾成本效益的 RAG 应用。基于 Neoverse 的处理器还可用于扩展 AI 工作负载,确保无缝集成各种 RAG 用例。

面向 AI 开发者的软件生态系统:对于希望在基于 Arm 架构的基础设施上利用最新 AI 功能的开发者,Arm Kleidi 技术能够显著提升 RAG 应用的性能和效率。Arm Kleidi 已经集成到 PyTorch、TensorFlow 和 llama.cpp 等开源 AI 和机器学习 (ML) 框架中,使开发者能够实现开箱即用的默认推理性能,而无需使用供应商插件或进行复杂的优化。

这些特性的结合带来了显著(zhe)的(de)性(xìng)能(néng)提(tí)升(shēng),首(shǒu)个(gè)基(jī)于(yú) Google Axion 的(de)云(yún)虚拟机 C4A 与 x86 同类方案相比,大幅提升了基于 CPU 的 AI 推理和通用云工作负载的性能,使 C4A 虚拟机成为在 Google Cloud 上运行 RAG 应用的理想选择。

Google Axion 性能基准测试

使用 RAG 系统进行推理涉及两个关键阶段:信息检索和生成响应。

信息检索:系统搜索向量数据库,根据用户的查询找到相关内容。

生成响应:检索到的内容与用户查询相结合,生成与上下文相关的准确回复。

一般来说,检索速度取决于数据库的大小和搜索算法的效率。在基于 Neoverse 平台的 CPU 上运行时,经优化的算法可在几毫秒内返回结果。然后,将检索到的信息与用户的输入相结合,构建新的提示词,并将其发送给 LLM 进行推理和生成响应。相较于检索阶段,生成响应阶段耗时更长,RAG 系统的整体推理延迟在很大程度上受 LLM 推理速度的影响。

本次测试使用 llama.cpp 基准和 Llama 3.1 8B 模型(Q4_0 量化方案)评估了多个 Google Cloud 虚拟机的 RAG 推理性能。使用 48 个线程进行了所有测试,输入(rù)词元(yuán) (token) 大小为 2058,输出词元大小为 256。以下是测试配置:

Google Axion (C4A, Neoverse V2): 在 c4a-standard-48 实例上进行了评估。

Intel Xeon (C4, Emerald Rapids): 在 c4-standard-48 上进行了性能测试。

AMD EPYC (C3D, Genoa): 在启用 48 个核心的 c3d-standard-60 上进行了测试。

Axion 处理器实现更快处理与更高效率

推理性能根据提示词处理速度和词元生成速度来测定。图表 1 的基准测试结果表明,与当前一代 x86 实例相比,基于 Google Axion 的 C4A 虚拟机在提示词处理和词元生成方面实现了高达 2.5 倍的性能提升。

8f3dca50-217d-11f0-9310-92fbcf53809c.jpg

图表 1:运行 Llama 3.1 8B/Q4 模型时,提示词处理(左)

和词元生成(右)与当前一代 x86 实例的性能比较

成本效益:降低 RAG 推理成本

为了评估推理任务的实例成本,还测量了从提交提示词到生成响应的延迟。有几个因素会影响延迟,包括检索速度、提示处理效率、词元生成速率、输入和输出词元大小以及用户批处理规模。由于信息检索延迟通常在毫秒级,与其他因素相比可以忽略不计,因此未纳入计算。批次大小选择为 1,以确保在单个用户级别进行公平的比较。为了与基准测试保持一致,测试中将输入和输出词元大小分别设置为 2048 和 256。首先通过提示词编码速度和词元生成速度计算提示词处理和词元生成的延迟,然后根据 Google Cloud 上的实例定价图表[3]计算每次请求的成本,再将这些数字归一化为所有三个实例的最大成本。

图表 2 中的结果表明,基于 Axion 的虚拟机可节省高达 64% 的成本,处理每次请求所需的成本仅为当前一代 x86 实例的三分之一左右。

8f4d955c-217d-11f0-9310-92fbcf53809c.jpg

图(tú)表(biǎo) 2:使(shǐ)用(yòng) RAG 处(chù)理(lǐ)推(tuī)理请求的归一化成本对比注

注:成本计算基于截至 2025 年 3 月 5 日公布的实例定价,可参见

https://cloud.google.com/compute/vm-instance-pricing

快速入门:基于 Arm 平台构建 RAG 应用

以 Neoverse 平台为核心,Google Axion 赋能的实例能以更低的成本提供高性能,助力企业构建可扩展且高效的 RAG 应用,同时与 x86 方案相比显著降低了基础设施开支。

为了帮助开发者快速入门,Arm 开发了分步演示和 Learning Path 教程,以便开发者使用自己选择的 LLM 和数据源构建基本的 RAG 系统。

以下资源能够帮助刚接触 Arm 生态系统的开发者顺利踏上开发旅程:

通过 Arm Learning Path 迁移到 Axion:依照详细的指南和最佳实践,简化向 Axion 实例的迁移进程。

Arm Software Ecosystem Dashboard:及时了解 Arm 平台上支持的最新软件信息。

Arm 开(kāi)发(fā)者(zhě)中(zhōng)心(xīn):无(wú)论(lùn)你(nǐ)是(shì)刚(gāng)接(jiē)触(chù) Arm 平(píng)台(tái),还(hái)是(shì)正(zhèng)在(zài)寻(xún)找(zhǎo)资(zī)源(yuán)来(lái)开(kāi)发(fā)高(gāo)性(xìng)能(néng)软(ruǎn)件(jiàn)解(jiě)决(jué)方(fāng)案(àn),Arm 开发者中心应有尽有,可以帮助开发者构建更好的软件,为数十亿设备提供丰富的体验。在 Arm 不断壮大的全球开发者社区中,开发者可以访问资源、交流学习和提问探讨。

还等什么?即刻开启你的迁移之旅,利用 Arm Neoverse 平台释放云和 AI 工作负载的全部潜力!

上一篇 查看列表 下一篇