CN112446476A - 神经网络模型压缩的方法、装置、存储介质和芯片 - Google Patents
神经网络模型压缩的方法、装置、存储介质和芯片 Download PDFInfo
- Publication number
- CN112446476A CN112446476A CN201910833833.9A CN201910833833A CN112446476A CN 112446476 A CN112446476 A CN 112446476A CN 201910833833 A CN201910833833 A CN 201910833833A CN 112446476 A CN112446476 A CN 112446476A
- Authority
- CN
- China
- Prior art keywords
- data
- neural network
- network model
- training
- server
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Pending
Links
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/06—Physical realisation, i.e. hardware implementation of neural networks, neurons or parts of neurons
- G06N3/063—Physical realisation, i.e. hardware implementation of neural networks, neurons or parts of neurons using electronic means
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
- G06N3/084—Backpropagation, e.g. using gradient descent
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/21—Design or setup of recognition systems or techniques; Extraction of features in feature space; Blind source separation
- G06F18/214—Generating training patterns; Bootstrap methods, e.g. bagging or boosting
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/24—Classification techniques
- G06F18/241—Classification techniques relating to the classification model, e.g. parametric or non-parametric approaches
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/045—Combinations of networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/40—Extraction of image or video features
- G06V10/44—Local feature extraction by analysis of parts of the pattern, e.g. by detecting edges, contours, loops, corners, strokes or intersections; Connectivity analysis, e.g. of connected components
- G06V10/443—Local feature extraction by analysis of parts of the pattern, e.g. by detecting edges, contours, loops, corners, strokes or intersections; Connectivity analysis, e.g. of connected components by matching or filtering
- G06V10/449—Biologically inspired filters, e.g. difference of Gaussians [DoG] or Gabor filters
- G06V10/451—Biologically inspired filters, e.g. difference of Gaussians [DoG] or Gabor filters with interaction between the filter responses, e.g. cortical complex cells
- G06V10/454—Integrating the filters into a hierarchical structure, e.g. convolutional neural networks [CNN]
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/70—Arrangements for image or video recognition or understanding using pattern recognition or machine learning
- G06V10/82—Arrangements for image or video recognition or understanding using pattern recognition or machine learning using neural networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/24—Classification techniques
- G06F18/241—Classification techniques relating to the classification model, e.g. parametric or non-parametric approaches
- G06F18/2415—Classification techniques relating to the classification model, e.g. parametric or non-parametric approaches based on parametric or probabilistic models, e.g. based on likelihood ratio or false acceptance rate versus a false rejection rate
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/25—Fusion techniques
- G06F18/253—Fusion techniques of extracted features
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/047—Probabilistic or stochastic networks
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Physics & Mathematics (AREA)
- Evolutionary Computation (AREA)
- General Physics & Mathematics (AREA)
- Artificial Intelligence (AREA)
- Data Mining & Analysis (AREA)
- Life Sciences & Earth Sciences (AREA)
- Health & Medical Sciences (AREA)
- General Health & Medical Sciences (AREA)
- Biomedical Technology (AREA)
- General Engineering & Computer Science (AREA)
- Molecular Biology (AREA)
- Computing Systems (AREA)
- Software Systems (AREA)
- Biophysics (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Computational Linguistics (AREA)
- Mathematical Physics (AREA)
- Multimedia (AREA)
- Bioinformatics & Cheminformatics (AREA)
- Bioinformatics & Computational Biology (AREA)
- Evolutionary Biology (AREA)
- Databases & Information Systems (AREA)
- Medical Informatics (AREA)
- Biodiversity & Conservation Biology (AREA)
- Neurology (AREA)
- Image Analysis (AREA)
Abstract
本申请提供了人工智能领域中的一种神经网络模型压缩的方法,包括:服务器获取用户设备上传的第一神经网络模型和第一神经网络的训练数据;根据第一神经网络的训练数据和服务器存储的未标记数据,得到PU分类器;使用PU分类器,从服务器存储的未标记数据中选择扩展数据,扩展数据具有与第一神经网络模型的训练数据相似属性与分布;根据扩展数据,利用知识蒸馏KD方法训练第二神经网络模型,第一神经网络模型作为教师网络模型,第二神经网络模型作为学生网络模型。采用PU分类器,从未标记数据中选择与第一神经网络模型的训练数据具有相似属性与分布的数据,提高神经网络模型压缩的准确性,避免大量正样本数据的传输。
Description
技术领域
本申请涉及人工智能领域,尤其涉及一种神经网络模型压缩的方法及装置。
背景技术
人工智能(artificial intelligence,AI)是利用数字计算机或者数字计算机控制的机器模拟、延伸和扩展人的智能,感知环境、获取知识并使用知识获得最佳结果的理论、方法、技术及应用系统。换句话说,人工智能是计算机科学的一个分支,它企图了解智能的实质,并生产出一种新的能以人类智能相似的方式作出反应的智能机器。人工智能也就是研究各种智能机器的设计原理与实现方法,使机器具有感知、推理与决策的功能
计算机视觉是各个应用领域,如制造业、检验、文档分析、医疗诊断,和军事等领域中各种智能/自主系统中不可分割的一部分,它是一门关于如何运用照相机/摄像机和计算机来获取我们所需的,被拍摄对象的数据与信息的学问。形象地说,就是给计算机安装上眼睛(照相机/摄像机)和大脑(算法)用来代替人眼对目标进行识别、跟踪和测量等,从而使计算机能够感知环境。因为感知可以看作是从感官信号中提取信息,所以计算机视觉也可以看作是研究如何使人工系统从图像或多维数据中“感知”的科学。总的来说,计算机视觉就是用各种成象系统代替视觉器官获取输入信息,再由计算机来代替大脑对这些输入信息完成处理和解释。计算机视觉的最终研究目标就是使计算机能像人那样通过视觉观察和理解世界,具有自主适应环境的能力。
卷积神经网络模型通常会存在大量的冗余参数。为了将CNN网络应用于智能手机等算力有限的终端设备,可以对现有的卷积神经网络模型进行压缩和加速。神经网络模型压缩技术需要提供大量训练数据才能使网络收敛到较好的结果,然而让用户上传大量数据到云端是非常费时的,会造成用户体验的下降。部分神经网络模型压缩技术仅适用少量训练数据来进行模型压缩,但压缩后的神经网络模型难以达到令人满意的效果。
发明内容
本申请提供一种神经网络模型压缩的方法、装置、存储介质和芯片,以减小传输的数据量,提高用户体验。
第一方面,提供一种神经网络模型压缩的方法,包括:服务器获取用户设备上传的第一神经网络模型和所述第一神经网络的训练数据;所述服务器根据所述第一神经网络的训练数据和所述服务器存储的未标记数据,利用正类与未标记PU学习算法得到PU分类器;所述服务器使用所述PU分类器,从所述服务器存储的未标记数据中选择出扩展数据,所述扩展数据为与所述第一神经网络模型的训练数据具有相似属性与分布的数据;所述服务器根据所述扩展数据,利用知识蒸馏KD方法训练第二神经网络模型,所述第一神经网络模型作为所述KD方法的教师网络模型,所述第二神经网络模型作为所述KD方法的学生网络模型。
利用第一神经网络的训练数据和未标记数据,训练得到PU分类器,对未标记数据进行分类,得到与第一神经网络模型的训练数据具有相似属性与分布的数据。根据这些与所述第一神经网络模型的训练数据具有相似属性与分布的数据,能够实现神经网络模型的压缩,在保证神经网络模型压缩准确性的同时,减小了传输的数据量。
结合第一方面,在一些可能的实现方式中,所述服务器根据所述第一神经网络的训练数据、所述服务器存储的未标记数据和比例信息,利用正类与未标记PU学习算法得到PU分类器,所述PU学习算法的损失函数为所述第一神经网络的训练数据和所述服务器存储的未标记数据的训练损失的期望,所述比例信息用于指示所述扩展数据在所述服务器存储的未标记数据中的占比,所述比例信息用于计算所述期望。
根据第二数据中正样本数据的比例信息,计算第一数据和所述第二数据的训练损失的期望,作为PU学习算法的损失函数,从而训练PU分类器。
结合第一方面,在一些可能的实现方式中,所述第一神经网络模型的训练数据为用于训练所述第一神经网络模型的训练数据中的部分训练数据。
用户设备上传用于训练所述第一神经网络模型的训练数据中的部分训练数据,减小上传数据的数量,提高用户体验。
结合第一方面,在一些可能的实现方式中,所述部分训练数据包括所述第一神经网络输出的多个类别中每个类别的数据。
用户上传的第一神经网络模型的训练数据包括第一神经网络能够处理的多类数据中每一类的数据,使得训练得到的第二神经网络能够对该多类数据进行处理,提高对第一神经网络模型压缩的准确性。
结合第一方面,在一些可能的实现方式中,所述PU分类器是根据第一特征和所述比例信息得到的,所述第一特征是基于多个第三特征融合获得的,所述多个第三特征是利用所述第一神经网络模型对所述第一神经网络的训练数据和所述服务器存储的未标记数据进行特征提取得到的,所述多个第三特征与所述第一神经网络的多个层一一对应,所述服务器使用所述PU分类器,从所述服务器存储的未标记数据中选择出扩展数据,包括:所述服务器利用所述第一神经网络模型对所述服务器存储的未标记数据进行特征提取,以得到第二特征;所述服务器将所述第二特征,输入所述PU分类器,以确定所述扩展数据。通过第一神经网络模型对训练PU分类器的数据进行特征的提取网络,减少对PU分类器进行训练的时间,提高效率。
结合第一方面,在一些可能的实现方式中,所述第一特征是利用经过第一权重调整之后的多个第三特征融合获得的,所述第一权重调整是基于根据所述比例信息进行的,所述第二特征是利用所述第一权重对多个第四特征融合得到的,所述多个第四特征与所述第一神经网络的多个层一一对应。
通过对第一神经网络模型不同层输出的特征进行权重的调整,根据调整后的权重对提取的特征进行融合,提高PU分类器分类结果的准确性。
结合第一方面,在一些可能的实现方式中,所述服务器根据所述扩展数据,利用KD方法训练第二神经网络模型,包括:所述服务器将所述扩展数据输入所述第一神经网络模型,以对所述扩展数据进行分类,得到多类扩展数据以及所述多类扩展数据中每类扩展数据的第二权重;所述服务器最小化所述KD算法的损失函数,以得到训练后的第二神经网络模型,其中,所述KD算法的损失函数为所述多类扩展数据中的各类扩展数据的训练误差与所述各类扩展数据的第二权重的乘积之和。
根据未标记数据中的正样本数据在第一神经网络模型能够处理的类别中的数量,调整KD算法的损失函数中每一个类的数据对应的权重,在正样本数据在不同的类别中分布不平衡的情况下,对于每一个类的数据,训练得到的神经网络模型均能够得到较好的分类结果。
结合第一方面,在一些可能的实现方式中,所述各类扩展数据的第二权重包括对所述各类扩展数据的初始权重进行随机扰动后得到的多个扰动后的权重,所述KD算法的损失函数包括所述多个扰动后的权重一一对应多个损失函数,其中,所述各类扩展数据的初始权重与所述各类扩展数据的数量负相关,所述服务器最小化所述KD算法的损失函数,以得到训练后的第二神经网络模型,包括:所述服务器最小化所述多个损失函数的最大值,以得到训练后的第二神经网络模型。
通过对KD算法的损失函数中的权重进行随机扰动,计算神经网络模型在不同扰动情况下的KD算法的损失函数,将使得所述多个损失函数的最大值最小的神经网络模型作为训练得到的神经网络模型。减小了教师网络模型分类误差对训练得到的神经网络模型准确性的影响,提高训练得到的神经网络模型的准确性。
第二方面,提供一种神经网络模型压缩装置,该装置包括用于执行上述第一方面中的方法中的各个模块。
第三方面,提供一种计算机设备,包括存储器,用于存储程序;处理器,用于执行所述存储器存储的程序,当所述存储器存储的程序被执行时,所述处理器用于执行上述第一方面中的方法。
应当理解,计算机设备可以是服务器,该服务器可以部署在云端。计算机设备具有一定的运算能力。
第四方面,提供一种计算机存储介质,该计算机可读存储介质存储有程序代码,该程序代码包括用于执行第一方面中的方法中的步骤的指令。
第五方面,提供一种包含指令的计算机程序产品,当该计算机程序产品在计算机上运行时,使得计算机执行上述第一方面中的方法。
第六方面,提供一种芯片,所述芯片包括至少一个处理器,当程序指令在所述至少一个处理器中执行时,使得所述芯片执行第一方面所述的方法。
可选地,作为一种实现方式,所述芯片还可以包括存储器,所述存储器中存储有指令,所述处理器用于执行所述存储器上存储的指令,当所述指令被执行时,所述处理器用于执行第一方面中的方法。
上述芯片具体可以是现场可编程门阵列FPGA或者专用集成电路ASIC。
应理解,本申请中,第一方面的方法具体可以是指第一方面以及第一方面中各种实现方式中的任意一种实现方式中的方法。
附图说明
图1是本申请实施例提供的系统架构的结构示意图
图2是一种卷积神经网络模型的示意图。
图3是一种神经网络模型压缩的方法的示意图。
图4是本申请一个实施例提供的一种神经网络模型压缩的方法的示意性流程图。
图5是本申请另一个实施例提供的一种神经网络模型压缩的方法的示意性流程图。
图6是本申请一个实施例提供的一种扩展正样本数据的方法的示意性流程图。
图7是本申请一个实施例提供的一种带注意力机制的多特征融合模型的示意图。
图8是本申请一个实施例提供的一种知识蒸馏的方法的示意图。
图9是本申请又一个实施例提供的一种神经网络模型压缩的方法的示意性流程图。
图10是本申请一个实施例提供的一种神经网络模型压缩的装置的示意性结构图。
图11是本申请一个实施例提供的一种神经网络模型压缩的装置的硬件结构示意图。
具体实施方式
下面将结合附图,对本申请中的技术方案进行描述。
由于本申请实施例涉及大量神经网络模型的应用,为了便于理解,下面先对本申请实施例涉及的相关术语及神经网络模型等相关概念进行介绍。
(1)神经网络模型
神经网络模型可以是由神经单元组成的,神经单元可以是指以xs和截距b为输入的运算单元,该运算单元的输出可以为:
其中,s=1、2、……n,n为大于1的自然数,Ws为xs的权重,b为神经单元的偏置。f为神经单元的激活函数(activation functions),用于将非线性特性引入神经网络模型中,来将神经单元中的输入信号转换为输出信号。该激活函数的输出信号可以作为下一层卷积层的输入。激活函数可以是sigmoid函数。神经网络模型是将许多个上述单一的神经单元联结在一起形成的网络,即一个神经单元的输出可以是另一个神经单元的输入。每个神经单元的输入可以与前一层的局部接受域相连,来提取局部接受域的特征,局部接受域可以是由若干个神经单元组成的区域。
(2)深度神经网络模型
深度神经网络模型(deep neural network,DNN),也称多层神经网络模型,可以理解为具有很多层隐含层的神经网络模型,这里的“很多”并没有特别的度量标准。从DNN按不同层的位置划分,DNN内部的神经网络模型可以分为三类:输入层,隐含层,输出层。一般来说第一层是输入层,最后一层是输出层,中间的层数都是隐含层。例如,全连接神经网络模型中层与层之间是全连接的,也就是说,第i层的任意一个神经元一定与第i+1层的任意一个神经元相连。虽然DNN看起来很复杂,但是就每一层的工作来说,其实并不复杂,简单来说就是如下线性关系表达式:其中,是输入向量,是输出向量,是偏移向量,W是权重矩阵(也称系数),α()是激活函数。每一层仅仅是对输入向量经过如此简单的操作得到输出向量由于DNN层数多,则系数W和偏移向量的数量也就很多了。这些参数在DNN中的定义如下所述:以系数W为例:假设在一个三层的DNN中,第二层的第4个神经元到第三层的第2个神经元的线性系数定义为上标3代表系数W所在的层数,而下标对应的是输出的第三层索引2和输入的第二层索引4。总结就是:第L-1层的第k个神经元到第L层的第j个神经元的系数定义为需要注意的是,输入层是没有W参数的。在深度神经网络模型中,更多的隐含层让网络更能够刻画现实世界中的复杂情形。理论上而言,参数越多的模型复杂度越高,“容量”也就越大,也就意味着它能完成更复杂的学习任务。训练深度神经网络模型的也就是学习权重矩阵的过程,其最终目的是得到训练好的深度神经网络模型的所有层的权重矩阵(由很多层的向量W形成的权重矩阵)。
(3)卷积神经网络模型
卷积神经网络模型(convolutional neuron network,CNN)是一种带有卷积结构的深度神经网络模型。卷积神经网络模型包含了一个由卷积层和子采样层构成的特征抽取器。该特征抽取器可以看作是滤波器,卷积过程可以看作是使用一个可训练的滤波器与一个输入的图像或者卷积特征平面(feature map)做卷积。卷积层是指卷积神经网络模型中对输入信号进行卷积处理的神经元层。在卷积神经网络模型的卷积层中,一个神经元可以只与部分邻层神经元连接。一个卷积层中,通常包含若干个特征平面,每个特征平面可以由一些矩形排列的神经单元组成。同一特征平面的神经单元共享权重,这里共享的权重就是卷积核。共享权重可以理解为提取图像信息的方式与位置无关。这其中隐含的原理是:图像的某一部分的统计信息与其他部分是一样的。即意味着在某一部分学习的图像信息也能用在另一部分上。所以对于图像上的所有位置,都能使用同样的学习得到的图像信息。在同一卷积层中,可以使用多个卷积核来提取不同的图像信息,一般地,卷积核数量越多,卷积操作反映的图像信息越丰富。
卷积核可以以随机大小的矩阵的形式初始化,在卷积神经网络模型的训练过程中卷积核可以通过学习得到合理的权重。另外,共享权重带来的直接好处是减少卷积神经网络模型各层之间的连接,同时又降低了过拟合的风险。
(4)损失函数
在训练深度神经网络模型的过程中,因为希望深度神经网络模型的输出尽可能的接近真正想要预测的值,所以可以通过比较当前网络的预测值和真正想要的目标值,再根据两者之间的差异情况来更新每一层神经网络模型的权重向量(当然,在第一次更新之前通常会有初始化的过程,即为深度神经网络模型中的各层预先配置参数),比如,如果网络的预测值高了,就调整权重向量让它预测低一些,不断的调整,直到深度神经网络模型能够预测出真正想要的目标值或与真正想要的目标值非常接近的值。因此,就需要预先定义“如何比较预测值和目标值之间的差异”,这便是损失函数(loss function)或目标函数(objective function),它们是用于衡量预测值和目标值的差异的重要方程。其中,以损失函数举例,损失函数的输出值(loss)越高表示差异越大,那么深度神经网络模型的训练就变成了尽可能缩小这个loss的过程。
(5)残差网络
在不断加神经网络模型的深度时,会出现退化的问题,即随着神经网络模型深度的增加,准确率先上升,然后达到饱和,再持续增加深度则会导致准确率下降。普通直连的卷积神经网络模型和残差网络(residual network,ResNet)的最大区别在于,ResNet有很多旁路的支线将输入直接连到后面的层,通过直接将输入信息绕道传到输出,保护信息的完整性,解决退化的问题。残差网络包括卷积层和/或池化层。
残差网络可以是:深度神经网络模型中多个隐含层之间除了逐层相连之外,例如第1层隐含层连接第2层隐含层,第2层隐含层连接第3层隐含层,第3层隐含层连接第4层隐含层(这是一条神经网络模型的数据运算通路,也可以形象的称为神经网络模型传输),残差网络还多了一条直连支路,这条直连支路从第1层隐含层直接连到第4层隐含层,即跳过第2层和第3层隐含层的处理,将第1层隐含层的数据直接传输给第4层隐含层进行运算。公路网络可以是:深度神经网络模型中除了有上面所述的运算通路和直连分支之外,还包括一条权重获取分支,这条支路引入传输门(transform gate)进行权重值的获取,并输出权重值T供上面的运算通路和直连分支后续的运算使用。
(6)反向传播算法
卷积神经网络模型可以采用误差反向传播(back propagation,BP)算法在训练过程中修正初始的神经网络模型中参数的大小,使得神经网络模型的重建误差损失越来越小。具体地,前向传递输入信号直至输出会产生误差损失,通过反向传播误差损失信息来更新初始的神经网络模型中参数,从而使误差损失收敛。反向传播算法是以误差损失为主导的反向传播运动,旨在得到最优的神经网络模型的参数,例如权重矩阵。
(7)注意力机制
注意力机制模仿了生物观察行为的内部过程,即一种将内部经验和外部感觉对齐从而增加部分区域的观察精细度的机制,能够利用有限的注意力资源从大量信息中快速筛选出高价值信息。注意力机制可以快速提取稀疏数据的重要特征,因而被广泛用于自然语言处理任务,特别是机器翻译。而自注意力机制(self-attention mechanism)是注意力机制的改进,其减少了对外部信息的依赖,更擅长捕捉数据或特征的内部相关性。注意力机制的本质思想可以改写为如下公式:
其中,Lx=||Source||代表Source的长度,公式含义即将Source中的构成元素想象成是由一系列的<Key,Value>数据对构成,此时给定目标Target中的某个元素Query,通过计算Query和各个Key的相似性或者相关性,得到每个Key对应Value的权重系数,然后对Value进行加权求和,即得到了最终的Attention数值。所以本质上Attention机制是对Source中元素的Value值进行加权求和,而Query和Key用来计算对应Value的权重系数。从概念上理解,把Attention可以理解为从大量信息中有选择地筛选出少量重要信息并聚焦到这些重要信息上,忽略大多不重要的信息。聚焦的过程体现在权重系数的计算上,权重越大越聚焦于其对应的Value值上,即权重代表了信息的重要性,而Value是其对应的信息。自注意力机制可以理解为内部Attention(intra attention),Attention机制发生在Target的元素Query和Source中的所有元素之间,自注意力机制指的是在Source内部元素之间或者Target内部元素之间发生的Attention机制,也可以理解为Target=Source这种特殊情况下的注意力计算机制,其具体计算过程是一样的,只是计算对象发生了变化而已。
(8)像素值
图像的像素值可以是一个红绿蓝(RGB)颜色值,像素值可以是表示颜色的长整数。例如,像素值为256×Red+100×Green+76×Blue,其中,Blue代表蓝色分量,Green代表绿色分量,Red代表红色分量。各个颜色分量中,数值越小,亮度越低,数值越大,亮度越高。对于灰度图像来说,像素值可以是灰度值。
(9)知识蒸馏
较大、较复杂的网络虽然通常具有很好的性能,但是也存在很多的冗余信息,因此运算量以及资源的消耗都非常多。神经网络模型的知识蒸馏(distilling the knowledgein a neural network)是将复杂网络中的有用信息提取出来迁移到一个更小的网络上,这样学习来的小网络可以具备和大的复杂网络想接近的性能效果,并且也大大的节省了计算资源。这个复杂的网络可以称为教师网络模型,而小的网络则可以称为学生网络模型。
(10)正类与未标记学习
正类与未标记学习(positive-unlabeled learning,PU learning)是一种半监督的机器学习方式。通过该方法,在给定的数据中,只有部分正样本数据被标记,其余正样本数据与所有的负样本数据均未被标记。在这种情况下通过学习,训练一个二类分类器,对未标记数据进行分类,确定其中的正样本数据和负样本数据。
以上对神经网络的一些基本内容做了简单介绍,下面针对图像数据处理时可能用到的一些特定神经网络进行介绍。
下面结合图1对本申请实施例的系统架构进行详细的介绍。
图1是本申请实施例的系统架构的示意图。如图1所示,系统架构100包括执行设备110、训练设备120、数据库130、客户设备140、数据存储系统150、以及数据采集系统160。
另外,执行设备110包括计算模块111、I/O接口112、预处理模块113和预处理模块114。其中,计算模块111中可以包括目标模型/规则101,预处理模块113和预处理模块114是可选的。
数据采集设备160用于采集训练数据。本申请实施例中训练数据包括:第一神经网络的训练数据、第一神经网络模型和服务器存储的未标记数据。并将训练数据存入数据库130,训练设备120基于数据库130中维护的训练数据训练得到目标模型/规则101。
下面对训练设备120基于训练数据得到目标模型/规则101进行描述。上述目标模型/规则101能够用于实现本申请实施例的神经网络模型压缩的方法,即,将第一神经网络的训练数据、第一神经网络模型输入该目标模型/规则101,即可得到训练得到的第二神经网络模型。本申请实施例中的目标模型/规则101具体可以为神经网络。需要说明的是,在实际应用中,数据库130中维护的训练数据不一定都来自于数据采集设备160的采集,也有可能是从其他设备接收得到的。另外需要说明的是,训练设备120也不一定完全基于数据库130维护的训练数据进行目标模型/规则101的训练,也有可能从云端或其他地方获取训练数据进行模型训练,上述描述不应该作为对本申请实施例的限定。根据训练设备120训练得到的目标模型/规则101可以应用于不同的系统或设备中,如应用于图3所示的执行设备110,所述执行设备110可以是终端,如手机终端,平板电脑,笔记本电脑,增强现实(augmented reality,AR)/虚拟现实(virtual reality,VR),车载终端等,还可以是服务器或者云端等。在图3中,执行设备110配置输入/输出(input/output,I/O)接口112,用于与外部设备进行数据交互,用户可以通过客户设备140向I/O接口112输入数据,所述输入数据在本申请实施例中可以包括:第一神经网络的训练数据、第一神经网络模型。这里的客户设备140具体可以是用户设备。
预处理模块113和预处理模块114用于根据I/O接口112接收到的输入数据(如待处理图像)进行预处理,在本申请实施例中,可以没有预处理模块113和预处理模块114或者只有的一个预处理模块。当不存在预处理模块113和预处理模块114时,可以直接采用计算模块111对输入数据进行处理。
在执行设备110对输入数据进行预处理,或者在执行设备110的计算模块111执行计算等相关的处理过程中,执行设备110可以调用数据存储系统150中的数据、代码等以用于相应的处理,也可以将相应处理得到的数据、指令等存入数据存储系统150中。
最后,I/O接口112将处理结果,如将目标模型/规则101计算得到的训练后的第二神经网络模型呈现给客户设备140,从而提供给用户。
具体地,经过计算模块111中的目标模型/规则101处理得到的训练后的第二神经网络模型可以通过预处理模块113(也可以再加上预处理模块114的处理)的处理后将处理结果送入到I/O接口,再由I/O接口将处理结果送入到客户设备140中。
应理解,当上述系统架构100中不存在预处理模块113和预处理模块114时,计算模块111还可以将处理得到的训练后的第二神经网络模型传输到I/O接口,然后再由I/O接口将处理结果送入到客户设备140中显示。
值得说明的是,训练设备120可以针对不同的目标或称不同的任务,基于不同的训练数据生成相应的目标模型/规则101,该相应的目标模型/规则101即可以用于实现上述目标或完成上述任务,从而为用户提供所需的结果。
在图3中所示情况下,用户可以手动给定输入数据,该手动给定可以通过I/O接口112提供的界面进行操作。另一种情况下,客户设备140可以自动地向I/O接口112发送输入数据,如果要求客户设备140自动发送输入数据需要获得用户的授权,则用户可以在客户设备140中设置相应权限。用户可以在客户设备140查看执行设备110输出的结果,具体的呈现形式可以是显示、声音、动作等具体方式。客户设备140也可以作为数据采集端,采集如图所示输入I/O接口112的输入数据及输出I/O接口112的输出结果作为新的样本数据,并存入数据库130。当然,也可以不经过客户设备140进行采集,而是由I/O接口112直接将如图所示输入I/O接口112的输入数据及输出I/O接口112的输出结果,作为新的样本数据存入数据库130。
值得注意的是,图1仅是本申请实施例提供的一种系统架构的示意图,图中所示设备、器件、模块等之间的位置关系不构成任何限制,例如,在图1中,数据存储系统150相对执行设备110是外部存储器,在其它情况下,也可以将数据存储系统150置于执行设备110中。
如图3所示,根据训练设备120训练得到目标模型/规则101,可以是本申请实施例中的神经网络,具体的,本申请实施例提供的神经网络可以是CNN以及深度卷积神经网络(deep convolutional neural networks,DCNN)等等。
由于CNN是一种非常常见的神经网络,下面结合图2重点对CNN的结构进行详细的介绍。如前文的基础概念介绍所述,卷积神经网络模型是一种带有卷积结构的深度神经网络模型,是一种深度学习(deep learning)架构,深度学习架构是指通过机器学习的算法,在不同的抽象层级上进行多个层次的学习。作为一种深度学习架构,CNN是一种前馈(feed-forward)人工神经网络模型,该前馈人工神经网络模型中的各个神经元可以对输入其中的图像作出响应。
如图2所示,卷积神经网络模型(CNN)200可以包括输入层210,卷积层/池化层220(其中池化层为可选的),以及神经网络模型层230。
卷积层/池化层220:
卷积层:
如图2所示卷积层/池化层220可以包括如示例221-226层,举例来说:在一种实现中,221层为卷积层,222层为池化层,223层为卷积层,224层为池化层,225为卷积层,226为池化层;在另一种实现方式中,221、222为卷积层,223为池化层,224、225为卷积层,226为池化层。即卷积层的输出可以作为随后的池化层的输入,也可以作为另一个卷积层的输入以继续进行卷积操作。
下面将以卷积层221为例,介绍一层卷积层的内部工作原理。
卷积层221可以包括很多个卷积算子,卷积算子也称为核,其在图像处理中的作用相当于一个从输入图像矩阵中提取特定信息的过滤器,卷积算子本质上可以是一个权重矩阵,这个权重矩阵通常被预先定义,在对图像进行卷积操作的过程中,权重矩阵通常在输入图像上沿着水平方向一个像素接着一个像素(或两个像素接着两个像素……这取决于步长stride的取值)的进行处理,从而完成从图像中提取特定特征的工作。该权重矩阵的大小应该与图像的大小相关,需要注意的是,权重矩阵的纵深维度(depth dimension)和输入图像的纵深维度是相同的,在进行卷积运算的过程中,权重矩阵会延伸到输入图像的整个深度。因此,和一个单一的权重矩阵进行卷积会产生一个单一纵深维度的卷积化输出,但是大多数情况下不使用单一权重矩阵,而是应用多个尺寸(行×列)相同的权重矩阵,即多个同型矩阵。每个权重矩阵的输出被堆叠起来形成卷积图像的纵深维度,这里的维度可以理解为由上面所述的“多个”来决定。不同的权重矩阵可以用来提取图像中不同的特征,例如一个权重矩阵用来提取图像边缘信息,另一个权重矩阵用来提取图像的特定颜色,又一个权重矩阵用来对图像中不需要的噪点进行模糊化等。该多个权重矩阵尺寸(行×列)相同,经过该多个尺寸相同的权重矩阵提取后的特征图的尺寸也相同,再将提取到的多个尺寸相同的特征图合并形成卷积运算的输出。
这些权重矩阵中的权重值在实际应用中需要经过大量的训练得到,通过训练得到的权重值形成的各个权重矩阵可以用来从输入图像中提取信息,从而使得卷积神经网络模型200进行正确的预测。
当卷积神经网络模型200有多个卷积层的时候,初始的卷积层(例如221)往往提取较多的一般特征,该一般特征也可以称之为低级别的特征;随着卷积神经网络模型200深度的加深,越往后的卷积层(例如226)提取到的特征越来越复杂,比如高级别的语义之类的特征,语义越高的特征越适用于待解决的问题。
池化层:
由于常常需要减少训练参数的数量,因此卷积层之后常常需要周期性的引入池化层,在如图1中220所示例的221-226各层,可以是一层卷积层后面跟一层池化层,也可以是多层卷积层后面接一层或多层池化层。在图像处理过程中,池化层的唯一目的就是减少图像的空间大小。池化层可以包括平均池化算子和/或最大池化算子,以用于对输入图像进行采样得到较小尺寸的图像。平均池化算子可以在特定范围内对图像中的像素值进行计算产生平均值作为平均池化的结果。最大池化算子可以在特定范围内取该范围内值最大的像素作为最大池化的结果。另外,就像卷积层中用权重矩阵的大小应该与图像尺寸相关一样,池化层中的运算符也应该与图像的大小相关。通过池化层处理后输出的图像尺寸可以小于输入池化层的图像的尺寸,池化层输出的图像中每个像素点表示输入池化层的图像的对应子区域的平均值或最大值。
神经网络模型层230:
在经过卷积层/池化层220的处理后,卷积神经网络模型200还不足以输出所需要的输出信息。因为如前所述,卷积层/池化层220只会提取特征,并减少输入图像带来的参数。然而为了生成最终的输出信息(所需要的类信息或其他相关信息),卷积神经网络模型200需要利用神经网络模型层230来生成一个或者一组所需要的类的数量的输出。因此,在神经网络模型层230中可以包括多层隐含层(如图1所示的231、232至23n)以及输出层240,该多层隐含层中所包含的参数可以根据具体的任务类型的相关训练数据进行预先训练得到,例如该任务类型可以包括图像识别,图像分类,图像超分辨率重建等等……
在神经网络模型层230中的多层隐含层之后,也就是整个卷积神经网络模型200的最后层为输出层240,该输出层240具有类似分类交叉熵的损失函数,具体用于计算预测误差,一旦整个卷积神经网络模型200的前向传播(如图1由210至240方向的传播为前向传播)完成,反向传播(如图1由240至210方向的传播为反向传播)就会开始更新前面提到的各层的权重值以及偏差,以减少卷积神经网络模型200的损失,及卷积神经网络模型200通过输出层输出的结果和理想结果之间的误差。
需要说明的是,如图2所示的卷积神经网络模型200仅作为一种卷积神经网络模型的示例,在具体的应用中,卷积神经网络模型还可以以其他网络模型的形式存在。CNN网络在计算机视觉领域中被广泛使用,并在图片分类、目标检测、语义分割等多个实际应用中获得了成功。为了得到更高的分类精度,CNN网络通常会存在大量的冗余参数,许多研究证明了可以在不影响网络分类性能的前提下去除这些冗余的参数。同时,为了将CNN网络应用于智能手机等算力有限的终端设备,我们需要对现有的CNN网络进行压缩和加速。考虑到云端存在有大量的计算资源,我们可以为用户提供云上模型压缩的服务。
通过CNN网络压缩和加速技术,例如可以利用知识蒸馏(knowledgedistillation,KD)、剪除相似神经元、权值离散化、哈希法等方法对神经网络模型进行压缩,能够在云端为用户提供模型压缩与加速服务。
图3是一种神经网络模型压缩的方法的示意图。
将训练数据输入教师网络模型和学生网络模型,根据教师网络模型的输出和学生网络模型的输出,确定交叉熵损失(cross entropy loss),即KD算法的损失函数LKD。KD算法的损失函数LKD可以用于衡量教师网络模型和学生网络模型对于训练数据的处理结果的相似度。调整学生网络模型的参数,以使得KD算法的损失函数LKD最小:
KD算法也可以称为CNN模型压缩与加速算法。通过调整学生网络模型的参数以使得损失函数LKD最小,即学生网络模型的输出尽可能的和教师网络模型的输出一致,这样,学生网络模型学习到了教师网络模型的相关特性。
教师网络模型是根据原始训练数据得到的,教师网络模型对于原始训练数据的输出较为准确。可以根据原始训练数据完成对学生网络模型的训练。
训练学生网络模型时,需要获取训练教师网络模型所需的全部训练数据,才能够使压缩后的网络具有好的分类性能。但相比于教师网络模型,训练数据通的文件大小很大。例如ResNet-50网络仅仅占用95MB的空间来存储模型参数,然而训练ResNet-50网络所需的训练数据集(image net)有超过120万张训练图片,并且需要超过120GB的空间来存放。因此在传输速度有限的情况下,用户需要花费很长的时间才能够将训练数据上传到云端。提供大量训练数据才能使网络收敛到较好的结果,然而让用户上传大量数据到云端是非常费时的,因此会带来很糟糕的用户体验。
教师网络模型对于与原始数据相同的类的图片能够获得较为准确的输出结果。通过与原始数据相同的类的图片,对学生网络模型进行训练,可以有效地利用教师网络模型,得较为准确的学生网络模型。
一种获取与原始训练数据的类别分布相同的图像的方式,可以参见《Data-FreeLearning of Student Networks》(Chen H,Wang Y,Xu C,et al.2019.),通过生成对抗网络生成图像。
生成式对抗网络(generative adversarial networks,GAN)是一种深度学习模型。该模型中至少包括两个模块:一个模块是生成模型(generative model),另一个模块是判别模型(discriminative model),通过这两个模块互相博弈学习,从而产生更好的输出。GAN的基本原理如下:以生成图片的GAN为例,假设有两个网络,G(generator)和D(discriminator),其中G是一个生成图片的网络,它接收一个随机的噪声z,通过这个噪声生成图片,记做G(z);D是一个判别网络,用于判别一张图片是不是“真实的”。在理想的状态下,G可以生成足以“以假乱真”的图片G(z),而D难以判定G生成的图片究竟是不是真实的,即D(G(z))=0.5。这样就得到了一个优异的生成模型G,它可以用来生成图片。
将教师网络模型作为判别模型。生成模型根据随机信号生成一组图像,将这些图像输入教师网络模型,得到这些图片所属的类别。例如,教师网络模型的输出是图片属于每种类别的概率,且图片属于所有类别的概率之和为1。
教师网络模型仅能够对特定的类别的图像进行处理,将不属于教师网络模型能够处理的类别的图像输入教师网络模型,教师网络模型的输出结果依然是该图像属于教师网络模型能够处理的所有类别的概率。
图像属于某一类别的概率大于预设值,认为该图像属于该类别。如果预设值较高,图像属于该类的可能较高,将不属于该类的图像归为该类的可能性较小,但很多属于该类的图像因为教师网络模型判断属于该类的可能小于该预设值,而被认为不属于该类。也就是说,预设值较高,获得的用于训练学生网络模型的图像仅仅是教师网络模型能够处理的部分图像,通过这些图像训练的学生网络模型,不能反映教师网络模型的全部性能。
如果预设值较低,图像属于该类的可能较低,将不属于该类的图像归为该类的可能性较大。将教师网络模型不能够处理的图像用于学生网络模型的训练,不能准确地获得能够对特定类别的图像进行处理的学生网络模型。
因此,通过教师网络模型对图像的处理结果判断该图像是否属于教师网络模型能够处理的类别,判断结果并不准确,从而对学生网络模型的训练产生不利影响,压缩后的神经网络模型效果难以令人满意。
图4是本申请实施例提供的一种神经网络模型压缩方法的示意性流程图。
在步骤S301,服务器获取用户设备上传的第一神经网络模型和所述第一神经网络的训练数据。
所述第一神经网络的训练数据也可以称为第一数据。第一数据包括第一神经网络模型的训练数据中的全部或部分数据。
用户设备上传用于训练所述第一神经网络模型的训练数据中的部分训练数据,减小上传数据的数量,提高用户体验。
所述部分训练数据包括所述第一神经网络输出的多个类别中每个类别的数据。
所述第一神经网络模型用于将输入的数据分为N类中的至少一类。所述第一数据可以包括所述N类中每类的数据。也就是说,第一数据可以包括第一神经网络模型能够处理的类别中每个类别的数据。第一神经网络模型是需要进行压缩的神经网络模型。第一神经网络模型可以用于数据的分类。第一数据可以包括用于训练第一神经网络模型的所有数据所属的类别中的每个类别的数据。在本申请实施例中,正样本数据属于第一神经网络模型能够处理的类别。
用户上传的第一神经网络模型的训练数据包括第一神经网络能够处理的多类数据中每一类的数据,使得训练得到的第二神经网络能够对该多类数据进行处理,提高对第一神经网络模型压缩的准确性。
第一神经网络模型可以实现对输入数据的分类。例如,第一神经网络模型可以对文本、语音、图像、特征等进行分类,如可以对输入的一句话里每个词的词性进行分类(名词、动词…),或可以根据输入一段人的语音判断其此人说话时的情绪,或可以对输入的图片确定图片中的人或物体的类别,或对已经提取的特征进行分类等等。
在步骤S302,服务器根据所述第一神经网络的训练数据和所述服务器存储的未标记数据,利用正类与未标记(positive-unlabeled,PU)学习算法得到PU分类器。
与所述第一神经网络模型的训练数据具有相似属性与分布的数据可以称为正类数据或正样本数据。第一神经网络模型可以用于处理正样本数据。第一数据为正样本(positive samples)数据。与所述第一神经网络模型的训练数据具有相似属性与分布的数据可以称为正样本数据。
所述服务器存储的未标记数据也可以称为第二数据。第二数据可以包括正样本数据和正样本数据之外的数据,即负样本(negative samples)数据。未标记数据即尚不确定该数据是否为正样本数据。
服务器获取PU分类器,服务器可以接收用户设备或其他服务器发送的PU分类器,服务器也可以从存储器中获取PU分类器。参见图9,PU分类器的训练可以由其他装置进行。PU分类器的训练也可以由对第一神经网络模型进行压缩的服务器进行。
服务器可以根据所述第一数据、所述第二数据、所述第二数据中正样本数据的比例信息,训练所述PU分类器。
服务器可以通过PU学习算法训练PU分类器。PU学习算法的损失函数可以是所述第一数据和所述第二数据的训练损失的期望,所述比例信息用于计算所述期望。
服务器可以获取第二数据中正样本数据的比例信息,该比例也可以称为先验概率信息,用于指示第二数据中正样本数据的比例。
PU学习算法提供了一种半监督学习的方式,服务器能够根据已标记的正样本数据,对未标记数据进行分类,确定未标记数据中的正样本数据和负样本数据。其中,正样本数据可以是第一神经网络模型可以处理的数据,负样本数据是第一神经网络模型可以处理的数据之外的数据。正样本数据属于第一神经网络模型可以处理的类别,负样本数据不属于第一神经网络模型可以处理的类别。
利用PU学习算法,服务器可以将第一数据和第二数据输入待训练的PU分类器,调整待训练的PU分类器的参数,以使得PU分类器对第二数据的分类结果满足第二数据中正样本数据的比例信息,从而得到调整后的PU分类器。调整后的PU分类器即训练得到的PU分类器。PU学习算法的原理可以参见图6的说明。
在步骤S303,服务器使用所述PU分类器,从所述服务器存储的未标记数据中选择出扩展数据。
扩展数据为与所述第一神经网络模型的训练数据具有相似属性与分布的数据,即扩展数据为正类数据。
用于得到扩展数据的未标记数据与用于训练PU分类器的未标记数据可以相同,用于得到扩展数据的未标记数据也可以包括与训练PU分类器的未标记数据不同的数据。用于得到扩展数据的未标记数据可以称为第三数据。第三数据可以包括所述第二数据中的全部或部分数据,或者也可以是第二数据之外的其他数据。服务器存储的数据量有限,训练PU分类器和使用PU分类器时,输入PU分类器的数据相同,可以增加用于训练第二神经网络的数据量,提高训练第二神经网络的准确性。
将第三数据输入PU分类器,PU分类器对第三数据进行分类,确定其中的正样本数据和负样本数据。扩展数据即第三数据中的正样本数据。
PU分类器可以对数据进行分类,也可以对数据的特征进行分类。PU分类器的分类的对象与PU分类器的训练过程有关。
可以将第一数据和第二数据输入待训练的特征提取网络,以得到第一特征。特征提取网络可以用于对输入的数据进行特征的提取,以得到第一特征。将第一特征输入待训练的PU分类器。根据第二数据中正样本数据的比例信息和提取的第一特征,对待训练的特征提取网络和待训练的PU分类器的参数进行调整,以得到特征提取网络和PU分类器。
第一特征包括特征提取网络提取的第一数据中每个数据的特征和第二数据中每个数据的特征。
根据PU分类器,可以确定所存储的第三数据中的正样本数据,即扩展数据。可以将第三数据输入特征提取网络,以得到第二特征。可以将第二特征输入PU分类器,以确定扩展数据。
第一特征可以是对特征提取网络多个层的输出多个第三特征进行融合得到的。多个第三特征与特征提取网络多个层一一对应。所述第一权重调整是基于根据所述比例信息进行的。
第三特征包括特征提取网络提取的第一数据中每个数据的特征和第二数据中每个数据的特征。
特征提取网络可以是CNN网络。
特征提取网络中的参数较多,对特征提取网络进行参数的调整需要较多的数据,占用较长的时间。
可以将第一神经网络作为特征提取网络。可以将第一数据和第二数据输入第一神经网络模型以提取第一神经网络模型的多个层输出的多个第三特征,对多个第三特征进行融合以得到第一特征。根据第一特征和第二数据中正样本数据的比例信息,得到PU分类器。
也就是说,PU分类器是根据第一特征和第二数据中正样本数据的比例信息得到的。第一特征是基于多个第三特征融合获得的。多个第三特征是利用所述第一神经网络模型对第一数据和第二数据进行特征提取得到的。多个第三特征与所述第一神经网络的多个层一一对应。
可以利用所述第一神经网络模型对所述第三数据进行特征提取,以得到第二特征。第二特征是利用所述第一权重对多个第四特征融合得到的。所述多个第四特征与所述第一神经网络的多个层一一对应。将所述第二特征输入所述PU分类器,以确定所述扩展数据。
第一神经网络模型是训练得到的神经网络模型,用于处理正样本数据,因此能够提取对于正样本数据的分类有效的特征。将第一神经网络模型可以作为特征提取网络,可以减小PU学习算法中需要调整的参数的数量,提高云端存储的未标记数据中得到正样本数据的效率。
根据所述第一神经网络模型输出的特征,利用正类与未标记PU学习算法,可以训练PU分类器。
对文字、图像等数据进行分类,可以通过特征提取网络提取数据的特征。通过将第一神经网络模型作为特征提取网络,无需再对特征提取网络进行训练,减少对未标记的数据进行分类的时间,提取提高效率。
特征提取网络的多个层输出的多个特征。可以对特征提取网络的多个层输出的多个特征进行融合,将融合的结果输入PU分类器。可以将特征提取网络多个层输出的特征以相同的权重输入PU分类器。也可以通过注意力机制的多特征网络对多个层的特征的权重进行调整。
第一特征是利用经过第一权重调整之后的多个第三特征融合获得的,所述第一权重调整是基于根据第二数据中正样本数据的比例信息进行的。多个第三特征与特征提取网络的多个层一一对应。
也就是说,可以将多个第三特征进行融合,将融合后的特征输入待训练的PU分类器。调整多个第三特征中每个第三特征的权重,并调整待训练的PU分类器的参数,以使得PU分类器的分类结果满足第二数据中正样本数据的比例信息。也就是所,根据第二数据中正样本数据的比例信息,确定第一权重,得到PU分类器。
将扩展数据输入特征提取网络,特征提取网络的多个层输出多个第四特征。根据第一权重,对特征提取网络的多个层输出的多个第四特征进行融合。多个第四特征与特征提取网络的多个层一一对应。
第三特征与特征提取网络的层的对应关系,和第四特征与特征提取网络的层的对应关系相同。或者说,特征提取网络的多个层的输出与第一权重中的多个权重值一一对应。
融合的方式有多种。
组合(combine)是一种融合的方式。采用combine的方式进行融合,可以将待融合的特征直接相加,或者加权相加。加权相加,即乘以一定系数即权重值之后相加。也就是说,采用combine的方式,可以将通道维度(channel wise)进行线性组合。
可以将特征提取网络的多个层输出的多个特征相加,例如,可以将特征提取网络的多个层输出的多个特征直接相加,也可以将特征提取网络的多个层输出的多个特征按照一定权重相加。T1和T2分别表示特征提取网络的两个层输出的特征,可以用T3表示融合后的特征,T3=a×T1+b×T2,其中,a和b分别为计算T3时T1和T2乘的系数,即权重值,a≠0,且b≠0。
级联(concatenate)和通道融合(channel fusion)是另一种融合的方式。采用concatenate&channel fusion的方式,可以将待融合的特征的维数直接拼接,或者乘以一定系数即权重值之后进行拼接。
以特征提取网络为第一神经网络模型为例进行说明。可以在所述第一神经网络模型中添加带注意力机制的多特征网络,以得到第三神经网络模型。带注意力机制的多特征网络利用所述注意力机制用于对所述第一神经网络模型的不同层输出的多个特征进行选择。将所述已标记数据和所述未标记数据输入所述第三神经网络模型,并利用所述PU学习算法得到所述正样本数据。图7示出了一种特征融合的方式。
通过带注意力机制的多特征网络,对第一神经网络模型不同层提取的特征进行选择,提高对未分类数据进行分类的准确性。
在PU分类器的训练过程中,可以调整多特征网络和PU分类器的参数,以得到较为准确的PU分类器,从而得到从第三数据中准确选择扩展数据。也就是说,可以在半监督学习的过程中,对多特征网络和PU分类器的参数进行调整。
在步骤S304,服务器根据所述扩展数据,利用KD方法训练第二神经网络模型。
所述第一神经网络模型作为所述KD方法的教师网络模型,所述第二神经网络模型作为所述KD方法的学生网络模型。KD方法也可以称为KD算法。
服务器还可以根据所述扩展数据和用户上传的第一神经网络的训练数据,利用知识蒸馏KD方法训练第二神经网络模型。
将第一神经网络的训练数据作为用于训练第二神经网络模型的训练数据,增加训练的数据量,提高训练得到的第二神经网络模型的准确性。
扩展数据可以包括第一神经网络模型能够处理的多个类别。每个类别中的数据的数量可能差异较大。
将扩展数据输入所述第一神经网络模型和第二神经网络模型,根据KD方法的损失函数,调整第二神经网络模型的参数,以使得所述损失函数满足预设条件。例如,调整第二神经网络模型的参数,以使得KD方法的损失函数最小化。调整后的第二神经网络模型即对第一神经网络模型进行压缩的结果。
第一神经网络将所述训练数据分为N类数据,N为正整数,所述N类数据中的第一类数据中的每个数据在所述第一神经网络模型和所述原始神经网络模型的所述损失函数中的第二权重与所述第一类数据的数量负相关。
如果KD方法的损失函数中每个数据的第二权重相等,可能出现这样的情况:对应于第一神经网络模型分类结果的某个类别的数据很少,对于该类别的数据,第一神经网络模型和训练得到的第二神经网络模型的输出差异较大,但损失函数满足预设要求。这种情况,训练得到的第二神经网络模型不能有效处理该类别的数据,分类不准确,即对于第一神经网络模型的压缩结果并不准确。
在根据扩展数据对第一神经网络模型进行压缩时,可以考虑已标记数据和未标记数据中的正样本数据中,每个类别的数据的数量。对损失函数进行调整,使得损失函数中的各类扩展数据的第二权重与第三数据中不同类的数据的数量负相关。通过对KD方法的损失函数的调整,使得压缩后的神经网络模型即训练得到的第二神经网络模型能够对每个类别的数据均与第一神经网络模型相似,实现对第一神经网络模型的较好压缩效果。
将扩展数据输入所述第一神经网络模型,以对所述扩展数据进行分类,得到多类扩展数据以及所述多类扩展数据中每类扩展数据的第二权重。服务器最小化所述KD算法的损失函数,以得到训练后的第二神经网络模型。
服务器根据所述第三数据和所述第一神经网络模型,利用知识蒸馏KD方法,得到训练后的第二神经网络模型,其中,所述KD方法的损失函数为所述多类扩展数据中的各类扩展数据的训练误差与所述各类扩展数据的第二权重的乘积之和。
每类扩展数据的训练误差可以理解为该类扩展数据中每个数据的训练误差之和。
根据未标记数据中的正样本数据在第一神经网络模型能够处理的类别中的数量,调整KD方法的损失函数中每一个类的数据对应的第二权重,在正样本数据在不同的类别中分布不平衡的情况下,对于每一个类的数据,压缩后的神经网络模型均能够得到较好的分类结果。
由于第一神经网络模型对于数据的分类可能并不准确,第一神经网络模型对于数据的分类结果存在误差。因此,可以对损失函数中的不同类的第二权重进行随机扰动。调整第二神经网络模型的参数,以使得所述损失函数在不同扰动情况下满足预设条件。调整后的第二神经网络模型即压缩后的神经网络模型。
将扩展数据输入所述第一神经网络模型和第二神经网络模型,所述第一神经网络将第三数据分为N类扩展数据,N为正整数,所述N类扩展数据中的各类扩展数据在KD方法的损失函数中的第二权重与各类扩展数据的数量负相关;对KD方法的损失函数中的各类扩展数据对应的第二权重进行随机扰动;调整所述第二神经网络模型的参数,以使得所述损失函数在不同扰动情况下满足预设条件,以得到调整后的第二神经网络模型。
各类扩展数据的第二权重包括对所述各类扩展数据的初始权重进行随机扰动后得到的多个扰动后的权重,KD方法的损失函数包括所述多个扰动后的权重一一对应多个损失函数。所述各类扩展数据的初始权重与所述各类扩展数据的数量负相关。最小化所述多个损失函数的最大值,以得到训练后的第二神经网络模型。训练后的第二神经网络模型使得所述多个损失函数的最大值最小。对于损失函数中第二权重的确定和扰动,可以参见图8的说明。
通过对KD方法的损失函数中的初始权重进行随机扰动,扰动后的权重在所述已标记数据和所述正样本数据中不同类的数据的数量负相关的数值周围随机波动,调整原始神经网络模型的参数,以使得在多个扰动后的权重情况下损失函数均满足预设条件。将使得所述多个损失函数的最大值最小的调整后的原始神经网络模型作为压缩后的第二神经网络模型,减小了第一神经网络模型分类结果中的误差对压缩后的神经网络模型准确性的影响,提高压缩后的神经网络模型对数据的分类结果的准确性。
通过步骤S301-S304,利用PU分类器,根据用户上传的第一神经网络的训练数据,对服务器存储的未标记数据进行分类,可以从云端存储的未标记数据中得到第一神经网络模型能够处理的扩展数据,即与所述第一神经网络模型的训练数据具有相似属性与分布的数据。根据扩展数据,能够实现神经网络模型的压缩,从而降低神经网络模型压缩对已标记数据的数量的要求,在保证神经网络模型压缩准确性的同时,减少传输的数据量,提高用户体验。
图5是本申请实施例提供的一种神经网络模型压缩方法的示意图。以对图像进行分类的神经网络模型为例进行说明。
用户设备向云服务设备发送第一数据和第一神经网络模型。云服务设备也可以称为云服务器或服务器。发送第一神经网络模型,也可以理解为发送第一神经网络模型的参数。云服务设备可以将第一神经网络模型作为教师网络模型,对第一神经网络模型进行压缩。第一神经网络模型用于处理正样本数据。第一神经网络模型能够确定第一数据中每个图像的类别。第一数据可以是第一神经网络模型的原始训练数据中的部分数据。第一数据可以是是已标记数据,第一数据可以是正样本数据。第一数据可以包括第一神经网络模型的原始训练数据中每个类别中的至少一个图像。第一神经网络模型用于将输入的数据分为N类中的至少一类,N为正整数。第一数据包括所述N类中每类的数据。
云服务设备根据用户设备发送的第一数据和第一神经网络模型,以及云服务设备存储的云上数据,将第一神经网络模型作为教师网络模型,对第一神经网络模型进行压缩,以获取作为压缩后的第二神经网络模型。云上数据即云端存储的数据,包括至少一个图像。
云服务设备根据用户设备上传的第一数据、云上数据,训练第一神经网络模型对应的PU分类器。
云服务设备可以根据云上数据中的正样本数据,压缩第一神经网络。云服务设备使用PU分类器,确定云上数据中的正样本数据。
云服务设备可以采用KD方法,进行第一神经网路模型的压缩。云服务设备可以将第一神经网路模型作为教师网络模型,利用KD方法对学生网络模型进行训练。云服务设备可以将经过训练的学生网络模型作为压缩后的第二神经网路模型。训练之前的学生网络模型也可以称为原始神经网络模型或原始模型。
云服务设备将云上数据中的正样本数据输入第一神经网络模型和学生网络模型,根据第一神经网络模型和学生网络模型的输出,确定KD方法的损失函数LKD。KD方法的损失函数LKD可以表示为
第一神经网络模型的输出的取值范围可能在负无穷到正无穷。可以对第一神经网络模型和学生网络模型的输出结果取值范围进行相同方式的调整,从而减小输出结果的位宽,从而减小计算交叉熵损失的计算量。例如,可以通过进行softmax变换,将第一神经网络模型和学生网络模型的输出进行标准化。标准化,就是将输出结果调整在0和1之间。参见《Distilling the Knowledge in a Neural Network》(Hinton G,Vinyals O,DeanJ.Computer Science,2015,14(7):38-39.),可以将第一神经网络模型和学生网络模型的输出除以相同的温度参数,并通过归一化指数函数softmax进行变换,以得到软化的概率,根据这两个软化的概率分布,计算KD方法的损失函数。通过温度参数,可以调整输出结果在0和1之间的分布情况。
云服务设备可以通过调整学生网络模型,最小化损失函数KD方法的损失函数LKD,从而得到调整后的学生网络模型。调整后的学生网络模型即第二神经网络模型。
云服务设备可以将将压缩后的第二神经网络模型发送至用户设备。
用户设备根据压缩后的神经网络模型对图像进行处理,从而减小处理时间,并减小资源占用,提高处理效率。
第一神经网络模型对云上数据中的正样本数据进行分类,分类结果的多个类中,某一类的图像的数量可能很少,而其他类的图像的数量较多。对于最小化的KD方法的损失函数,如果分类结果的多个类中某一类的数据数量很少,对于该类数据,第二神经网络模型与第一神经网络模型的输出差异较大,则第二神经网络模型不能有效处理该类数据,神经网络压缩的准确性较低。参见图8,可以对KD方法的损失函数进行调整,以获得鲁棒的知识蒸馏方法。
图6是本申请实施例提供的一种扩展正样本数据的方法的示意图。
属于原始训练数据的类别的图像,为正样本数据。不属于原始训练数据的类别的图像,为负样本数据。正样本数据可以用于对第一神经网络模型进行压缩以得到第二神经网络模型。第一神经网络模型即教师网络模型,第二神经网络模型即学生网络模型,对第一神经网络模型进行压缩,以得到第二神经网络模型,可以理解为训练第二神经网络模型。因此,第一神经网络模型也可以称为预训练网络模型。
云服务设备获取第一数据。第一数据包括至少一个图像,至少一张图像中的每张图像均为正样本数据。第一数据可以包括用于训练第一神经网络模型的所有图像所属类中的每一个类的图像。第一数据可以是第一神经网络模型原始训练数据中的部分数据。
云服务设备基于第一数据和第一神经网络模型,以及第二数据,对正样本数据进行扩展。第二数据为云上的大量的未标记数据,第二数据包括多个图像。
通过PU学习算法,根据第一数据、第二数据以及第二数据中正样本数据的比例信息,进行半监督的学习,并通过此学习过程生成PU分类器,以便为大量未标记数据进行标注。正类与未标记(positive-unlabeled,PU)分类器的训练数据包括第一数据、第二数据以及第二数据中正样本数据的比例信息。第一数据为正样本数据,第二数据为未标记数据。贝叶斯决策论在相关概率已知的情况下利用误判损失来选择最优的类别分类。
将第一数据和第二数据输入第一神经网络模型。第一神经网络模型对第一数据和第二数据中每张图像进行特征的提取,待训练的PU分类器根据第一神经网络模型提取的特征对图像进行分类,将输入的图像分类为正样本数据或负样本数据。
可以根据第二数据中正样本数据的比例信息对PU分类器的参数进行调整,以保证输出结果的准确性。根据第二数据中正样本数据的比例信息可以确定PU学习算法的损失函数。也就是说,可以根据PU学习算法的损失函数对PU分类器的参数进行调整。
第一数据中的每张图像的标签均为正样本数据,假设未标记数据中的每张图像的标签均为负样本数据。
xi为PU分类器的训练样本,其中,X表示xi的训练样本的集合,Rd表示图像的表示方式。yi为xi对应的标签,例如,yi∈Y={-1,1},Y表示标签的集合,“+1”表示正样本数据,“-1”表示负样本数据。xi∈T,T表示训练样本的集合,T可以表示为:
其中,L为已标记数据集,即第一数据;U为未标记数据集,即第二数据;nl为第一数据的数量,nu为第二数据的数量。第一数据均为正样本数据,标签为“+1”,第二数据种的未标记数据的标签可以表示为yu,yu∈Y={-1,1},即yu为未标记数据的真实标签。
定义决策函数f和判别式函数F。决策函数f用于表示输入的图像xi与PU分类器的输出zi之间的关系,即f:xi→zi。根据PU分类器的输出zi,可以确定图像xi对应的标签yi,例如,yi与zi的对应关系可以是,当zi>0时yi=1,当zi≤0时yi=-1。判别式函数F用于表示输入的图像xi与标签yi的关系,即F:xi→yi。
对于传统二分类问题,考虑训练数据中正样本数据和负样本数据的比例,损失函数可以表示为训练数据的训练损失的期望(mean)。期望也可以称为数学期望,是试验中每次可能结果的概率乘以其结果的总和。可以根据通过如下损失函数对二分类器的输出进行调整:
其中,表示传统二分类器将训练数据中正样本数据分类为负样本数据带来的损失,表示传统二分类器将训练数据中负样本数据分类为正样本数据带来的损失,πp表示正样本数据的先验概率,πn表示负样本数据的先验概率,πp与πn具有如下关系:
πp+πn=1
在传统二分类中,若训练数据是从自然数据中完全随机挑选出来的,则可以用训练数据中正样本数据的比例代表正样本数据的先验概率,用负样本数据的比例代表负样本数据的先验概率。也就是说,πp和πn可以是统计概率。当然,πp和πn也可以是对第二数据中正类样本的比例进行估计得到的。
输入PU分类器的数据中,我们假设第二数据即未标记数据是从自然数据中完全随机挑选出来的,因此可以估计未标记数据中正样本数据的比例,该比例也可以称为正样本数据的先验概率,用πp表示。但由于输入PU分类器的数据中不存在标记的负样本数据,因此无法直接得到Rpn(f)表达式中的第二项。
PU分类器也可以根据图像中提取的特征对图像进行分类。PU分类器将第二数据中正样本数据分类为负样本数据的可能性与PU分类器将第一数据中正样本数据分类为负样本数据的可能性相同。相同也可以是近似相同。
第一数据中均为正样本数据。根据PU的输出确定的第一数据中的图像的标签与标签“+1”进行比较,确定PU分类器对第一数据的分类结果带来的损失。标签“+1”为正样本数据的标签。
其中,l为损失函数,l(F(x),+1)用于确定根据PU分类器得到的一个正样本数据的标签带来的损失;Ep为风险函数,用于确定PU分类器对第一数据的整体分类结果带来的损失,Ep可以是根据PU分类器得到的第一数据中每一个数据的标签带来的损失的和。
定义Rx(f)为第二数据即未标记数据对应的损失。假设未标记数据为负样本数据。则PU分类器将未标记数据中的图像分为正样本数据,就存在损失。Rx(f)表示PU分类器将未标记数据中的图像分类为正样本数据的损失。将根据PU分类器的输出确定的未标记数据的标签与标签“-1”进行比较,确定PU分类器对未标记数据的分类带来的损失Rx(f)。标签“-1”为负样本数据的标签。
Rx(f)可以表示为:
Rx(f)=Ex[l(F(x),-1)]
其中,Ex为风险函数,用于确定PU分类器对未标记数据的整体分类结果带来的损失,Ex可以是根据PU分类器得到的每一个图像的标签带来的损失的和。Ep与Ex的下标p、x仅仅表示损失的来源,Ep与Ex的计算方式相同,即Ep与Ex表达式可以相同。
未标记数据包括正样本数据和负样本数据。因此,Rx(f)包括将未标记数据中正样本数据分类为正样本数据带来的损失,以及将未标记数据中负样本数据分类为正样本数据带来的损失。Rx(f)可以表示为
其中,
表示PU分类器将未标记数据中正样本数据分类为正样本数据带来的损失,表示PU分类器将未标记数据中负样本数据分类为正样本数据带来的损失。仅在未标记数据中存在负样本数据,因此,PU算法的损失函数可以表示为:
表达式中,第一项可以通过决策函数f在已标记数据集L上的分类损失表示,第二项可以通过决策函数f在未标记数据集U上的分类损失以及决策函数f在已标记数据集U上的分类损失表示。第一项可以表示决策函数f在已标记数据集L上的分类损失,第二项可以表示决策函数f在未标记数据集U上的分类损失。两项之和即为决策函数f在训练样本的集合T上的整体损失。
假设T代表了真实数据的分布,因此最小化决策函数f在T上的损失,就代表了最小化决策函数f在真实数据上的损失,就意味着学习到了最优的决策函数f。
可以将扩展数据作为对第一神经网络模型压缩的训练数据的一部分。
图7是本申请实施例提供的一种带注意力机制的多特征融合模型的示意图。
PU分类器根据第一神经网络模型提取的特征对输入第一神经网络模型的数据进行分类。第一神经网络模型提取的特征可以包括第一神经网络模型的一个或多个层的输出,例如可以是最后一个层的输出,也可以是对多个层的输出进行融合或其他变换的结果。
在训练PU分类器时,可以将第一数据和第二数据输入第一神经网络模型。将第一神经网络模型多个层提取的特征输入多特征模型。多特征模型对第一神经网络模型多个层提取的特征进行处理,得到变换后的特征。将变换后的特征输入PU分类器。PU分类器对融合后的特征进行处理,以获取图像的标签。
融合的方式可以有多种。
组合(combine)是一种融合的方式。采用combine的方式进行融合,可以将待融合的特征直接相加,或者加权相加。加权相加,即乘以一定系数即权重值之后相加。也就是说,采用combine的方式,可以将通道维度(channel wise)进行线性组合。特征中的每一位乘以的权重值可以相同或不同。
级联(concatenate)和通道融合(channel fusion)是另一种融合的方式。采用concatenate&channel fusion的方式,可以将待融合的特征的维数直接拼接,或者乘以一定系数即权重值之后进行拼接。特征中的每一位乘以的权重值可以相同或不同。
可以通过注意力机制,对多个层提取的特征进行调整,将调整后的特征作为PU分类器的输入。
可以对第一神经网络多个层的输出进行全局平均池化,以得到与该多个层的输出一一对应的多个特征值。可以将该多个特征值进行拼接,得到特征描述子o。
池化(pooling)也称为欠采样或下采样。主要用于特征降维,压缩数据和参数的数量,减小过拟合,同时提高模型的容错性。全局平均池化不以窗口的形式取均值,而是以一个层的输出为单位求平均值。即一个层的输出转换为一个值。全局平均池化的计算结果为一个层的输出矩阵中每个点的平均值。
对特征描述子o进行注意力机制变化机制,得到对应的特征o'。
例如,特征描述子o的权重参数w可以表示为:
w=Attention(o,W)=σ(W2δ(W1o))
其中,δ、σ为非线性变换函数,W1、W2为两个全连接的参数。W1、W2是两个通过机器学习得到的参数矩阵,通过W1、W2进行线性变化。δ例如可以是线性整流单元(rectifiedlinear unit,ReLU)。
通过线性变化与非线性变换的组合,使用注意力机制挑选特征。对于给定的输入数据,网络在多层之间输出了不同的特征,这些特征分别代表了原始数据在不同层面上的表达。例如,输入一辆汽车的图片,底层网络输出的特征为比较通用的像是边缘线条、轮廓等基本的特征,高层网络输出的则是和该图片高度相关的例如车轮、车牌等特征。通过注意力机制,可以挑选哪些层的特征对输出结果更加重要。
对于特征描述子中的第j个特征值oj,其权重参数为wj,多特征模型的输出中与特征值oj对应的字段oj'为:
oj'=wjoj
其中,wj的值的大小表示了对应特征的重要性。参数wj数值越大代表对应的特征越重要。
图8是本申请实施例提供的一种知识蒸馏的方法的示意图。
该知识蒸馏的方法可以称为鲁棒的知识蒸馏(robust knowledge distillation,RKD)的方法。
通过PU分类器对第三数据进行分类,得到扩展数据。扩展数据存在数据量不平衡的问题。第一神经网络模型对扩展数据进行分类。扩展数据即服务器存储的未标记数据中与所述第一神经网络模型的训练数据具有相似属性与分布的数据。扩展数据可以包括云上的未标记数据中部分数据,也可以包括用户设备上传的第一神经网络模型的训练数据。扩展数据中,某一类的图像的数量可能很少,而其他类的图像的数量较多。对于图像的数量较少的类,应当尽可能保证第二神经网络模型与第一神经网络模型的输出相同,从而使得第二神经网络模型更好的学习到第一神经网络模型的相关特性,即第二神经网络模型与第一神经网络模型能够实现更加接近的功能。
例如,在扩展后的正样本数据的5005个训练图像中,对于其中的3个图像,训练得到的第二神经网络模型的输出和第一神经网络模型的分类结果不同,对于其他图像,训练得到的第二神经网络模型的输出和第一神经网络模型的分类结果相同。在这5005个训练图像中,有5张属于第一类,5000张属于第二类。如果这3张图像均属于第一类,则训练得到的第二神经网络模型对于第一类图像有60%的结果与第一神经网络模型的分类结果不同,训练得到的第二神经网络模型与第一神经网络模型相关特性差异较大,对于这类图像的分类的准确性较低,即神经网络模型压缩的准确性较低。而如果这3张图像均属于第二类,则训练得到的第二神经网络模型对于第二类图像有0.06%的结果与第一神经网络模型的分类结果不同,训练得到的第二神经网络模型与第一神经网络模型相关特性差异很小,神经网络模型压缩的准确性较高。
为了解决由于正样本数据在不同的类中的数据不平衡造成的神经网络模型压缩不准确的问题,可以利用第一神经网络模型对扩展数据进行分类。根据分类结果,可以对KD方法的损失函数中数据进行权重的调整,使得数量越少的类别中的每个数据对应的权重越大,即KD方法的损失函数中第k类的权重与所述已标记数据和所述正样本数据中不同类的数据的数量负相关。
另外,在数据量较大的某一类别中,数据量的少量增加或减小,对神经网络模型压缩准确性的影响较小。而在数据量较小的某一类别中,数据量的少量增加或减小,对神经网络模型压缩准确性可能也会产生很大的影响。因此,可以对KD方法的损失函数中数据进行权重的调整,使得数据量越少的类别中数据量的变化使得权重的变化越大,而数据量越多的类别中数据量的变化使得权重的变化较小。
K个类的中每个类的权重形成权重向量:
KD方法的损失函数可以表示为:
其中,Lc表示第i数据的损失,n为输入的数据量,
wi表示是第i数据所属的类别对应的权重,
可以是第一神经网络模型输出的向量,向量中的每一位与一个第一神经网络模型能够处理的类别对应,表示输入图片属于该类别的概率。或者,可以是软化的概率分布,即可以是对第一神经网络模型输出的向量进行标准化得到的,即,
与传统的知识蒸馏方法的不同,在本申请实施例中,计算KD方法的损失函数时添加了参数wi,对于每个训练数据添加对应的权重来区分不同数据的重要性。其中,数据如果越稀少,它的权重就越大,表示它的重要性越高。一旦这种稀少的数据被分类错误,就会给分类器施加一个较大的惩罚。相应的,数量越多的数据,对于分类器的重要性也就越低。
对权重或数据量yk进行随机扰动,得到多个扰动后的权重。第一神经网络模型对扩展数据的分类结果中各类的数据量与扩展数据实际所属的各类的数据量差异较小。因此,权重与实际的权重情况差异不会太大。设定一个预设范围,对yk的数量进行调整的范围小于该预设范围,例如,对于每个yk,增加或减少的数量或比例小于预设值,或权重增加或减少的数量或比例小于预设值。
扰动后的权重向量组成权重向量矩阵:
W={wkd_1,wkd_2,...,wkd_n}
其中,wkd_1,wkd_2,...,wkd_n分别为对应于N种扰动情况的权重向量。
基于这个权重矩阵,对原始神经网络模型的参数进行调整,训练第二神经网络模型。通过求解如下方程,可以得到最终优化后的第二神经网络模型Nst:
权重向量为w时,第二神经网络模型Nst与第一神经网络模型输出结果之间的损失函数为:
LKD(Nst,w)
对于N种扰动情况,即N中不同的w,损失函数的最大值可以表示为:
调整第二神经网络模型的参数,以使得不同的扰动情况下损失函数的最大值尽可能的减小,即最小化KD方法损失函数的最大值。可以将调整后的学生网络模型Nst作为压缩后的第二神经网络模型。
对于不同的扰动情况,KD方法的损失函数的最大值小于预设值,即对于不同的扰动,第二神经网络模型与第一神经网络模型对图像的处理结果均不会出现较大的差异。因此,通过上述方式,可以降低正样本数据扩展引起的噪声对神经网络模型压缩的影响,提高压缩后的神经网络模型的准确性。
图9是本申请实施例提供的一种神经网络压缩的方法的示意性流程图。
第一服务器获取用户设备上传的第一数据。第一数据为第一神经网络模型的训练数据。第一服务器可以根据所述第一数据、所存储的第二数据,训练PU分类器。
第二服务器使用PU分类器对所存储的第三数据进行选择,以得到扩展数据。
第二服务器获取用户设备上传的第一神经网络模型,根据扩展数据,利用KD方法,训练第二神经网络模型。第一神经网络模型作为KD方法的教师网络模型,以得到压缩后的作为KD方法的教师网络模型。
第二服务器可以向用户设备发送第二神经网络模型。
第一数据是第一神经网络模型的全部或部分训练数据。第一数据包括第一神经网络模型输出的多个类别中每个类别的数据。第一服务器可以根据PU算法,训练PU分类器,PU算法的原理可以参见图5的说明。
属于第一神经网络模型输出的类别的数据可以称为正样本数据。第二数据包括未标记数据。第二数据包括正样本数据和正样本数据之外的数据。正样本数据具有与与所述第一神经网络模型的训练数据具有相似属性与分布。
神经网络模型压缩过程可以由两个专用服务器进行。两个服务器中的一个服务器可以专用于集中化处理PU分类器的训练,另一个服务器可以专用于根据获取神经网络模型和对应的PU分类器,进行正样本数据的选择和知识蒸馏过程。对PU分类器的训练和知识蒸馏过程进行集中化处理,可以提高神经网络模型压缩的速度。
对本申请实施例提供的神经网络模型压缩方法进行验证。在数据集ImageNet、Flicker1M和修改后的国家标准技术研究数据集(modified national institute ofstandards and technology database,MNIST)上进行了测试。
表1是对ImageNet数据集中的图片进行分类得到的测试结果。
ResNet-34网络作为用户上传的第一神经网络模型,对于ResNet-34网络能够处理的每个类别,随机在加拿大前沿研究院公布的数据集(Canadian institute for advancedresearch,CIFAR-10)中选择nl张图片,将选择的所有类的图片作为用户上传的第一数据,并将ImageNet数据集作为云上的未标记数据集,ResNet-18网络作为学生网络模型。
表1
其中,教师网络模型为未压缩的第一神经网络模型,KD为使用知识蒸馏方法在完整的CIFAR-10上训练得到的分类结果。人工分类方法,通过人工分类,从ImageNet数据集上选取正样本数据,并将数据送入RKD训练第二神经网络模型。PU-s1为通过PU方法选取正样本数据,并将全部数据送入RKD训练第二神经网络模型。PU-s2为通过PU方法选取正样本数据,并随机挑选和原始训练集CIFAR-10数量相同的训练数据,并送入RKD训练第二神经网络模型。结果显示,本申请实施例提供的方法对图片分类的准确率甚至高于手工挑选数据的方法。
表2是对Flicker1M数据集中的图片进行分类得到的测试结果。
ResNet-34网络作为用户上传的第一神经网络模型,对于ResNet-34网络能够处理的每个类别,随机在ImageNet数据集中选择nl张图片,将选择的所有类的图片作为用户上传的第一数据,并将Flicker1M数据集作为云上的未标记数据集,ResNet-18网络作为学生网络模型。
表2
方法 | n<sub>t</sub> | 数据来源 | Top-1acc(%) | Top-5acc(%) |
教师网络模型 | 1,281,167 | 原始数据 | 73.27 | 91.26 |
KD-all | 1,281,167 | 原始数据 | 68.67 | 88.76 |
KD-500k | 500,000 | 原始数据 | 63.90 | 85.88 |
PU-s1 | 690,978 | PU数据 | 61.92 | 86.00 |
PU-s1 | 500,000 | PU数据 | 61.21 | 85.33 |
其中,教师网络模型为未压缩的第一神经网络模型,KD-all为使用知识蒸馏方法在完整的ImageNet数据集上训练得到的分类结果,KD-500k为使用知识蒸馏方法并在ImageNet数据集上随机挑选50万数据训练得到的分类结果。PU-s1和PU-s2同上。Top-1acc(%)表示在预测的标签向量中取概率最大的1个作为预测结果,若结果和真实标签相同则为正确。Top-5acc(%)表示在预测的标签向量中取概率最大的5个作为预测结果,若结果中的任意一个和真实标签相同则为正确。结果显示,本申请实施例提供的方法确定的正样本数据相比于使用原始数据集中的训练数据得到的第二神经网络模型的分类结果的top5准确率更高。
表3是对EMNIST数据集中的图片进行分类得到的测试结果。
卷积神经网络模型LeNet-5网络作为教师网络模型,从MNIST数据集中教师网络模型能够处理的每类数据中分别随机挑选1,2,5,10,20张图片,组成第一数据。EMNIST数据集作为云上未标记数据。将LeNet-5网络所有层的通道层减少一半作为第二神经网络模型。
表3
方法 | 1 | 2 | 5 | 10 | 20 |
PU-s1 | 98.5 | 98.7 | 98.7 | 98.8 | 98.9 |
PU-s1 | 98.3 | 98.5 | 98.5 | 98.6 | 98.6 |
结果显示,随着第一数据中每一类的图片数量增加,神经网络模型压缩的准确率提高。即使在第一数据中每一类的图片数量很小(仅使用1张)的情况下,神经网络模型压缩也能够达到较好的效果(准确率高于98%)。
图10是本申请实施例提供的一种电子装置的示意性结构图。装置800包括获取模块801,处理模块802。
获取模块801,用于获取用户设备上传的第一神经网络模型和所述第一神经网络的训练数据;
处理模块802,用于根据所述第一神经网络的训练数据和所述服务器存储的未标记数据,利用正类与未标记PU学习算法得到PU分类器。
处理模块802还用于,使用所述PU分类器,从所述服务器存储的未标记数据中选择出扩展数据,所述扩展数据为与所述第一神经网络模型的训练数据具有相似属性与分布的数据。
处理模块802还用于,根据所述扩展数据,利用知识蒸馏KD方法训练第二神经网络模型,所述第一神经网络模型作为所述KD方法的教师网络模型,所述第二神经网络模型作为所述KD方法的学生网络模型。
可选地,处理模块802还用于,根据所述第一神经网络的训练数据、所述服务器存储的未标记数据和比例信息,利用正类与未标记PU学习算法得到PU分类器,所述PU学习算法的损失函数为所述第一神经网络的训练数据和所述服务器存储的未标记数据的训练损失的期望,所述比例信息用于指示所述扩展数据在所述服务器存储的未标记数据中的占比,所述比例信息用于计算所述期望。
可选地,所述PU分类器是根据第一特征和所述比例信息得到的,所述第一特征是基于多个第三特征融合获得的,所述多个第三特征是利用所述第一神经网络模型对所述第一神经网络的训练数据和所述服务器存储的未标记数据进行特征提取得到的,所述多个第三特征与所述第一神经网络的多个层一一对应。
处理模块802还用于,利用所述第一神经网络模型对所述服务器存储的未标记数据进行特征提取,以得到第二特征;
处理模块802还用于,将所述第二特征,输入所述PU分类器,以确定所述扩展数据。
可选地,所述第一特征是利用经过第一权重调整之后的多个第三特征融合获得的,所述第一权重调整是基于根据所述比例信息进行的,所述第二特征是利用所述第一权重对多个第四特征融合得到的,所述多个第四特征与所述第一神经网络的多个层一一对应。
可选地,所述第一神经网络模型的训练数据为用于训练所述第一神经网络模型的训练数据中的部分训练数据。
可选地,所述部分训练数据包括所述第一神经网络输出的多个类别中每个类别的数据。
可选地,处理模块802还用于,将所述扩展数据输入所述第一神经网络模型,以对所述扩展数据进行分类,得到多类扩展数据以及所述多类扩展数据中每类扩展数据的第二权重。
处理模块802还用于,最小化所述KD方法的损失函数,以得到训练后的第二神经网络模型,其中,所述KD方法的损失函数为所述多类扩展数据中的各类扩展数据的训练误差与所述各类扩展数据的第二权重的乘积之和。
可选地,所述各类扩展数据的第二权重包括对所述各类扩展数据的初始权重进行随机扰动后得到的多个扰动后的权重,所述KD方法的损失函数包括所述多个扰动后的权重一一对应多个损失函数,其中,所述各类扩展数据的初始权重与所述各类扩展数据的数量负相关,
处理模块802还用于,最小化所述多个损失函数的最大值,以得到训练后的第二神经网络模型。
图11是本申请实施例提供的一种电子装置的硬件结构示意图。图10所示的电子装置1000(该装置1000具体可以是一种计算机设备)包括存储器1001、处理器1002、通信接口1003以及总线1004。其中,存储器1001、处理器1002、通信接口1003通过总线1004实现彼此之间的通信连接。
存储器1001可以是只读存储器(read only memory,ROM),静态存储设备,动态存储设备或者随机存取存储器(random access memory,RAM)。存储器1001可以存储程序,当存储器1001中存储的程序被处理器1002执行时,处理器1002和通信接口1003用于执行本申请实施例的神经网络模型压缩的方法的各个步骤。
处理器1002可以采用通用的中央处理器(central processing unit,CPU),微处理器,应用专用集成电路(application specific integrated circuit,ASIC),图形处理器(graphics processing unit,GPU)或者一个或多个集成电路,用于执行相关程序,以实现本申请实施例的神经网络模型压缩的装置中的单元所需执行的功能,或者执行本申请方法实施例的神经网络模型压缩的方法。
处理器1002还可以是一种集成电路芯片,具有信号的处理能力。在实现过程中,本申请的神经网络模型压缩的方法的各个步骤可以通过处理器1002中的硬件的集成逻辑电路或者软件形式的指令完成。上述的处理器1002还可以是通用处理器、数字信号处理器(digital signal processing,DSP)、专用集成电路(ASIC)、现成可编程门阵列(fieldprogrammable gate array,FPGA)或者其他可编程逻辑器件、分立门或者晶体管逻辑器件、分立硬件组件。可以实现或者执行本申请实施例中的公开的各方法、步骤及逻辑框图。通用处理器可以是微处理器或者该处理器也可以是任何常规的处理器等。结合本申请实施例所公开的方法的步骤可以直接体现为硬件译码处理器执行完成,或者用译码处理器中的硬件及软件模块组合执行完成。软件模块可以位于随机存储器,闪存、只读存储器,可编程只读存储器或者电可擦写可编程存储器、寄存器等本领域成熟的存储介质中。该存储介质位于存储器1001,处理器1002读取存储器1001中的信息,结合其硬件完成本申请实施例的神经网络模型压缩的装置中包括的单元所需执行的功能,或者执行本申请方法实施例的神经网络模型压缩的方法。
通信接口1003使用例如但不限于收发器一类的收发装置,来实现装置1000与其他设备或通信网络之间的通信。例如,可以通过通信接口1003获取第一数据、第二数据、所述第二数据中正样本数据的比例信息、第一神经网络模型的参数、PU分类器中的一种或多种。
总线1004可包括在装置1000各个部件(例如,存储器1001、处理器1002、通信接口1003)之间传送信息的通路。
本申请实施例还提供一种神经网络模型压缩装置,其特征在于,包括:至少一个处理器和通信接口,所述通信接口用于所述神经网络模型压缩装置与其他装置进行信息交互,当程序指令在所述至少一个处理器中执行时,使得所述神经网络模型压缩装置执行上文中的方法。
本申请实施例还提供一种计算机可读介质,其特征在于,所述计算机可读介质具有程序指令,当所述程序指令被直接或者间接执行时,使得前文中的方法得以实现。
本申请实施例还提供一种芯片,其特征在于,所述芯片包括至少一个处理器,当程序指令在所述至少一个处理器中执行时,使得前文中的方法得以实现。
本领域普通技术人员可以意识到,结合本文中所公开的实施例描述的各示例的单元及算法步骤,能够以电子硬件、或者计算机软件和电子硬件的结合来实现。这些功能究竟以硬件还是软件方式来执行,取决于技术方案的特定应用和设计约束条件。专业技术人员可以对每个特定的应用来使用不同方法来实现所描述的功能,但是这种实现不应认为超出本申请的范围。
所属领域的技术人员可以清楚地了解到,为描述的方便和简洁,上述描述的系统、装置和单元的具体工作过程,可以参考前述方法实施例中的对应过程,在此不再赘述。
在本申请所提供的几个实施例中,应该理解到,所揭露的系统、装置和方法,可以通过其它的方式实现。例如,以上所描述的装置实施例仅仅是示意性的,例如,所述单元的划分,仅仅为一种逻辑功能划分,实际实现时可以有另外的划分方式,例如多个单元或组件可以结合或者可以集成到另一个系统,或一些特征可以忽略,或不执行。另一点,所显示或讨论的相互之间的耦合或直接耦合或通信连接可以是通过一些接口,装置或单元的间接耦合或通信连接,可以是电性,机械或其它的形式。
所述作为分离部件说明的单元可以是或者也可以不是物理上分开的,作为单元显示的部件可以是或者也可以不是物理单元,即可以位于一个地方,或者也可以分布到多个网络单元上。可以根据实际的需要选择其中的部分或者全部单元来实现本实施例方案的目的。
另外,在本申请各个实施例中的各功能单元可以集成在一个处理单元中,也可以是各个单元单独物理存在,也可以两个或两个以上单元集成在一个单元中。
所述功能如果以软件功能单元的形式实现并作为独立的产品销售或使用时,可以存储在一个计算机可读取存储介质中。基于这样的理解,本申请的技术方案本质上或者说对现有技术做出贡献的部分或者该技术方案的部分可以以软件产品的形式体现出来,该计算机软件产品存储在一个存储介质中,包括若干指令用以使得一台计算机设备(可以是个人计算机,服务器,或者网络设备等)执行本申请各个实施例所述方法的全部或部分步骤。而前述的存储介质包括:U盘、移动硬盘、只读存储器(Read-Only Memory,ROM)、随机存取存储器(Random Access Memory,RAM)、磁碟或者光盘等各种可以存储程序代码的介质。
以上所述,仅为本申请的具体实施方式,但本申请的保护范围并不局限于此,任何熟悉本技术领域的技术人员在本申请揭露的技术范围内,可轻易想到变化或替换,都应涵盖在本申请的保护范围之内。因此,本申请的保护范围应以所述权利要求的保护范围为准。
Claims (18)
1.一种神经网络模型压缩的方法,其特征在于,包括:
服务器获取用户设备上传的第一神经网络模型和所述第一神经网络模型的训练数据;
所述服务器根据所述第一神经网络模型的训练数据和所述服务器存储的未标记数据,利用正类与未标记PU学习算法得到PU分类器;
所述服务器使用所述PU分类器,从所述服务器存储的未标记数据中选择出扩展数据,所述扩展数据为与所述第一神经网络模型的训练数据具有相似属性与分布的数据;
所述服务器根据所述扩展数据,利用知识蒸馏KD方法训练第二神经网络模型,所述第一神经网络模型作为所述KD方法的教师网络模型,所述第二神经网络模型作为所述KD方法的学生网络模型。
2.根据权利要求1所述的方法,其特征在于,所述服务器根据所述第一神经网络的训练数据和所述服务器存储的未标记数据,利用正类与未标记PU学习算法得到PU分类器,包括:
所述服务器根据所述第一神经网络的训练数据、所述服务器存储的未标记数据和比例信息,利用正类与未标记PU学习算法得到所述PU分类器,所述PU学习算法的损失函数为所述第一神经网络的训练数据和所述服务器存储的未标记数据的训练损失的期望,所述比例信息用于指示所述扩展数据在所述服务器存储的未标记数据中的占比,所述比例信息用于计算所述期望。
3.根据权利要求2所述的方法,其特征在于,
所述PU分类器是根据第一特征和所述比例信息得到的,所述第一特征是基于多个第三特征融合获得的,所述多个第三特征是利用所述第一神经网络模型对所述第一神经网络的训练数据和所述服务器存储的未标记数据进行特征提取得到的,所述多个第三特征与所述第一神经网络的多个层一一对应,
所述服务器使用所述PU分类器,从所述服务器存储的未标记数据中选择出扩展数据,包括:
所述服务器利用所述第一神经网络模型对所述服务器存储的未标记数据进行特征提取,以得到第二特征;
所述服务器将所述第二特征,输入所述PU分类器,以确定所述扩展数据。
4.根据权利要求3所述的方法,其特征在于,所述第一特征是利用经过第一权重调整之后的多个第三特征融合获得的,所述第一权重调整是基于根据所述比例信息进行的,所述第二特征是利用所述第一权重对多个第四特征融合得到的,所述多个第四特征与所述第一神经网络的所述多个层一一对应。
5.根据权利要求1-4中任一项所述的方法,其特征在于,所述第一神经网络模型的训练数据为用于训练所述第一神经网络模型的训练数据中的部分训练数据。
6.根据权利要求5所述的方法,其特征在于,所述部分训练数据包括所述第一神经网络输出的多个类别中每个类别的数据。
7.根据权利要求1-6中任一项所述的方法,其特征在于,所述服务器根据所述扩展数据,利用KD方法训练第二神经网络模型,包括:
所述服务器将所述扩展数据输入所述第一神经网络模型,以对所述扩展数据进行分类,得到多类扩展数据以及所述多类扩展数据中每类扩展数据的第二权重;
所述服务器最小化所述KD方法的损失函数,以得到训练后的第二神经网络模型,其中,所述KD方法的损失函数为所述多类扩展数据中的各类扩展数据的训练误差与所述各类扩展数据的第二权重的乘积之和。
8.根据权利要求7所述的方法,其特征在于,所述各类扩展数据的第二权重包括对所述各类扩展数据的初始权重进行随机扰动后得到的多个扰动后的权重,所述KD方法的损失函数包括所述多个扰动后的权重一一对应多个损失函数,其中,所述各类扩展数据的初始权重与所述各类扩展数据的数量负相关,
所述服务器最小化所述KD方法的损失函数,以得到训练后的第二神经网络模型,包括:所述服务器最小化所述多个损失函数的最大值,以得到训练后的第二神经网络模型。
9.一种神经网络模型压缩的装置,其特征在于,包括:
获取模块,用于获取用户设备上传的第一神经网络模型和所述第一神经网络的训练数据;
处理模块,用于:
根据所述第一神经网络的训练数据和所述服务器存储的未标记数据,利用正类与未标记PU学习算法得到PU分类器;
使用所述PU分类器,从所述服务器存储的未标记数据中选择出扩展数据,所述扩展数据为与所述第一神经网络模型的训练数据具有相似属性与分布的数据;
根据所述扩展数据,利用知识蒸馏KD方法训练第二神经网络模型,所述第一神经网络模型作为所述KD方法的教师网络模型,所述第二神经网络模型作为所述KD方法的学生网络模型。
10.根据权利要求9所述的装置,其特征在于,
所述处理模块还用于,根据所述第一神经网络的训练数据、所述服务器存储的未标记数据和比例信息,利用正类与未标记PU学习算法得到所述PU分类器,所述PU学习算法的损失函数为所述第一神经网络的训练数据和所述服务器存储的未标记数据的训练损失的期望,所述比例信息用于指示所述扩展数据在所述服务器存储的未标记数据中的占比,所述比例信息用于计算所述期望。
11.根据权利要求10所述的装置,其特征在于,
所述PU分类器是根据第一特征和所述比例信息得到的,所述第一特征是基于多个第三特征融合获得的,所述多个第三特征是利用所述第一神经网络模型对所述第一神经网络的训练数据和所述服务器存储的未标记数据进行特征提取得到的,所述多个第三特征与所述第一神经网络的多个层一一对应,
所述处理模块还用于:
利用所述第一神经网络模型对所述服务器存储的未标记数据进行特征提取,以得到第二特征;
将所述第二特征,输入所述PU分类器,以确定所述扩展数据。
12.根据权利要求11所述的装置,其特征在于,所述第一特征是利用经过第一权重调整之后的多个第三特征融合获得的,所述第一权重调整是基于根据所述比例信息进行的,所述第二特征是利用所述第一权重对多个第四特征融合得到的,所述多个第四特征与所述第一神经网络的多个层一一对应。
13.根据权利要求9-12中任一项所述的装置,其特征在于,所述第一神经网络模型的训练数据为用于训练所述第一神经网络模型的训练数据中的部分训练数据。
14.根据权利要求13所述的装置,其特征在于,所述部分训练数据包括所述第一神经网络输出的多个类别中每个类别的数据。
15.根据权利要求9-14中任一项所述的装置,其特征在于,所述处理模块还用于:
将所述扩展数据输入所述第一神经网络模型,以对所述扩展数据进行分类,得到多类扩展数据以及所述多类扩展数据中每类扩展数据的第二权重;
最小化所述KD方法的损失函数,以得到训练后的第二神经网络模型,其中,所述KD方法的损失函数为所述多类扩展数据中的各类扩展数据的训练误差与所述各类扩展数据的第二权重的乘积之和。
16.根据权利要求15所述的装置,其特征在于,所述各类扩展数据的第二权重包括对所述各类扩展数据的初始权重进行随机扰动后得到的多个扰动后的权重,所述KD方法的损失函数包括所述多个扰动后的权重一一对应多个损失函数,其中,所述各类扩展数据的初始权重与所述各类扩展数据的数量负相关,
所述处理模块还用于,最小化所述多个损失函数的最大值,以得到训练后的第二神经网络模型。
17.一种计算机可读介质,其特征在于,所述计算机可读介质存储用于设备执行的程序代码,该程序代码包括用于执行如权利要求1-8中任一项所述的训练方法。
18.一种芯片,其特征在于,包括至少一个处理器,当程序指令在所述至少一个处理器中执行时,所述至少一个处理器用于执行如权利要求1-8中任一项的方法被执行。
Priority Applications (3)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN201910833833.9A CN112446476A (zh) | 2019-09-04 | 2019-09-04 | 神经网络模型压缩的方法、装置、存储介质和芯片 |
PCT/CN2020/097957 WO2021042828A1 (zh) | 2019-09-04 | 2020-06-24 | 神经网络模型压缩的方法、装置、存储介质和芯片 |
US17/680,630 US20220180199A1 (en) | 2019-09-04 | 2022-02-25 | Neural network model compression method and apparatus, storage medium, and chip |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN201910833833.9A CN112446476A (zh) | 2019-09-04 | 2019-09-04 | 神经网络模型压缩的方法、装置、存储介质和芯片 |
Publications (1)
Publication Number | Publication Date |
---|---|
CN112446476A true CN112446476A (zh) | 2021-03-05 |
Family
ID=74734048
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN201910833833.9A Pending CN112446476A (zh) | 2019-09-04 | 2019-09-04 | 神经网络模型压缩的方法、装置、存储介质和芯片 |
Country Status (3)
Country | Link |
---|---|
US (1) | US20220180199A1 (zh) |
CN (1) | CN112446476A (zh) |
WO (1) | WO2021042828A1 (zh) |
Cited By (16)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN112990296A (zh) * | 2021-03-10 | 2021-06-18 | 中科人工智能创新技术研究院(青岛)有限公司 | 基于正交相似度蒸馏的图文匹配模型压缩与加速方法及系统 |
CN113219357A (zh) * | 2021-04-28 | 2021-08-06 | 东软睿驰汽车技术(沈阳)有限公司 | 电池包健康状态计算方法、系统及电子设备 |
CN113298710A (zh) * | 2021-05-10 | 2021-08-24 | 天津大学 | 基于外部注意力机制的光学相干层析超分辨率成像方法 |
CN113326852A (zh) * | 2021-06-11 | 2021-08-31 | 北京百度网讯科技有限公司 | 模型训练方法、装置、设备、存储介质及程序产品 |
CN113343803A (zh) * | 2021-05-26 | 2021-09-03 | 北京百度网讯科技有限公司 | 模型训练方法、装置、设备和存储介质 |
CN113642727A (zh) * | 2021-08-06 | 2021-11-12 | 北京百度网讯科技有限公司 | 神经网络模型的训练方法和多媒体信息的处理方法、装置 |
CN113743474A (zh) * | 2021-08-10 | 2021-12-03 | 扬州大学 | 基于协同半监督卷积神经网络的数字图片分类方法与系统 |
CN113780468A (zh) * | 2021-09-28 | 2021-12-10 | 中国人民解放军国防科技大学 | 一种基于少量神经元连接的健壮模型训练方法 |
CN114202733A (zh) * | 2022-02-18 | 2022-03-18 | 青岛海信网络科技股份有限公司 | 一种基于视频的交通故障检测方法及设备 |
CN114422515A (zh) * | 2021-12-13 | 2022-04-29 | 广东电网有限责任公司 | 一种适配电力行业的边缘计算架构设计方法及系统 |
CN114466082A (zh) * | 2022-01-29 | 2022-05-10 | 上海阵量智能科技有限公司 | 数据压缩、数据解压方法、系统及人工智能ai芯片 |
CN115169556A (zh) * | 2022-07-25 | 2022-10-11 | 美的集团(上海)有限公司 | 模型剪枝方法及装置 |
WO2023024406A1 (zh) * | 2021-08-27 | 2023-03-02 | 上海商汤智能科技有限公司 | 数据蒸馏的方法、装置、设备、存储介质、计算机程序及产品 |
WO2023113693A3 (en) * | 2021-12-17 | 2023-10-05 | Lemon Inc. | Optimal knowledge distillation scheme |
CN117437459A (zh) * | 2023-10-08 | 2024-01-23 | 昆山市第一人民医院 | 基于决策网络实现用户膝关节髌骨软化状态分析方法 |
CN113642727B (zh) * | 2021-08-06 | 2024-05-28 | 北京百度网讯科技有限公司 | 神经网络模型的训练方法和多媒体信息的处理方法、装置 |
Families Citing this family (11)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US11640527B2 (en) * | 2019-09-25 | 2023-05-02 | Salesforce.Com, Inc. | Near-zero-cost differentially private deep learning with teacher ensembles |
EP3997625A4 (en) * | 2019-10-29 | 2022-11-09 | Samsung Electronics Co., Ltd. | ELECTRONIC DEVICE AND ASSOCIATED CONTROL METHOD |
US20220301012A1 (en) * | 2021-03-18 | 2022-09-22 | At&T Intellectual Property I, L.P. | Apparatuses and methods for facilitating a generation and use of models |
CN113240580B (zh) * | 2021-04-09 | 2022-12-27 | 暨南大学 | 一种基于多维度知识蒸馏的轻量级图像超分辨率重建方法 |
CN113793265A (zh) * | 2021-09-14 | 2021-12-14 | 南京理工大学 | 一种基于深度特征关联性的图像超分辨率方法及系统 |
KR102500341B1 (ko) | 2022-02-10 | 2023-02-16 | 주식회사 노타 | 신경망 모델에 대한 정보를 제공하는 방법 및 이를 수행하는 전자 장치 |
CN114817742B (zh) * | 2022-05-18 | 2022-09-13 | 平安科技(深圳)有限公司 | 基于知识蒸馏的推荐模型配置方法、装置、设备、介质 |
CN115481692B (zh) * | 2022-09-23 | 2023-10-10 | 江苏安控智汇科技股份有限公司 | 一种基于sgan的水泵机组故障诊断方法 |
CN115859220B (zh) * | 2022-12-23 | 2023-08-25 | 北京瑞莱智慧科技有限公司 | 数据处理方法、相关装置及存储介质 |
CN115861997B (zh) * | 2023-02-27 | 2023-05-16 | 松立控股集团股份有限公司 | 一种关键前景特征引导知识蒸馏的车牌检测识别方法 |
CN116843553B (zh) * | 2023-07-11 | 2024-01-02 | 太原理工大学 | 一种基于核不确定学习和退化嵌入的盲超分辨率重建方法 |
Family Cites Families (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US10402701B2 (en) * | 2017-03-17 | 2019-09-03 | Nec Corporation | Face recognition system for face recognition in unlabeled videos with domain adversarial learning and knowledge distillation |
CN110059717A (zh) * | 2019-03-13 | 2019-07-26 | 山东大学 | 用于乳腺钼靶数据集的卷积神经网络自动分割方法及系统 |
CN110059189B (zh) * | 2019-04-11 | 2021-03-05 | 厦门点触科技股份有限公司 | 一种游戏平台消息的分类系统及方法 |
-
2019
- 2019-09-04 CN CN201910833833.9A patent/CN112446476A/zh active Pending
-
2020
- 2020-06-24 WO PCT/CN2020/097957 patent/WO2021042828A1/zh active Application Filing
-
2022
- 2022-02-25 US US17/680,630 patent/US20220180199A1/en active Pending
Cited By (24)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN112990296B (zh) * | 2021-03-10 | 2022-10-11 | 中科人工智能创新技术研究院(青岛)有限公司 | 基于正交相似度蒸馏的图文匹配模型压缩与加速方法及系统 |
CN112990296A (zh) * | 2021-03-10 | 2021-06-18 | 中科人工智能创新技术研究院(青岛)有限公司 | 基于正交相似度蒸馏的图文匹配模型压缩与加速方法及系统 |
CN113219357A (zh) * | 2021-04-28 | 2021-08-06 | 东软睿驰汽车技术(沈阳)有限公司 | 电池包健康状态计算方法、系统及电子设备 |
CN113298710A (zh) * | 2021-05-10 | 2021-08-24 | 天津大学 | 基于外部注意力机制的光学相干层析超分辨率成像方法 |
CN113298710B (zh) * | 2021-05-10 | 2024-04-16 | 天津大学 | 基于外部注意力机制的光学相干层析超分辨率成像方法 |
CN113343803A (zh) * | 2021-05-26 | 2021-09-03 | 北京百度网讯科技有限公司 | 模型训练方法、装置、设备和存储介质 |
CN113343803B (zh) * | 2021-05-26 | 2023-08-22 | 北京百度网讯科技有限公司 | 模型训练方法、装置、设备和存储介质 |
CN113326852A (zh) * | 2021-06-11 | 2021-08-31 | 北京百度网讯科技有限公司 | 模型训练方法、装置、设备、存储介质及程序产品 |
CN113642727A (zh) * | 2021-08-06 | 2021-11-12 | 北京百度网讯科技有限公司 | 神经网络模型的训练方法和多媒体信息的处理方法、装置 |
CN113642727B (zh) * | 2021-08-06 | 2024-05-28 | 北京百度网讯科技有限公司 | 神经网络模型的训练方法和多媒体信息的处理方法、装置 |
CN113743474A (zh) * | 2021-08-10 | 2021-12-03 | 扬州大学 | 基于协同半监督卷积神经网络的数字图片分类方法与系统 |
CN113743474B (zh) * | 2021-08-10 | 2023-09-26 | 扬州大学 | 基于协同半监督卷积神经网络的数字图片分类方法与系统 |
WO2023024406A1 (zh) * | 2021-08-27 | 2023-03-02 | 上海商汤智能科技有限公司 | 数据蒸馏的方法、装置、设备、存储介质、计算机程序及产品 |
CN113780468A (zh) * | 2021-09-28 | 2021-12-10 | 中国人民解放军国防科技大学 | 一种基于少量神经元连接的健壮模型训练方法 |
CN114422515A (zh) * | 2021-12-13 | 2022-04-29 | 广东电网有限责任公司 | 一种适配电力行业的边缘计算架构设计方法及系统 |
CN114422515B (zh) * | 2021-12-13 | 2023-09-15 | 广东电网有限责任公司 | 一种适配电力行业的边缘计算架构设计方法及系统 |
WO2023113693A3 (en) * | 2021-12-17 | 2023-10-05 | Lemon Inc. | Optimal knowledge distillation scheme |
CN114466082A (zh) * | 2022-01-29 | 2022-05-10 | 上海阵量智能科技有限公司 | 数据压缩、数据解压方法、系统及人工智能ai芯片 |
CN114466082B (zh) * | 2022-01-29 | 2024-01-09 | 上海阵量智能科技有限公司 | 数据压缩、数据解压方法、系统及人工智能ai芯片 |
CN114202733A (zh) * | 2022-02-18 | 2022-03-18 | 青岛海信网络科技股份有限公司 | 一种基于视频的交通故障检测方法及设备 |
CN115169556B (zh) * | 2022-07-25 | 2023-08-04 | 美的集团(上海)有限公司 | 模型剪枝方法及装置 |
CN115169556A (zh) * | 2022-07-25 | 2022-10-11 | 美的集团(上海)有限公司 | 模型剪枝方法及装置 |
CN117437459B (zh) * | 2023-10-08 | 2024-03-22 | 昆山市第一人民医院 | 基于决策网络实现用户膝关节髌骨软化状态分析方法 |
CN117437459A (zh) * | 2023-10-08 | 2024-01-23 | 昆山市第一人民医院 | 基于决策网络实现用户膝关节髌骨软化状态分析方法 |
Also Published As
Publication number | Publication date |
---|---|
US20220180199A1 (en) | 2022-06-09 |
WO2021042828A1 (zh) | 2021-03-11 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN112446476A (zh) | 神经网络模型压缩的方法、装置、存储介质和芯片 | |
CN110378381B (zh) | 物体检测方法、装置和计算机存储介质 | |
CN110188795B (zh) | 图像分类方法、数据处理方法和装置 | |
CN112446270B (zh) | 行人再识别网络的训练方法、行人再识别方法和装置 | |
CN110084281B (zh) | 图像生成方法、神经网络的压缩方法及相关装置、设备 | |
CN110532871B (zh) | 图像处理的方法和装置 | |
CN110222717B (zh) | 图像处理方法和装置 | |
CN112639828A (zh) | 数据处理的方法、训练神经网络模型的方法及设备 | |
CN111291809B (zh) | 一种处理装置、方法及存储介质 | |
US20220215227A1 (en) | Neural Architecture Search Method, Image Processing Method And Apparatus, And Storage Medium | |
CN112446398A (zh) | 图像分类方法以及装置 | |
CN112529146B (zh) | 神经网络模型训练的方法和装置 | |
CN113705769A (zh) | 一种神经网络训练方法以及装置 | |
CN110222718B (zh) | 图像处理的方法及装置 | |
CN112668366B (zh) | 图像识别方法、装置、计算机可读存储介质及芯片 | |
CN111882031A (zh) | 一种神经网络蒸馏方法及装置 | |
CN113326930A (zh) | 数据处理方法、神经网络的训练方法及相关装置、设备 | |
CN111797882A (zh) | 图像分类方法及装置 | |
CN113191489B (zh) | 二值神经网络模型的训练方法、图像处理方法和装置 | |
CN112818764A (zh) | 一种基于特征重建模型的低分辨率图像人脸表情识别方法 | |
CN112464930A (zh) | 目标检测网络构建方法、目标检测方法、装置和存储介质 | |
CN113536970A (zh) | 一种视频分类模型的训练方法及相关装置 | |
CN110705564B (zh) | 图像识别的方法和装置 | |
CN114492634A (zh) | 一种细粒度装备图片分类识别方法及系统 | |
CN112132207A (zh) | 基于多分支特征映射目标检测神经网络构建方法 |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination |