用裁剪替代丢弃,能显著提升文本嵌入自监督训练效果
Cropping outperforms dropout as an augmentation strategy for self-supervised training of text embeddings
- 用文本裁剪作为数据增强,比传统丢弃法更有效
- 在特定领域数据上,短时间微调即可达到高质量嵌入
- 只需微调最后几层变压器,就能获得接近全模型的效果
文本嵌入在信息检索增强生成、聚类和文本数据探索等任务中至关重要。当前高性能嵌入模型多基于预训练语言模型,通过有监督对比微调获得,依赖外部相似性定义和标注数据构建正样本对。本文研究自监督微调,系统比较了两种主流数据增强策略。在MTEB及特定领域数据集上的评估显示,裁剪增强显著优于基于丢弃的方法。尽管在跨域数据上性能仍低于有监督最优模型,但在特定领域数据上,极短时间的自监督微调即可生成高质量嵌入。此外,发现表示质量随变压器最后一层提升,且仅微调这些层即可达到相近性能。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。