2024年11月14日消息,國家知識產(chǎn)權(quán)局信息顯示,小紅書科技有限公司申請一項名為“訓(xùn)練圖預(yù)訓(xùn)練模型的方法、對圖進行分類的方法和裝置”的專利,公開號CN 118940097 A,申請日期為2023年5月。
專利摘要顯示,本申請公開了訓(xùn)練圖預(yù)訓(xùn)練模型的方法、對圖進行分類的方法、裝置。訓(xùn)練圖預(yù)訓(xùn)練模型的方法包括:獲取步驟,獲取多個圖的多個向量表示,每個向量表示包括多個節(jié)點和多個邊;掩碼步驟,利用多個任務(wù),分別對多個向量表示進行掩碼處理,得到與多個任務(wù)分別對應(yīng)的多組掩碼向量表示;構(gòu)建步驟,利用多組掩碼向量表示,構(gòu)建與多個任務(wù)分別對應(yīng)的多個損失函數(shù);訓(xùn)練步驟,利用多個損失函數(shù)對圖預(yù)訓(xùn)練模型進行訓(xùn)練以得到訓(xùn)練后的圖預(yù)訓(xùn)練模型本發(fā)明的訓(xùn)練方法極大的提升了圖預(yù)訓(xùn)練模型的性能。