中国科学院机构知识库网格
Chinese Academy of Sciences Institutional Repositories Grid
MoDE-CoTD: Chain-of-Thought Distillation for Complex Reasoning Tasks with Mixture of Decoupled LoRA-Experts

文献类型:会议论文

作者Xiang Li2,3; Shizhu He2,3; Jiayu Wu2,3; Zhao Yang2,3; Yao Xu2,3; Yang Jun1; Haifeng Liu1; Kang Liu2,3; Jun Zhao2,3
出版日期2024-05-20
会议日期2024.5.20 - 2024.5.25
会议地点Torino (Italia)
英文摘要

Chain-of-thought Distillation (CoTD) aims at distilling Chain-of-thought (CoT) reasoning ability of large language models (LLMs) to much smaller student models. The core of CoTD is using a large teacher model to generate rationales and fine-tune smaller student models. However, current Chain-of-thought Distillation works have the following limitations: 1) Student models are separately distilled from specific reasoning tasks and lack a collaboration mechanism, hindering the enhancement of reasoning performance through collaboration among various reasoning tasks. 2) The parameter update of student models severely harms the CoT reasoning ability on other unseen reasoning tasks not included in the distillation process. In this work, we introduce a novel CoT Distillation method, MoDE-CoTD, which decouples the CoT reasoning abilities out of the student model by distilling multiple LoRA-Experts and freezing the parameters of the student model. Sequentially, LoRA-Experts are combined and adapted to handle both seen and unseen reasoning tasks, enabling collaboration among diverse reasoning tasks to further enhance CoT reasoning performance. Experimental results on 14 datasets (including 4 unseen datasets) demonstrate the strength of MoDE-CoTD, with an average accuracy gain of 6.3% on seen datasets and 7.8% on unseen datasets.

会议录出版者ELRA and ICCL
源URL[http://ir.ia.ac.cn/handle/173211/57448]  
专题复杂系统认知与决策实验室
通讯作者Shizhu He
作者单位1.Guangdong OPPO Mobile Telecommunications Corp.,Ltd
2.Institute of Automation, Chinese Academy of Sciences
3.School of Artificial Intelligence, University of Chinese Academy of Sciences
推荐引用方式
GB/T 7714
Xiang Li,Shizhu He,Jiayu Wu,et al. MoDE-CoTD: Chain-of-Thought Distillation for Complex Reasoning Tasks with Mixture of Decoupled LoRA-Experts[C]. 见:. Torino (Italia). 2024.5.20 - 2024.5.25.

入库方式: OAI收割

来源:自动化研究所

浏览0
下载0
收藏0
其他版本

除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。