arXiv · 2208.06610
MetricBERT: Text Representation Learning via Self-Supervised Triplet Training
Abstract
We present MetricBERT, a BERT-based model that learns to embed text under a well-defined similarity metric while simultaneously adhering to the ``traditional'' masked-language task. We focus on downstream tasks of learning similarities for recommendations where we show that MetricBERT outperforms state-of-the-art alternatives, sometimes by a substantial margin. We conduct extensive evaluations of our method and its different variants, showing that our training objective is highly beneficial over a traditional contrastive loss, a standard cosine similarity objective, and six other baselines. As an additional contribution, we publish a dataset of video games descriptions along with a test set of similarity annotations crafted by a domain expert.
Explore related subjects
Keep this discovery
Itzik Malkiel, Dvir Ginzburg, Oren Barkan, Avi Caciularu, Yoni Weill, Noam Koenigstein. 2022-08-13. MetricBERT: Text Representation Learning via Self-Supervised Triplet Training. https://arxiv.org/abs/2208.06610
Cite the original work for its findings. Save a collection to share your selection of sources.