模型托管常见问题解答 - Amazon SageMaker
Amazon Web Services 文档中描述的 Amazon Web Services 服务或功能可能因区域而异。要查看适用于中国区域的差异,请参阅 中国的 Amazon Web Services 服务入门 (PDF)

模型托管常见问题解答

有关 SageMaker Inference 托管的常见问题,请参阅以下常见问题解答项。

通用托管

以下常见问题解答项回答了有关 SageMaker Inference 的常见一般问题。

答:在您构建和训练模型之后,Amazon SageMaker 提供了四个部署模型的选项,以便您开始进行预测。实时推理适用于要求毫秒级延迟、负载大小最大 6 MB、处理时间最长 60 秒的工作负载。批量转换非常适合对预先提供的大批量数据进行离线预测。异步推理设计用于没有亚秒延迟要求、负载大小不超过 1 GB、处理时间最多 15 分钟的工作负载。借助无服务器推理,您可以快速部署用于推理的机器学习模型,无需配置或管理底层基础设施,而且您只需为处理推理请求所用的计算容量付费,这非常适合间歇性工作负载。

答:下图可以帮助您选择 SageMaker 托管模型部署选项。

解释如何在 SageMaker 中选择模型部署选项的流程图,如下段所述。

上图引导您了解以下决策过程。如果要批量处理请求,您可能需要选择批量转换。否则,如果您想接收发送到模型的每个请求的推理,则可能需要选择异步推理、无服务器推理或实时推理。如果您的处理时间较长或负载较大,并且想要对请求进行排队,则可以选择异步推理。如果您的工作负载具有不可预测或间歇性的流量,则可以选择无服务器推理。如果您有持续的流量并且需要较低且一致的请求延迟,则可以选择实时推理。

答:要使用 SageMaker Inference 优化成本,您应该为自己的使用案例选择合适的托管选项。您还可以使用推理功能,例如 Amazon SageMaker 节省计划、使用 SageMaker Neo 的模型优化、多模型端点多容器端点或者自动扩缩。有关如何优化推理成本的提示,请参阅推理成本优化最佳实践

答:如果您需要有关合适端点配置的建议,以提高性能和降低成本,则应使用 Amazon SageMaker Inference Recommender。以前,想要部署模型的数据科学家必须运行手动基准测试,以选择合适的端点配置。首先,他们必须根据模型和示例负载的资源要求,从 70 多种可用实例类型中选择合适的机器学习实例类型,然后优化模型以考虑不同的硬件。接下来,他们必须进行广泛的负载测试,以验证是否满足了延迟和吞吐量要求并且成本很低。Inference Recommender 可帮助您实现:

  • 提供实例推荐,在几分钟内即可开始使用。

  • 对各种实例类型进行负载测试,以便在数小时内获得有关端点配置的建议。

  • 自动调整容器和模型服务器参数,并针对给定实例类型执行模型优化。

答:SageMaker 端点是使用容器化 Web 服务器(包括模型服务器)的 HTTP REST 端点。这些容器负责加载和处理机器学习模型的请求。它们实施在端口 8080 上响应 /invocations/ping 的 Web 服务器。

常见的模型服务器包括 TensorFlow Serving、TorchServe 和多模型服务器。SageMaker 框架容器内置了这些模型服务器。

答:SageMaker Inference 中的所有内容都是容器化的。SageMaker 为 TensorFlow、SKlearn 和 HuggingFace 等热门框架提供托管容器。有关这些映像的最新完整列表,请参阅可用映像

有时,您可能需要为一些自定义框架构建容器。这种方法被称为自带容器,简称 BYOC。使用 BYOC 方法,您可以提供 Docker 映像来设置框架或库。然后,您将映像推送到 Amazon Elastic Container Registry (Amazon ECR),这样便可在 SageMaker 中使用映像。有关 BYOC 方法的示例,请参阅 Amazon SageMaker 容器概述

或者,您可以扩展容器,而不是从头开始构建映像。您可以获取 SageMaker 提供的基础映像之一,然后在 Dockerfile 中在其上添加依赖项。

答:SageMaker 提供了功能,让您可以将已在 SageMaker 之外训练的框架模型引入,并将其部署到任何 SageMaker 托管选项上。

SageMaker 要求您将模型打包到 model.tar.gz 文件中并采用特定的目录结构。每个框架都有自己的模型结构(有关示例结构,请参阅以下问题)。有关更多信息,请参阅 SageMaker Python SDK 文档 中的 TensorFlowPyTorchMXNet

虽然您可以从 TensorFlow、PyTorch 和 MXNet 等预先构建的框架映像中进行选择来托管您的经过训练的模型,但您也可以构建自己的容器,在 SageMaker 端点上托管经过训练的模型。有关演练,请参阅 Jupyter 笔记本构建自己的算法容器

答:SageMaker 要求将模型构件压缩到 .tar.gz 文件或压缩到 tarball 中。SageMaker 会自动将此 .tar.gz 文件提取到您的容器中的目录 /opt/ml/model/ 中。tarball 不应包含任何符号链接或不必要的文件。如果您使用了框架容器之一,例如 TensorFlow、PyTorch 或 MXNet,则该容器需要的 TAR 结构如下所示:

TensorFlow

model.tar.gz/ |--[model_version_number]/ |--variables |--saved_model.pb code/ |--inference.py |--requirements.txt

PyTorch

model.tar.gz/ |- model.pth |- code/ |- inference.py |- requirements.txt # only for versions 1.3.1 and higher

MXNet

model.tar.gz/ |- model-symbol.json |- model-shapes.json |- model-0000.params |- code/ |- inference.py |- requirements.txt # only for versions 1.6.0 and higher

答:ContentType 是请求正文中输入数据的 MIME 类型(您发送到端点的数据的 MIME 类型)。模型服务器使用 ContentType 来确定是否可以处理所提供的类型。

Accept 是推理响应的 MIME 类型(端点返回的数据的 MIME 类型)。模型服务器使用 Accept 类型来确定是否可以处理返回的类型。

常见的 MIME 类型包括 text/csvapplication/jsonapplication/jsonlines

答:SageMaker 将任何请求传递到模型容器而不进行任何修改。容器必须包含反序列化请求的逻辑。有关为内置算法定义的格式的信息,请参阅用于推理的常用数据格式。如果您正在构建自己的容器或使用 SageMaker 框架容器,则可以包含接受所选请求格式的逻辑。

同样,SageMaker 也会返回响应而不进行修改,然后客户端必须反序列化响应。对于内置算法,它们会以特定格式返回响应。如果您正在构建自己的容器或使用 SageMaker 框架容器,则可以包含以您选择的格式返回响应的逻辑。

使用 Invoke Endpoint API 调用对您的端点进行推理。

将输入作为负载传递给 InvokeEndpoint API 时,您必须提供模型所需的正确输入数据类型。在 InvokeEndpoint API 调用中传递负载时,请求字节会直接转发到模型容器。例如,对于图像,您可以为 ContentType 使用 application/jpeg,并确保您的模型可以对此类数据进行推理。这适用于 JSON、CSV、视频或任何其他可能处理的输入类型。

另一个需要考虑的因素是负载大小限制。对于实时端点和无服务器端点,负载限制为 6 MB。您可以将视频拆分成多个帧,并为每个帧分别调用端点。或者,如果您的使用案例允许,您可以使用支持最多 1 GB 负载的异步端点,在负载中发送整个视频。

有关展示如何使用异步推理对大型视频运行计算机视觉推理的示例,请参阅此博客文章

实时推理

以下常见问题解答项回答了有关 SageMaker 实时推理的常见问题。

答:您可以通过 Amazon 支持的工具(例如 Amazon SDK、SageMaker Python SDK、Amazon Web Services Management Console、Amazon CloudFormation 和 Amazon Cloud Development Kit (Amazon CDK))来创建 SageMaker 端点。

端点创建中有三个关键实体:SageMaker 模型、SageMaker 端点配置和 SageMaker 端点。SageMaker 模型指向您正在使用的模型数据和映像。端点配置定义您的生产变体,其中可以包括实例类型和实例数量。然后,您可以使用 SageMaker 的 create_endpoint API 调用或 .deploy() 调用,使用模型和端点配置中的元数据创建端点。

答:不需要,您可以使用各种 Amazon SDK(有关可用的 SDK,请参阅调用/创建),甚至可以直接调用相应的 Web API。

答:多模型端点是 SageMaker 提供的实时推理选项。借助多模型端点,您可以在一个端点上托管数千个模型。多模型服务器是一个用于提供机器学习模型的开源框架。它提供了多模型端点所需的 HTTP 前端和模型管理功能,以在单个容器中托管多个模型,在容器中动态加载和卸载模型,以及对指定的加载模型执行推理。

答:SageMaker 实时推理支持各种模型部署架构,例如多模型端点、多容器端点和串行推理管道。

多模型端点 (MME) – 通过 MME,客户能够以经济高效的方式部署上千种超个性化模型。所有模型都部署在共享资源的实例集上。当模型的大小和延迟相似且属于同一个 ML 框架时,MME 效果最好。当您不需要始终调用相同的模型时,这些端点是理想的选择。您可以将相应的模型动态加载到 SageMaker 端点以服务您的请求。

多容器端点 (MCE) – 通过 MCE,客户可以部署 15 个不同的容器,这些容器具有不同的 ML 框架和功能,无需冷启动,而且只使用一个 SageMaker 端点。您可以直接调用这些容器。MCE 最适合需要将所有模型保存在内存中的情况。

串行推理管道 (SIP) – 您可以使用 SIP 将 2-15 个容器串连在一个端点上。SIP 最适合将预处理和模型推理结合在一个端点中,也适用于低延迟操作。

无服务器推理

以下常见问题解答项回答了有关 Amazon SageMaker 无服务器推理的常见问题。

答:无服务器推理是专用的无服务器模型服务选项,可用于轻松部署和扩展 ML 模型。无服务器推理端点会自动启动计算资源,并根据流量横向扩展和缩减,而无需选择实例类型、运行预置容量或管理扩展。您也可以选择为无服务器端点指定内存要求。您只需为推理代码的运行时间和处理的数据量付费,无需为空闲时间付费。

答:无服务器推理无需预先预置容量和管理扩展策略,从而简化了开发人员的体验。根据使用模式,无服务器推理可以在几秒钟内立即从数十个扩展到数千个推理,这使其非常适合间歇性或不可预测流量的 ML 应用程序。例如,薪资处理公司使用的聊天机器人服务在月底的查询量会增加,而在月中其余时间的流量是间歇性的。在这种情况下,为整个月预置实例并不划算,因为这样就需要为空闲时间付费。

无服务器推理为您提供开箱即用的自动快速横向扩展,无需您预先预测流量或管理扩展策略,即可满足这些类型使用案例的需求。此外,您只需为运行推理代码的计算时间和数据处理量付费,因此这非常适合间歇性流量的工作负载。

答:您的无服务器端点的最小 RAM 大小为 1024 MB (1 GB),您可以选择的最大 RAM 大小为 6144 MB (6 GB)。您可以选择的内存大小为 1024 MB、2048 MB、3072 MB、4096 MB、5120 MB 或 6144 MB。无服务器推理会自动分配与您选择的内存成比例的计算资源。如果您选择更大的内存大小,则容器可以访问更多 vCPU。

根据您的模型大小选择端点内存大小。一般来说,内存大小至少应与模型大小相同。您可能需要进行基准测试,以便根据延迟 SLA 为模型选择合适的内存大小。内存大小增量有不同的定价;有关更多信息,请参阅 Amazon SageMaker 定价页面

批量转换

以下常见问题解答项回答了有关 SageMaker 批量推理的常见问题。

答:对于 CSV、RecordIO 和 TFRecord 等特定文件格式,SageMaker 可以将您的数据拆分为单条记录或多条记录的小批次,然后将其作为负载发送到模型容器。当 BatchStrategy 的值为 MultiRecord 时,SageMaker 在每个请求中发送最大记录数,直至达到 MaxPayloadInMB 限制。当 BatchStrategy 的值为 SingleRecord 时,SageMaker 在每个请求中发送单独的记录。

答:批量转换的最大超时时间为 3600 秒。记录(每个小批次)的最大负载大小为 100 MB。

答:如果您使用 CreateTransformJob API,您可以使用最佳的参数值(例如 MaxPayloadInMBMaxConcurrentTransformsBatchStrategy)来减少完成批量转换作业所需的时间。MaxConcurrentTransforms 的理想值等于批量转换作业中的计算工作线程数。如果您使用的是 SageMaker 控制台,则可以在批量转换作业配置页面的其他配置部分中指定这些最佳参数值。SageMaker 会自动为内置算法找到最佳参数设置。对于自定义算法,通过 execution-parameters 端点提供这些值。

答:批量转换支持 CSV 和 JSON。

异步推理

以下常见问题解答项回答了有关 SageMaker 异步推理的常见一般问题。

答:异步推理队列对传入的请求进行排队并异步处理。此选项非常适合具有较大负载的请求,或者处理时间较长且需要在收到时进行处理的请求。或者,您可以配置自动扩缩设置,以便在没有主动处理请求时将实例个数缩减为零。

答:Amazon SageMaker 支持自动扩缩异步端点。自动扩缩动态调整为模型预置的实例数,以响应工作负载的变化。与 SageMaker 支持的其他托管模型不同,使用异步推理,您还可以将异步端点实例个数缩减为零。在实例数为零个时收到的请求将排队等待,直到端点纵向扩展后再处理这些请求。有关更多信息,请参阅自动扩缩异步端点

Amazon SageMaker 无服务器推理还可自动缩减到零。您不会看到这一点,因为 SageMaker 管理扩展您的无服务器端点的扩缩,但是在您没有任何流量时,同样的基础设施也适用。