Batch-Softmax Contrastive Loss for Pairwise Sentence Scoring Tasks

Anton Chernyavskiy, Dmitry Ilvovsky, Pavel Kalinin, Preslav Nakov


Abstract
The use of contrastive loss for representation learning has become prominent in computer vision, and it is now getting attention in Natural Language Processing (NLP).Here, we explore the idea of using a batch-softmax contrastive loss when fine-tuning large-scale pre-trained transformer models to learn better task-specific sentence embeddings for pairwise sentence scoring tasks.We introduce and study a number of variations in the calculation of the loss as well as in the overall training procedure; in particular, we find that a special data shuffling can be quite important.Our experimental results show sizable improvements on a number of datasets and pairwise sentence scoring tasks including classification, ranking, and regression.Finally, we offer detailed analysis and discussion, which should be useful for researchers aiming to explore the utility of contrastive loss in NLP.
Anthology ID:
2022.naacl-main.9
Volume:
Proceedings of the 2022 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies
Month:
July
Year:
2022
Address:
Seattle, United States
Venue:
NAACL
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
116–126
Language:
URL:
https://aclanthology.org/2022.naacl-main.9
DOI:
10.18653/v1/2022.naacl-main.9
Bibkey:
Cite (ACL):
Anton Chernyavskiy, Dmitry Ilvovsky, Pavel Kalinin, and Preslav Nakov. 2022. Batch-Softmax Contrastive Loss for Pairwise Sentence Scoring Tasks. In Proceedings of the 2022 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies, pages 116–126, Seattle, United States. Association for Computational Linguistics.
Cite (Informal):
Batch-Softmax Contrastive Loss for Pairwise Sentence Scoring Tasks (Chernyavskiy et al., NAACL 2022)
Copy Citation:
PDF:
https://aclanthology.org/2022.naacl-main.9.pdf