公开(公告)号：CN114819140B

公开(公告)日：2025-05-13

申请号：CN202210330396.0

申请日：2022-03-31

Applicant: 支付宝(杭州)信息技术有限公司

Inventor： 余昉 ,  王萌 ,  黄堃 ,  程远

IPC: G06N3/082 ,  G06N3/0455 ,  G06N3/0464

Abstract: 本说明书实施例公开了一种模型剪枝方法、装置和计算机设备。所述方法包括：根据剪枝参数，确定掩码信息；所述掩码信息用于指示目标模型中剪枝对象的有效状态；将样本输入至增添了掩码信息后的目标模型，得到目标模型的第一输出；根据第一输出，优化参数信息；所述参数信息包括目标模型的模型参数和剪枝参数；迭代执行以上步骤，直至满足结束条件；根据掩码信息，对剪枝对象进行剪枝处理。本说明书实施例可以对目标模型进行剪枝处理，以减少存储资源和计算资源的占用。

公开(公告)号：CN118013812A

公开(公告)日：2024-05-10

申请号：CN202211402991.7

申请日：2022-11-10

Applicant: 支付宝(杭州)信息技术有限公司

Inventor： 黄堃 ,  程远 ,  王萌

IPC: G06F30/27 ,  G06F119/02

Abstract: 本说明书实施例公开了一种基于布局传播的机器学习模型转化方法、装置以及设备，采用了布局传播法，区别于传统的算子插入法。方案包括：将原始机器学习模型转化为框架转移中间模型；对框架转移中间模型进行解析，得到包含多个算子的计算图；在多个算子中确定具有隐式布局的算子；判断确定的算子的隐式布局是否与其输入张量的布局变量一致，布局变量是根据目标机器学习模型的目标布局为输入张量显式地赋予的；若否，则对确定的算子的属性进行调整，以使其适应于输入张量；根据调整后的算子，生成目标机器学习模型。

公开(公告)号：CN114819140A

公开(公告)日：2022-07-29

申请号：CN202210330396.0

申请日：2022-03-31

Applicant: 支付宝(杭州)信息技术有限公司

Inventor： 余昉 ,  王萌 ,  黄堃 ,  程远

IPC: G06N3/08 ,  G06N3/04

Abstract: 本说明书实施例公开了一种模型剪枝方法、装置和计算机设备。所述方法包括：根据剪枝参数，确定掩码信息；所述掩码信息用于指示目标模型中剪枝对象的有效状态；将样本输入至增添了掩码信息后的目标模型，得到目标模型的第一输出；根据第一输出，优化参数信息；所述参数信息包括目标模型的模型参数和剪枝参数；迭代执行以上步骤，直至满足结束条件；根据掩码信息，对剪枝对象进行剪枝处理。本说明书实施例可以对目标模型进行剪枝处理，以减少存储资源和计算资源的占用。