用两层对比学习优化BERT,提升句子嵌入的语义表达能力
2-Tier SimCSE: Elevating BERT for Robust Sentence Embeddings
- 采用两层对比学习框架,融合无监督与有监督训练提升语义捕捉
- 在STS任务上达到0.742的平均测试得分,优于单层模型
- 适合需要高质量句子表示的文本匹配、情感分析等场景
有效的句子嵌入需精准捕捉语义细微差别并具备跨场景泛化能力。本文基于SimCSE方法,对minBERT模型在情感分析、语义文本相似度(STS)和同义句检测任务上进行微调。通过实验比较标准丢弃、课程丢弃和自适应丢弃三种策略以缓解过拟合,提出一种新型两层SimCSE微调模型,结合无监督与有监督的SimCSE训练范式,并探索了在同义句和SST任务上的迁移学习潜力。结果表明,该模型在STS任务上表现最优,三项下游任务平均测试得分为0.742。错误分析显示其在处理复杂情感时仍存挑战,且同义句检测依赖词汇重叠。消融实验发现,在单任务无监督SimCSE模型中移除自适应丢弃反而提升了性能,说明额外参数导致过拟合。从STS任务向同义句和SST任务迁移知识未带来性能提升,表明知识迁移能力有限。
原文摘要 · Abstract (English)
Effective sentence embeddings that capture semantic nuances and generalize well across diverse contexts are crucial for natural language processing tasks. We address this challenge by applying SimCSE (Simple Contrastive Learning of Sentence Embeddings) using contrastive learning to fine-tune the minBERT model for sentiment analysis, semantic textual similarity (STS), and paraphrase detection. Our contributions include experimenting with three different dropout techniques, namely standard dropout, curriculum dropout, and adaptive dropout, to tackle overfitting, proposing a novel 2-Tier SimCSE Fine-tuning Model that combines both unsupervised and supervised SimCSE on STS task, and exploring transfer learning potential for Paraphrase and SST tasks. Our findings demonstrate the effectiveness of SimCSE, with the 2-Tier model achieving superior performance on the STS task, with an average test score of 0.742 across all three downstream tasks. The results of error analysis reveals challenges in handling complex sentiments and reliance on lexical overlap for paraphrase detection, highlighting areas for future research. The ablation study revealed that removing Adaptive Dropout in the Single-Task Unsupervised SimCSE Model led to improved performance on the STS task, indicating overfitting due to added parameters. Transfer learning from SimCSE models on Paraphrase and SST tasks did not enhance performance, suggesting limited transferability of knowledge from the STS task.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。