Recursively Summarizing Books with Human Feedback
arXiv:2109.10862
Abstract
A major challenge for scaling machine learning is training models to perform tasks that are very difficult or time-consuming for humans to evaluate. We present progress on this problem on the task of abstractive summarization of entire fiction novels. Our method combines learning from human feedback with recursive task decomposition: we use models trained on smaller parts of the task to assist humans in giving feedback on the broader task. We collect a large volume of demonstrations and comparisons from human labelers, and fine-tune GPT-3 using behavioral cloning and reward modeling to do summarization recursively. At inference time, the model first summarizes small sections of the book and then recursively summarizes these summaries to produce a summary of the entire book. Our human labelers are able to supervise and evaluate the models quickly, despite not having read the entire books themselves. Our resulting model generates sensible summaries of entire books, even matching the quality of human-written summaries in a few cases ( of books). We achieve state-of-the-art results on the recent BookSum dataset for book-length summarization. A zero-shot question-answering model using these summaries achieves state-of-the-art results on the challenging NarrativeQA benchmark for answering questions about books and movie scripts. We release datasets of samples from our model.
References in corpus (14)
- Language Models are Few-Shot Learners
- An Actor-Critic Algorithm for Sequence Prediction
- Way Off-Policy Batch Deep Reinforcement Learning of Implicit Human Preferences in Dialog
- Generating Wikipedia by Summarizing Long Sequences
- HIBERT: Document Level Pre-training of Hierarchical Bidirectional Transformers for Document Summarization
- Hierarchical Transformers for Multi-Document Summarization
- Asking and Answering Questions to Evaluate the Factual Consistency of Summaries
- BIGPATENT: A Large-Scale Dataset for Abstractive and Coherent Summarization
- SUPERT: Towards New Frontiers in Unsupervised Evaluation Metrics for Multi-Document Summarization
- SEAL: Segment-wise Extractive-Abstractive Long-form Text Summarization
- Hidden Incentives for Auto-Induced Distributional Shift
- Learning to Compare for Better Training and Evaluation of Open Domain Natural Language Generation Models
- On Generating Extended Summaries of Long Documents
- Frustratingly Hard Evidence Retrieval for QA Over Books