arXiv cs.LG

GATベースの協調フィルタリングのためのベクトル化されたコンテキスト対応埋め込み

Vectorized Context-Aware Embeddings for GAT-Based Collaborative Filtering

http://arxiv.org/abs/2510.26461v1


本論文では、データのスパース性やコールドスタート問題に対処するために、グラフアテンションネットワーク(GAT)に基づく協調フィルタリング(CF)フレームワークが提案されています。このフレームワークは、大規模言語モデル(LLM)を活用したコンテキスト対応埋め込みによって強化されています。具体的には、テキスト形式のユーザープロフィールを生成し、アイテムのメタデータを統合してリッチなテキスト埋め込みを作成し、バイパーティートユーザーアイテムグラフに最初のノード特徴量として注入します。また、ランキング性能を最適化するため、ベイジアン個別化ランキング(BPR)とコサイン類似度項、ロバストなネガティブサンプリングを組み合わせたハイブリッド損失関数を導入し、明示的なネガティブフィードバックと未観測データを区別します。実験結果は、MovieLens 100kおよび1Mデータセットにおいて、精度、NDCG、MAPで最先端のベースラインに対して一貫した改善を示しました。最終的に、LLMに基づくコンテキスト理解を統合することで、スパース性やコールドスタートの制約を効果的に軽減しています。