2024

Understanding Chain-of-Thought in LLMs through Information Theory

Ton, Jean-Francois, Taufiq, Muhammad Faaiz, Liu, Yang

Understand

Large Language Models (LLMs) have shown impressive performance in complex reasoning tasks through the use of Chain-of-Thought (CoT) reasoning, allowing models to break down problems into manageable sub-tasks.

  • However, existing CoT evaluation techniques either require annotated CoT data or fall short in accurately assessing intermediate reasoning steps, leading to high rates of false positives.
  • In this paper, we formalize CoT reasoning in LLMs through an information-theoretic lens.
  • Specifically, our framework quantifies the `information-gain' at each reasoning step, enabling the identification of failure modes in LLMs without the need for expensive annotated datasets.

Reading the bibliography…