BERT Goes Shopping: Comparing Distributional Models for Product Representations
Abstract
Word embeddings (e.g., word2vec) have been applied successfully to eCommerce products through~\textit{prod2vec}. Inspired by the recent performance improvements on several NLP tasks brought by contextualized embeddings, we propose to transfer BERT-like architectures to eCommerce: our model -- ~\textit{Prod2BERT} -- is trained to generate representations of products through masked session modeling. Through extensive experiments over multiple shops, different tasks, and a range of design choices, we systematically compare the accuracy of~\textit{Prod2BERT} and~\textit{prod2vec} embeddings: while~\textit{Prod2BERT} is found to be superior in several scenarios, we highlight the importance of resources and hyperparameters in the best performing models. Finally, we provide guidelines to practitioners for training embeddings under a variety of computational and data constraints.
Cite
@article{arxiv.2012.09807,
title = {BERT Goes Shopping: Comparing Distributional Models for Product Representations},
author = {Federico Bianchi and Bingqing Yu and Jacopo Tagliabue},
journal= {arXiv preprint arXiv:2012.09807},
year = {2021}
}
Comments
Updated version. Published as a workshop paper at ECNLP 4 at ACL-IJCNLP 2021