arXiv:2602.03563cs.CL2026-02被引 1

提出ACL框架,让对比学习与分类损失协同工作,提升BERT微调效果。

ACL: Aligned Contrastive Learning Improves BERT and Multi-exit BERT Fine-tuning

  • 用标签嵌入作为增强样本,通过对比学习对齐样本表示与标签
  • 引入ACL-Grad机制,在目标冲突时自动忽略矛盾的对比项
  • 跨层引导浅层出口优化,显著提升多出口BERT的推理速度与精度

尽管对比学习在自监督中表现优异,但在监督设置下研究较少。本文通过预实验发现,交叉熵损失(CE)与对比学习目标常相互冲突,阻碍其在监督任务中的应用。为此,提出新型对齐对比学习(ACL)框架:首先,ACL-Embed将标签嵌入视为带不同标签的增强样本,利用对比学习对齐样本表示与标签嵌入;其次,设计ACL-Grad机制,在两目标冲突时舍弃对比项以促进优化。为提升多出口BERT中间出口性能,进一步提出跨层ACL(ACL-CL),由教师出口指导学生浅层出口优化。在GLUE基准上的大量实验表明:(a) ACL-BERT在多数任务上优于或等同于传统方法;(b) 特别是跨层版本在多出口微调中显著超越基线,为低延迟应用提供更优的质量-速度权衡。

原文摘要 · Abstract (English)

Despite its success in self-supervised learning, contrastive learning is less studied in the supervised setting. In this work, we first use a set of pilot experiments to show that in the supervised setting, the cross-entropy loss objective (CE) and the contrastive learning objective often conflict with each other, thus hindering the applications of CL in supervised settings. To resolve this problem, we introduce a novel \underline{A}ligned \underline{C}ontrastive \underline{L}earning (ACL) framework. First, ACL-Embed regards label embeddings as extra augmented samples with different labels and employs contrastive learning to align the label embeddings with its samples' representations. Second, to facilitate the optimization of ACL-Embed objective combined with the CE loss, we propose ACL-Grad, which will discard the ACL-Embed term if the two objectives are in conflict. To further enhance the performances of intermediate exits of multi-exit BERT, we further propose cross-layer ACL (ACL-CL), which is to ask the teacher exit to guide the optimization of student shallow exits. Extensive experiments on the GLUE benchmark results in the following takeaways: (a) ACL-BRT outperforms or performs comparably with CE and CE+SCL on the GLUE tasks; (b) ACL, especially CL-ACL, significantly surpasses the baseline methods on the fine-tuning of multi-exit BERT, thus providing better quality-speed tradeoffs for low-latency applications.

对比学习BERT微调多出口模型高效推理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。