推奨システムは、ユーザーの嗜好を予測して商品やコンテンツを提案する重要な技術である。近年、深層学習を用いた協調フィルタリングが注目されている。
推奨システムの主要な3つの要素は、(1)ユーザーと商品の相互作用をエンコードするモジュール、(2)損失関数、(3)ネガティブサンプリング戦略である。
従来の研究では、複雑なエンコーディングモジュールの開発に注力されてきたが、最近では損失関数とネガティブサンプリング戦略の改善に注目が集まっている。特に、対照学習(contrastive learning)は、類似したペアを近づけ、異なるペアを離すことで表現を学習する手法として注目されている。
対照学習では、大きなバッチサイズや多数のネガティブサンプルを必要とするが、メモリ使用量が増大するという課題がある。
本研究では、Multi-Margin Cosine Loss (MMCL)を提案する。MMCLは、複数のマージンと異なるウェイトを用いることで、最も難しいネガティブサンプルだけでなく、他の非自明なネガティブサンプルも効果的に活用できる。
実験の結果、MMCLは少数のネガティブサンプルを使用する場合でも、従来の損失関数と比較して最大20%の性能向上を示した。
Naar een andere taal
vanuit de broninhoud
arxiv.org
Belangrijkste Inzichten Gedestilleerd Uit
by Makbule Gulc... om arxiv.org 09-11-2024
https://arxiv.org/pdf/2405.04614.pdfDiepere vragen