CN111667004B - 数据生成方法、装置、设备以及存储介质 - Google Patents

数据生成方法、装置、设备以及存储介质 Download PDF

Info

Publication number
CN111667004B
CN111667004B CN202010506988.4A CN202010506988A CN111667004B CN 111667004 B CN111667004 B CN 111667004B CN 202010506988 A CN202010506988 A CN 202010506988A CN 111667004 B CN111667004 B CN 111667004B
Authority
CN
China
Prior art keywords
model
data set
training data
target task
training
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.)
Active
Application number
CN202010506988.4A
Other languages
English (en)
Other versions
CN111667004A (zh
Inventor
希滕
张刚
温圣召
Current Assignee (The listed assignees may be inaccurate. Google has not performed a legal analysis and makes no representation or warranty as to the accuracy of the list.)
Xiaogan Sichuang Information Technology Co ltd
Original Assignee
Xiaogan Sichuang Information Technology Co ltd
Priority date (The priority date 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 date listed.)
Filing date
Publication date
Application filed by Xiaogan Sichuang Information Technology Co ltd filed Critical Xiaogan Sichuang Information Technology Co ltd
Priority to CN202010506988.4A priority Critical patent/CN111667004B/zh
Publication of CN111667004A publication Critical patent/CN111667004A/zh
Application granted granted Critical
Publication of CN111667004B publication Critical patent/CN111667004B/zh
Active legal-status Critical Current
Anticipated expiration legal-status Critical

Links

Classifications

    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06FELECTRIC DIGITAL DATA PROCESSING
    • G06F18/00Pattern recognition
    • G06F18/20Analysing
    • G06F18/21Design or setup of recognition systems or techniques; Extraction of features in feature space; Blind source separation
    • G06F18/214Generating training patterns; Bootstrap methods, e.g. bagging or boosting
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/04Architecture, e.g. interconnection topology
    • G06N3/045Combinations of networks
    • GPHYSICS
    • G06COMPUTING; CALCULATING OR COUNTING
    • G06NCOMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
    • G06N3/00Computing arrangements based on biological models
    • G06N3/02Neural networks
    • G06N3/08Learning methods

Landscapes

  • Engineering & Computer Science (AREA)
  • Theoretical Computer Science (AREA)
  • Physics & Mathematics (AREA)
  • Data Mining & Analysis (AREA)
  • Life Sciences & Earth Sciences (AREA)
  • Artificial Intelligence (AREA)
  • General Physics & Mathematics (AREA)
  • General Engineering & Computer Science (AREA)
  • Evolutionary Computation (AREA)
  • Biophysics (AREA)
  • Computational Linguistics (AREA)
  • Software Systems (AREA)
  • Mathematical Physics (AREA)
  • Health & Medical Sciences (AREA)
  • Biomedical Technology (AREA)
  • Computing Systems (AREA)
  • Molecular Biology (AREA)
  • General Health & Medical Sciences (AREA)
  • Evolutionary Biology (AREA)
  • Bioinformatics & Cheminformatics (AREA)
  • Bioinformatics & Computational Biology (AREA)
  • Computer Vision & Pattern Recognition (AREA)
  • Image Analysis (AREA)

Abstract

本申请实施例公开了数据生成方法、装置、设备以及存储介质,涉及人工智能、深度学习和图像处理技术领域。该方法的一具体实施方式包括:获取目标任务的第一训练数据集;构建生成式对抗网络的搜索空间;在搜索空间中搜索生成式对抗网络的模型结构;利用第一训练数据集对生成式对抗网络的模型结构进行训练,得到收敛生成式对抗网络;利用收敛生成式对抗网络生成目标任务的第二训练数据集。该实施方式通过自动搜索的方式生成训练数据,能够应用于训练数据量稀缺的场景,降低了生成训练数据的成本。

Description

数据生成方法、装置、设备以及存储介质
技术领域
本申请实施例涉及计算机技术领域,具体涉及人工智能、深度学习和图像处理技术领域,尤其涉及数据生成方法、装置、设备以及存储介质。
背景技术
深度学习技术在很多方向上都取得了巨大的成功。在深度学习技术中,训练数据对模型的性能和效果有着非常重要的影响。通常,深度神经网络的训练过程中的训练数据量会影响模型的收敛速度和精度。因此,对于训练数据稀缺的场景,通常需要花费高昂的成本来生成训练数据。
发明内容
本申请实施例提出了数据生成方法、装置、设备以及存储介质。
第一方面,本申请实施例提出了一种数据生成方法,包括:获取目标任务的第一训练数据集;构建生成式对抗网络的搜索空间;在搜索空间中搜索生成式对抗网络的模型结构;利用第一训练数据集对生成式对抗网络的模型结构进行训练,得到收敛生成式对抗网络;利用收敛生成式对抗网络生成目标任务的第二训练数据集。
第二方面,本申请实施例提出了一种数据生成装置,包括:获取模块,被配置成获取目标任务的第一训练数据集;构建模块,被配置成构建生成式对抗网络的搜索空间;搜索模块,被配置成在搜索空间中搜索生成式对抗网络的模型结构;训练模块,被配置成利用第一训练数据集对生成式对抗网络的模型结构进行训练,得到收敛生成式对抗网络;生成模块,被配置成利用收敛生成式对抗网络生成目标任务的第二训练数据集。
第三方面,本申请实施例提出了一种电子设备,包括:至少一个处理器;以及与至少一个处理器通信连接的存储器;其中,存储器存储有可被至少一个处理器执行的指令,指令被至少一个处理器执行,以使至少一个处理器能够执行如第一方面中任一实现方式描述的方法。
第四方面,本申请实施例提出了一种存储有计算机指令的非瞬时计算机可读存储介质,计算机指令用于使计算机执行如第一方面中任一实现方式描述的方法。
第五方面,本申请实施例提供了一种计算机程序产品,包括计算机程序,计算机程序在被处理器执行时实现根据如第一方面中任一实现方式描述的方法。
本申请实施例提供的数据生成方法、装置、设备以及存储介质,首先获取目标任务的第一训练数据集;之后构建生成式对抗网络的搜索空间;而后在搜索空间中搜索生成式对抗网络的模型结构;然后利用第一训练数据集对生成式对抗网络的模型结构进行训练,得到收敛生成式对抗网络;最后利用收敛生成式对抗网络生成目标任务的第二训练数据集。通过自动搜索的方式生成训练数据,能够应用于训练数据量稀缺的场景,降低了生成训练数据的成本。此外,在第一训练数据集是图像集的情况下,该方法还可以应用于图像处理领域,从搜索空间搜索出GAN的较优模型结构,进而利用图像集训练出来扩充样本图像的收敛GAN。
从搜索空间中搜索GAN的模型结构,能够快速地确定出GAN的较优的模型结构。仅对从搜索空间中搜索到的模型结构进行训练,即可生成能够用于扩充训练数据的收敛GAN。无需对GAN的所有可能的模型结构进行训练,即可得到效果较优的收敛GAN,使得执行该方法的硬件减少了在模型训练中的运算量,提高了硬件的处理速度。进而,降低了模型训练对硬件环境的要求,使得该方法能够适配于更低的硬件配置的任务执行端设备,并且能够确保获取到效果较优的收敛GAN。
应当理解,本部分所描述的内容并非旨在标识本公开的实施例的关键或重要特征,也不用于限制本公开的范围。本公开的其它特征将通过以下的说明书而变得容易理解。
附图说明
通过阅读参照以下附图所作的对非限制性实施例所作的详细描述,本申请的其它特征、目的和优点将会变得更明显。附图用于更好地理解本方案,不构成对本申请的限定。其中:
图1是本申请可以应用于其中的示例性系统架构;
图2是根据本申请的数据生成方法的一个实施例的流程图;
图3是根据本申请的数据生成方法的又一个实施例的流程图;
图4是可以实现本申请实施例的数据生成方法的场景图;
图5是根据本申请的数据生成装置的一个实施例的结构示意图;
图6是用来实现本申请实施例的数据生成方法的电子设备的框图。
具体实施方式
以下结合附图对本申请的示范性实施例做出说明,其中包括本申请实施例的各种细节以助于理解,应当将它们认为仅仅是示范性的。因此,本领域普通技术人员应当认识到,可以对这里描述的实施例做出各种改变和修改,而不会背离本申请的范围和精神。同样,为了清楚和简明,以下的描述中省略了对公知功能和结构的描述。
需要说明的是,在不冲突的情况下,本申请中的实施例及实施例中的特征可以相互组合。下面将参考附图并结合实施例来详细说明本申请。
图1示出了可以应用本申请的数据生成方法或数据生成装置的实施例的示例性系统架构100。
如图1所示,系统架构100中可以包括存储设备101、网络102和服务器103。网络102用以在存储设备101和服务器103之间提供通信链路的介质。网络102可以包括各种连接类型,例如有线、无线通信链路或者光纤电缆等等。
存储设备101可以通过网络102与服务器103交互,以接收或发送消息等。存储设备101中可以提供目标任务的第一训练数据集,包括但不限于数据库、用户终端等等。
服务器103可以是具有模型计算能力的独立服务器,也可以是具有计算能力的处理模块。例如,服务器103可以是模型服务器。模型服务器可以对从存储设备101获取到的第一训练数据集等数据进行分析等处理,并生成处理结果(例如第二训练数据集)。
需要说明的是,服务器103可以是硬件,也可以是软件。当服务器103为硬件时,可以实现成多个服务器组成的分布式服务器集群,也可以实现成单个服务器。当服务器103为软件时,可以实现成多个软件或软件模块(例如用来提供分布式服务),也可以实现成单个软件或软件模块。在此不做具体限定。
需要说明的是,本申请实施例所提供的数据生成方法一般由服务器103执行,相应地,数据生成装置一般设置于服务器103中。
应该理解,图1中的存储设备、网络和服务器的数目仅仅是示意性的。根据实现需要,可以具有任意数目的存储设备、网络和服务器。
继续参考图2,其示出了根据本申请的数据生成方法的一个实施例的流程200。该数据生成方法包括以下步骤:
步骤201,获取目标任务的第一训练数据集。
在本实施例中,数据生成方法的执行主体(例如图1所示的服务器103)可以获取目标任务的第一训练数据集。
其中,目标任务可以是任意特定任务,例如,人体检测人任务、人脸识别任务、目标跟踪任务等等。通常,利用特定的训练数据集可以训练出特定的模型,来执行特定的任务。例如,利用标注人体位置的训练数据集可以训练出人体检测模型,人体检测模型可以执行人体检测任务。
其中,目标任务的第一训练数据集可以包括通过传统方式收集的训练数据。以训练数据是图像为例,传统方式可以包括但不限于利用摄像头采集、从网络下载等方式。而通过传统方式获取的训练数据量通常非常有限。由于模型训练过程中的训练数据对模型的收敛速度和精度至关重要,若仅利用训练数据量有限的第一训练数据集训练模型,训练出的模型的收敛速度较慢,且精度较低。
步骤202,构建生成式对抗网络的搜索空间。
在本实施例中,上述执行主体可以构建GAN(Generative Adversarial Networks,生成式对抗网络)的搜索空间。
通常,上述执行主体可以基于预设的搜索空间设计规则来构建GAN的搜索空间。其中,预设的搜索空间设计规则可以用于指导构建适用于本申请确定GAN的最优模型结构的搜索空间。为实现这一目的,搜索空间中至少应包括GAN的多个模型结构,这是能够确定GAN的较优模型结构的基础。在常规情况下,搜索空间中可以包括GAN的所有可能的模型结构。应当理解的是,搜索空间设计规则可以表现为以文字和字符的组合,也可以直接为便于计算机识别和调用的代码,此处不做具体限定。
步骤203,在搜索空间中搜索生成式对抗网络的模型结构。
在本实施例中,上述执行主体可以在搜索空间中搜索GAN的模型结构。
其中,在搜索空间中搜索出的GAN的模型结构通常是GAN的最优模型结构。在自动搜索过程中,在搜索空间中得到GAN的一个模型结构,然后用这个模型结构在训练集上训练,并在验证集上测试这个模型结构的效果(例如精度)。随后,将这个模型结构的效果回传给搜索空间,在搜索空间中得到GAN的另一个模型结构,继续进行训练。如此反复进行直至得到最优的模型结构。
其中,GAN是一种深度学习模型,由两大模块组成,即生成模型(GenerativeModel,简称为G模型)和判别模型(Discriminative Model,简称为D模型)。给定一系列的数据(例如图像),G模型能够生成新的数据。给定一个数据,D模型能够判定其真实性。通过G模型和D模型的相互博弈学习,能够产生相当好的输出。需要说明的是,在搜索空间中可以同时搜索GAN的G模型和D模型,也可以仅搜索GAN的G模型。也就是说,搜索空间中的GAN的模型结构可以既包括生成模型又包括判别模型,也可以仅包括生成模型。
步骤204,利用第一训练数据集对生成式对抗网络的模型结构进行训练,得到收敛生成式对抗网络。
在本实施例中,上述执行主体可以利用第一训练数据集对GAN的模型结构进行训练,得到收敛GAN。
在训练过程中,G模型的目标就是尽量生成真实的数据去欺骗D模型。而D模型的目标就是尽量把G模型生成的数据和真实的数据区分开来。这样,G模型和D模型就构成了一个动态的博弈过程。理想状态下的博弈结果是,G模型可以生成足以以假乱真的数据。对于D模型来说,它难以判定G模型生成的数据究竟是不是真实的。这样,就得到了一个收敛GAN。
步骤205,利用收敛生成式对抗网络生成目标任务的第二训练数据集。
在本实施例中,上述执行主体可以利用收敛GAN生成目标任务的第二训练数据集。其中,用于生成第二训练数据集的具体是收敛GAN中的G模型。
本申请实施例提供的数据生成方法,首先获取目标任务的第一训练数据集;之后构建生成式对抗网络的搜索空间;而后在搜索空间中搜索生成式对抗网络的模型结构;然后利用第一训练数据集对生成式对抗网络的模型结构进行训练,得到收敛生成式对抗网络;最后利用收敛生成式对抗网络生成目标任务的第二训练数据集。通过自动搜索的方式生成训练数据,能够应用于训练数据量稀缺的场景,降低了生成训练数据的成本。此外,在第一训练数据集是图像集的情况下,该方法还可以应用于图像处理领域,从搜索空间搜索出GAN的较优模型结构,进而利用图像集训练出来扩充样本图像的收敛GAN。
从搜索空间中搜索GAN的模型结构,能够快速地确定出GAN的较优的模型结构。仅对从搜索空间中搜索到的模型结构进行训练,即可生成能够用于扩充训练数据的收敛GAN。无需对GAN的所有可能的模型结构进行训练,即可得到效果较优的收敛GAN,使得执行该方法的硬件减少了在模型训练中的运算量,提高了硬件的处理速度。进而,降低了模型训练对硬件环境的要求,使得该方法能够适配于更低的硬件配置的任务执行端设备,并且能够确保获取到效果较优的收敛GAN。
进一步参考图3,其示出了根据本申请的数据生成方法的又一个实施例的流程300。该数据生成方法包括以下步骤:
步骤301,获取目标任务的第一训练数据集、测试数据集和模型结构。
在本实施例中,数据生成方法的执行主体(例如图1所示的服务器103)可以获取目标任务的第一训练数据集、测试数据集和模型结构。
需要说明的是,目标任务的测试数据集也可以是通过传统方式收集的测试数据。目标任务的模型结构可以是为了完成目标任务而创建的多个相互关联和相互作用的网络层,包括但不限于输入层、卷积层、池化层、输出层等等。
步骤302,构建生成式对抗网络的搜索空间。
在本实施例中,步骤302具体操作已在图2所示的实施例中步骤202进行了详细的介绍,在此不再赘述。
步骤303,根据搜索空间初始化模型编码生成器。
在本实施例中,上述执行主体可以根据搜索空间初始化模型编码生成器。其中,搜索空间中的模型编码生成器可以用于生成GAN的模型结构的模型编码。
通常,模型编码生成器中的参数可以用一些不同的小随机数进行初始化。小随机数可以用来保证模型编码生成器不会因权值过大而进入饱和状态,从而导致训练失败,不同随机数可以用来保证模型编码生成器正常地学习。
步骤304,根据模型编码生成器生成模型编码。
在本实施例中,上述执行主体可以根据模型编码生成器生成模型编码。这里,一种参数的模型编码生成器可以生成一个模型编码,一个模型编码可以对应GAN的一个模型结构。
步骤305,在搜索空间中将模型编码解码成生成式对抗网络的模型结构。
在本实施例中,上述执行主体可以在搜索空间中将模型编码解码成GAN的模型结构。这里,采用模型编码生成器的编码方式对应的解码方式进行解码,一个模型编码可以解码成一个模型结构。
步骤306,利用第一训练数据集对生成式对抗网络的模型结构进行训练,得到收敛生成式对抗网络。
步骤307,利用收敛生成式对抗网络生成目标任务的第二训练数据集。
在本实施例中,步骤306-307具体操作已在图2所示的实施例中步骤204-205进行了详细的介绍,在此不再赘述。
步骤308,利用第一训练数据集和第二训练数据集对目标任务的模型结构进行训练,得到目标任务模型。
在本实施例中,上述执行主体可以利用第一训练数据集和第二训练数据集对目标任务的模型结构进行训练,得到目标任务模型。这里,利用第二训练数据集扩充用于模型训练的数据量,从而提升模型收敛速度和精度。
步骤309,利用测试数据集对目标任务模型进行测试,得到目标任务模型的性能。
在本实施例中,上述执行主体可以利用测试数据集对目标任务模型进行测试,得到目标任务模型的性能。而目标任务模型的性能可以作为激励。
步骤310,基于目标任务模型的性能更新模型编码生成器。
在本实施例中,上述执行主体可以基于目标任务模型的性能更新模型编码生成器。通常,更新的模型编码生成器可以用于再次生成一个新的模型编码。
步骤311,确定模型编码生成器的迭代次数是否小于预设迭代次数。
在本实施例中,上述执行主体可以确定模型编码生成器的迭代次数是否小于预设迭代次数。若小于预设迭代次数,返回继续执行步骤304。即,利用第一训练数据集和最新生成的第二训练数据集再次训练目标任务模型,以及基于再次训练的目标任务模型的性能再次更新所述模型编码器。若不小于预设迭代次数,继续执行步骤312。即,利用最新生成的第二训练数据集扩充用于模型训练的数据量。此外,将利用第一训练数据集和最新生成的第二训练数据集训练得到目标任务模型,能够作为最终的目标任务模型。其中,最终得到的目标任务模型的性能与迭代次数相关,迭代次数越多,目标任务模型的性能越高,因此预设迭代次数可以是根据对目标任务模型的性能的实际需求来确定。通过设置合理的预设迭代次数,从而确保得到的目标任务模型的性能。
步骤312,利用最新生成的第二训练数据集扩充用于模型训练的数据量。
在本实施例中,若模型编码生成器的迭代次数不小于预设迭代次数,上述执行主体可以利用最新生成的第二训练数据集扩充用于模型训练的数据量。
从图3中可以看出,与图2对应的实施例相比,本实施例中的数据生成方法的流程300突出了自动搜索步骤。由此,本实施例描述的方案增强了生成训练数据与目标任务之间的关联性,将生成的训练数据在目标任务模型的性能作为指导生成GAN的模型结构的激励,从而提升利用加入生成的训练数据训练出的目标任务模型的性能。
为了便于理解,下面提供可以实现本申请实施例的数据生成方法的场景,具体如图4所示:
步骤401,获取人脸识别任务的第一训练样本图像集、测试样本图像集和模型结构。
步骤402,构建GAN的搜索空间。
步骤403,根据搜索空间初始化模型编码生成器。
步骤404,根据模型编码生成器生成模型编码。
步骤405,在搜索空间中将模型编码解码成GAN的模型结构。
步骤406,利用第一训练样本图像集对GAN的模型结构进行训练,得到收敛GAN。
步骤407,利用收敛GAN生成人脸识别任务的第二训练样本图像集。
步骤408,利用第一训练样本图像集和第二训练样本图像集对人脸识别任务的模型结构进行训练,得到人脸识别模型。
步骤409,利用测试样本图像集对人脸识别模型进行测试,得到人脸识别模型的性能。
步骤410,基于人脸识别模型的性能更新模型编码生成器,直至模型编码生成器的迭代次数达到5次,利用最新生成的第二训练图像集扩充用于模型训练的数据量。
进一步参考图5,作为对上述各图所示方法的实现,本申请提供了一种数据生成装置的一个实施例,该装置实施例与图2所示的方法实施例相对应,该装置具体可以应用于各种电子设备中。
如图5所示,本实施例的数据生成装置500可以包括:获取模块501、构建模块502、搜索模块503、训练模块504和生成模块505。其中,获取模块501,被配置成获取目标任务的第一训练数据集;构建模块502,被配置成构建生成式对抗网络的搜索空间;搜索模块503,被配置成在搜索空间中搜索生成式对抗网络的模型结构;训练模块504,被配置成利用第一训练数据集对生成式对抗网络的模型结构进行训练,得到收敛生成式对抗网络;生成模块505,被配置成利用收敛生成式对抗网络生成目标任务的第二训练数据集。
在本实施例中,数据生成装置500中:获取模块501、构建模块502、搜索模块503、训练模块504和生成模块505的具体处理及其所带来的技术效果可分别参考图2对应实施例中的步骤201-205的相关说明,在此不再赘述。
在本实施例的一些可选的实现方式中,搜索模块503包括:初始化子模块(图中未示出),被配置成根据搜索空间初始化模型编码生成器;生成子模块(图中未示出),被配置成根据模型编码生成器生成模型编码;解码子模块(图中未示出),被配置成在搜索空间中将模型编码解码成生成式对抗网络的模型结构。
在本实施例的一些可选的实现方式中,搜索模块503还包括:获取子模块(图中未示出),被配置成获取目标任务的测试数据集和模型结构;第一训练子模块(图中未示出),被配置成利用第一训练数据集和第二训练数据集对目标任务的模型结构进行训练,得到目标任务模型;测试子模块(图中未示出),被配置成利用测试数据集对目标任务模型进行测试,得到目标任务模型的性能;更新子模块(图中未示出),被配置成基于目标任务模型的性能更新模型编码生成器,以及根据更新的模型编码生成器再次生成模型编码。
在本实施例的一些可选的实现方式中,搜索模块503还包括:第二训练子模块(图中未示出),被配置成若模型编码生成器的迭代次数小于预设迭代次数,利用所述第一训练数据集和最新生成的第二训练数据集再次训练所述目标任务模型,以及基于再次训练的目标任务模型的性能再次更新所述模型编码器;扩充子模块(图中未示出),被配置成若所述模型编码器的迭代次数不小于所述预设迭代次数,利用最新生成的第二训练数据集扩充所述第一训练数据集。
在本实施例的一些可选的实现方式中,生成式对抗网络的模型结构既包括生成模型又包括判别模型,或者模型结构仅包括生成模型。
根据本申请的实施例,本申请还提供了一种电子设备和一种可读存储介质。
如图6所示,是根据本申请实施例数据生成方法的电子设备的框图。电子设备旨在表示各种形式的数字计算机,诸如,膝上型计算机、台式计算机、工作台、个人数字助理、服务器、刀片式服务器、大型计算机、和其它适合的计算机。电子设备还可以表示各种形式的移动装置,诸如,个人数字处理、蜂窝电话、智能电话、可穿戴设备和其它类似的计算装置。本文所示的部件、它们的连接和关系、以及它们的功能仅仅作为示例,并且不意在限制本文中描述的和/或者要求的本申请的实现。
如图6所示,该电子设备包括:一个或多个处理器601、存储器602,以及用于连接各部件的接口,包括高速接口和低速接口。各个部件利用不同的总线互相连接,并且可以被安装在公共主板上或者根据需要以其它方式安装。处理器可以对在电子设备内执行的指令进行处理,包括存储在存储器中或者存储器上以在外部输入/输出装置(诸如,耦合至接口的显示设备)上显示GUI的图形信息的指令。在其它实施方式中,若需要,可以将多个处理器和/或多条总线与多个存储器和多个存储器一起使用。同样,可以连接多个电子设备,各个设备提供部分必要的操作(例如,作为服务器阵列、一组刀片式服务器、或者多处理器系统)。图6中以一个处理器601为例。
存储器602即为本申请所提供的非瞬时计算机可读存储介质。其中,所述存储器存储有可由至少一个处理器执行的指令,以使所述至少一个处理器执行本申请所提供的数据生成方法。本申请的非瞬时计算机可读存储介质存储计算机指令,该计算机指令用于使计算机执行本申请所提供的数据生成方法。
存储器602作为一种非瞬时计算机可读存储介质,可用于存储非瞬时软件程序、非瞬时计算机可执行程序以及模块,如本申请实施例中的数据生成方法对应的程序指令/模块(例如,附图5所示的获取模块501、构建模块502、搜索模块503、训练模块504和生成模块505)。处理器601通过运行存储在存储器602中的非瞬时软件程序、指令以及模块,从而执行服务器的各种功能应用以及数据处理,即实现上述方法实施例中的数据生成方法。
存储器602可以包括存储程序区和存储数据区,其中,存储程序区可存储操作系统、至少一个功能所需要的应用程序;存储数据区可存储根据数据生成方法的电子设备的使用所创建的数据等。此外,存储器602可以包括高速随机存取存储器,还可以包括非瞬时存储器,例如至少一个磁盘存储器件、闪存器件、或其他非瞬时固态存储器件。在一些实施例中,存储器602可选包括相对于处理器601远程设置的存储器,这些远程存储器可以通过网络连接至数据生成方法的电子设备。上述网络的实例包括但不限于互联网、企业内部网、局域网、移动通信网及其组合。
数据生成方法的电子设备还可以包括:输入装置603和输出装置604。处理器601、存储器602、输入装置603和输出装置604可以通过总线或者其他方式连接,图6中以通过总线连接为例。
输入装置603可接收输入的数字或字符信息,以及产生与数据生成方法的电子设备的用户设置以及功能控制有关的键信号输入,例如触摸屏、小键盘、鼠标、轨迹板、触摸板、指示杆、一个或者多个鼠标按钮、轨迹球、操纵杆等输入装置。输出装置604可以包括显示设备、辅助照明装置(例如,LED)和触觉反馈装置(例如,振动电机)等。该显示设备可以包括但不限于,液晶显示器(LCD)、发光二极管(LED)显示器和等离子体显示器。在一些实施方式中,显示设备可以是触摸屏。
此处描述的系统和技术的各种实施方式可以在数字电子电路系统、集成电路系统、专用ASIC(专用集成电路)、计算机硬件、固件、软件、和/或它们的组合中实现。这些各种实施方式可以包括:实施在一个或者多个计算机程序中,该一个或者多个计算机程序可在包括至少一个可编程处理器的可编程系统上执行和/或解释,该可编程处理器可以是专用或者通用可编程处理器,可以从存储系统、至少一个输入装置、和至少一个输出装置接收数据和指令,并且将数据和指令传输至该存储系统、该至少一个输入装置、和该至少一个输出装置。
这些计算程序(也称作程序、软件、软件应用、或者代码)包括可编程处理器的机器指令,并且可以利用高级过程和/或面向对象的编程语言、和/或汇编/机器语言来实施这些计算程序。如本文使用的,术语“机器可读介质”和“计算机可读介质”指的是用于将机器指令和/或数据提供给可编程处理器的任何计算机程序产品、设备、和/或装置(例如,磁盘、光盘、存储器、可编程逻辑装置(PLD)),包括,接收作为机器可读信号的机器指令的机器可读介质。术语“机器可读信号”指的是用于将机器指令和/或数据提供给可编程处理器的任何信号。
为了提供与用户的交互,可以在计算机上实施此处描述的系统和技术,该计算机具有:用于向用户显示信息的显示装置(例如,CRT(阴极射线管)或者LCD(液晶显示器)监视器);以及键盘和指向装置(例如,鼠标或者轨迹球),用户可以通过该键盘和该指向装置来将输入提供给计算机。其它种类的装置还可以用于提供与用户的交互;例如,提供给用户的反馈可以是任何形式的传感反馈(例如,视觉反馈、听觉反馈、或者触觉反馈);并且可以用任何形式(包括声输入、语音输入或者、触觉输入)来接收来自用户的输入。
可以将此处描述的系统和技术实施在包括后台部件的计算系统(例如,作为数据服务器)、或者包括中间件部件的计算系统(例如,应用服务器)、或者包括前端部件的计算系统(例如,具有图形用户界面或者网络浏览器的用户计算机,用户可以通过该图形用户界面或者该网络浏览器来与此处描述的系统和技术的实施方式交互)、或者包括这种后台部件、中间件部件、或者前端部件的任何组合的计算系统中。可以通过任何形式或者介质的数字数据通信(例如,通信网络)来将系统的部件相互连接。通信网络的示例包括:局域网(LAN)、广域网(WAN)和互联网。
计算机系统可以包括客户端和服务器。客户端和服务器一般远离彼此并且通常通过通信网络进行交互。通过在相应的计算机上运行并且彼此具有客户端-服务器关系的计算机程序来产生客户端和服务器的关系。
根据本申请的技术方案,首先获取目标任务的第一训练数据集;之后构建生成式对抗网络的搜索空间;而后在搜索空间中搜索生成式对抗网络的模型结构;然后利用第一训练数据集对生成式对抗网络的模型结构进行训练,得到收敛生成式对抗网络;最后利用收敛生成式对抗网络生成目标任务的第二训练数据集。通过自动搜索的方式生成训练数据,能够应用于训练数据量稀缺的场景,降低了生成训练数据的成本。
应该理解,可以使用上面所示的各种形式的流程,重新排序、增加或删除步骤。例如,本发申请中记载的各步骤可以并行地执行也可以顺序地执行也可以不同的次序执行,只要能够实现本申请公开的技术方案所期望的结果,本文在此不进行限制。
上述具体实施方式,并不构成对本申请保护范围的限制。本领域技术人员应该明白的是,根据设计要求和其他因素,可以进行各种修改、组合、子组合和替代。任何在本申请的精神和原则之内所作的修改、等同替换和改进等,均应包含在本申请保护范围之内。

Claims (6)

1.一种图像生成方法,包括:
获取目标任务的第一训练数据集,其中,所述第一训练数据集是图像集;
构建生成式对抗网络的搜索空间;
在所述搜索空间中搜索生成式对抗网络的模型结构;
利用所述第一训练数据集对所述生成式对抗网络的模型结构进行训练,得到收敛生成式对抗网络,其中,所述生成式对抗网络包括生成模型和判别模型,将所述第一训练数据集中的图像输入至所述生成模型,生成新的图像,将所述新的图像输入至所述判别模型,判定所述新的图像的真实性;
利用所述收敛生成式对抗网络生成所述目标任务的第二训练数据集,其中,将所述第一训练数据集中的图像输入至所述收敛生成式对抗网络中的生成模型,生成所述第二训练数据集中的图像;
其中,所述在所述搜索空间中搜索生成式对抗网络的模型结构,包括:
根据所述搜索空间初始化模型编码生成器;
根据所述模型编码生成器生成模型编码;
在所述搜索空间中将所述模型编码解码成所述生成式对抗网络的模型结构;
获取所述目标任务的测试数据集和模型结构;
利用所述第一训练数据集和所述第二训练数据集对所述目标任务的模型结构进行训练,得到目标任务模型;
利用所述测试数据集对所述目标任务模型进行测试,得到所述目标任务模型的性能;
基于所述目标任务模型的性能更新所述模型编码生成器,以及根据更新的模型编码生成器再次生成模型编码。
2.根据权利要求1所述的方法,其中,所述在所述搜索空间中搜索生成式对抗网络的模型结构,还包括:
若所述模型编码生成器的迭代次数小于预设迭代次数,利用所述第一训练数据集和最新生成的第二训练数据集再次训练所述目标任务模型,以及基于再次训练的目标任务模型的性能再次更新所述模型编码器;
若所述模型编码器的迭代次数不小于所述预设迭代次数,利用最新生成的第二训练数据集扩充所述第一训练数据集。
3.一种图像生成装置,包括:
获取模块,被配置成获取目标任务的第一训练数据集,其中,所述第一训练数据集是图像集;
构建模块,被配置成构建生成式对抗网络的搜索空间;
搜索模块,被配置成在所述搜索空间中搜索生成式对抗网络的模型结构;
训练模块,被配置成利用所述第一训练数据集对所述生成式对抗网络的模型结构进行训练,得到收敛生成式对抗网络,其中,所述生成式对抗网络包括生成模型和判别模型,将所述第一训练数据集中的图像输入至所述生成模型,生成新的图像,将所述新的图像输入至所述判别模型,判定所述新的图像的真实性;
生成模块,被配置成利用所述收敛生成式对抗网络生成所述目标任务的第二训练数据集,其中,将所述第一训练数据集中的图像输入至所述收敛生成式对抗网络中的生成模型,生成所述第二训练数据集中的图像;
其中,所述搜索模块包括:
初始化子模块,被配置成根据所述搜索空间初始化模型编码生成器;
生成子模块,被配置成根据所述模型编码生成器生成模型编码;
解码子模块,被配置成在所述搜索空间中将所述模型编码解码成所述生成式对抗网络的模型结构;
获取子模块,被配置成获取所述目标任务的测试数据集和模型结构;
第一训练子模块,被配置成利用所述第一训练数据集和所述第二训练数据集对所述目标任务的模型结构进行训练,得到目标任务模型;
测试子模块,被配置成利用所述测试数据集对所述目标任务模型进行测试,得到所述目标任务模型的性能;
更新子模块,被配置成基于所述目标任务模型的性能更新所述模型编码生成器,以及根据更新的模型编码生成器再次生成模型编码。
4.根据权利要求3所述的装置,其中,所述搜索模块还包括:
第二训练子模块,被配置成若所述模型编码生成器的迭代次数小于预设迭代次数,利用所述第一训练数据集和最新生成的第二训练数据集再次训练所述目标任务模型,以及基于再次训练的目标任务模型的性能再次更新所述模型编码器;
扩充子模块,被配置成若所述模型编码器的迭代次数不小于所述预设迭代次数,利用最新生成的第二训练数据集扩充所述第一训练数据集。
5.一种电子设备,包括:
一个或多个处理器;
存储装置,其上存储有一个或多个程序,
当所述一个或多个程序被所述一个或多个处理器执行,使得所述一个或多个处理器实现如权利要求1-2中任一所述的方法。
6.一种计算机可读存储介质,其上存储有计算机程序,其中,所述计算机程序被处理器执行时实现如权利要求1-2中任一所述的方法。
CN202010506988.4A 2020-06-05 2020-06-05 数据生成方法、装置、设备以及存储介质 Active CN111667004B (zh)

Priority Applications (1)

Application Number Priority Date Filing Date Title
CN202010506988.4A CN111667004B (zh) 2020-06-05 2020-06-05 数据生成方法、装置、设备以及存储介质

Applications Claiming Priority (1)

Application Number Priority Date Filing Date Title
CN202010506988.4A CN111667004B (zh) 2020-06-05 2020-06-05 数据生成方法、装置、设备以及存储介质

Publications (2)

Publication Number Publication Date
CN111667004A CN111667004A (zh) 2020-09-15
CN111667004B true CN111667004B (zh) 2024-05-31

Family

ID=72386586

Family Applications (1)

Application Number Title Priority Date Filing Date
CN202010506988.4A Active CN111667004B (zh) 2020-06-05 2020-06-05 数据生成方法、装置、设备以及存储介质

Country Status (1)

Country Link
CN (1) CN111667004B (zh)

Citations (10)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN108229657A (zh) * 2017-12-25 2018-06-29 杭州健培科技有限公司 一种基于演化算法的深度神经网络训练与优化算法
CN109815928A (zh) * 2019-01-31 2019-05-28 中国电子进出口有限公司 一种基于对抗学习的人脸图像合成方法和装置
CN110110862A (zh) * 2019-05-10 2019-08-09 电子科技大学 一种基于适应性模型的超参数优化方法
CN110189253A (zh) * 2019-04-16 2019-08-30 浙江工业大学 一种基于改进生成对抗网络的图像超分辨率重建方法
CN110297218A (zh) * 2019-07-09 2019-10-01 哈尔滨工程大学 基于生成对抗网络的雷达信号未知调制方式检测方法
CN110443369A (zh) * 2019-08-02 2019-11-12 首都师范大学 基于机器学习的定理证明方法、装置、电子设备及存储介质
CN110909877A (zh) * 2019-11-29 2020-03-24 百度在线网络技术(北京)有限公司 神经网络模型结构搜索方法、装置、电子设备及存储介质
CN110930308A (zh) * 2019-11-15 2020-03-27 东南大学 一种图像超分辨生成网络的结构搜索方法
CN111127364A (zh) * 2019-12-26 2020-05-08 吉林大学 图像数据增强策略选择方法及人脸识别图像数据增强方法
CN111178316A (zh) * 2020-01-06 2020-05-19 武汉大学 一种基于深度架构自动搜索的高分辨率遥感影像土地覆盖分类方法

Family Cites Families (1)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
WO2018053340A1 (en) * 2016-09-15 2018-03-22 Twitter, Inc. Super resolution using a generative adversarial network

Patent Citations (10)

* Cited by examiner, † Cited by third party
Publication number Priority date Publication date Assignee Title
CN108229657A (zh) * 2017-12-25 2018-06-29 杭州健培科技有限公司 一种基于演化算法的深度神经网络训练与优化算法
CN109815928A (zh) * 2019-01-31 2019-05-28 中国电子进出口有限公司 一种基于对抗学习的人脸图像合成方法和装置
CN110189253A (zh) * 2019-04-16 2019-08-30 浙江工业大学 一种基于改进生成对抗网络的图像超分辨率重建方法
CN110110862A (zh) * 2019-05-10 2019-08-09 电子科技大学 一种基于适应性模型的超参数优化方法
CN110297218A (zh) * 2019-07-09 2019-10-01 哈尔滨工程大学 基于生成对抗网络的雷达信号未知调制方式检测方法
CN110443369A (zh) * 2019-08-02 2019-11-12 首都师范大学 基于机器学习的定理证明方法、装置、电子设备及存储介质
CN110930308A (zh) * 2019-11-15 2020-03-27 东南大学 一种图像超分辨生成网络的结构搜索方法
CN110909877A (zh) * 2019-11-29 2020-03-24 百度在线网络技术(北京)有限公司 神经网络模型结构搜索方法、装置、电子设备及存储介质
CN111127364A (zh) * 2019-12-26 2020-05-08 吉林大学 图像数据增强策略选择方法及人脸识别图像数据增强方法
CN111178316A (zh) * 2020-01-06 2020-05-19 武汉大学 一种基于深度架构自动搜索的高分辨率遥感影像土地覆盖分类方法

Non-Patent Citations (3)

* Cited by examiner, † Cited by third party
Title
AutoGAN:Neural Architecture Search for Generative Adversarial Networks;Xinyu Gong,Shiyu Chang,Yifan Jiang,Zhangyang Wan;https://doi.org/10.48550/arXiv.1908.03835,AutoGAN:Neural Architecture Search for Generative Adversarial Networks;正文第2-6页 *
Xinyu Gong,Shiyu Chang,Yifan Jiang,Zhangyang Wan.AutoGAN:Neural Architecture Search for Generative Adversarial Networks.https://doi.org/10.48550/arXiv.1908.03835,AutoGAN:Neural Architecture Search for Generative Adversarial Networks.2019,正文第2-6页. *
监控视频人脸考勤与陌生人鉴别方法及系统实现;陈柱良;《中国优秀硕士学位论文全文数据库 信息科技辑》,监控视频人脸考勤与陌生人鉴别方法及系统实现;正文第1-13 *

Also Published As

Publication number Publication date
CN111667004A (zh) 2020-09-15

Similar Documents

Publication Publication Date Title
CN111539514B (zh) 用于生成神经网络的结构的方法和装置
CN111639710B (zh) 图像识别模型训练方法、装置、设备以及存储介质
CN111667057B (zh) 用于搜索模型结构的方法和装置
CN111582479B (zh) 神经网络模型的蒸馏方法和装置
CN111667056B (zh) 用于搜索模型结构的方法和装置
CN111582477B (zh) 神经网络模型的训练方法和装置
CN111488740B (zh) 一种因果关系的判别方法、装置、电子设备及存储介质
CN112149829B (zh) 确定网络模型剪枝策略的方法、装置、设备以及存储介质
CN111695519B (zh) 关键点定位方法、装置、设备以及存储介质
CN112149634B (zh) 图像生成器的训练方法、装置、设备以及存储介质
CN111324715B (zh) 问答机器人的生成方法和装置
CN111695698B (zh) 用于模型蒸馏的方法、装置、电子设备及可读存储介质
CN111563593B (zh) 神经网络模型的训练方法和装置
CN111914994B (zh) 多层感知机的生成方法、装置、电子设备及存储介质
CN111539224B (zh) 语义理解模型的剪枝方法、装置、电子设备和存储介质
CN111241838B (zh) 文本实体的语义关系处理方法、装置及设备
CN111680600B (zh) 人脸识别模型处理方法、装置、设备和存储介质
CN111967591B (zh) 神经网络自动剪枝方法、装置及电子设备
CN111582452A (zh) 生成神经网络模型的方法和装置
CN111652354B (zh) 用于训练超网络的方法、装置、设备以及存储介质
CN111680597B (zh) 人脸识别模型处理方法、装置、设备和存储介质
CN111753964A (zh) 神经网络的训练方法和装置
CN112580723B (zh) 多模型融合方法、装置、电子设备和存储介质
CN111553169B (zh) 语义理解模型的剪枝方法、装置、电子设备和存储介质
CN111667428B (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
TA01 Transfer of patent application right

Effective date of registration: 20240322

Address after: 510000 No. 106 Fengze East Road, Nansha District, Guangzhou City, Guangdong Province

Applicant after: Guangzhou dinghang Information Technology Service Co.,Ltd.

Country or region after: China

Address before: 2 / F, baidu building, 10 Shangdi 10th Street, Haidian District, Beijing 100085

Applicant before: BEIJING BAIDU NETCOM SCIENCE AND TECHNOLOGY Co.,Ltd.

Country or region before: China

TA01 Transfer of patent application right
TA01 Transfer of patent application right

Effective date of registration: 20240424

Address after: 432000 No.1 Wenchang Avenue, High tech Development Zone, Xiaogan City, Hubei Province*

Applicant after: Xiaogan Sichuang Information Technology Co.,Ltd.

Country or region after: China

Address before: 510000 No. 106 Fengze East Road, Nansha District, Guangzhou City, Guangdong Province

Applicant before: Guangzhou dinghang Information Technology Service Co.,Ltd.

Country or region before: China

TA01 Transfer of patent application right
GR01 Patent grant
GR01 Patent grant