Cropping outperforms dropout as an augmentation strategy for self-supervised training of text embeddings

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: González-Márquez, Rita, Berens, Philipp, Kobak, Dmitry
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917346305638400
author González-Márquez, Rita
Berens, Philipp
Kobak, Dmitry
author_facet González-Márquez, Rita
Berens, Philipp
Kobak, Dmitry
contents Text embeddings, i.e. vector representations of entire texts, play an important role in many NLP applications, such as retrieval-augmented generation, clustering, or visualizing collections of texts for data exploration. Currently, top-performing embedding models are derived from pre-trained language models via supervised contrastive fine-tuning. This fine-tuning strategy relies on an external notion of similarity and annotated data for generation of positive pairs. Here we study self-supervised fine-tuning and systematically compare the two most well-known augmentation strategies used for fine-tuning text embeddings models. We assess embedding quality on MTEB and additional in-domain evaluations and show that cropping augmentation strongly outperforms the dropout-based approach. We find that on out-of-domain data, the quality of resulting embeddings is substantially below the supervised state-of-the-art models, but for in-domain data, self-supervised fine-tuning can produce high-quality text embeddings after very short fine-tuning. Finally, we show that representation quality increases towards the last transformer layers, which undergo the largest change during fine-tuning; and that fine-tuning only those last layers is sufficient to reach similar embedding quality.
format Preprint
id arxiv_https___arxiv_org_abs_2508_03453
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Cropping outperforms dropout as an augmentation strategy for self-supervised training of text embeddings
González-Márquez, Rita
Berens, Philipp
Kobak, Dmitry
Computation and Language
Machine Learning
Text embeddings, i.e. vector representations of entire texts, play an important role in many NLP applications, such as retrieval-augmented generation, clustering, or visualizing collections of texts for data exploration. Currently, top-performing embedding models are derived from pre-trained language models via supervised contrastive fine-tuning. This fine-tuning strategy relies on an external notion of similarity and annotated data for generation of positive pairs. Here we study self-supervised fine-tuning and systematically compare the two most well-known augmentation strategies used for fine-tuning text embeddings models. We assess embedding quality on MTEB and additional in-domain evaluations and show that cropping augmentation strongly outperforms the dropout-based approach. We find that on out-of-domain data, the quality of resulting embeddings is substantially below the supervised state-of-the-art models, but for in-domain data, self-supervised fine-tuning can produce high-quality text embeddings after very short fine-tuning. Finally, we show that representation quality increases towards the last transformer layers, which undergo the largest change during fine-tuning; and that fine-tuning only those last layers is sufficient to reach similar embedding quality.
title Cropping outperforms dropout as an augmentation strategy for self-supervised training of text embeddings
topic Computation and Language
Machine Learning
url https://arxiv.org/abs/2508.03453