PACT: Pretraining with Adversarial Contrastive Learning for Text Classification

作者: Md Tawkat Islam Khondaker, Muhammad Abdul-Mageed, Laks Lakshmanan. V.S.

摘要

We present PACT ( P retraining with A dversarial C ontrastive Learning for T ext Classification), a novel self-supervised framework for text classification. Instead of contrasting against in-batch negatives, a popular approach in the literature, PACT mines negatives closer to the anchor representation. PACT operates by en-dowing the standard pretraining mechanisms of BERT with adversarial contrastive learning objectives, allowing for effective joint optimization of token-and sentence-level pretraining of the BERT model. Our experiments on 13 diverse datasets including token-level, single-sentence, and sentence-pair text classification tasks show that PACT achieves consistent improvements over SOTA baselines. We further show that PACT regularizes both token-level and sentence-level embedding spaces into more uniform representations, thereby alleviating the undesirable anisotropic phenomenon of language models. 1

PDF打开原文