2024年11月14日消息,国家知识产权局信息显示,小红书科技有限公司申请一项名为“训练图预训练模型的方法、对图进行分类的方法和装置”的专利,公开号CN 118940097 A,申请日期为2023年5月。
专利摘要显示,本申请公开了训练图预训练模型的方法、对图进行分类的方法、装置。训练图预训练模型的方法包括:获取步骤,获取多个图的多个向量表示,每个向量表示包括多个节点和多个边;掩码步骤,利用多个任务,分别对多个向量表示进行掩码处理,得到与多个任务分别对应的多组掩码向量表示;构建步骤,利用多组掩码向量表示,构建与多个任务分别对应的多个损失函数;训练步骤,利用多个损失函数对图预训练模型进行训练以得到训练后的图预训练模型本发明的训练方法极大的提升了图预训练模型的性能。