PFDial: A Structured Dialogue Instruction Fine-tuning Method Based on UML Flowcharts
Created by
Haebom
저자
Ming Zhang, Yuhui Wang, Yujiong Shen, Tingyi Yang, Changhao Jiang, Yilong Wu, Shihan Dou, Qinhao Chen, Zhiheng Xi, Zhihao Zhang, Yi Dong, Zhen Wang, Zhihui Fei, Mingyang Wan, Tao Liang, Guojun Ma, Qi Zhang, Tao Gui, Xuanjing Huang
개요
본 논문은 엄격한 사전 정의된 프로세스 제약 조건 하에서 작동하는 프로세스 기반 대화 시스템을 다룹니다. 대규모 언어 모델(LLM)이 대화 및 추론에서 상당한 발전을 보였지만, 엄격하게 제약된 대화 작업 해결에는 여전히 어려움을 겪습니다. 이러한 문제를 해결하기 위해, 440개의 플로우차트(5,055개의 프로세스 노드 포함)에서 파생된 12,705개의 고품질 중국어 대화 지침으로 구성된 Process Flow Dialogue (PFDial) 데이터셋을 구축했습니다. PlantUML 사양을 기반으로 각 UML 플로우차트는 원자 대화 단위(구조화된 5-튜플)로 변환됩니다. 실험 결과, 800개의 샘플로만 학습된 7B 모델과 전체 데이터로 학습된 0.5B 모델 모두 90% 이상의 정확도를 달성했습니다. 또한, 8B 모델은 GPT-4보다 평균 11.00% 향상되어 최대 43.88%까지 성능을 능가했습니다. 플로우차트의 어려운 역방향 전환에 대한 모델 성능을 추가로 평가하고, 다양한 데이터셋 형식이 의사 결정 및 순차적 분기 처리에서 모델 성능에 미치는 영향을 심층적으로 분석했습니다. 데이터는 https://github.com/KongLongGeFDU/PFDial 에서 공개됩니다.