arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2106.04080cs.CL
收藏

RewardsOfSum: Exploring Reinforcement Learning Rewards for Summarisation

University of Technology Sydney(悉尼科技大学);RoZetta Technology(罗泽塔科技公司)

Jacob Parnell, Inigo Jauregi Unanue, Massimo Piccardi

英文摘要:

To date, most abstractive summarisation models have relied on variants of the negative log-likelihood (NLL) as their training objective. In some cases, reinforcement learning has been added to train the models with an objective that is closer to their evaluation measures (e.g. ROUGE). However, the reward function to be used within the reinforcement learning approach can play a key role for performance and is still partially unexplored. For this reason, in this paper, we propose two reward functions for the task of abstractive summarisation: the first function, referred to as RwB-Hinge, dynamically selects the samples for the gradient update. The second function, nicknamed RISK, leverages a small pool of strong candidates to inform the reward. In the experiments, we probe the proposed approach by fine-tuning an NLL pre trained model over nine summarisation datasets of diverse size and nature. The experimental results show a consistent improvement over the negative log-likelihood baselines.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑