arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.03006cs.AI
收藏

将大语言模型的推理过程蒸馏为概念预测图

Distilling LLM Reasoning into Graph of Concept Predictors

Ziyang Yu, Liang Zhao

AI总结:

本文提出GCP框架,通过外部化教师决策过程为有向无环图,提升样本效率和训练稳定性,实验证明在有限标注预算下提升性能并增强可解释性。

摘要:

部署大型语言模型(LLMs)进行判别性工作负载时,通常受限于推理延迟、计算和扩展规模的API成本。主动蒸馏通过查询LLM oracle训练紧凑的判别学生模型来减少这些成本,但大多数流程仅蒸馏最终标签,丢弃中间推理信号,限制了对缺失推理和错误来源的诊断。我们提出概念预测图(GCP),一种推理感知的主动蒸馏框架,将教师的决策过程外部化为有向无环图,并在学生中用模块化概念预测器镜像。GCP通过图感知的获取策略提升样本效率,针对关键推理节点的不确定性和分歧进行目标获取。此外,通过针对子模块的重新训练提升训练稳定性和效率,将下游损失归因于特定的概念预测器,并仅更新最具影响的模块。在八个NLP分类基准测试中,GCP在有限标注预算下提升性能,同时产生更可解释和可控的训练动态。代码可在:https://github.com/Ziyang-Yu/GCP。

英文摘要:

Deploying Large Language Models (LLMs) for discriminative workloads is often limited by inference latency, compute, and API costs at scale. Active distillation reduces these costs by querying an LLM oracle to train small discriminative students, but most pipelines distill only final labels, discarding intermediate reasoning signals and offering limited diagnostics of what reasoning is missing and where errors arise. We propose Graph of Concept Predictors (GCP), a reasoning-aware active distillation framework in which the teacher's reasoning is elicited as a directed acyclic graph of intermediate concepts and mirrored in the student. GCP enhances sample efficiency through a graph-aware acquisition strategy that weights per-concept uncertainty, gradient diversity, and coverage by node centrality. Additionally, it improves training stability and efficiency by performing targeted sub-module retraining, which attributes downstream loss to specific concept predictors and updates only the most influential modules. Experiments on eight NLP classification benchmarks demonstrate that GCP enhances performance under limited annotation budgets while yielding more interpretable and controllable training dynamics. Code is available at https://github.com/Ziyang-Yu/GCP.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑