神经网络结构探测方法、结构探测模型的训练方法及装置

基本信息

申请号 CN202010331224.6 申请日 -
公开(公告)号 CN111582474A 公开(公告)日 2020-08-25
申请公布号 CN111582474A 申请公布日 2020-08-25
分类号 G06N3/08(2006.01)I;G06N3/04(2006.01)I 分类 -
发明人 赵恒 申请(专利权)人 中科物栖(北京)科技有限责任公司
代理机构 北京和信华成知识产权代理事务所(普通合伙) 代理人 胡剑辉
地址 100086北京市海淀区科学院南路6号中国科学院计算技术研究所科研综合楼12层1248室
法律状态 -

摘要

摘要 本发明实施例涉及一种神经网络结构探测方法、结构探测模型的训练方法及装置,所述神经网络结构探测方法包括:获得参考网络在多核平台上运行时每一层的计算完成时间;将所述每一层的计算完成时间输入至已训练的结构探测模型,得到与所述参考网络运行在同一多核平台上的目标网络的结构信息;依据所述目标网络的结构信息进行拓扑结构重建,得到所述目标网络的结构。由此,可以实现通过较低的成本即可探测出目标网络的结构,并且探测结果具有较高的鲁棒性和准确性。