[发明专利]一种分布式环境下的人工智能模型管理系统及管理方法

说明书

本发明公开了一种人工智能模型管理系统、方法，其中系统包括：注册信息添加模块，用于添加模型数据以及使用该模型的在线服务的服务信息；模型上传模块，用于上传模型文件并存储；模型数据获取模块，用于获取模型数据；特征值计算模块，用于根据模型数据，计算得到一特征值；存储模块，用于根据特征值，并通过Hash算法将模型文件存储至存储集群中的对应的存储节点中；服务信息查询模块，用于根据特征值，查询使用该模型的服务信息；模型推送指令生成模块，用于在查询到服务信息后形成一模型推送指令并输出；模型推送模块，用于根据模型推送指令，将存储于存储节点中的模型文件推送给对应的在线服务，本发明解决了人工部署模型容易出错的问题。

技术领域

本发明涉及数据处理技术领域，具体涉及一种分布式环境下的人工智能模型管理系统及管理方法。

背景技术

在人工智能图像识别技术领域，现阶段通常采用集中训练一次识别模型，然后将该识别模型通过人为部署方式部署给需要的服务项目，以实现对图像的识别检测。上述的人为部署方式对于单个或为数不多的模型是可行的，但当一个机器学习服务平台所要支撑的项目较多需要多个模型或者模型需要经常更新的情况下，针对模型的人为部署方式就变得繁杂又容易出错，所以需要一种模型智能管理系统，以对模型进行智能管理，以满足不同的服务平台对模型使用的不同需求。

发明内容

本发明的目的在于提供一种分布式环境下的人工智能模型管理系统，以解决上述技术问题。

为达此目的，本发明采用以下技术方案：

提供一种分布式环境下的人工智能模型管理系统，用于对机器学习服务平台所使用的模型进行更新管理，所述人工智能模型管理系统包括：

注册信息添加模块，用于提供给用户添加所述模型的模型数据以及使用所述模型的在线服务的服务信息，并将所述用户添加的所述模型数据和所述服务信息存储于一数据库中；

模型上传模块，用于提供给所述用户将更新后的所述模型对应的模型文件上传至一模型库中存储；

模型数据获取模块，连接所述数据库，用于从所述数据库中获取关联于所述模型的所述模型数据；

特征值计算模块，连接所述模型数据获取模块，用于根据所获取的关联于所述模型的所述模型数据，计算得到一可用于表示所述模型的特征值；

存储模块，分别连接所述特征值计算模块和所述模型库，用于根据所述特征值，并通过Hash算法将暂存于所述模型库中的关联于所述模型的所述模型文件存储至存储集群中的对应的存储节点中；

服务信息查询模块，分别连接所述数据库和所述特征值计算模块，用于根据所计算的关联于所述模型的所述特征值，在所述数据库中查询使用所述模型的所述服务信息；

模型推送指令生成模块，连接所述服务信息查询模块，用于在查询到所述服务信息后形成一模型推送指令并输出；

模型推送模块，连接各所述存储节点和所述模型推送指令生成模块，用于根据所述模型推送指令，将存储于所述存储节点中的所述模型文件推送给对应的所述在线服务，所述在线服务根据新接收的所述模型文件完成对服务模型的更新。

作为本发明的一种优选方案，所述人工智能模型管理系统还包括：

回滚模块，通信连接各所述在线服务、所述数据库和所述模型库，用于在需要从各所述在线服务中回退所述模型时，提供给所述用户输入需要回滚的所述模型的所述模型数据，并根据所输入的所述模型数据生成模型回退指令发送给对应的所述在线服务，所述在线服务根据接收到的所述模型回退指令将所述模型回滚到模型更新前的状态；

所述回滚模块中具体包括：

回滚模型信息输入单元，用于提供给所述用户输入需要回滚的所述模型对应的所述模型数据；