大模型时代,计算创新如何为应用性能提升开启新路径
引言:
数据库一直是 IT 基础设施的核心组件之一,在云计算和大数据时代,数据库的重要性更加凸显。随着生成式 AI 应用开始广泛流行,企业更加需要海量数据来为大模型提供充足的数据养分。
对于大模型而言,模型所需的数据都经过了向量化(Vector embedding)过程,经过向量化的数据可以大幅提升模型的知识查找效率,使模型可以支持更长的上下文,同时降低训练和推理成本。因此,传统数据库不适合作为大模型的知识库,这就需要新一代向量数据库来满足大模型的数据需求。如今,行业内已经涌现了一批能力出色、为大模型应用做了充分优化的向量数据库产品,腾讯云向量数据库就是其中的佼佼者之一。作为一款全托管的自研企业级分布式数据库服务,其能为多维向量数据提供高效的存储、检索和分析能力,具备完善的嵌入功能,兼具高性能、高可用性,稳定性、可靠性,且使用简单,成本低廉。借助上述优势与特性,腾讯云向量数据库正成为用户构建能力时的强力后援团。
对于向量数据库而言,只有具备极高的性能水平,才能为规模迅速扩大的模型和上层应用提供充足的知识查询吞吐量和极低的查询延迟。然而,仅靠软件端的优化来提升性能是不够的,软硬协同才是增强向量数据库性能的最佳路径。为此,腾讯云与英特尔公司展开了深度合作,将英特尔第五代可扩展至强处理器的诸多优势特性融入腾讯云向量数据库的软件设计中,从而显著提升性能水平。
腾讯云向量数据库能与 LLM 模型配合使用,为 LLM 模型提供知识库
图片及相关介绍援引自腾讯云官网介绍,详情请参阅:https://cloud.tencent.com/product/vdb
对于向量数据库而言,只有具备极高的性能水平,才能为规模迅速扩大的模型和上层应用提供充足的知识查询吞吐量和极低的查询延迟。然而,仅靠软件端的优化来提升性能是不够的,软硬协同才是增强向量数据库性能的最佳路径。为此,腾讯云与英特尔公司展开了深度合作,将英特尔第五代可扩展至强处理器的诸多优势特性融入腾讯云向量数据库的软件设计中,从而显著提升性能水平。
AI 任务中所用的各类数据通常会以向量形式表示。向量(Vector)是一种具有大小和方向的数据表示,其包含多个维度的信息,且每个维度都会用来表示一个特征或属性。正因如此,以向量形式来构建的模型知识库无疑将更具效率,例如在图像处理任务中,图像可表示为像素值的向量;而自然语言处理任务中,文本可表示为词向量或句子向量。
数据的向量化是借助词向量模型、卷积神经网络等模型,经过嵌入(Embedding )环节将文本、图像、音视频等不同数据转换为向量后存入向量数据库中,而向量查询则是通过对向量之间的相似度计算来完成。在实践中,向量数据库可作为知识库,通过与语言、图像等 AI 模型的结合来有效降低用户的模型训练成本,提升 AIGC 等应用的信息输出准确度和及时性。同时向量数据库还可用于模型预训练数据的分类、去重和清洗,相比传统方式实现效率的大幅提升。
向量数据库检索向量的方法主要是相似度度量,也就是通过计算来确定向量数据库中两个向量的相似程度,并最终找出与给定查询向量最相似的向量。常见的计算方法包括内积、欧式距离和余弦相似度,而处理器在运行这些计算方法时,主要涉及大量的密集矩阵向量运算过程,同时还需要提供足够高的访存吞吐量。当处理器有很多核心时,核心之间的互联指标也会影响整体的计算效率。相比之下,传统数据库的查询操作主要是非向量化计算过程,属于非密集运算,其对处理器的需求与向量数据库有着巨大差异。显然,为传统数据库优化的处理器和软件设计对于向量数据库来说效率就很难令人满意。为此,处理器需要全新的硬件设计,大幅加强密集向量运算性能、内存吞吐量、多核互联性能等指标,结合软件层面的深度优化,才能为向量数据库提供充足的能力基础。
2017 年,计算机体系架构宗师 David Patterson 与 John Hennessy 在斯坦福大学发表著名演讲,指出针对特定应用场景的针对性性能加速设计将是下一个十年中芯片领域的主要命题。两位宗师将这种定制加速设计称为 DSA 领域加速计算。正如这次演讲预言的那样,过去几年来行业内的芯片性能提升更多来自各类定制化加速方案,诸如 GPGPU、TPU 等。而在 CPU 层面,扩展指令集和专用加速引擎也已成为 CPU 和上层软件提升性能的快捷路径。进入大模型时代,领域加速计算创新更能大大缓解处理器通用计算能力遭遇瓶颈带来的挑战,使得 CPU 同样可以在 AI 时代焕发全新动能,继续承担算力基础设施的核心角色。
作为 CPU 行业的领军企业,英特尔公司一直非常重视 CPU 扩展指令集和专用加速引擎的研发和创新探索,英特尔第五代可扩展至强处理器内置的英特尔 AVX-512 指令集与英特尔 AMX 高级矩阵扩展加速引擎就是这些探索的最新成果。这两种加速设计可以帮助腾讯云向量数据库在更高的性价比前提下构筑更高密度的算力输出,深度优化其向量检索算法的执行效率,大幅提升检索性能。腾讯云向量数据库支持适合不同场景下的多种算法,而 AVX512 可以为包含 FP32 在内的多种数据格式的向量检索的计算提供 SIMD 的加速支持, AMX 更是支持 BF16 和 int8 数据格式的矩阵运算 从而进一步加速向量检索,实现更高的性能。二者可以适用于不同的场景下数据计算需求, 为客户提供了更多的选择。
另一项可为腾讯云向量数据库带来显著加速的是英特尔 AMX 加速引擎。这一技术引入了一种用于矩阵处理的新框架(包括了两个新的组件,一个二维寄存器文件,其中包含称为 tile 的寄存器,以及一组能在这些 tile 上操作的加速器),从而能高效地处理各类 AI 任务所需的大量矩阵乘法运算,提升其在训练和推理时的效能。例如在向量检索的过程中,如存在 n 个 batch 任务,进行相似度计算时就需要对 n 个输入向量 x 和 n 个数据库中向量 y 进行比对,这其中的距离计算会产生大量的矩阵乘法, 而英特尔 AMX 能针对这一场景实现有效加速。
硬件性能的巨大进步,需要软件层面的充分优化才能在实际应用中发挥作用。为此,腾讯云与英特尔一起,针对腾讯云向量数据库常用的计算库进行了专门优化。包括:
FAISS:方案中针对其不同的索引提出了不同的优化方案,包括面向 IVF-FLAT 算法的单次读取和离散化两种优化思路,以及借助英特尔 AVX-512 加速 IVF-PQFastScan 算法和 IVF-SQ 索引的优化方案;
HNSWlib:方案借助英特尔 AVX-512,对 HNSWlib 的向量检索性能进行加速。同时方案也针对增删数据后的性能和召回率抖动的问题进行了专向优化,使 HNSWlib 的性能和召回率可以保持较平稳状态。
英特尔还为腾讯云向量数据库提供了英特尔 FMAL 加速库。暴力搜索在海量向量数据场景非常常用,但这一场景对算力需求非常高。作为针对向量暴力搜索场景开发的算法库,英特尔 FMAL 在英特尔 AVX-512 和英特尔 AMX 的加持下,能对相似度计算进行加速并提供了相似度计算和 top-K 查询的 API 接口,英特尔 AMX 还能帮助英特尔 FMAL 对 INT8 数据类型提供更好的性能输出。同时,英特尔 FMAL 还能在多线程并发下对处理器资源进行合理调配,以便让用户充分挖掘最新处理器所具备的多核心优势。除此之外,加速库也提供了对内存的非一致内存访问架构优化和缓存数据对齐功能,这些都进一步提升了腾讯云向量数据库的性能。
为验证第五代英特尔至强可扩展处理器基于英特尔 AVX-512 及英特尔 AMX 为腾讯云向量数据库中向量检索任务提供的助力,腾讯云与英特尔携手开展了验证测试,测试分为两个场景:第一个场景中,使用英特尔 AVX-512 优化后,使用 IVF-PQFastscan 算法执行向量检索时的 QPS 性能相比基于第三代至强可扩展处理器的基准组提升了高达 230%。第二个场景中,同样使用第五代至强可扩展处理器的算力平台上,使用英特尔 AMX 加速数据格式为 INT8 的测试场景对比使用英特尔 AVX-512 加速数据格式为 FP32 的测试场景,性能提升高达 5.8 倍之多,显示英特尔 AMX 可以进一步大幅提升 INT8 数据格式下的腾讯云向量数据库向量检索效率。
以上实测对比显示,领域加速技术已经成为新一代处理器大幅增强 AI 等创新应用场景性能的秘籍。值得一提的是,英特尔的加速技术不仅可以显著提升数据库吞吐量,还可以有效降低查询延迟,缩短每个查询需要的时间。除了复杂度较高的向量计算外,腾讯云向量数据库同时还提供了 AI 套件实现一站式文档检索解决方案, 包含自动化文档解析、信息补充、向量化、内容检索等能力,并拥有丰富的可配置项。向量数据库的应用场景还会涉及到不同的集合大小、维度数、数据类型、查询召回精度、延迟的要求, 这对 CPU 通用算力、加速算力、访存性能各项综合能力都提出了更高更快的要求。 而英特尔第五代至强处理器的单核通用算力、内存速度和带宽、L3 缓存等指标相比此前的至强都有了广泛的提升,以满足向量数据库在不同场景下的算力要求,结合极强的领域加速能力,助力向量数据库打通性能瓶颈,为大模型的高效训练和推理工作奠定坚实基础。
在软件优化层面,英特尔为客户提供了模块化的代码优化方案,非常方便客户根据自身的实际场景快速集成,已经有一些客户将这些优化代码成功集成到自身产品中,给出了良好的反馈。
向量数据库并不是领域加速计算大展身手的唯一场景。事实上,英特尔第五代至强可扩展处理器就凭借英特尔 AVX-512 和英特尔 AMX,在诸多科学计算、AI 推理、AI 训练等场景中取得了非常优秀的表现。例如,某互联网厂商的实际测试中,第五代至强可扩展处理器的 AI 推理和训练性能相比上代就有 40% 提升,HPC 应用则有 39% 提升。如果与未引入上述加速技术的第三代至强可扩展处理器相比,第五代至强可扩展处理器的 AI 推理性能最高提升可达 14 倍之多。
腾讯云与英特尔公司长期以来一直在软硬件优化方面有着深度合作,腾讯云也在与英特尔携手将更多先进计算产品和技术应用到 AI 等领域中,在竞争激烈的市场中取得更为领先的优势。与腾讯云类似,还有更多企业都在与英特尔公司展开类似合作,不仅利用英特尔 AVX-512 和英特尔 AMX 加速技术提升 AI 场景效能,还会应用英特尔 TDX 机密计算技术来改善 AI 大模型云端部署时的企业敏感数据安全性,或者利用第五代至强可扩展处理器在内存扩展性方面的优势增强智能推荐等场景的性能,等等。英特尔公司在软件生态层面的深耕成果,以及英特尔公司与各家客户的深度合作,也帮助这些企业以更低的资源投入就能快速享受到硬件新特性的收益。
正是包括英特尔 AVX-512 和英特尔 AMX 在内的一系列计算创新技术,让 CPU 在生成式 AI 大规模普及的时代依旧能够一马当先,为企业带来显著的生产力提升和竞争优势。可以认为,引入领域加速技术等创新的 CPU 在生成式 AI 浪潮中仍将肩负基础算力底座的重任,发挥不可或缺的作用。在 CPU 的创新探索支持下,我们也将看到生成式 AI 应用更快走入社会生产生活,在各行各业中焕发新的光彩。
第五代英特尔® 至强® 可扩展处理器助腾讯云向量数据库成为大模型时代数据中枢,点击【阅读原文】,即刻查看详情!
微信扫码关注该文公众号作者