UNIFIEDQA: Crossing Format Boundaries with a Single QA System
Written for the Research library (claude-opus-5-5) from the paper text. Original — aclanthology.org/2020.findings-emnlp.171.pdf
Summary
QA benchmarks come in several formats: span extraction, free-form generation, multiple choice and yes/no. Models are usually built for just one of them. The authors argue that this split is artificial, because the reasoning a model needs does not depend on how the answer is presented. They build UnifiedQA, a single text-to-text model based on T5, with BART variants, and train it jointly on seed datasets covering all four formats. Each example becomes plain text: the question, then the paragraph and/or lettered answer options, separated by "\n". No task or format prefix is added, so the model must work out the expected answer type from the input itself. Sampling is weighted so that each dataset contributes about equally to a batch, whatever its size. The single model matches eight separately trained dataset-specific systems and transfers well to twelve held-out datasets in familiar formats. With further fine-tuning, it sets new state-of-the-art results on ten datasets. This makes it a practical general-purpose starting point for new QA tasks, especially when training data is limited.
Key points
- The study covers 20 QA datasets in four formats: 5 extractive (e.g., SQuAD 1.1/2.0, NewsQA, Quoref, ROPES), 3 abstractive (NarrativeQA, DROP, NatQA), 9 multiple-choice (e.g., RACE, MCTest, OBQA, ARC, QASC, CommonsenseQA, PIQA, SIQA, WinoGrande) and 3 yes/no (BoolQ, NP-BoolQ, MultiRC).
- Unlike the original T5 multi-task setup, the input carries no dataset or format tags. Restricting the mix to closely related QA tasks is what makes this multi-task training succeed where broader mixes did not.
- Fine-tuning from UnifiedQA beats fine-tuning from plain T5 or BART on every target. This gives new best scores on 10 factoid and commonsense datasets, some of which use extra retrieved context (Aristo corpus for ARC/OBQA, DPR for NatQA).
- In a leave-one-out ablation, BoolQ, SQuAD 2.0, OBQA and NarrativeQA contributed most, one from each format. SQuAD 1.1 mattered least, likely because SQuAD 2.0 largely covers it. This suggests a small set of datasets is enough if every format is represented.
- Dataset size and domain similarity don't reliably predict how well training transfers to another dataset. For example, BoolQ and SQuAD both use Wikipedia paragraphs, yet barely help each other. Code and models are released at github.com/allenai/unifiedqa.
Cited by
- Muppet: Massive Multi-task Representations with Pre-Finetuning
- Explaining Answers with Entailment Trees
- A Dataset of Information-Seeking Questions and Answers Anchored in Research Papers
- QAFactEval: Improved QA-Based Factual Consistency Evaluation for Summarization
- PPT: Pre-trained Prompt Tuning for Few-shot Learning
- Evaluating Large Language Models: A Comprehensive Survey
- Measuring Massive Multitask Language Understanding
- Measuring Mathematical Problem Solving With the MATH Dataset
- Surface Form Competition: Why the Highest Probability Answer Isn't Always Right
- Editing Models with Task Arithmetic
- OPT-IML: Scaling Language Model Instruction Meta Learning through the Lens of Generalization
- Maieutic Prompting: Logically Consistent Reasoning with Recursive Explanations
- TellMeWhy: A Dataset for Answering Why-Questions in Narratives
- Large Language Models with Controllable Working Memory
- From Quantity to Quality: Boosting LLM Performance with Self-Guided Data Selection for Instruction Tuning
- TruthfulQA: Measuring How Models Mimic Human Falsehoods
- Pre-train, Prompt, and Predict: A Systematic Survey of Prompting Methods in Natural Language Processing
- The Flan Collection: Designing Data and Methods for Effective Instruction Tuning
- Dynamic Prompt Learning via Policy Gradient for Semi-structured Mathematical Reasoning
- Chameleon: Plug-and-Play Compositional Reasoning with Large Language Models
- Augmented Language Models: a Survey
- Recent Advances in Natural Language Processing via Large Pre-Trained Language Models: A Survey
- Rethinking the Role of Demonstrations: What Makes In-Context Learning Work?
- SPARTQA: A Textual Question Answering Benchmark for Spatial Reasoning
- LILA: A Unified Benchmark for Mathematical Reasoning
- REFINER: Reasoning Feedback on Intermediate Representations
- Is Reinforcement Learning (Not) for Natural Language Processing: Benchmarks, Baselines, and Building Blocks for Natural Language Policy Optimization
- Leveraging Large Language Models for Multiple Choice Question Answering
- Multitask Prompted Training Enables Zero-Shot Task Generalization
- RAPTOR: Recursive Abstractive Processing for Tree-Organized Retrieval
- Language Models are Multilingual Chain-of-Thought Reasoners
- Aya Dataset: An Open-Access Collection for Multilingual Instruction Tuning
- Beyond the Imitation Game: Quantifying and extrapolating the capabilities of language models
- UL2: Unifying Language Learning Paradigms
- Aya Model: An Instruction Finetuned Open-Access Multilingual Language Model
- A Stitch in Time Saves Nine: Detecting and Mitigating Hallucinations of LLMs by Validating Low-Confidence Generation
- Self-Consistency Improves Chain of Thought Reasoning in Language Models
- Recursively Summarizing Books with Human Feedback
- UnifiedSKG: Unifying and Multi-Tasking Structured Knowledge Grounding with Text-to-Text Language Models
- LinkBERT: Pretraining Language Models with Document Links
- ZeroGen: Efficient Zero-shot Learning via Dataset Generation
- Instruction Tuning for Large Language Models: A Survey
- LLaMA-Adapter: Efficient Fine-tuning of Language Models with Zero-init Attention
- Multimodal Chain-of-Thought Reasoning in Language Models
Cite
@inproceedings{khashabi2020unifiedqa,
title = {{UNIFIEDQA: Crossing Format Boundaries with a Single QA System}},
author = {Khashabi, Daniel and Min, Sewon and Khot, Tushar and Sabharwal, Ashish and Tafjord, Oyvind and Clark, Peter and Hajishirzi, Hannaneh},
booktitle = {Findings of the Association for Computational Linguistics: EMNLP 2020},
year = {2020},
doi = {10.18653/v1/2020.findings-emnlp.171},
eprint = {2005.00700},
archivePrefix = {arXiv},
url = {https://doi.org/10.18653/v1/2020.findings-emnlp.171}
}References
- Naveen Arivazhagan, Ankur Bapna, Orhan Firat, Dmitry Lepikhin, Melvin Johnson, Maxim Krikun, Mia Xu Chen, Yuan Cao, George Foster, Colin Cherry, et al. 2019. Massively multilingual neural machine translation in the wild: Findings and challenges. In NAACL. find our card
- Pratyay Banerjee and Chitta Baral. 2020. Knowledge fusion and semantic knowledge ranking for open domain question answering. arXiv preprint arXiv:2004.03101. — [arXiv:2004.03101](https://arxiv.org/abs/2004.03101) find
- Yonatan Bisk, Rowan Zellers, Ronan Le Bras, Jianfeng Gao, and Yejin Choi. 2020. Piqa: Reasoning about physical commonsense in natural language. In AAAI. find our card
- Rich Caruana. 1997. Multitask learning. Machine learning, 28(1):41–75. find
- Christopher Clark, Kenton Lee, Ming-Wei Chang, Tom Kwiatkowski, Michael Collins, and Kristina Toutanova. 2019a. Boolq: Exploring the surprising difficulty of natural yes/no questions. In NAACL-HLT. find our card
- Kevin Clark, Minh-Thang Luong, Urvashi Khandelwal, Christopher D Manning, and Quoc Le. 2019b. BAM! Born-again multi-task networks for natural language understanding. In ACL, pages 5931–5937. find
- Peter Clark, Isaac Cowhey, Oren Etzioni, Tushar Khot, Ashish Sabharwal, Carissa Schoenick, and Oyvind Tafjord. 2018. Think you have solved question answering? Try ARC, the AI2 reasoning challenge. ArXiv, abs/1803.05457. find
- Peter Clark, Oren Etzioni, Tushar Khot, Bhavana Dalvi Mishra, Kyle Richardson, Ashish Sabharwal, Carissa Schoenick, Oyvind Tafjord, Niket Tandon, Sumithra Bhakthavatsalam, et al. 2019c. From ’F’ to ’A’ on the NY Regents science exams: An overview of the Aristo project. ArXiv, abs/1909.01958. find
- Peter Clark, Oren Etzioni, Tushar Khot, Ashish Sabharwal, Oyvind Tafjord, Peter Turney, and Daniel Khashabi. 2016. Combining retrieval, statistics, and inference to answer elementary science questions. In AAAI. find
- Pradeep Dasigi, Nelson F. Liu, Ana Marasović, Noah A. Smith, and Matt Gardner. 2019. Quoref: A reading comprehension dataset with questions requiring coreferential reasoning. In EMNLP/IJCNLP. find our card
- Jacob Devlin, Ming-Wei Chang, Kenton Lee, and Kristina Toutanova. 2019. BERT: Pre-training of deep bidirectional transformers for language understanding. In NAACL. find
- Dheeru Dua, Ananth Gottumukkala, Alon Talmor, Matt Gardner, and Sameer Singh. 2019a. Comprehensive multi-dataset evaluation of reading comprehension. In 2nd Workshop on Machine Reading for Question Answering. find
- Dheeru Dua, Yizhong Wang, Pradeep Dasigi, Gabriel Stanovsky, Sameer Singh, and Matt Gardner. 2019b. DROP: A reading comprehension benchmark requiring discrete reasoning over paragraphs. In NAACL. find our card
- Adam Fisch, Alon Talmor, Robin Jia, Minjoon Seo, Eunsol Choi, and Danqi Chen. 2019. find
- MRQA 2019 shared task: Evaluating generalization in reading comprehension. In 2nd Workshop on Machine Reading for Question Answering, at EMNLP. find our card
- Matt Gardner, Yoav Artzi, Victoria Basmova, Jonathan Berant, Ben Bogin, Sihao Chen, Pradeep Dasigi, Dheeru Dua, Yanai Elazar, Ananth Gottumukkala, et al. 2020. Evaluating models’ local decision boundaries via contrast sets. In EMNLP - Findings. find our card
- Vladimir Karpukhin, Barlas Oğuz, Sewon Min, Ledell Wu, Sergey Edunov, Danqi Chen, and Wen-tau Yih. 2020. Dense passage retrieval for open-domain question answering. In EMNLP. find our card
- Nitish Shirish Keskar, Bryan McCann, Caiming Xiong, and Richard Socher. 2019. Unifying question answering, text classification, and regression via span extraction. arXiv preprint arXiv:1904.09286. — [arXiv:1904.09286](https://arxiv.org/abs/1904.09286) find
- Daniel Khashabi, Snigdha Chaturvedi, Michael Roth, Shyam Upadhyay, and Dan Roth. 2018. Looking beyond the surface: A challenge set for reading comprehension over multiple sentences. In NAACL-HLT. find our card
- Daniel Khashabi, Tushar Khot, and Ashish Sabharwal. 2020. More bang for your buck: Natural perturbation for robust question answering. In EMNLP. find
- Tushar Khot, Peter Clark, Michal Guerquin, Peter Jansen, and Ashish Sabharwal. 2019. QASC: A dataset for question answering via sentence composition. In AAAI. find
- Tomás Kociský, Jonathan Schwarz, Phil Blunsom, Chris Dyer, Karl Moritz Hermann, Gábor Melis, and Edward Grefenstette. 2018. The narrativeqa reading comprehension challenge. TACL, 6:317–328. find our card
- Tom Kwiatkowski, Jennimaria Palomaki, Olivia Red-field, Michael Collins, Ankur P. Parikh, Chris Alberti, Danielle Epstein, Illia Polosukhin, Jacob Devlin, Kenton Lee, Kristina Toutanova, Llion Jones, Matthew Kelcey, Ming-Wei Chang, Andrew M. Dai, Jakob Uszkoreit, Quoc Le, and Slav Petrov. 2019. Natu… find our card
- Guokun Lai, Qizhe Xie, Hanxiao Liu, Yiming Yang, and Eduard H. Hovy. 2017. RACE: Large-scale reading comprehension dataset from examinations. In EMNLP. find our card
- Zhenzhong Lan, Mingda Chen, Sebastian Goodman, Kevin Gimpel, Piyush Sharma, and Radu Soricut. 2019. Albert: A lite bert for self-supervised learning of language representations. In ICLR. find
- Hector J. Levesque, Ernest Davis, and Leora Morgenstern. 2011. The winograd schema challenge. In KR. find
- Mike Lewis, Yinhan Liu, Naman Goyal, Marjan Ghazvininejad, Abdelrahman Mohamed, Omer Levy, Ves Stoyanov, and Luke Zettlemoyer. 2020. BART: Denoising sequence-to-sequence pre-training for natural language generation, translation, and comprehension. In ACL. find our card
- Chin-Yew Lin, Guihong Cao, Jianfeng Gao, and Jian-Yun Nie. 2006. An information-theoretic approach to automatic evaluation of summaries. In NAACL. find
- Kevin Lin, Oyvind Tafjord, Peter Clark, and Matt Gardner. 2019. Reasoning over paragraph effects in situations. In 2nd Workshop on Machine Reading for Question Answering, at EMNLP. find
- Yinhan Liu, Myle Ott, Naman Goyal, Jingfei Du, Mandar Joshi, Danqi Chen, Omer Levy, Mike Lewis, Luke Zettlemoyer, and Veselin Stoyanov. 2019. Roberta: A robustly optimized bert pretraining approach. arXiv preprint arXiv:1907.11692. find
- Bryan McCann, Nitish Shirish Keskar, Caiming Xiong, and Richard Socher. 2018. The natural language decathlon: Multitask learning as question answering. arXiv preprint arXiv:1806.08730. find our card
- Todor Mihaylov, Peter Clark, Tushar Khot, and Ashish Sabharwal. 2018. Can a suit of armor conduct electricity? a new dataset for open book question answering. In EMNLP. find
- Sewon Min, Danqi Chen, Hannaneh Hajishirzi, and Luke Zettlemoyer. 2019. A discrete hard EM approach for weakly supervised question answering. In EMNLP/IJCNLP. find
- Sewon Min, Julian Michael, Hannaneh Hajishirzi, and Luke Zettlemoyer. 2020. AmbigQA: Answering ambiguous open-domain questions. In EMNLP. find
- Arindam Mitra, Pratyay Banerjee, Kuntal Kumar Pal, Swaroop Mishra, and Chitta Baral. 2020. How additional knowledge can improve natural language commonsense question answering. arXiv: Computation and Language. find
- Kosuke Nishida, Kyosuke Nishida, Masaaki Nagata, Atsushi Otsuka, Itsumi Saito, Hisako Asano, and Junji Tomita. 2019. Answering while summarizing: Multi-task learning for multi-hop qa with evidence extraction. In ACL. find
- Haoruo Peng, Daniel Khashabi, and Dan Roth. 2015. Solving hard coreference problems. In NAACL, pages 809–819. find
- Alec Radford, Jeff Wu, Rewon Child, David Luan, Dario Amodei, and Ilya Sutskever. 2019a. Language models are unsupervised multitask learners. find
- Alec Radford, Jeffrey Wu, Rewon Child, David Luan, Dario Amodei, and Ilya Sutskever. 2019b. Language models are unsupervised multitask learners. OpenAI Blog, 1(8):9. find
- Colin Raffel, Noam Shazeer, Adam Roberts, Katherine Lee, Sharan Narang, Michael Matena, Yanqi Zhou, Wei Li, and Peter J Liu. 2020. Exploring the limits of transfer learning with a unified text-to-text transformer. JMLR. find
- Pranav Rajpurkar, Robin Jia, and Percy Liang. 2018. Know what you don’t know: Unanswerable questions for squad. In ACL. find our card
- Pranav Rajpurkar, Jian Zhang, Konstantin Lopyrev, and Percy Liang. 2016. SQuAD: 100,000+ questions for machine comprehension of text. In EMNLP. find our card
- Matthew Richardson, Christopher J. C. Burges, and Erin Renshaw. 2013. Mctest: A challenge dataset for the open-domain machine comprehension of text. In EMNLP. find our card
- Adam Roberts, Colin Raffel, and Noam Shazeer. 2020. How much knowledge can you pack into the parameters of a language model? In EMNLP. find
- Mrinmaya Sachan and Eric Xing. 2016. Easy questions first? a case study on curriculum learning for question answering. In ACL, pages 453–463. find
- Keisuke Sakaguchi, Ronan Le Bras, Chandra Bhagavatula, and Yejin Choi. 2020. WINOGRANDE: an adversarial winograd schema challenge at scale. In AAAI. find
- Maarten Sap, Hannah Rashkin, Derek Chen, Ronan Le Bras, and Yejin Choi. 2019. Social iqa: Commonsense reasoning about social interactions. In EMNLP-IJCNLP, pages 4453–4463. find
- Elad Segal, Avia Efrat, Mor Shoham, Amir Globerson, and Jonathan Berant. 2019. A simple and effective model for answering multi-span questions. In EMNLP. find
- Alon Talmor and Jonathan Berant. 2019. Multiqa: An empirical investigation of generalization and transfer in reading comprehension. In ACL. find
- Alon Talmor, Jonathan Herzig, Nicholas Lourie, and Jonathan Berant. 2019. Commonsenseqa: A question answering challenge targeting commonsense knowledge. In NAACL-HLT. find
- Adam Trischler, Tong Wang, Xingdi Yuan, Justin Harris, Alessandro Sordoni, Philip Bachman, and Kaheer Suleman. 2017. Newsqa: A machine comprehension dataset. In Rep4NLP@ACL. find our card
- Zhuosheng Zhang, Junjie Yang, and Hai Zhao. 2020. Retrospective reader for machine reading comprehension. ArXiv, abs/2001.09694. — [arXiv:2001.09694](https://arxiv.org/abs/2001.09694) find
- Chen Zhu, Yu Cheng, Zhe Gan, Siqi Sun, Tom Goldstein, and Jingjing Liu. 2020. Freelb: Enhanced adversarial training for natural language understanding. In ICLR. find our card