Rewarding What Matters: Step-by-Step Reinforcement Learning for Task-Oriented Dialogue
Huifang Du, Shuqin Li, Minghao Wu, Xuejing Feng, Yuan-Fang Li, Haofen Wang
Abstract
Reinforcement learning (RL) is a powerful approach to enhance task-oriented dialogue (TOD) systems. However, existing RL methods tend to mainly focus on generation tasks, such as dialogue policy learning (DPL) or response generation (RG), while neglecting dialogue state tracking (DST) for understanding. This narrow focus limits the systems to achieve globally optimal performance by overlooking the interdependence between understanding and generation. Additionally, RL methods face challenges with sparse and delayed rewards, which complicates training and optimization. To address these issues, we extend RL into both understanding and generation tasks by introducing step-by-step rewards throughout the token generation. The understanding reward increases as more slots are correctly filled in DST, while the generation reward grows with the accurate inclusion of user requests. Our approach provides a balanced optimization aligned with task completion. Experimental results demonstrate that our approach effectively enhances the performance of TOD systems and achieves new state-of-the-art results on three widely used datasets, including MultiWOZ2.0, MultiWOZ2.1, and In-Car. Our approach also shows superior few-shot ability in low-resource settings compared to current models.- Anthology ID:
- 2024.findings-emnlp.472
- Volume:
- Findings of the Association for Computational Linguistics: EMNLP 2024
- Month:
- November
- Year:
- 2024
- Address:
- Miami, Florida, USA
- Editors:
- Yaser Al-Onaizan, Mohit Bansal, Yun-Nung Chen
- Venue:
- Findings
- SIG:
- Publisher:
- Association for Computational Linguistics
- Note:
- Pages:
- 8030–8046
- Language:
- URL:
- https://preview.aclanthology.org/ingest_wac_2008/2024.findings-emnlp.472/
- DOI:
- 10.18653/v1/2024.findings-emnlp.472
- Cite (ACL):
- Huifang Du, Shuqin Li, Minghao Wu, Xuejing Feng, Yuan-Fang Li, and Haofen Wang. 2024. Rewarding What Matters: Step-by-Step Reinforcement Learning for Task-Oriented Dialogue. In Findings of the Association for Computational Linguistics: EMNLP 2024, pages 8030–8046, Miami, Florida, USA. Association for Computational Linguistics.
- Cite (Informal):
- Rewarding What Matters: Step-by-Step Reinforcement Learning for Task-Oriented Dialogue (Du et al., Findings 2024)
- PDF:
- https://preview.aclanthology.org/ingest_wac_2008/2024.findings-emnlp.472.pdf