本文提出了一種新的損失函數 Multi-Margin Cosine Loss (MMCL),用於推薦系統中的協同過濾深度學習技術。推薦系統由三個主要組件組成:交互編碼器、損失函數和負採樣策略。近年來,研究人員轉向改進損失函數和負採樣策略,增加了對比學習的興趣。對比學習旨在通過拉近相似對、拉遠不相似對來學習表示。但對比學習可能會帶來高內存需求和部分負樣本利用不足的挑戰。
MMCL通過引入多個邊界和為不同難度的負樣本分配不同權重來解決這些挑戰。它能夠有效利用不僅是最困難的負樣本,也包括其他非平凡的負樣本,提供一種簡單而有效的損失函數,在資源受限的情況下特別能夠優於更複雜的方法。
在兩個著名的數據集上的實驗表明,與基線損失函數相比,當使用較少的負樣本時,MMCL可以實現高達20%的性能提升。
A otro idioma
del contenido fuente
arxiv.org
Ideas clave extraídas de
by Makbule Gulc... a las arxiv.org 09-11-2024
https://arxiv.org/pdf/2405.04614.pdfConsultas más profundas