一种基于对比学习的自监督深度表征方法

    公开(公告)号:CN115759189A

    公开(公告)日:2023-03-07

    申请号:CN202211399532.8

    申请日:2022-11-09

    Abstract: 本发明公开了一种基于对比学习的自监督深度表征方法,包括以下步骤:S1:获取待训练的数据集V;S2:一次批处理训练N条数据,对每一条数据v进行数据增强;S3:将数据v与增强后的数据vs传进受限玻尔兹曼机进行正向传播,获得其在隐藏层对应的数据h与数据hs;S4:进行对比学习,提高正样本之间的相似度;S5:进行反向传播,获得数据v在可视层的重构数据v';S6:最小化所述数据v'与所述数据v的距离;S7:重复步骤S2‑S6,直至重构损失函数收敛,实现受限玻尔兹曼机的训练;S8:利用S7训练好的受限玻尔兹曼机对数据集V进行深度表征学习,获得其深度特征表示。本发明能够克服传统受限玻尔兹曼机无目的学习的缺点,从而提高模型的表征能力。

Patent Agency Ranking