TIAO: Token Importance-Aware Policy Optimization for Text Summarization

arXiv:2609.16748 · cs.CL · Submitted 2026-09-15 · Read on arXiv

cs.CL

Submitted: 2026-09-15

Updated: 2026-09-15

Code: https://github.com/TechCloud-x/TIAO

License: http://creativecommons.org/licenses/by/4.0/

The gist: Text summarization requires models to condense content while preserving key qualities such as consistency and coherence.

Terminology

Abstract

Text summarization requires models to condense content while preserving key qualities such as consistency and coherence. Large language models (LLMs) have shown strong performance on this task and can be further improved through reinforcement learning (RL). However, most existing methods apply reward signals directly to undifferentiated token sequences, overlooking the varying importance of individual tokens to word and sentence level quality in summarization. In this paper, we propose Token Importance-Aware Policy Optimization (TIAO), a novel reinforcement learning strategy that explicitly leverages token-importance awareness. Specifically, TIAO identifies core tokens based on token dependency and reweights a trajectory's advantage according to its overall dependencies. Experiments on the real world dataset show that our TIAO achieves highly competitive results, and that a 7B foundation model enhanced by TIAO performs comparably to GPT-4 and GPT-5-nano. Code is available at https://github.com/TechCloud-x/TIAO

Sources

Related papers