当前位置: 首页 > 专利查询>清华大学专利>正文

一种兼容可信执行环境TEE的深度神经网络DNN模型训练方法及框架、电子设备和存储介质技术

技术编号:43984204 阅读:23 留言:0更新日期:2025-01-10 20:07
本公开涉及计算机技术领域,尤其涉及一种兼容可信执行环境TEE的深度神经网络DNN模型训练方法及框架、电子设备和存储介质,所述方法包括:将预设训练样本图像数据输入DNN模型,DNN模型对应有目标计算图,目标计算图中包括多个计算节点、以及每个计算节点的硬件执行位置信息,每个计算节点的硬件执行位置信息用于指示异构硬件平台中的NNA或TEE;通过动态优化目标计算图,控制DNN模型在异构硬件平台上对预设训练样本图像数据执行数据处理;在数据处理完成之后,得到DNN模型输出的图像处理结果,以及根据图像处理结果,调整DNN模型的网络参数。本公开实施例有效实现DNN模型训练过程中对计算效率和数据隐私保护的兼容。

【技术实现步骤摘要】

本公开涉及计算机,尤其涉及一种兼容可信执行环境tee的深度神经网络dnn模型训练方法及框架、电子设备和存储介质。


技术介绍

1、目前,深度神经网络(deep declarative networks,dnn)正在向更复杂的模型架构和更庞大的模型规模发展。一方面,为了处理不断增长的计算,开发了各种神经网络加速器(neural network accelerators,nna),例如先进的图形处理器(graphics processingunit,gpu)、神经网络处理器(neural-network processing unit,npu)、张量处理单元(tensor processing unit,tpu)等,以使得通过专门的硬件优化来加速dnn计算。另一方面,由于训练dnn模型需要大量的用户数据,广泛的dnn应用引发了重大的安全和隐私问题。在人工智能时代,如何在保证dnn模型训练过程中的计算效率的同时保护数据隐私,成为一个迫切的需求和关键问题。


技术实现思路

1、本公开提出了一种兼容可信执行环境tee的深度本文档来自技高网...

【技术保护点】

1.一种兼容可信执行环境TEE的深度神经网络DNN模型训练方法,其特征在于,包括:

2.根据权利要求1所述的方法,其特征在于,所述方法还包括:

3.根据权利要求1所述的方法,其特征在于,所述通过动态优化所述目标计算图,控制所述DNN模型在所述异构硬件平台上对所述预设训练样本图像数据执行数据处理,包括:

4.根据权利要求3所述的方法,其特征在于,每个未执行的计算节点的运行时间包括:计算耗时和通信耗时;

5.根据权利要求3所述的方法,其特征在于,所述根据优化后的所述目标计算图,控制所述DNN模型在所述异构硬件平台上执行每个未执行的计算节点,以对...

【技术特征摘要】

1.一种兼容可信执行环境tee的深度神经网络dnn模型训练方法,其特征在于,包括:

2.根据权利要求1所述的方法,其特征在于,所述方法还包括:

3.根据权利要求1所述的方法,其特征在于,所述通过动态优化所述目标计算图,控制所述dnn模型在所述异构硬件平台上对所述预设训练样本图像数据执行数据处理,包括:

4.根据权利要求3所述的方法,其特征在于,每个未执行的计算节点的运行时间包括:计算耗时和通信耗时;

5.根据权利要求3所述的方法,其特征在于,所述根据优化后的所述目标计算图,控制所述dnn模型在所述异构硬件平台上执行每个未执行的计算节点,以对所述预设训练样本图像数据执行数据处理,包括:

6.根据权利要求5所述的方法,其特征在于,所述基于预设加密...

【专利技术属性】
技术研发人员:任炬李秋实张焱张尧学
申请(专利权)人:清华大学
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1