Attention Temperature Matters in Abstractive Summarization Distillation
Not on the map. Why?
Abstract (arXiv)
Recent progress of abstractive text summarization largely relies on large pre-trained sequence-to-sequence Transformer models, which are computationally expensive. This paper aims to distill these large models into smaller ones for faster inference and minimal performance loss. Pseudo-labeling based methods are popular in sequence-to-sequence model distillation. In this paper, we find simply manipulating attention temperatures in Transformers can make pseudo labels easier to learn for student models. Our experiments on three summarization datasets show our proposed method consistently improves over vanilla pseudo-labeling based methods. We also find that both the pseudo labels and summaries produced by our students are shorter and more abstractive. Our code is available at urlhttps://github.com/Shengqiang-Zhang/plate.
Key facts. Cited by 14 papers in the map (data through 7 Oct 2026). Builds on 38 of its 51 references that are in the map. First posted on arXiv 7 Jun 2021.
Builds on
38 of 51 references.
References · 38
- 2017Attention Is All You Needcited by 59,932
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2019BART: Denoising Sequence-to-Sequence Pre-training for Natural Language Generation, Translation, and Comprehensioncited by 5,874
- 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,854
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,243
- 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,115
- 2015Neural Machine Translation of Rare Words with Subword Unitscited by 2,898
- 2019The Curious Case of Neural Text Degenerationcited by 2,881
- 2014FitNets: Hints for Thin Deep Netscited by 1,619
26 more
- 2019fairseq: A Fast, Extensible Toolkit for Sequence Modelingcited by 1,486
- 2020MiniLM: Deep Self-Attention Distillation for Task-Agnostic Compression of Pre-Trained Transformerscited by 1,414
- 2018Don't Give Me the Details, Just the Summary! Topic-Aware Convolutional Neural Networks for Extreme Summarizationcited by 1,308
- 2019Self-training with Noisy Student improves ImageNet classificationcited by 1,304
- 2017Get To The Point: Summarization with Pointer-Generator Networkscited by 1,281
- 2019TinyBERT: Distilling BERT for Natural Language Understandingcited by 1,153
- 2019PEGASUS: Pre-training with Extracted Gap-sentences for Abstractive Summarizationcited by 1,100
- 2016Paying More Attention to Attention: Improving the Performance of Convolutional Neural Networks via Attention Transfercited by 1,059
- 2015Teaching Machines to Read and Comprehendcited by 1,039
- 2016Sequence-Level Knowledge Distillationcited by 761
- 2019Text Summarization with Pretrained Encoderscited by 553
- 2018Born Again Neural Networkscited by 539
- 2020MobileBERT: a Compact Task-Agnostic BERT for Resource-Limited Devicescited by 476
- 2013Do Deep Nets Really Need to be Deep?cited by 466
- 2017Regularizing Neural Networks by Penalizing Confident Output Distributionscited by 461
- 2017Non-Autoregressive Neural Machine Translationcited by 457
- 2019MASS: Masked Sequence to Sequence Pre-training for Language Generationcited by 345
- 2020UniLMv2: Pseudo-Masked Language Models for Unified Language Model Pre-Trainingcited by 204
- 2019Revisiting Self-Training for Neural Sequence Generationcited by 154
- 2017Sobolev Training for Neural Networkscited by 136
- 2020BERT-of-Theseus: Compressing BERT by Progressive Module Replacingcited by 112
- 2020Pre-trained Summarization Distillationcited by 70
- 2016Learning-Based Single-Document Summarization with Compression and Anaphoricity Constraintscited by 34
- 2020Understanding Neural Abstractive Summarization Models via Uncertaintycited by 34
- 2020Noisy Self-Knowledge Distillation for Text Summarizationcited by 31
- 2017Stronger Baselines for Trustable Results in Neural Machine Translationcited by 12
Built on by
Citing papers per year: 2023: 5; 2024: 4; 2025: 3; 2026: 2.
14 papers.
Sort:
By area:
- 2024Learning to Summarize from LLM-generated Feedbackcited by 24
- 2023A Systematic Study of Knowledge Distillation for Natural Language Generation with Pseudo-Target Trainingcited by 16
- 2023Should We Attend More or Less? Modulating Attention for Fairnesscited by 11
- 2025Mitigating Hallucinations in Diffusion Models through Adaptive Attention Modulationcited by 11
- 2024TriSum: Learning Summarization Ability from Large Language Models with Structured Rationalecited by 10
- 2023Enhancing Abstractiveness of Summarization Models through Calibrated Distillationcited by 8
- 2024A Bayesian Flow Network Framework for Chemistry Taskscited by 6
- 2026Attention Smoothing Is All You Need For Unlearningcited by 6
- 2023Task-agnostic Distillation of Encoder-Decoder Language Modelscited by 4
- 2023Modular Transformers: Compressing Transformers into Modularized Layers for Flexible Efficient Inferencecited by 4
- 2024Understanding Position Bias Effects on Fairness in Social Multi-Document Summarizationcited by 4
- 2025Optimal Attention Temperature Improves the Robustness of In-Context Learning under Distribution Shift in High Dimensionscited by 1
- 2026CoANeRV: Coordinate-Aware Token-Space Neural Video Representation14 Aug 2026
- 2026Attention Smoothing Is All You Need For Unlearning1 Mar 2026
- 2025Optimal Attention Temperature Improves the Robustness of In-Context Learning under Distribution Shift in High Dimensions3 Nov 2025
- 2025InforME: Improving Informativeness of Abstractive Text Summarization With Informative Attention Guided by Named Entity Salience7 Oct 2025
- 2025Mitigating Hallucinations in Diffusion Models through Adaptive Attention Modulation24 Feb 2025
- 2024Learning to Summarize from LLM-generated Feedback17 Oct 2024
- 2024A Bayesian Flow Network Framework for Chemistry Tasks28 Jul 2024
- 2024Understanding Position Bias Effects on Fairness in Social Multi-Document Summarization3 May 2024
- 2024TriSum: Learning Summarization Ability from Large Language Models with Structured Rationale15 Mar 2024
- 2023Enhancing Abstractiveness of Summarization Models through Calibrated Distillation20 Oct 2023
- 2023Modular Transformers: Compressing Transformers into Modularized Layers for Flexible Efficient Inference4 Jun 2023
- 2023Should We Attend More or Less? Modulating Attention for Fairness22 May 2023
- 2024Learning to Summarize from LLM-generated Feedbackcited by 24
- 2024Learning to Summarize from LLM-generated Feedback17 Oct 2024
- 2026Attention Smoothing Is All You Need For Unlearningcited by 6
- 2026Attention Smoothing Is All You Need For Unlearning1 Mar 2026
- 2023A Systematic Study of Knowledge Distillation for Natural Language Generation with Pseudo-Target Trainingcited by 16
- 2023Should We Attend More or Less? Modulating Attention for Fairnesscited by 11
- 2024TriSum: Learning Summarization Ability from Large Language Models with Structured Rationalecited by 10
- 2023Enhancing Abstractiveness of Summarization Models through Calibrated Distillationcited by 8
- 2024A Bayesian Flow Network Framework for Chemistry Taskscited by 6
- 2023Task-agnostic Distillation of Encoder-Decoder Language Modelscited by 4
- 2023Modular Transformers: Compressing Transformers into Modularized Layers for Flexible Efficient Inferencecited by 4
- 2024Understanding Position Bias Effects on Fairness in Social Multi-Document Summarizationcited by 4
- 2025Optimal Attention Temperature Improves the Robustness of In-Context Learning under Distribution Shift in High Dimensionscited by 1
- 2025InforME: Improving Informativeness of Abstractive Text Summarization With Informative Attention Guided by Named Entity Saliencecited by 0
- 2026CoANeRV: Coordinate-Aware Token-Space Neural Video Representationcited by 0
- 2026CoANeRV: Coordinate-Aware Token-Space Neural Video Representation14 Aug 2026
- 2025Optimal Attention Temperature Improves the Robustness of In-Context Learning under Distribution Shift in High Dimensions3 Nov 2025
- 2025InforME: Improving Informativeness of Abstractive Text Summarization With Informative Attention Guided by Named Entity Salience7 Oct 2025
- 2024A Bayesian Flow Network Framework for Chemistry Tasks28 Jul 2024
- 2024Understanding Position Bias Effects on Fairness in Social Multi-Document Summarization3 May 2024
- 2024TriSum: Learning Summarization Ability from Large Language Models with Structured Rationale15 Mar 2024
- 2023Enhancing Abstractiveness of Summarization Models through Calibrated Distillation20 Oct 2023
- 2023Modular Transformers: Compressing Transformers into Modularized Layers for Flexible Efficient Inference4 Jun 2023
- 2023Should We Attend More or Less? Modulating Attention for Fairness22 May 2023
- 2023Task-agnostic Distillation of Encoder-Decoder Language Models21 May 2023
- 2023A Systematic Study of Knowledge Distillation for Natural Language Generation with Pseudo-Target Training3 May 2023