Recommending Related Products Using Graph Neural Networks in Directed Graphs

TL;DR

This paper introduces DAEMON, a graph neural network framework that effectively recommends related products in e-commerce by modeling asymmetry, leveraging multi-modal data, and addressing selection bias, significantly outperforming existing methods.

Contribution

The paper proposes DAEMON, a novel GNN-based approach that models product asymmetry, incorporates multi-modal data, and mitigates selection bias for improved related product recommendations.

Findings

DAEMON outperforms baselines by 30-160% in HitRate and MRR.

Employs asymmetric loss and dual embeddings for product asymmetry.

Leverages multi-modal data to improve cold-start recommendations.

Abstract

Related product recommendation (RPR) is pivotal to the success of any e-commerce service. In this paper, we deal with the problem of recommending related products i.e., given a query product, we would like to suggest top-k products that have high likelihood to be bought together with it. Our problem implicitly assumes asymmetry i.e., for a phone, we would like to recommend a suitable phone case, but for a phone case, it may not be apt to recommend a phone because customers typically would purchase a phone case only while owning a phone. We also do not limit ourselves to complementary or substitute product recommendation. For example, for a specific night wear t-shirt, we can suggest similar t-shirts as well as track pants. So, the notion of relatedness is subjective to the query product and dependent on customer preferences. Further, various factors such as product price, availability lead to presence of selection bias in the historical purchase data, that needs to be controlled for while training related product recommendations model. These challenges are orthogonal to each other deeming our problem nontrivial. To address these, we propose DAEMON, a novel Graph Neural Network (GNN) based framework for related product recommendation, wherein the problem is formulated as a node recommendation task on a directed product graph. In order to capture product asymmetry, we employ an asymmetric loss function and learn dual embeddings for each product, by appropriately aggregating features from its neighborhood. DAEMON leverages multi-modal data sources such as catalog metadata, browse behavioral logs to mitigate selection bias and generate recommendations for cold-start products. Extensive offline experiments show that DAEMON outperforms state-of-the-art baselines by 30-160% in terms of HitRate and MRR for the node recommendation task.