ELECTRA is a Zero-Shot Learner, Too

Shiwen Ni, Hung-Yu Kao
7/17/2022
cs.CL

No official code found. Best implementation: nishiwen1214/Zero-Few-Shot-ELECTRA (9 stars, MIT, last commit 2022-08-16).

Abstract

Recently, for few-shot or even zero-shot learning, the new paradigm "pre-train, prompt, and predict" has achieved remarkable achievements compared with the "pre-train, fine-tune" paradigm. After the success of prompt-based GPT-3, a series of masked language model (MLM)-based (e.g., BERT, RoBERTa) prompt learning methods became popular and widely used. However, another efficient pre-trained discriminative model, ELECTRA, has probably been neglected. In this paper, we attempt to accomplish several NLP tasks in the zero-shot scenario using a novel our proposed replaced token detection (RTD)-based prompt learning method. Experimental results show that ELECTRA model based on RTD-prompt learning achieves surprisingly state-of-the-art zero-shot performance. Numerically, compared to MLM-RoBERTa-large and MLM-BERT-large, our RTD-ELECTRA-large has an average of about 8.4% and 13.7% improvement on all 15 tasks. Especially on the SST-2 task, our RTD-ELECTRA-large achieves an astonishing 90.1% accuracy without any training data. Overall, compared to the pre-trained masked language models, the pre-trained replaced token detection model performs better in zero-shot learning. The source code is available at: https://github.com/nishiwen1214/RTD-ELECTRA.

View on arXivView PDF

Code Implementations(1)

MIT

Code of paper: "ELECTRA is a Zero-shot learner, Too" -- Prompt-based ELECTRA for zero-shot learning.

90PythonDec 5, 20214 years agoMIT
bert4keraselectraprompt-learningrtdzero-shot-learning

Similar Papers

Cite this paper

@article{ni2022electra,
  title  = {ELECTRA is a Zero-Shot Learner, Too},
  author = {Shiwen Ni and Hung-Yu Kao},
  year   = {2022},
  eprint = {2207.08141},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2207.08141v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2207.08141.svg)](https://codeofpaper.com/paper/2207.08141)

Discussion