TextBox 2.0: A Text Generation Library with Pre-trained Language Models
Not on the map. Why?
Abstract (arXiv)
To facilitate research on text generation, this paper presents a comprehensive and unified library, TextBox 2.0, focusing on the use of pre-trained language models (PLMs). To be comprehensive, our library covers 13 common text generation tasks and their corresponding 83 datasets and further incorporates 45 PLMs covering general, translation, Chinese, dialogue, controllable, distilled, prompting, and lightweight PLMs. We also implement 4 efficient training strategies and provide 4 generation objectives for pre-training new PLMs from scratch. To be unified, we design the interfaces to support the entire research pipeline (from data loading to training and evaluation), ensuring that each step can be fulfilled in a unified way. Despite the rich functionality, it is easy to use our library, either through the friendly Python API or command line. To validate the effectiveness of our library, we conduct extensive experiments and exemplify four types of research scenarios. The project is released at the link: https://github.com/RUCAIBox/TextBox.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Builds on 47 of its 67 references that are in the map. First posted on arXiv 26 Dec 2022.
Builds on
47 of 67 references.
References · 47
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2019BART: Denoising Sequence-to-Sequence Pre-training for Natural Language Generation, Translation, and Comprehensioncited by 5,874
- 2019BERTScore: Evaluating Text Generation with BERTcited by 5,462
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,243
- 2019Parameter-Efficient Transfer Learning for NLPcited by 4,195
- 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,115
- 2021Prefix-Tuning: Optimizing Continuous Prompts for Generationcited by 3,878
- 2020Don't Stop Pretraining: Adapt Language Models to Domains and Taskscited by 1,548
- 2019fairseq: A Fast, Extensible Toolkit for Sequence Modelingcited by 1,486
- 2018Don't Give Me the Details, Just the Summary! Topic-Aware Convolutional Neural Networks for Extreme Summarizationcited by 1,308
- 2017Get To The Point: Summarization with Pointer-Generator Networkscited by 1,281
- 2019PEGASUS: Pre-training with Extracted Gap-sentences for Abstractive Summarizationcited by 1,100
35 more
- 2019Cross-lingual Language Model Pretrainingcited by 1,088
- 2015A Diversity-Promoting Objective Function for Neural Conversation Modelscited by 1,068
- 2020Multilingual Denoising Pre-training for Neural Machine Translationcited by 1,028
- 2019DialoGPT: Large-Scale Generative Pre-training for Conversational Response Generationcited by 842
- 2018Personalizing Dialogue Agents: I have a dog, do you have pets too?cited by 820
- 2019Plug and Play Language Models: A Simple Approach to Controlled Text Generationcited by 793
- 2018MultiWOZ -- A Large-Scale Multi-Domain Wizard-of-Oz Dataset for Task-Oriented Dialogue Modellingcited by 760
- 2019CTRL: A Conditional Transformer Language Model for Controllable Generationcited by 754
- 2020Recipes for building an open-domain chatbotcited by 593
- 2018Texygen: A Benchmarking Platform for Text Generation Modelscited by 484
- 2016A Corpus and Evaluation Framework for Deeper Understanding of Commonsense Storiescited by 413
- 2019MASS: Masked Sequence to Sequence Pre-training for Language Generationcited by 345
- 2021The GEM Benchmark: Natural Language Generation, its Evaluation and Metricscited by 239
- 2018Dear Sir or Madam, May I introduce the GYAFC Dataset: Corpus, Benchmarks and Metrics for Formality Style Transfercited by 209
- 2017A Survey on Dialogue Systems: Recent Advances and New Frontierscited by 162
- 2017ParlAI: A Dialog Research Software Platformcited by 139
- 2020MinTL: Minimalist Transfer Learning for Task-Oriented Dialogue Systemscited by 103
- 2021CPT: A Pre-Trained Unbalanced Transformer for Both Chinese Language Understanding and Generationcited by 81
- 2020Pre-trained Summarization Distillationcited by 70
- 2015LCSTS: A Large Scale Chinese Short Text Summarization Datasetcited by 65
- 2019Long and Diverse Text Generation with Planning-based Hierarchical Variational Modelcited by 62
- 2020CPM: A Large-scale Generative Chinese Pre-trained Language Modelcited by 60
- 2021JointGT: Graph-Text Joint Representation Learning for Text Generation from Knowledge Graphscited by 58
- 2021ExplainaBoard: An Explainable Leaderboard for NLPcited by 50
- 2020Syntax-guided Controlled Generation of Paraphrasescited by 47
- 2020GLGE: A New General Language Generation Evaluation Benchmarkcited by 47
- 2021Long Text Generation by Modeling Sentence-Level and Discourse-Level Coherencecited by 45
- 2021Thank you BART! Rewarding Pre-Trained Models Improves Formality Style Transfercited by 28
- 2018OpenNMT: Neural Machine Translation Toolkitcited by 27
- 2021Training Dynamics for Text Summarization Modelscited by 25
- 2022Learning to Transfer Prompts for Text Generationcited by 22
- 2021TextBox: A Unified, Modularized, and Extensible Framework for Text Generationcited by 15
- 2021Attention-guided Generative Models for Extractive Question Answeringcited by 14
- 2021FastSeq: Make Sequence Generation Fastercited by 13
- 2022Context-Tuning: Learning Contextualized Prompts for Natural Language Generationcited by 11
Built on by
Citing papers per year: 2023: 2; 2024: 1.
3 papers.