We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Computation and Language

Title: 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.
Subjects: Computation and Language (cs.CL)
Cite as: arXiv:2208.06610 [cs.CL]
  (or arXiv:2208.06610v1 [cs.CL] for this version)

Submission history

From: Itzik Malkiel [view email]
[v1] Sat, 13 Aug 2022 09:52:58 GMT (5651kb,D)

Link back to: arXiv, form interface, contact.