Matryoshka Representation Learning for Recommendation with Layer- and Hardness-Adaptive Negative Sampling

Representation learning is essential for deep-neural-network-based recommender systems to capture user preferences and item features within fixed-dimensional user and item vectors. Unlike existing representation learning methods that either treat each user preference and item feature uniformly or categorize them into discrete clusters, we argue that in the real world, user preferences and item features are naturally expressed and organized in a hierarchical manner, leading to a new direction for representation learning.
In this paper, we introduce a novel matryoshka representation learning method for recommendation (MRL4Rec), by which we restructure user and item vectors into matryoshka representations with nested vector spaces to explicitly represent user preferences and item features at different hierarchical layers. We theoretically establish that training with the same triplets for each sliced vector cannot guarantee representation learning with hierarchical structures. Subsequently, we propose the layer- and hardness-adaptive negative sampling (LHANS) mechanism to construct training triplets, which further ensures the soundness of learned matryoshka representations in capturing hierarchical user preferences and item features. The experiments demonstrate that MRL4Rec can consistently and substantially outperform a number of state-of-the-art competitors on several real-life datasets. Our code is publicly available at https://github.com/Riwei-HEU/MRL.