引言#

在深入探究 AI 编译原理之后,将进一步迈向一个与日常生活紧密相连的新领域。这个领域无处不在,无论是日常使用的购物应用、观看在线视频的平台,还是钟爱的游戏,它们都与这个领域息息相关。该领域,便是推理系统与推理引擎。

那么,推理系统与推理引擎究竟是什么呢?它们之间又存在着怎样的差异?推理的具体工作流程是怎样的?在实际应用中又该如何操作?这些问题都亟待去解答。本节将围绕推理系统与推理引擎这两个核心概念展开,详细解释它们的内涵与区别。随后,将聚焦于推理引擎,探讨如何将其模型小型化,如何进行离线优化与压缩,并最终探讨推理引擎的部署与运行优化策略。

推理系统介绍#

在深入探讨推理系统与推理引擎之前,首先需要明确“推理”这一概念。推理,简单来说,就是在利用大量数据训练好模型的结构和参数后,使用小批量数据进行一次前向传播,从而得到模型输出的过程。在此过程中,并不涉及模型梯度和损失的优化。推理的最终目标,便是将训练好的模型部署到实际的生产环境中,使 AI 真正运行起来,服务于日常生活。

推理系统,是一个专门用于部署人工智能模型,执行推理预测任务的人工智能系统。它类似于传统的 Web 服务或移动端应用系统,但专注于 AI 模型的部署与运行。推理系统会加载模型到内存,并进行版本管理,确保新版本能够顺利上线,旧版本能够安全回滚。此外,它还会对输入数据进行批量尺寸(Batch Size)的动态优化,以提高处理效率。通过提供服务接口(如 HTTP、gRPC 等),推理系统使得客户端能够方便地调用模型进行推理预测。同时,推理系统还可以作为一个微服务,在数据中心中与其他微服务协同工作,共同完成复杂的请求处理任务。

推理引擎,则是推理系统中的重要组成部分,它主要负责 AI 模型的加载与执行。推理引擎可分为调度与执行两层,聚焦于 Runtime 执行部分和 Kernel 算子内核层,为不同的硬件提供更加高效、快捷的执行引擎。它可以看作是一个基础软件,提供了一组 API,使得开发者能够在特定的加速器平台(如 CPU、GPU 和 TPU)上轻松地进行推理任务。目前市场上已有多种推理引擎,如字节跳动的 LightSeq、Meta AI 的 AITemplate、英伟达的 TensorRT,以及华为的 MindSpore Lite 和腾讯的 NCNN 等。

在本节中,将深入探讨推理系统与推理引擎的概念及其区别。在了解推理系统的工作流程,同时,也将深入剖析推理引擎的整体架构,理解其在推理过程中的核心作用与运行机制。通过这一节的学习,将对推理系统与推理引擎有更加全面、深入的认识,为后续的实际应用打下坚实基础。

模型小型化#

在端侧推理引擎中,模型小型化、轻量化是至关重要的环节。由于端侧设备资源有限,执行轻量的模型结构能够确保高效且稳定的推理性能。模型小型化的核心思想在于设计出更为高效的网络计算方式,从而在减少模型参数量的同时,保持网络精度,并进一步提升模型的执行效率。

在该节中,将重点关注模型小型化过程中的关键参数和指标。这些参数和指标不仅有助于评估模型的小型化程度,还能指导如何更有效地进行模型优化。将深入探讨模型大小(通常以参数量来衡量)、计算复杂度(如 FLOPs,即浮点运算次数)等指标,并分析它们之间的权衡关系。

接下来,将介绍一些在模型小型化领域取得显著成果的主干网络(Backbone)或 SOTA(state of the art)网络模型。这些模型通过采用创新的网络结构和优化策略,实现了在保证精度的同时,大幅减少模型参数量和计算复杂度。将详细分析这些模型的设计思路、网络结构,为读者提供宝贵的参考和启示。

此外,还将重点关注 CNN(卷积神经网络)结构下的小型化工作。CNN 是计算机视觉领域中最常用的网络结构之一,其小型化研究具有广泛的应用价值。将介绍一些针对 CNN 的小型化技术和方法,包括轻量级卷积核设计、网络剪枝、量化等方法,并分析它们在减少模型大小和提高推理速度方面的实际效果。

最后,还将简要介绍 Transformer 结构中的小型化工作。Transformer 在自然语言处理领域取得了巨大成功,其小型化研究同样具有重要意义。将探讨一些针对 Transformer 的小型化策略,如采用更高效的自注意力机制、压缩嵌入层等。

通过本节的学习,读者将深入了解模型小型化的重要性、关键指标、以及常见的小型化技术和方法,这将有助于读者在实际应用中更好地进行模型优化和推理性能提升。

离线优化压缩#

推理系统作为类似于传统 Web 服务的存在,需要高效响应用户请求并维持高标准的服务等级协议,如响应时间低于 100ms 等。为了实现这一目标,离线优化压缩在端侧推理引擎中发挥着至关重要的作用。与轻量化网络模型设计不同,离线优化压缩主要通过对轻量化或非轻量化模型应用剪枝、蒸馏、量化等压缩算法和手段,使模型体积更小、更轻便,从而提高执行效率。

在本节中,将围绕离线优化压缩展开详细介绍。首先,来探讨低比特量化。低比特量化是一种将模型权重和激活值从浮点数转换为低比特整数(如 8 位、4 位甚至更低)的技术。通过减少表示每个数值所需的比特数,可以显著减少模型的大小和内存占用,同时加速推理过程。然而,低比特量化也可能导致精度损失,因此需要在压缩率和精度之间找到平衡。

接下来,介绍二值化网络。二值化网络是一种极端的量化方法,它将模型权重和激活值限制为两个可能的值(通常是+1 和-1)。这种方法可以进一步减小模型大小并提高推理速度,但可能导致更大的精度损失。因此,在设计二值化网络时,需要精心选择网络结构和训练策略,以在保持精度的同时实现高效的压缩。

除了量化和二值化,模型剪枝也是一种常用的压缩方法。模型剪枝通过移除网络中的冗余连接或神经元来减小模型大小。这可以通过设定阈值来删除权重较小的连接或神经元实现。剪枝后的模型不仅更小,而且往往具有更快的推理速度。然而,剪枝过程需要谨慎处理,以避免过度剪枝导致精度大幅下降。

最后介绍知识蒸馏。知识蒸馏是一种将大型教师模型的知识转移到小型学生模型中的技术。通过让教师模型指导学生模型的学习过程,可以在保持较高精度的同时实现模型的小型化。这种方法的关键在于设计有效的蒸馏策略,以确保学生模型能够充分吸收教师模型的知识。

在实际应用中,这些优化压缩方法通常需要根据具体任务和模型特点进行选择和调整。通过综合运用这些方法,可以在满足服务需要的同时,实现模型的高效推理和部署。

在线部署和优化#

推理引擎的在线部署和优化是确保 AI 模型能够在实际应用中高效运行的关键环节。在模型部署的过程中,推理引擎需要应对多种挑战,包括适配多样的 AI 框架、处理不同部署硬件的兼容性问题,以及实现持续集成和持续部署的模型上线发布等软件工程问题。为了应对这些挑战,推理引擎的在线部署和优化显得尤为重要。

首先,推理引擎需要支持不同 AI 框架训练得到的模型的转换。由于市场上存在多种 AI 框架,如 TensorFlow、PyTorch 等,每种框架都有其独特的模型格式和存储方式。因此,推理引擎需要具备模型格式的解析和转换能力,确保不同框架下的模型能够统一地部署到推理引擎中。

其次,推理引擎需要对转换后的模型进行计算图的优化。计算图优化是提升模型推理效率的关键步骤。通过算子融合、算子替换、布局调整、内存分配等方式,可以减少计算冗余、优化内存访问、提高计算并行度,从而显著提升模型的推理速度。

最后,本节对推理引擎的 Kernel 优化方面做了细致的介绍。卷积 kernel 算子的优化是一个重要的方向,卷积操作是神经网络模型中计算密集且耗时的部分,因此对其进行优化能够显著提升推理性能。其中,对于卷积 kernel 算子的优化主要关注 Im2Col、Winograd 等算法的应用。这些算法通过特定的数学变换和近似,减少了卷积操作的计算复杂度,从而提升了推理速度。

除了算法层面的优化,内存布局也对 kernel 性能产生重要影响。在本节中,将介绍 NC1HWC0 和 NCHW4 等不同的内存布局方式,并阐述它们对 kernel 优化的作用和意义。通过合理选择内存布局,可以减少内存访问的延迟和冲突,提高数据访问的效率,从而进一步提升推理性能。在这之后将会介绍汇编上的优化特别是在指令和汇编层面上的优化,并介绍通过 MNN 的预推理模块介绍调度优化。

此外,汇编层面的优化也是提升推理性能的重要手段。将深入探讨指令和汇编层面的优化技术,通过循环优化(Loop Optimization)、指令优化(Instructions Optimization)、存储优化(Memory Optimization)的方式,减少指令执行的时间开销,提高处理器的利用率。这将有助于进一步挖掘硬件的性能潜力,提升推理速度。

随后,本节还将介绍通过 MNN 的预推理模块实现调度优化的方法。调度优化是一种在推理引擎执行过程中进行任务调度和资源分配的技术,通过合理的调度策略,可以充分利用硬件资源,提高推理效率。

推理应用#

在本小节中,将通过具体实例来展示推理系统的实际应用,包括人脸 landmark 的应用以及利用华为 HMS Core 实现的人脸和手势检测等端侧应用。同时,也将探讨维护推理系统所面临的问题和挑战。

人脸 Landmark#

如图所示,这款应用在移动终端上实现了精准的人脸 landmark 识别功能。它通过先进的算法技术,能够迅速捕捉并准确识别拍摄者脸部的轮廓、五官位置等关键面部信息。这些信息被实时处理并以一种直观且易于理解的方式显示出来,使用户能够清晰地看到自己脸部的各个特征点。

人脸 landmark 识别技术是人脸识别领域的重要组成部分,它在多个方面发挥着关键作用。首先,在人脸对齐方面,通过识别面部特征点,可以实现对人脸图像的精确对齐,从而提高后续人脸处理和分析的准确性。其次,在人脸重建方面,landmark 信息为三维人脸模型的构建提供了重要依据,使得能够以更真实的方式还原人脸的形态和细节。此外,人脸 landmark 还在身份鉴别、人脸编辑以及人脸 AR 等领域发挥着重要作用,为这些应用提供了精确、可靠的人脸特征数据。

图示的这款移动终端上的人脸 landmark 识别应用具有高度的准确性和实时性,它使得用户可以方便地获取自己的面部信息,并在多种场景下进行扩展应用,如美妆试妆、虚拟形象创建、人脸特效等。随着技术的不断进步和应用场景的不断拓展,人脸 landmark 识别技术将在未来发挥更加广泛和重要的作用。

人脸检测与手势识别#

面这两张图也是应用于移动终端上的推理系统,左图是使用华为 HMS Core 实现人脸检测,具体来说是使用人脸检测来获取人脸的位置,然后利用这个坐标来控制游戏中的飞船进行移动。而右图是华为 HMS Core 实现手势检测,与左图类似,右图是将左图的面部坐标换成了手的坐标进行飞船的移动,并配合手势去做相应的动作。 s

上面这两张图也是展示在移动终端上应用推理系统的实际案例,通过华为 HMS Core 实现的人脸检测和手势检测功能,为用户带来了新颖而富有互动性的体验。

左图展示了使用华为 HMS Core 进行人脸检测的场景。在这一应用中,推理系统通过调用 HMS Core 的人脸检测 API,实时地捕捉和识别用户的人脸位置。一旦获取到人脸的坐标信息,系统便能够利用这些坐标来控制游戏中的飞船进行移动。这种将人脸检测与游戏控制相结合的方式,不仅提升了游戏的趣味性和互动性,还为用户提供了一种全新的操作方式。

右图则展示了华为 HMS Core 实现手势检测的应用场景。与左图类似,这一应用也是通过推理系统来实时捕捉和识别用户的手势。不同的是,这次是将手势的坐标信息用于控制游戏中的飞船移动。用户可以通过不同的手势来执行不同的动作,如前进、发射导弹等,从而实现更加自然和直观的游戏控制。这种手势控制的方式不仅使得游戏操作更加便捷,还为用户带来了更加丰富和多样的交互体验。

这两张图所展示的应用案例充分展示了推理系统在移动终端上的强大功能和广泛应用前景。通过利用华为 HMS Core 提供的 AI 能力,可以轻松地实现各种复杂的人脸和手势检测功能,并将其应用于各种实际场景中,为用户带来更加智能、便捷和富有创新性的体验。

人工客服应用#

推理引擎或推理系统在人工客服和人工智能对话方面有广泛的应用。以下是一些相关的内容:

智能客服:推理引擎可以用于实现智能客服系统,能够理解用户的问题并提供准确的答案。通过对大量的语料库和知识库进行训练,推理引擎可以学习到不同的问题模式和解决方案,从而能够快速准确地回答用户的问题。

对话管理:在人工智能对话中,推理引擎可以帮助系统理解用户的意图和需求,并根据这些信息来引导对话的流向。它可以根据用户的输入和历史对话记录,预测用户可能的问题和需求,并提供相应的回答和建议。

情感分析:推理引擎可以对用户的语言进行情感分析,判断用户的情绪状态。这对于人工客服来说非常重要,因为它可以帮助客服人员更好地理解用户的需求和问题,并提供更合适的解决方案。

知识图谱:结合知识图谱,推理引擎可以利用实体和关系的信息来进行更深入的推理和回答。它可以根据用户的问题,从知识图谱中检索相关的信息,并以更自然和准确的方式呈现给用户。

多轮对话:推理引擎可以支持多轮对话,使系统能够与用户进行连续的交互。它可以根据用户的回答和反馈,动态地调整对话策略和回答内容,以提供更个性化和有效的服务。

实时响应:推理引擎需要具备快速的推理能力,以实现实时响应。它可以在短时间内处理用户的输入,并给出及时的回答,提高用户体验和满意度。

优化和改进:通过对推理引擎的性能进行评估和分析,可以不断优化和改进系统的回答准确性和效率。通过收集用户的反馈和评价,系统可以不断学习和改进,以提供更好的服务。

以下是一个具体的应用场景示例:

  • 用户:我的订单显示已发货,但我还没有收到货物。

  • 智能客服系统:好的,我可以帮您查询订单状态。请告诉我您的订单号。

  • 用户:[订单号]

  • 智能客服系统:根据您提供的订单号,我查询到您的订单已于[发货日期]发货,预计在[预计送达日期]到达。请您耐心等待。

  • 用户:好的,谢谢。

如图所示,在这个示例中,智能客服系统通过推理引擎理解用户的问题,并根据订单号查询相关的订单信息,然后给出准确的回答。推理引擎的应用使得智能客服能够快速、准确地回答用户的问题,提供高效的服务。

推理系统思考点#

在实际维护推理系统的过程中,需要全面考虑并解决以下问题:

首先,如何设计并生成用户友好、易于调用的 API 接口,以便用户能够便捷地与推理系统进行交互。其次,关于数据的生成,需要明确数据的来源、生成方式以及质量保障措施,确保推理系统能够依赖准确、可靠的数据进行运算。

再者,在网络环境的影响下,如何实现低延迟的用户反馈是一个关键挑战。需要优化网络传输机制,减少数据传输的延迟,确保用户能够及时获得推理结果。同时,充分利用手机上的各种加速器或 SoC 加速资源对于提升推理系统的性能至关重要。需要深入研究手机硬件的特性,合理利用加速资源,提高推理的运算速度和效率。

另外,当用户访问量增大时,如何确保服务的稳定性和流畅性是一个必须面对的问题。需要设计合理的负载均衡策略,优化系统架构,提高系统的并发处理能力。此外,为了应对潜在的风险和故障,需要制定冗灾措施和扩容方案,确保在突发情况下推理系统能够稳定运行。

最后,随着技术的不断发展,未来可能会有新的网络模型上线。需要考虑如何平滑地集成这些新模型,并制定 AB 测试策略,以评估新模型的性能和效果。

总之,维护推理系统需要综合考虑多个方面的问题,从 API 接口设计、数据生成、网络延迟优化、硬件加速资源利用、服务稳定性保障、冗灾与扩容措施,到新模型上线与测试等方面,都需要进行深入研究与精心规划。

小结与思考#

  • 推理系统与推理引擎基础:推理系统是专门部署 AI 模型并执行预测任务的系统,类似于 Web 服务但专注于 AI;推理引擎则是推理系统中负责模型加载与执行的核心组件,提供高效快捷的执行环境。

  • 模型优化技术:包括模型小型化、离线优化压缩等技术,通过量化、剪枝、知识蒸馏等手段减少模型大小、提高执行效率,同时保持模型精度,以适应端侧设备有限的资源。

  • 推理系统应用与挑战:推理系统广泛应用于日常生活,如人脸 landmark 识别、人脸和手势检测等,同时面临 API 设计、数据质量、网络延迟、硬件加速、服务稳定性等维护挑战。

本节视频#