yuanyimeng | 21 Jun 2022 | ( words)

有赞算法平台之模型部署演进

一、 前言

模型部署作为算法工程落地的最后一公里，其天然对算法团队而言具有较高的复杂性，不仅要考虑如何高效地部署、管理不同框架模型，还需要考虑分布式服务的负载均衡、故障容错、可扩展性、资源隔离、限流、核心指标监控等问题。 这些都极大的依赖于工程团队的能力，不是算法团队的强项，如何解决这最后一公里，让焦点聚焦在模型开发上，是模型部署服务模块需要解决的问题。 alt

二、 原有架构

2.1 架构设计

在有赞算法平台Sunfish包含算法训练和模型部署两部分， 模型部署的模块称为ABox（小盒子）。 alt

ABox 主要提供将模型部署为在线服务的功能， 主要包含以下功能：
1. 提供 tensorflow 模型的服务加载和版本管理、弹性部署
2. 提供 tensorflow 模型和其他模型服务（自己部署在额外服务器上）的路由管理
3. 提供模型输入和输出的自定义处理逻辑执行
4. 提供服务主机的负载均衡管理
5. 收集的 Metric 写入上报到 kafka，通过 druid 做监控，并提供指标界面查询

其整体的架构设计如下图所示分为 3 个模块: master、 worker 和 manager， 各自主要职责为： alt master：

  1. 业务请求的路由
    根据 zookeeper 上的动态路由选择将请求直接路由给可以访问的服务（这里包括TF-Serving 服务和第三方注册REST服务，多个服务之间采用轮询的方式）, 请求不会经过worker。
  2. 自定义 jar 包（udl: user defined lib）执行
    在模型预测前和预测后可以加载自定义处理逻辑，可以对模型的输入数据和输出数据进行预处理

worker:

  1. 注册本机信息，负责上报心跳给 manager, 心跳包含本机上的算法服务的健康状态
  2. 负责算法模型的本地拉取， 由 tf-serving 服务加载模型

manager:

  1. 负责服务器（master 和 worker）的注册和下线
  2. 负责算法和模型的创建
  3. 提供 udl 的更新接口
  4. 提供集群、业务组管理的接口
  5. 提供模型的部署和上下线功能
  6. 提供第三方算法的注册和上下线功能

2.2 痛点问题

基于以上架构，ABox 能较好的支持 tensorflow cpu 模型的部署，但存在以下问题：

痛点1 运维投入大

  • 扩缩容需要人工调用接口
  • 模型服务最多实例个数取决于 worker 节点个数， 横向扩容需要增加机器
  • 除 tensorflow 模型部署， 其他需要人工调用接口注册 URL 到 master 来提供路由能力
  • tfserving 采用容器化部署，模型加载过多易 OOM，无法自动拉起

痛点2 负载不均衡

  • 模型按照一定的资源调度策略分布在各个 worker 节点上，各 worker 节点资源使用容易不均衡

痛点3 资源未隔离

  • 热点模型占用大量的CPU/内存，容易影响其他模型服务

痛点4 缺乏通用性

  • 模型服务无法统一管理， tensorflow 模型和其他框架模型管理割裂
  • 没有GPU模型服务部署功能

三、 改造升级

3.1 seldon 介绍

基于前述问题，为了统一不同框架模型部署服务的管理， 为了增加 cpu 和 gpu 模型的统一部署功能， 为了简化运维操作，我们引入了 seldon 这个开源的基于云原生的模型部署服务。 seldon 是一个基于 K8S 的集成的模型部署方案， 内置了很多通用的例如 tfserving、 sklearn server、mlflow server、triton这样的模型推理服务器(inference server)。 除了这些还可以通过自己实现自定义的 inference server 提供一些额外的模型服务支持。通过将你的模型部署为K8S上的微服务，来提供 HTTP/GRPC 接口给业务调用。
我们来看下 seldon 的核心模块， 如下图所示。seldon 的核心模块是它的 Seldon Core Controller, 即 operator 模块， 该模块用来管理 Seldon Deployment 资源( Custom Resource )。 seldon 通过 CRD 向 K8S 注册自定义资源，Seldon Core Controller 负责处理该资源的 CRUD， 我们通过创建/删除/修改 Seldon Deployment 来达到创建/删除/修改模型服务的目的。 除此之外， seldon 还可以集成 Prometheus 和 Jaeger 来提供模型服务的指标监控和调用追踪能力。 alt seldon 的另一核心概念是 Model Server , 即加载模型用来提供HTTP接口的模型推理服务器。Model Server 有两种类型， 分为 Reusable Model Servers 和 Non-Reusable Model Servers 。 他们的差别从名字就可以看出， 前者是不同模型之间可复用的，后者是固定模型的。
Reusable Model Servers 通过配置的模型地址，从外部的模型仓库下载模型， seldon 模型预置了较多的开源模型推理服务器， 包含 tfserving , triton 都属于 Reusable Model Servers。
Non-Resuable Model Servers 将模型打入自定义镜像内部，不需要单独从外部加载模型， 适合做一些特殊化处理以及不需要频繁更新的场景。
alt

例如启动一个 tensorflow 的模型服务， mnist.yaml 文件定义如下：

apiVersion: machinelearning.seldon.io/v1alpha2  
kind: SeldonDeployment  
metadata:  
  name: tfserving
spec:  
  name: mnist
  predictors:
  - graph:
      children: []
      implementation: TENSORFLOW_SERVER
      modelUri: gs://seldon-models/tfserving/mnist-model
      name: mnist-model
      parameters:
        - name: signature_name
          type: STRING
          value: predict_images
        - name: model_name
          type: STRING
          value: mnist-model
    name: default
    replicas: 1

这里声明一个名为 minist 的 SeldonDeployment 资源，主要的配置参数就是 implementation 和 modelUri。 implementation 指定此次 Model Server 使用预置的 tfserving 服务器， 并且需要指定模型的 modelUri 地址。 通过 kubectl create -f mnist.yaml 我们就可以创建一个 Seldon Deployment 资源。 通过 kubectl get sdep 可以看到的 Seldon Deployment minist。 该 Seldon Deployment 资源会管理对应的所需要的 Deployment、 Service 和 Virtual Service。

3.2 设计方案

基于公司内部 K8S 环境，在商量了如何部署seldon的后，我们最后决定的架构如图所示： alt

在引入 seldon 管理模型服务部署的同时，进行了以下的改造：

  1. 保留 ABox master 作为 Dubbo 请求接入入口， master 负责将请求根据协议封装成 http 请求 K8S 集群的 ingress controller。
  2. 修改 Ingress Controller 为 Nginx Ingress Controller， 通过为每个模型服务生成 Ingress 规则路由到指定的 K8S service。
  3. 增加 HDFS-Intializer 用于 Reusable Model Server 中的 hdfs:// 协议的 modelUri
  4. 基于腾讯云的 GpuManager 方案实现GPU的虚拟化和共享
  5. 通过在算法平台集成 K8S client 进行 Seldon Deployment 和 Ingress 的 CRUD
  6. 为自定义镜像提供日志收集服务
  7. 为模型服务增加资源使用展示

下面来介绍下主要的部分。

3.1.1 Ingress Controller 替换

seldon 通过 Ingress Controller 来提供集群外服务的统一访问， 然后通过 istio 或者 ambassador 提供服务的路由和分流， 默认的 istio 服务较为强大，但由于公司内部 K8S 不提供 istio 服务的维护，为了便于运维， 我们在部署 seldon operator 的时候关闭了 istio 和 ambassador 的功能。然后通过自己的客户端实现在 seldon deployment 和 ingress 的统一创建/删除/更新管理。

3.1.2 Reusable Model Server模型初始化

seldon 的默认模型下载协议只支持 s3://、 gs://， 我们的模型文件目前存储于 HDFS 管理， 所以实现了自定义的 HDFS-INITILIZER 负责HDFS模型文件的拉取。

3.1.3 GPU方案

我们都知道在k8s上使用GPU资源有 NVIDIA 的 k8s device plugin ，但是这种方案的缺点是不支持GPU的共享和隔离， 也就是一个pod 的 container 必须占用整数个GPU。在我们的实际使用中， 有一些小模型需要GPU加速但是只占用小部分显卡资源。 目前主流的方案有 alt

由于公司使用的腾讯云并且 GpuManager 方案功能较强大，所以我们考察了腾讯云的开源 GpuManager 方案和腾讯云新升级的 vGpu 方案，在权衡后， 选择了 GpuManager。 理由是：

  1. 由于 vGpu 实现内核级别的 cuda 请求拦截，需要依赖腾讯自研的 TencentOS， 不利于服务器内部统一运维，另外 TencentOS 基于 Centos 8, 在公司目前普遍还是 Centos 7 的情况下， 无法安装基础监控服务，也不利于运维。
  2. vGPU 在积极研发推进阶段，但是目前还不支持服务配置独占 N 张 GPU 卡的情况。

虽然GpuManager满足功能需求，但是也存在约5%的性能损耗。

3.1.4 日志管理

对于自定义的镜像的模型服务， 我们定制了 cpu 和 gpu 版本 （主要支持 pytorch）的基础镜像，通过在基础镜像内置 databus(filebeat) 实现日志上传到 kafka, 再通过自研 log-server 服务消费 kafka 写入日志文件，并且对日志文件进行定期清理，备份到 HDFS。 log-server 提供了 HTTP 的访问接口可以获取到最新的日志。

3.1.4 资源监控

通过定时收集每个 pod 的 cpu, memory 使用量， 我们粗略的统计了每个服务的最小/最大/平均资源使用情况， 并且在界面提供实时资源使用的展示。 alt

3.1.5 服务迁移

在完成基本功能开发后， 我们依次对 QA、预发、生产环境进行了已有的模型服务的迁移， 这个过程比较长。 为了不影响生产的模型服务， 我们定制了流量切换开关， 逐步按比例切流线上服务到新的集群， 保证业务调用无痛无感知。目前已基本完成线上服务的切流。 alt

3.3 注意点

tfserving 依赖 avx 指令集

在某些 KVM 虚机上无法启动 tfserving 服务， Container 提示状态为 CrashLoopBackOff。 进入容器直接运行 /usr/bin/tensorflow_model_server --port=9000 --model_name=xxx --model_base_path=/path/to/model 可以看到错误输出

Illegal instruction (core dumped)

tfserving 默认需要 avx 、avx2 的支持, 可以通过 lscpu 检查机器支持的对应指令集。

四、 总结与展望

第一阶段模型服务改造已经告一段落，我们具备了通用化的模型部署能力， 并且依托于 K8S 简化了运维。但是目前模型服务部署仍有更多的优化点， 比如：

  1. 支持推理图
    我们目前的模型服务为单模型的服务， seldon 支持更复杂的图推理的结果， 可以在模型服务前后配置 INPUT TRANSFORMER 和 OUTPUT TRANSFORMER 进行输入输出的处理
  2. 支持多种发布策略、分流策略
    支持灰度发布可以更为稳妥的支持线上模型在线推理服务的更新， 分流策略的支持可以用来衡量多版本模型之间的性能差异和效果比对
  3. 支持自定义 Model Server 镜像的自动打包更新
  4. 支持更多模型 intializer
    如对象存储的支持
欢迎关注我们的公众号