“The Chain Lens [interim]” by vgel, Raymond Douglas
<p> TLDR: This is an early draft of our new paper introducing a technique for recovering multi-token concepts from model activations. Our technique takes single-token methods like the J-lens and extends them by contrastively decoding on an MTP head. This gives improvements on WorkspaceBench, overall beating J-lens (0.42 vs 0.32 across all 28 families) and even a hindsight composite of the best untrained lens per task family (0.49 vs 0.41 across the 21 families with published scores). It especially improves on computational intermediate tasks, beating the trained oracle lens on arithmetic chain intermediates (0.42 vs 0.25).</p><p> We introduce the chain lens, a new technique for extracting multi-token concepts from a model's latent representations. Our method takes single-token techniques like the logit lens or the J-lens and uses contrastive decoding on an MTP head to expand their readouts into bags of multi-token sequences. Generally speaking, it is good at surfacing specific multi-token concepts or pieces the model computes (such as arithmetic intermediates or phrases), although less good at rendering how those pieces relate: like the logit lens and J-lens, chain lens readouts are fragmentary.</p><p> On WorkspaceBench, a benchmark designed to identify good multi-token activation-to-text techniques, the chain lens beats a hindsight [...]</p> <p>---</p><p><strong>Outline:</strong></p><p>(02:26) Background and related work</p><p>(05:33) Method</p><p>(07:35) Results and Discussion</p><p>(07:39) WorkspaceBench</p><p>(12:49) Workspace Sharing in Poetry Continuations</p><p>(15:44) Neuron Lens</p><p>(18:28) The Loom Lens</p><p>(19:35) Ablations</p><p>(21:56) Conclusion</p> <p>---</p> <p><b>First published:</b><br/> November 11th, 2026 </p> <p><b>Source:</b><br/> <a href="https://www.lesswrong.com/posts/Nr6fksEqqxNpdz6rk/the-chain-lens-interim?utm_source=TYPE_III_AUDIO&utm_medium=Podcast&utm_content=Source+URL+in+episode+description&utm_campaign=ai_narration" rel="noopener noreferrer" target="_blank">https://www.lesswrong.com/posts/Nr6fksEqqxNpdz6rk/the-chain-lens-interim</a> </p> <p>---</p> <p>Narrated by <a href="https://type3.audio/?utm_source=TYPE_III_AUDIO&utm_medium=Podcast&utm_content=Narrated+by+TYPE+III+AUDIO&utm_term=lesswrong&utm_campaign=ai_narration" rel="noopener noreferrer" target="_blank">TYPE III AUDIO</a>.</p> <p>---</p><div style="max-width: 100%";><p><strong>Images from the article:</strong></p><a href="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794359765/lexical_client_uploads/qeem7jnyzrkab3vhg4ty.png" target="_blank"><img src="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794359765/lexical_client_uploads/qeem7jnyzrkab3vhg4ty.png" alt="Diagram of MTP head architecture with Qwen poem generation and lens decoding comparison." style="max-width: 100%;" /></a><hr style="margin-top: 24px; margin-bottom: 24px;" /><a href="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794359843/lexical_client_uploads/l393lzrtclgwhoetrzf4.png" target="_blank"><img src="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794359843/lexical_client_uploads/l393lzrtclgwhoetrzf4.png" alt="Adapted from the Deepseek v3 technical report, figure 3" style="max-width: 100%;" /></a><hr style="margin-top: 24px; margin-bottom: 24px;" /><a href="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794360385/lexical_client_uploads/rbro0ilprhbmmtb38vct.png" target="_blank"><img src="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794360385/lexical_client_uploads/rbro0ilprhbmmtb38vct.png" alt="Hallucination and performance both scale with chain length, measured on 19 WorkspaceBench families. The loom lens is on the Pareto frontier." style="max-width: 100%;" /></a><hr style="margin-top: 24px; margin-bottom: 24px;" /><a href="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794360494/lexical_client_uploads/vbntsk7y95ajgcmnfxtb.png" target="_blank"><img src="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794360494/lexical_client_uploads/vbntsk7y95ajgcmnfxtb.png" alt="We sample the three most likely first lines for each topic, then three most likely next lines, and so forth, until we get poems per topic. At each stage of this sampling, we see multiple possible next lines from the poetic prefix, since the distribution never collapses." style="max-width: 100%;" /></a><hr style="margin-top: 24px; margin-bottom: 24px;" /><a href="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794360773/lexical_client_uploads/gxp1fsiqx5tbkbtpebb8.png" target="_blank"><img src="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794360773/lexical_client_uploads/gxp1fsiqx5tbkbtpebb8.png" alt="Percentage of layer neurons with chain lens readouts that round-trip, by layer." style="max-width: 100%;" /></a><hr style="margin-top: 24px; margin-bottom: 24px;" /><a href="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794360906/lexical_client_uploads/j0wl0nblcm1txbjlxcj3.png" target="_blank"><img src="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794360906/lexical_client_uploads/j0wl0nblcm1txbjlxcj3.png" alt="Judge accuracy picking the steered text that matches a provided readout, by layer band. We exclude very early layers, where we couldn’t coherently steer neurons." style="max-width: 100%;" /></a><hr style="margin-top: 24px; margin-bottom: 24px;" /><a href="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794360310/lexical_client_uploads/logkfubnxpph5wwfrzkx.png" target="_blank"><img src="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794360310/lexical_client_uploads/logkfubnxpph5wwfrzkx.png" alt="Scatter plot of WorkspaceBench macro score versus inference FLOPs per readout for various lens methods." style="max-width: 100%;" /></a><hr style="margin-top: 24px; margin-bottom: 24px;" /><a href="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794360349/lexical_client_uploads/zpxrredo9jed5n45xxfo.png" target="_blank"><img src="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794360349/lexical_client_uploads/zpxrredo9jed5n45xxfo.png" alt="Scatter plot of hallucination rate versus WorkspaceBench macro, showing Pareto frontiers for various lens methods." style="max-width: 100%;" /></a><hr style="margin-top: 24px; margin-bottom: 24px;" /><a href="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794360573/lexical_client_uploads/jboxenkbw4j5wtrtywo8.png" target="_blank"><img src="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794360573/lexical_client_uploads/jboxenkbw4j5wtrtywo8.png" alt="Diagram comparing J-lens and Chain lens outputs for AI poem generation." style="max-width: 100%;" /></a><hr style="margin-top: 24px; margin-bottom: 24px;" /><a href="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794360604/lexical_client_uploads/vorzhasdqfe5wmsajmjm.png" target="_blank"><img src="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794360604/lexical_client_uploads/vorzhasdqfe5wmsajmjm.png" alt="Annotated poem couplets with color-highlighted words showing text matching analysis." style="max-width: 100%;" /></a><hr style="margin-top: 24px; margin-bottom: 24px;" /><a href="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794360674/lexical_client_uploads/wm5gy06gc0rzxqosj9zr.png" target="_blank"><img src="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794360674/lexical_client_uploads/wm5gy06gc0rzxqosj9zr.png" alt="Line graph comparing chain lens, Oracle Lens, and J-lens coverage AUC across layers." style="max-width: 100%;" /></a><hr style="margin-top: 24px; margin-bottom: 24px;" /><a href="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794360716/lexical_client_uploads/ta1pwlnj5wkyypzvgzvs.png" target="_blank"><img src="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794360716/lexical_client_uploads/ta1pwlnj5wkyypzvgzvs.png" alt="Scatter line plot comparing coverage AUC versus readout content share for three lens methods." style="max-width: 100%;" /></a><hr style="margin-top: 24px; margin-bottom: 24px;" /><a href="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794361297/lexical_client_uploads/maeh4qbexuksrwx3cdtk.png" target="_blank"><img src="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794361297/lexical_client_uploads/maeh4qbexuksrwx3cdtk.png" alt="Forest plot showing paired differences from reference across 20 families with 95% confidence intervals." style="max-width: 100%;" /></a><hr style="margin-top: 24px; margin-bottom: 24px;" /><a href="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794361381/lexical_client_uploads/o8gbmmfbqp17c3wwvhji.png" target="_blank"><img src="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794361381/lexical_client_uploads/o8gbmmfbqp17c3wwvhji.png" alt="Forest plot "Further method variants vs. the full method (paired)" showing paired differences with confidence intervals." style="max-width: 100%;" /></a><hr style="margin-top: 24px; margin-bottom: 24px;" /><a href="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794361423/lexical_client_uploads/wmwypzkjskpersyjo5eg.png" target="_blank"><img src="https://res.cloudinary.com/lesswrong-2-0/image/upload/v1794361423/lexical_client_uploads/wmwypzkjskpersyjo5eg.png" alt="Forest plot showing paired differences from reference across lens and seed conditions with confidence intervals." style="max-width: 100%;" /></a><p><em>Apple Podcasts and Spotify do not show images in the episode description. Try <a href="https://pocketcasts.com/" target="_blank" rel="noreferrer">Pocket Casts</a>, or another podcast app.</em></p></div>




