Guess the Instruction! Flipped Learning Makes Language Models Stronger Zero-Shot Learners

6 Oct 2022  ·  Seonghyeon Ye, Doyoung Kim, Joel Jang, Joongbo Shin, Minjoon Seo ·

Meta-training, which fine-tunes the language model (LM) on various downstream tasks by maximizing the likelihood of the target label given the task instruction and input instance, has improved the zero-shot task generalization performance. However, meta-trained LMs still struggle to generalize to challenging tasks containing novel labels unseen during meta-training. In this paper, we propose Flipped Learning, an alternative method of meta-training which trains the LM to generate the task instruction given the input instance and label. During inference, the LM trained with Flipped Learning, referred to as Flipped, selects the label option that is most likely to generate the task instruction. On 14 tasks of the BIG-bench benchmark, the 11B-sized Flipped outperforms zero-shot T0-11B and even a 16 times larger 3-shot GPT-3 (175B) on average by 8.4% and 9.7% points, respectively. Flipped gives particularly large improvements on tasks with unseen labels, outperforming T0-11B by up to +20% average F1 score. This indicates that the strong task generalization of Flipped comes from improved generalization to novel labels. We release our code at https://github.com/seonghyeonye/Flipped-Learning.

PDF Abstract

Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Result Benchmark
Natural Language Inference ANLI test Flipped-3B A1 39.99 # 12
A2 37.05 # 18
A3 37.73 # 19
Question Answering COPA Flipped-3B Accuracy 89.88 # 19
Sentence Completion HellaSwag Flipped-3B Accuracy 41.6 # 70
Natural Language Inference RTE Flipped-3B Accuracy 71.05 # 52
Question Answering StoryCloze Flipped-3B Accuracy 95.88 # 2
Common Sense Reasoning WinoGrande Flipped-3B Accuracy 58.56 # 52

Results from Other Papers


Task Dataset Model Metric Name Metric Value Rank Source Paper Compare
Coreference Resolution Winograd Schema Challenge Flipped-3B Accuracy 58.37 # 62
Word Sense Disambiguation Words in Context Flipped-3B Accuracy 50.42 # 33

Methods