arXiv · 2506.05017
Controlling Summarization Length Through EOS Token Weighting
Abstract
Controlling the length of generated text can be crucial in various text-generation tasks, including summarization. Existing methods often require complex model alterations, limiting compatibility with pre-trained models. We address these limitations by developing a simple approach for controlling the length of automatic text summaries by increasing the importance of correctly predicting the EOS token in the cross-entropy loss computation. The proposed methodology is agnostic to architecture and decoding algorithms and orthogonal to other inference-time techniques to control the generation length. We tested it with encoder-decoder and modern GPT-style LLMs, and show that this method can control generation length, often without affecting the quality of the summary.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Zeno Belligoli, Emmanouil Stergiadis, Eran Fainman, Ilya Gusev. 2025-06-05. Controlling Summarization Length Through EOS Token Weighting. https://arxiv.org/abs/2506.05017
Cite the original work for its findings. Save a collection to share your selection of sources.