Patent search ap:("北京理工大学") AND inv:"尚楠" Page 1

1.

发明公开
基于知识蒸馏的图对比学习方法审中-实审

公开(公告)号：CN116720547A

公开(公告)日：2023-09-08

申请号：CN202310357741.4

申请日：2023-04-06

Applicant: 北京理工大学

Inventor： 尚楠 , 李荣华 , 秦宏超 , 王国仁

IPC: G06N3/045 , G06N3/088

Abstract: 本发明公开了一种基于知识蒸馏的图对比学习方法，包括以下步骤：步骤1，数据准备：准备合适的图数据集；步骤2，基于图对比学习进行教师模型的设计及其预训练：步骤21，数据增强；步骤22，生成正负样例对；步骤23，计算模型损失；步骤24，重复步骤21‑23降低模型计算损失；步骤3，定义并设计轻量化学生模型；步骤4，进行离线知识蒸馏；步骤5，在测试样本集上进行测试，调整相关参数使模型达到最佳效果。本发明采用上述基于知识蒸馏的图对比学习方法，重点解决了模型复杂程度过高导致的计算时间长、响应变慢的问题，同时极大的降低了图表示学习中对于数据量的依赖。

Patent Agency Ranking