一种等变网络训练方法、装置、电子设备及存储介质

基本信息

申请号 CN202210082336.1 申请日 -
公开(公告)号 CN114528977A 公开(公告)日 2022-05-24
申请公布号 CN114528977A 申请公布日 2022-05-24
分类号 G06N3/04(2006.01)I;G06N3/08(2006.01)I 分类 计算;推算;计数;
发明人 陈智强;余山;陈阳 申请(专利权)人 北京智源人工智能研究院
代理机构 北京辰权知识产权代理有限公司 代理人 -
地址 100083北京市海淀区中关村东路1号院8号楼三层B201D-1
法律状态 -

摘要

摘要 本发明公开了一种等变网络训练方法、装置、电子设备及存储介质,方法包括:构建由等变卷积层组成的等变网络;对等变网络中的等变卷积层进行高斯调制;利用训练图像集中的图像对等变网络进行训练,直至等变网络收敛结束。由于传统的卷积层是在离散空间中定义的,只有采样格点上有参数,而通过对等变网络中的等变卷积层进行高斯调制,经调制后的等变卷积层在空间位置和角度上是可导的,可以优化位置和角度参数,因此可以在连续空间中定义参数,使得训练完成的等变网络能够客服离散采样的限制,从而获得在连续变换群上的等变性。