Mechanistic Interpretability of Cognitive Complexity in LLMs via Linear Probing using Bloom's Taxonomy

· Source: cs.CL updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Data Science & Analytics · Depth: Expert, extended

Summary

This study investigates the internal neural representations of cognitive complexity in Large Language Models (LLMs) using Bloom's Taxonomy as a hierarchical framework. By analyzing high-dimensional activation vectors from the residual streams of four state-of-the-art open-weight LLMs, the research demonstrates that linear classifiers achieve approximately 95% mean accuracy across all six Bloom levels (Remember, Understand, Apply, Analyze, Evaluate, Create). This indicates that cognitive level is encoded in a linearly accessible subspace of the model's representations. The findings show that this "Cognitive Separability Onset" (CSO) emerges early in the forward pass, specifically around layer 5 for most models, and remains stable in deeper layers. A curated dataset of 1,128 questions, ranging from 3 to 39 words, was used. Control experiments confirmed that this separability is not attributable to surface-level lexical patterns, as TF-IDF and sentence embedding baselines achieved significantly lower accuracies of 73% and 61% respectively. Misclassifications primarily occur between adjacent Bloom levels, reflecting the taxonomy's ordinal structure.

Key takeaway

For AI Scientists and Machine Learning Engineers evaluating LLM capabilities, recognize that models internally encode cognitive complexity aligning with Bloom's Taxonomy. You should consider probing internal representations, especially in early-to-mid layers (e.g., around layer 5), to understand how models process different cognitive demands. This approach helps distinguish between superficial performance and genuine reasoning, informing the development of more transparent and robust AI systems. Future work could involve causal intervention studies.

Key insights

LLMs encode cognitive complexity in a linearly separable latent space, emerging early in processing.

Principles

Method

Extract activation vectors from LLM residual streams, then train layer-wise linear probes (Logistic Regression) to classify Bloom levels, evaluating accuracy and confusion matrices.

In practice

Topics

Best for: Research Scientist, AI Scientist, Machine Learning Engineer, AI Student

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CL updates on arXiv.org.