一种基于元学习的神经网络模型检索方法技术

技术编号:40036080 阅读:18 留言:0更新日期:2024-01-16 18:59
本发明专利技术实现了一种基于元学习的神经网络模型检索方法。包括两个阶段:离线训练阶段和在线检索阶段,分为五个步骤,其中步骤一至步骤三为离线训练阶段,步骤四至步骤五为在线检索阶段,步骤一为元特征提取步骤,输入图像数据集、文本任务描述和模型库,进行模型库中任务元特征的离线提取,步骤二为元数据集构建;步骤三为元学习器训练;步骤四为多模态用户任务所述元特征的在线提取;步骤五为元模型预测,最终得到在当前任务上性能表现好的模型序列。本申请方案能够解决现有方法难以处理深度学习多模态复杂输入的问题,并且能够高效准确地为先前已见和未见任务检索性能表现好的深度神经网络模型,从而提高模型检索和复用的准确性和效率。

【技术实现步骤摘要】

本专利技术涉及人工智能,尤其涉及一种基于元学习的神经网络模型检索方法


技术介绍

1、随着机器学习的兴起和发展,深度学习在许多问题和领域中展现出了显著的优势,如图像分类与识别、自然语言处理等。然而,为每个任务从头构建和训练深度神经网络(deep neural network,dnn)模型面临着成本高、时间长等难题。一方面,这需要同时具备领域知识和深度学习经验的专家进行模型的构建和开发;另一方面,模型的训练调参过程也要耗费大量的时间和算力。已有研究调研显示,随着人工智能模型规模的扩大,需要更多的计算资源,尤其是更多的能耗。近来随着模型规模变大,参数更多,训练所花费的时间、算力和能耗也进一步增加。过去几十年来,软件重用被视为降低软件开发成本的一种非常有效的思想方法。受此启发,模型重用已被广泛研究并应用于快速构建目标任务的复杂dnn模型。随着深度学习的广泛应用和开源社区的蓬勃发展,有大量深度学习模型被构建和公开共享。例如有“机器学习界的github”之称的hugging face,模型库中提供了数十万个机器学习模型的共享。这促使开发人员通过重用现有深度学习模型以本文档来自技高网...

【技术保护点】

1.一种基于元学习的神经网络模型检索方法,其特征在于:包括两个阶段:离线训练阶段和在线检索阶段,分为五个步骤,其中步骤一至步骤三为离线训练阶段,步骤四至步骤五为在线检索阶段。具体如下:

2.如权利要求1所述的一种基于元学习的神经网络模型检索方法,其特征在于:所述元特征提取步骤中进行模型库中任务元特征的离线提取是对模型库中所有任务分别进行元特征提取,其中所述特征提取方法具体为:输入任务作为查询来检索DNN模型,将其记作q,将要检索的DNN模型记作Mi,其中Mi∈{M1,M2,…,Mn},假设模型Mi在任务Ti上的性能表现为P(Mi,Ti),而在新任务Tj上的性能表现为P(Mi,...

【技术特征摘要】

1.一种基于元学习的神经网络模型检索方法,其特征在于:包括两个阶段:离线训练阶段和在线检索阶段,分为五个步骤,其中步骤一至步骤三为离线训练阶段,步骤四至步骤五为在线检索阶段。具体如下:

2.如权利要求1所述的一种基于元学习的神经网络模型检索方法,其特征在于:所述元特征提取步骤中进行模型库中任务元特征的离线提取是对模型库中所有任务分别进行元特征提取,其中所述特征提取方法具体为:输入任务作为查询来检索dnn模型,将其记作q,将要检索的dnn模型记作mi,其中mi∈{m1,m2,…,mn},假设模型mi在任务ti上的性能表现为p(mi,ti),而在新任务tj上的性能表现为p(mi,tj),则模型性能差异δp可表示为:δp=p(mi,tj)-p(mi,ti)∝δ(tj,ti)。对于新的未见任务tj,通过计算预测模型的性能,其中δ(tj,ti)定义为计算数据集差异δ(dj,di)和任务描述差异δ(lj,li),δ(tj,ti)=δ(dj,di)+δ(lj,li),通过基于多模态嵌入的用户任务表示方法得到元特征。

3.如权利要求2所述的一种基于元学习的神经网络模型检索方法,其特征在于:所述多模态嵌入的用户任务表示方法使用clip作为嵌入层,为文本和图像提供嵌入空间,给定一个包含图像数据集和任务描述的用户任务查询...

【专利技术属性】
技术研发人员:孙海龙李兆天
申请(专利权)人:北京航空航天大学
类型:发明
国别省市:

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

1