Push and Pull: Training Sentence Encoders with Contrastive Losses for Distance-Based Multi-Label Text Classification

Jens Van Nooten, Andriy Kosar


Abstract
Despite the potential of Distance-Based Classification (DBC), a method that assigns labels to text by measuring semantic similarity between the text and the label representations, it has received very little attention for Multi-Label Text Classification (MLTC). Previous studies have focused on determining optimal thresholds, reaching promising results with contextual sentence encoders. We demonstrate that the performance of these models can be further improved by training them with contrastive losses, i.e., by bringing text representations closer to the corresponding true label representations in an embedding space. Using three supervised contrastive losses and three sentence encoders (Stella, GIST-Large, and BGE), we evaluated our approach on five English datasets (SemEval, BioTech, Reuters, AAPD, and LitCovid) and one Dutch dataset (EventDNA). The results show consistent substantial improvements over base sentence encoders, thereby narrowing the gap between DBC methods and fine-tuned or zero-shot approaches.
Anthology ID:
2026.lrec-main.583
Volume:
Proceedings of the Fifteenth Language Resources and Evaluation Conference
Month:
May
Year:
2026
Address:
Palma de Mallorca, Spain
Editors:
Stelios Piperidis, Núria Bel, Henk van den Heuvel, Nancy Ide, Simon Krek, Antonio Toral
Venue:
LREC
SIG:
Publisher:
ELRA Language Resource Association
Note:
Pages:
7359–7379
Language:
URL:
https://preview.aclanthology.org/ingest-lrec/2026.lrec-main.583/
DOI:
Bibkey:
Cite (ACL):
Jens Van Nooten and Andriy Kosar. 2026. Push and Pull: Training Sentence Encoders with Contrastive Losses for Distance-Based Multi-Label Text Classification. International Conference on Language Resources and Evaluation, main:7359–7379.
Cite (Informal):
Push and Pull: Training Sentence Encoders with Contrastive Losses for Distance-Based Multi-Label Text Classification (Van Nooten & Kosar, LREC 2026)
Copy Citation:
PDF:
https://preview.aclanthology.org/ingest-lrec/2026.lrec-main.583.pdf