CNIPA.AI
검색으로 돌아가기
기록

利用对比知识蒸馏训练神经网络

발명유효
1조회수
20청구항 · 3 독립항
§ Ⅰ

개요

발명자

姚安邦; 范家伟; 李超; 宋淼; 陈玉荣; 刘晓龙; G·菲尔兹; S·里奇曼

IPC 분류

G06N 3/0464 (2006.01)

通过对比知识蒸馏使用支持神经网络可以训练目标神经网络。从支持神经网络提取第一特征图。从目标神经网络提取特征图并且将特征图转换成第二特征图。通过将第一特征图划分成支持样本并且将第二特征图划分成目标样本可以形成对比对。对比对可以包括表示支持特征图中的像素的支持样本和表示已转换的目标特征图中的像素的目标样本。基于正对和负对可以确定对比损失。基于支持特征图和目标特征图可以确定特征蒸馏损失。至少使用对比损失和特征蒸馏损失可以更新目标神经网络的内部参数。