ConSERT: A Contrastive Framework for Self-Supervised Sentence Representation Transfer
Yuanmeng Yan Thanks: Work done during internship at Meituan Inc. The first two authors contribute equally. Weiran Xu is the corresponding author. Affiliation: Beijing University of Posts and Telecommunications, Beijing, China Rumei Li Sirui Wang Affiliation: Meituan Inc., Beijing, Fuzheng Zhang Affiliation: Meituan Inc., Beijing, Wei Wu Affiliation: Meituan Inc., Beijing, Weiran Xu Affiliation: Beijing University of Posts and Telecommunications, Beijing, China
Abstract
Learning high-quality sentence representations benefits a wide range of natural language processing tasks. Though BERT-based pre-trained language models achieve high performance on many downstream tasks, the native derived sentence representations are proved to be collapsed and thus produce a poor performance on the semantic textual similarity (STS) tasks. In this paper, we present ConSERT, a Contrastive Framework for Self-Supervised SEntence Representation Transfer, that adopts contrastive learning to fine-tune BERT in an unsupervised and effective way. By making use of unlabeled texts, ConSERT solves the collapse issue of BERT-derived sentence representations and make them more applicable for downstream tasks. Experiments on STS datasets demonstrate that ConSERT achieves an 8% relative improvement over the previous state-of-the-art, even comparable to the supervised SBERT-NLI. And when further incorporating NLI supervision, we achieve new state-of-the-art performance on STS tasks. Moreover, ConSERT obtains comparable results with only 1000 samples available, showing its robustness in data scarcity scenarios.
原文 arXiv:2105.11741;中英对照 + 大白话阅读 https://aha.fim.ai/paper/2105.11741v1