[发明专利]稀疏参数的获取方法、装置、电子设备及存储介质

说明书

本公开提出一种稀疏参数的获取方法、装置、电子设备及存储介质。被电子设备执行，电子设备包括：中央处理器CPU以及图形处理器GPU，图形处理器GPU包括：多个初始稀疏参数，方法包括：获取待处理稀疏特征的待处理存储编号，其中，待处理存储编号表示待处理稀疏特征在中央处理器CPU中的存储位置，根据待处理存储编号，从多个初始稀疏参数中确定目标稀疏参数，由此，能够基于待处理稀疏特征在中央处理器CPU中的待处理存储编号，实现从图形处理器GPU快速调用与待处理稀疏特征对应的目标稀疏参数，在不引入其他资源消耗的情况下，有效地提升稀疏参数的获取效率。

技术领域

本公开涉及计算机技术领域，尤其涉及一种稀疏参数的获取方法、装置、电子设备及存储介质。

背景技术

稀疏特征，是指对模型训练过程中的数据进行ID化处理(ID化处理，例如,采用哈希算法对数据进行映射)，并ID化处理结果映射为整型后所得字符串特征，而稀疏参数，是指采用Embedding技术将稀疏特征转化为多维稠密实数向量特征，在深度学习模型的训练场景中(例如，点击率预估模型(Click Through Rate，CTR))，需要从图形处理器(GraphicsProcessing Unit，GPU)获取大量的稀疏参数去支持模型的训练过程。

相关技术中，稀疏参数和稀疏特征关联存储在中央处理器(Central ProcessingUnit，CPU)中，从而在获取稀疏参数时，需要根据稀疏特征去CPU中查询稀疏参数，再将查询所得稀疏参数复制至GPU，以支持模型训练过程中的稀疏参数调用需求。

这种方式下，稀疏参数的复制过程，会引入较大的资源消耗，并且稀疏参数的获取效率较低。

发明内容

本公开旨在至少在一定程度上解决相关技术中的技术问题之一。

为此，本公开的目的在于提出一种稀疏参数的获取方法、装置、电子设备及存储介质，能够基于待处理稀疏特征在中央处理器CPU中的待处理存储编号，实现从图形处理器GPU快速调用与所述待处理稀疏特征对应的目标稀疏参数，在不引入其他资源消耗的情况下，有效地提升稀疏参数的获取效率。

本公开第一方面实施例提出的稀疏参数的获取方法，被电子设备执行，所述电子设备包括：中央处理器CPU以及图形处理器GPU，所述图形处理器GPU包括：多个初始稀疏参数，所述方法包括：获取待处理稀疏特征的待处理存储编号，其中，所述待处理存储编号表示所述待处理稀疏特征在所述中央处理器CPU中的存储位置；根据所述待处理存储编号，从所述多个初始稀疏参数中确定目标稀疏参数。

本公开第一方面实施例提出的稀疏参数的获取方法，通过获取待处理稀疏特征的待处理存储编号，其中，所述待处理存储编号表示所述待处理稀疏特征在所述中央处理器CPU中的存储位置，根据所述待处理存储编号，从所述多个初始稀疏参数中确定目标稀疏参数，由此，能够基于待处理稀疏特征在中央处理器CPU中的待处理存储编号，实现从图形处理器GPU快速调用与所述待处理稀疏特征对应的目标稀疏参数，在不引入其他资源消耗的情况下，有效地提升稀疏参数的获取效率。

本公开第二方面实施例提出的稀疏参数的获取装置，包括：第一获取模块，用于获取待处理稀疏特征的待处理存储编号，其中，所述待处理存储编号表示所述待处理稀疏特征在所述中央处理器CPU中的存储位置；确定模块，用于根据所述待处理存储编号，从所述多个初始稀疏参数中确定目标稀疏参数。

本公开第二方面实施例提出的稀疏参数的获取装置，通过获取待处理稀疏特征的待处理存储编号，其中，所述待处理存储编号表示所述待处理稀疏特征在所述中央处理器CPU中的存储位置，根据所述待处理存储编号，从所述多个初始稀疏参数中确定目标稀疏参数，由此，能够基于待处理稀疏特征在中央处理器CPU中的待处理存储编号，实现从图形处理器GPU快速调用与所述待处理稀疏特征对应的目标稀疏参数，在不引入其他资源消耗的情况下，有效地提升稀疏参数的获取效率。