English

Ehrenfeucht-Haussler Rank and Chain of Thought

Machine Learning 2025-08-12 v2 Artificial Intelligence

Abstract

The notion of \emph{rank} of a Boolean function has been a cornerstone in PAC learning theory, enabling quasipolynomial-time learning algorithms for polynomial-size decision trees. We present a novel characterization of rank, grounded in the well-known Transformer architecture. We show that the rank of a function ff corresponds to the minimum number of \emph{Chain of Thought} (CoT) steps required by a single-layer Transformer with hard attention to compute ff. Based on this characterization we establish tight bounds on the number of CoT steps required for specific problems, showing that \ell-fold function composition necessitates exactly \ell CoT steps. Furthermore, we analyze the problem of identifying the position of the kk-th occurrence of 1 in a Boolean sequence, proving that it requires kk CoT steps. Finally, we introduce the notion of the multi-head rank that captures multi-head single-layer transformers, and perform the analysis of PAC-learnability of the classes of functions with bounded multi-head rank.

Keywords

Cite

@article{arxiv.2501.12997,
  title  = {Ehrenfeucht-Haussler Rank and Chain of Thought},
  author = {Pablo Barceló and Alexander Kozachinskiy and Tomasz Steifer},
  journal= {arXiv preprint arXiv:2501.12997},
  year   = {2025}
}

Comments

Changes to the previous version: new results about PAC learning for functions of bounded multi-head rank are addes