arXiv:2409.02686cs.CLcs.AI2024-09被引 1

通过因果分析提升大模型推理能力,仅用120万参数实现高效微调。

Deconfounded Causality-aware Parameter-Efficient Fine-Tuning for Problem-Solving Improvement of LLMs

  • 构建因果框架解释模型推理缺陷,定位知识理解瓶颈。
  • 在多个基准上超越基线,仅用1.2M参数即达领先效果。
  • 适合需要提升推理能力的模型优化场景,尤其关注效率与可解释性。

大型语言模型(LLMs)在基于人类指令的任务中表现出色,但在数学、物理等需推理的任务上仍显不足。本文探究其是否真正理解知识,而非仅模仿文本分布。通过注意力与表征层面的生成过程可视化,发现模型存在认知偏差;进一步将推理过程形式化为因果框架,解释观测到的问题。基于此,提出去混淆因果适配(DCA)方法,一种参数高效的微调策略,引导模型提取通用解题技能并迁移应用。实验表明,该方法在多个基准上持续优于基线,仅使用120万可调参数,即可达到或超过其他微调方法的效果,验证了其在提升大模型准确率与可靠性方面的有效性与高效性。

原文摘要 · Abstract (English)

Large Language Models (LLMs) have demonstrated remarkable efficiency in tackling various tasks based on human instructions, but studies reveal that they often struggle with tasks requiring reasoning, such as math or physics. This limitation raises questions about whether LLMs truly comprehend embedded knowledge or merely learn to replicate the token distribution without a true understanding of the content. In this paper, we delve into this problem and aim to enhance the reasoning capabilities of LLMs. First, we investigate if the model has genuine reasoning capabilities by visualizing the text generation process at the attention and representation level. Then, we formulate the reasoning process of LLMs into a causal framework, which provides a formal explanation of the problems observed in the visualization. Finally, building upon this causal framework, we propose Deconfounded Causal Adaptation (DCA), a novel parameter-efficient fine-tuning (PEFT) method to enhance the model's reasoning capabilities by encouraging the model to extract the general problem-solving skills and apply these skills to different questions. Experiments show that our method outperforms the baseline consistently across multiple benchmarks, and with only 1.2M tunable parameters, we achieve better or comparable results to other fine-tuning methods. This demonstrates the effectiveness and efficiency of our method in improving the overall accuracy and reliability of LLMs.

推理增强因果建模参数高效

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。