Source-linked AI summary
A Unified Model for Extractive and Abstractive Summarization using Inconsistency Loss
Wan-Ting Hsu, Chieh-Kai Lin, Ming-Ying Lee, Kerui Min, Jing Tang, Min Sun
TL;DR
Extractive summarization can select important content but may be less readable, while abstractive summarization can be readable yet lose or mistake facts. The paper unifies both through cross-level attention modulation and inconsistency loss, achieving state-of-the-art ROUGE scores and the strongest reported human-evaluation informativity and readability on CNN/Daily Mail.
Problem
Extractive summaries can be less readable, whereas abstractive summaries may lose or mistake factual details.
Source: §1 · p. 1Method
The model combines sentence-level and word-level attention, modulates word attention using sentence attention, and trains with inconsistency loss.
Source: §1 · p. 2Results
Takeaways & Limitations
Abstract
from arXiv · showhide
We propose a unified model combining the strength of extractive and abstractive summarization. On the one hand, a simple extractive model can obtain sentence-level attention with high ROUGE scores but less readable. On the other hand, a more complicated abstractive model can obtain word-level dynamic attention to generate a more readable paragraph. In our model, sentence-level attention is used to modulate the word-level attention such that words in less attended sentences are less likely to be generated. Moreover, a novel inconsistency loss function is introduced to penalize the inconsistency between two levels of attentions. By end-to-end training our model with the inconsistency loss and original losses of extractive and abstractive models, we achieve state-of-the-art ROUGE scores while being the most informative and readable summarization on the CNN/Daily Mail dataset in a solid human evaluation.
1 Introduction
Text summarization condenses text while preserving important points, with extractive methods favoring simplicity and abstractive methods favoring readable, concise generation. The paper unifies both approaches by combining sentence- and word-level attention and adding inconsistency loss.
- Text summarization condenses text while maintaining important points for applications including news digests, search results, and reports.Source: §1 · p. 1
- Extractive methods select source sentences and are typically simpler, whereas abstractive methods can generate novel words and phrases.Source: §1 · p. 1
- Abstractive summaries can be more coherent and concise than extractive summaries, but abstractive systems may inaccurately reproduce factual details.Source: §1 · pp. 1–2
- The unified model uses sentence-level attention to modulate word-level attention, making words in less attended sentences less likely to be generated.Source: §1 · p. 2
- The proposed inconsistency loss encourages consistency between attention levels without requiring additional human annotation.Source: §1 · p. 2
- The model achieves the best ROUGE scores on CNN/Daily Mail and outperforms recent state-of-the-art methods in informativity and readability in human evaluation.Source: §1 · p. 2
2 Related Work
Related work covers neural extractive and abstractive summarization, including mechanisms for handling unknown words and repetition. The paper combines sentence-level attention from an extractive model with word-level attention from an abstractive model and adds inconsistency loss.
Extractive summarization
- Prior work includes neural models that represent sentences and select them for extractive summarization.Source: §2 · p. 2
Abstractive summarization
- Pointer-generator and related models address out-of-vocabulary words and repeated phrases in generated summaries.Source: §2 · p. 3
Unified model
- The proposed model combines sentence-level attention from Nallapati et al. (2017) with word-level attention from See et al. (2017).Source: §2 · p. 3
- An inconsistency loss is introduced to enhance cooperation between the extractive and abstractive models.Source: §2 · p. 3
3 Our Unified Model
The unified model combines sentence-level extraction with word-level abstractive generation, using sentence attention to modulate word attention and an inconsistency loss to align them during training.
3 Our Unified Model
- The model combines sentence-level attention from an extractor with dynamic word-level attention from an abstracter.The extractor produces sentence probabilities, while the abstracter generates summary text word by word.Source: §3 · p. 3
3.1 Combining Attentions
- Sentence-level attention modulates word-level attention so words in less attended sentences are less likely to be generated.The attentions are combined by scalar multiplication and renormalization; high word attention requires both levels to be high.Source: §3.1 · pp. 3–4
3.2 Inconsistency Loss
- The inconsistency loss encourages sentence attention to be high when word attention is high and makes the two attention levels mutually beneficial.It uses the top K attended words and summary length T, while the original extractor, abstracter, and coverage losses prevent degenerate attention distributions.Source: §3.2 · p. 4
3.3 Extractor
- The extractor uses a hierarchical bidirectional GRU and predicts sentence-level attention, with labels selected to favor highly informative sentences.Ground-truth labels are constructed using ROUGE-L recall against the reference abstractive summary, prioritizing sentences that increase collective informativity.Source: §3.3 · p. 4
3.4 Abstracter
- The updated word attention affects decoding by producing a context vector that updates the final word distribution.The abstracter uses a pointer-generator network that can copy article words or generate words from a fixed vocabulary, including OOV words.Source: Figure 4 · p. 5, §3.4 · p. 5
3.5 Training Procedure
- End-to-end training minimizes extractor, abstractive, coverage, and inconsistency losses together.The authors also pre-train both components; during end-to-end training, sentence attention is soft and the extractor loss may receive a larger weight.Source: §3.5 · pp. 5–6
4 Experiments
The experiments use the CNN/Daily Mail news dataset and describe the model’s training and implementation settings, including pre-training and two-stage training.
- The CNN/Daily Mail dataset contains news stories paired with human-written multi-sentence summaries, with anonymized and non-anonymized versions.Source: §4.1 · p. 6
- The extractor and abstracter use 128-dimensional word embeddings, vocabularies of 50k, hidden dimensions of 200 and 256, Adagrad, early stopping, and summaries limited to 120 tokens.Source: §4.2 · p. 6
Pre-training
- During pre-training, the extractor runs for 27k iterations, while the abstracter uses ground-truth extracted sentences and trains with and without coverage mechanisms.Source: §4.2 · p. 6
Two-stages training
- During two-stage training, the abstracter receives sentences whose pre-trained extractor probabilities βn exceed 0.5 before fine-tuning.Source: §4.2 · p. 6
5 Results
The experiments evaluate extracted sentences, generated summaries, attention consistency, and human judgments. End-to-end training with inconsistency loss achieves strong ROUGE results, reduces inconsistency, and improves reported human-evaluation quality.
Extracted sentences
- The extractor is evaluated using full-length ROUGE recall between selected sentences and reference abstractive summaries, selecting sentences with β greater than 0.5.Source: §5.1 · p. 6
Abstractive summarization
- The two-stage model outperforms the pointer-generator model on ROUGE-1 and ROUGE-2, while end-to-end training with inconsistency loss exceeds the lead-3 baseline.Source: Table 2 · p. 7
Extracted sentences
- The ground-truth extracted sentences are an upper bound but cannot reach 100% recall because abstractive references may contain words absent from the article.Source: §5.1 · p. 7, Table 1 · p. 7
- The end-to-end extractor performs best when trained with inconsistency loss.Source: §5.1 · p. 7
Abstractive summarization
- The end-to-end model without inconsistency loss already improves ROUGE through cooperation between the extractor and abstracter, while inconsistency loss yields state-of-the-art ROUGE scores.Source: §5.2 · p. 7
Attention consistency
- The inconsistency rate counts decoder steps whose maximally attended word belongs to a sentence with below-mean sentence attention.Source: §5.2 · p. 7
- 20% to 4%: inconsistency loss significantly decreases the average inconsistency rate on the test set.Source: §5.2 · p. 7
Human evaluation
- Human evaluation scores informativity, conciseness, and readability on a 1-to-5 scale, using a random-summary trap to reject unreliable evaluations.Source: §5.3 · p. 8
6 Conclusion
The paper concludes that its unified model combines extractive and abstractive summarization, with inconsistency loss aligning their attentions. End-to-end training achieves strong ROUGE performance and favorable human-evaluation results on CNN/Daily Mail.
- The unified model combines the strengths of extractive and abstractive summarization through sentence-level and word-level attention.Source: §6 · p. 9
- The inconsistency loss penalizes disagreement between the two attention levels and enables extractive and abstractive summarization to be mutually beneficial.Source: §6 · p. 9
- End-to-end training achieves the best reported ROUGE-recall and ROUGE results while producing the most informative and readable summaries in the CNN/Daily Mail human evaluation.Source: §6 · p. 9
Related papers
This paper cites
- Attention Is All You Need2017
- Neural Machine Translation by Jointly Learning to Align and Translate2014
- Teaching Machines to Read and Comprehend2015
- Get To The Point: Summarization with Pointer-Generator Networks2017
- A Neural Attention Model for Abstractive Sentence Summarization2015
- Incorporating Copying Mechanism in Sequence-to-Sequence Learning2016