Self-supervised learning through data augmentation for recommendation systems

Abstract

A computer-implemented method to train a machine-learning model to recommend virtual experiences to a user. The method includes receiving training data that includes original training examples corresponding to a set of virtual experiences, where individual training examples comprise user features and item features. The method further generating augmented training examples by modifying one or more of a user feature or an item feature from corresponding original training examples. The method further extracting respective representation embeddings from the original training examples and the augmented training examples. The method further includes determining a loss function such that: a first loss between representation embeddings of individual original training examples and representation embeddings of corresponding augmented training examples is minimized and a second loss between the representation embeddings of individual original training examples and representation embeddings of non-corresponding augmented training examples is maximized.