$\textit{SKIntern}$: Internalizing Symbolic Knowledge for Distilling Better CoT Capabilities into Small Language Models

Huanxuan Liao, Shizhu He, Yupu Hao, Xiang Li, Yuanzhe Zhang +2 more
9/20/2024
cs.CL

Official code: Xnhyacinth/SKIntern (6 stars, Apache-2.0, last commit 2025-02-18).

Abstract

Small Language Models (SLMs) are attracting attention due to the high computational demands and privacy concerns of Large Language Models (LLMs). Some studies fine-tune SLMs using Chains of Thought (CoT) data distilled from LLMs, aiming to enhance their reasoning ability. Furthermore, Some CoT distillation methods introduce external symbolic knowledge into the generation process to improve the limited knowledge memory, reasoning ability and out-of-domain (OOD) generalization of SLMs. However, the introduction of symbolic knowledge increases computational overhead and introduces potential noise. In this paper, we introduce $\textit{SKIntern}$, an innovative approach that empowers SLMs to internalize symbolic knowledge and few-shot examples gradually through a progressive fine-tuning process, guided by a predefined linear decay schedule under curriculum learning. By efficiently internalizing knowledge, $\textit{SKIntern}$ reduces computational overhead and speeds up the reasoning process by focusing solely on the question during inference. It outperforms state-of-the-art baselines by over 5\%, while reducing inference costs (measured in FLOPs) by up to $4\times$ across a wide range of SLMs in both in-domain (ID) and out-of-domain (OOD) tasks. Our code will be available at \url{https://github.com/Xnhyacinth/SKIntern}.

View on arXivView PDF

Code Implementations(1)

Apache-2.0

[COLING 2025] SKIntern: Internalizing Symbolic Knowledge for Distilling Better CoT Capabilities into Small Language Models

60Shell, PythonSep 11, 20241 years agoApache-2.0

Similar Papers

Cite this paper

@article{liao2024textitskintern,
  title  = {$\textit{SKIntern}$: Internalizing Symbolic Knowledge for Distilling Better CoT Capabilities into Small Language Models},
  author = {Huanxuan Liao and Shizhu He and Yupu Hao and Xiang Li and Yuanzhe Zhang and Jun Zhao and Kang Liu},
  year   = {2024},
  eprint = {2409.13183},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2409.13183v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2409.13183.svg)](https://codeofpaper.com/paper/2409.13183)

Discussion