BACK_TO_FEEDAICRIER_2
llmri maps self CKA in 3D
OPEN_SOURCE ↗
REDDIT · REDDIT// 20d agoRESEARCH PAPER

llmri maps self CKA in 3D

llmri is an exploratory notebook that turns self-CKA measurements into a 3D, MRI-like visualization of LLM internals using Q-head subspace projections. The author compares GPT-2, RWKV, Mamba, Qwen, and the Prisma research model, and is still looking for a place to host the interactive HTML.

// ANALYSIS

This is the kind of interpretability demo that looks like art first and tooling second, which is exactly why it might end up useful. The visuals are compelling, but the current evidence is still correlational and hypothesis-driven.

  • Self-CKA is a sensible lens for comparing representational structure across layers instead of just staring at activations.
  • The Q-head subspace choice is the sharpest move here because it tries to inspect structure before FFN mixing muddies it.
  • Cross-model comparisons are the real payoff: the project already shows family resemblance across GPT-style, RWKV, Mamba, and Qwen models.
  • The “circuits within circuits” framing is intriguing, but the blob shapes alone do not prove causal mechanistic circuits.
  • A hosted interactive viewer would matter more than another static screenshot; the method needs rotation and exploration to be persuasive.
// TAGS
llmrillmresearchopen-sourcedata-tools

DISCOVERED

20d ago

2026-03-22

PUBLISHED

20d ago

2026-03-22

RELEVANCE

7/ 10

AUTHOR

y3i12