Search papers, labs, and topics across Lattice.
This paper introduces a pairwise data augmentation technique for topic-controlled summarization, where contexts from different documents are combined to create contrastive training examples. By training small language models (sLMs) on these augmented datasets, the models learn the relationship between topics and summaries more effectively. Experiments on the SciTLDR dataset demonstrate that increasing the augmentation scale leads to improved win rate and semantic alignment, allowing a T5-base model to achieve performance competitive with larger models while using fewer parameters and real training examples.
Small language models can punch above their weight in topic-controlled summarization, rivaling larger models, simply by training on cleverly mixed-and-matched document contexts.
Topic-controlled summarisation enables users to generate summaries focused on specific aspects of source documents. This paper investigates a data augmentation strategy for training small language models (sLMs) to perform topic-controlled summarisation. We propose a pairwise data augmentation method that combines contexts from different documents to create contrastive training examples, enabling models to learn the relationship between topics and summaries more effectively. Using the SciTLDR dataset enriched with Wikipedia-derived topics, we systematically evaluate how augmentation scale affects model performance. Results show consistent improvements in win rate and semantic alignment as the augmentation scale increases, while the amount of real training data remains fixed. Consequently, a T5-base model trained with our augmentation approach achieves competitive performance relative to larger models, despite using significantly fewer parameters and substantially fewer real training examples.
