LinkFND: Simple Framework for False Negative Detection in Recommendation Tasks With Graph Contrastive Learning

초록

Self-supervised learning has been shown to be effective in various fields, proving its usefulness in contrastive learning. Recently, graph contrastive learning has shown state-of-the-art performance in the recommendation task. They created two views and learned node embeddings so that target nodes in the two views attract each other based on the target node, and non-target nodes in the two views repel each other. However, they overlooked the fact that false negatives can occur when negative pairs are repelled. It has been shown through various studies that false negatives in contrastive learning in various fields can have a negative impact on model training, but research on the impact of false negatives in link prediction tasks, such as recommendation tasks, where classes cannot be clearly defined, is still hardly explored. In this paper, we propose an approach to define false negatives in link prediction tasks and fully utilize them in learning. Learning by defining false negatives and removing them from negative pairs showed consistent improvements over existing graph contrastive learning on five benchmark datasets. In addition, we found through comprehensive experimental studies that learning by removing false negatives is of great advantage, especially for low-density datasets. On top of these advantages, our false negative detection and false negative elimination can be naturally integrated into any graph contrastive learning architecture.

키워드