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