Parameter-Efficient Fine-Tuning Method for Task-Oriented Dialogue Systemsopen access
- Authors
- Mo, Yunho; Yoo, Joon; Kang, Sangwoo
- Issue Date
- Jul-2023
- Publisher
- MDPI
- Keywords
- natural language processing; task-oriented dialogue system; PEFT; fine-tuning; training efficiency
- Citation
- MATHEMATICS, v.11, no.14
- Journal Title
- MATHEMATICS
- Volume
- 11
- Number
- 14
- URI
- https://scholarworks.bwise.kr/gachon/handle/2020.sw.gachon/88833
- DOI
- 10.3390/math11143048
- ISSN
- 2227-7390
2227-7390
- Abstract
- The use of Transformer-based pre-trained language models has become prevalent in enhancing the performance of task-oriented dialogue systems. These models, which are pre-trained on large text data to grasp the language syntax and semantics, fine-tune the entire parameter set according to a specific task. However, as the scale of the pre-trained language model increases, several challenges arise during the fine-tuning process. For example, the training time escalates as the model scale grows, since the complete parameter set needs to be trained. Furthermore, additional storage space is required to accommodate the larger model size. To address these challenges, we propose a new new task-oriented dialogue system called PEFTTOD. Our proposal leverages a method called the Parameter-Efficient Fine-Tuning method (PEFT), which incorporates an Adapter Layer and prefix tuning into the pre-trained language model. It significantly reduces the overall parameter count used during training and efficiently transfers the dialogue knowledge. We evaluated the performance of PEFTTOD on the Multi-WOZ 2.0 dataset, a benchmark dataset commonly used in task-oriented dialogue systems. Compared to the traditional method, PEFTTOD utilizes only about 4% of the parameters for training, resulting in a 4% improvement in the combined score compared to the existing T5-based baseline. Moreover, PEFTTOD achieved an efficiency gain by reducing the training time by 20% and saving up to 95% of the required storage space.
- Files in This Item
- There are no files associated with this item.
- Appears in
Collections - IT융합대학 > 소프트웨어학과 > 1. Journal Articles
![qrcode](https://api.qrserver.com/v1/create-qr-code/?size=55x55&data=https://scholarworks.bwise.kr/gachon/handle/2020.sw.gachon/88833)
Items in ScholarWorks are protected by copyright, with all rights reserved, unless otherwise indicated.