A Non-Linear Structural Probe
Not on the map. Why?
Abstract (arXiv)
Probes are models devised to investigate the encoding of knowledge -- e.g. syntactic structure -- in contextual representations. Probes are often designed for simplicity, which has led to restrictions on probe design that may not allow for the full exploitation of the structure of encoded information; one such restriction is linearity. We examine the case of a structural probe (Hewitt and Manning, 2019), which aims to investigate the encoding of syntactic structure in contextual representations through learning only linear transformations. By observing that the structural probe learns a metric, we are able to kernelize it and develop a novel non-linear variant with an identical number of parameters. We test on 6 languages and find that the radial-basis function (RBF) kernel, in conjunction with regularization, achieves a statistically significant improvement over the baseline in all languages -- implying that at least part of the syntactic knowledge is encoded non-linearly. We conclude by discussing how the RBF kernel resembles BERT's self-attention layers and speculate that this resemblance leads to the RBF-based probe's stronger performance.
Key facts. Cited by 31 papers in the map (data through 7 Oct 2026). Builds on 13 of its 30 references that are in the map. First posted on arXiv 21 May 2021.
Builds on
13 of 30 references.
References · 13
- 2017Attention Is All You Needcited by 59,932
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2018Deep contextualized word representationscited by 2,285
- 2016Understanding intermediate layers using linear classifier probescited by 1,279
- 2019Analyzing Multi-Head Self-Attention: Specialized Heads Do the Heavy Lifting, the Rest Can Be Prunedcited by 844
- 2019Designing and Interpreting Probes with Control Taskscited by 608
- 2020Information-Theoretic Probing with Minimum Description Lengthcited by 230
- 2019Visualizing and Measuring the Geometry of BERTcited by 215
- 2020Information-Theoretic Probing for Linguistic Structurecited by 204
- 2020Pareto Probing: Trading Off Accuracy for Complexitycited by 55
- 2020A Tale of a Probe and a Parsercited by 40
1 more
Built on by
Citing papers per year: 2021: 3; 2022: 4; 2023: 4; 2024: 7; 2025: 7; 2026: 6.
31 papers.
Sort:
By area:
- 2021Conditional probing: measuring usable information beyond a baselinecited by 61
- 2022Kernelized Concept Erasurecited by 47
- 2021Structural Persistence in Language Models: Priming as a Window into Abstract Language Representationscited by 44
- 2021The Dangers of Underclaiming: Reasons for Caution When Reporting How NLP Systems Failcited by 38
- 2025The Non-Linear Representation Dilemma: Is Causal Abstraction Enough for Mechanistic Interpretability?cited by 34
- 2023Sociodemographic Bias in Language Models: A Survey and Forward Pathcited by 27
- 2024Mechanistic?cited by 25
- 2024A polar coordinate system represents syntax in large language modelscited by 24
- 2022AST-Probe: Recovering abstract syntax trees from hidden representations of pre-trained language modelscited by 16
- 2025Beyond Linear Probes: Dynamic Safety Monitoring for Language Modelscited by 14
- 2023Arithmetic with Language Models: from Memorization to Computationcited by 12
- 2024Non-Linear Inference Time Intervention: Improving LLM Truthfulnesscited by 9
19 more
- 2024How Reliable are Causal Probing Interventions?cited by 9
- 2025Linguistic Interpretability of Transformer-based Language Models: a systematic reviewcited by 7
- 2022The Architectural Bottleneck Principlecited by 5
- 2025Seed-Induced Uniqueness in Transformer Models: Subspace Alignment Governs Subliminal Transfercited by 5
- 2024A Critical Study of What Code-LLMs (Do Not) Learncited by 4
- 2024Incremental Sentence Processing Mechanisms in Autoregressive Transformer Language Modelscited by 4
- 2025Probing Syntax in Large Language Models: Successes and Remaining Challengescited by 4
- 2022Emergent Linguistic Structures in Neural Networks are Fragilecited by 2
- 2026The Grammar of Transformers: A Systematic Review of Interpretability Research on Syntactic Knowledge in Language Modelscited by 1
- 2026Perturbation: A simple and efficient adversarial tracer for representation learning in language modelscited by 1
- 2026Implicit Representations of Grammaticality in Language Modelscited by 1
- 2023Disentangling the Linguistic Competence of Privacy-Preserving BERTcited by 0
- 2023Rethinking the Construction of Effective Metrics for Understanding the Mechanisms of Pretrained Language Modelscited by 0
- 2024Hitting "Probe"rty with Non-Linearity, and Morecited by 0
- 2025Derivational Probing: Unveiling the Layer-wise Derivation of Syntactic Structures in Neural Language Modelscited by 0
- 2025Freeze, Diffuse, Decode: Task-Aware Adaptation of Transformer Embeddings for Antimicrobial Peptide Designcited by 0
- 2026Probing for Reading Timescited by 0
- 2026D2-Monitor: Dynamic Safety Monitoring for Diffusion LLMs via Hesitation Signalscited by 0
- 2026A Survey on the Linear Representation Hypothesiscited by 0
- 2026A Survey on the Linear Representation Hypothesis19 Sep 2026
- 2026D2-Monitor: Dynamic Safety Monitoring for Diffusion LLMs via Hesitation Signals25 May 2026
- 2026Implicit Representations of Grammaticality in Language Models6 May 2026
- 2026Probing for Reading Times20 Apr 2026
- 2026Perturbation: A simple and efficient adversarial tracer for representation learning in language models25 Mar 2026
- 2026The Grammar of Transformers: A Systematic Review of Interpretability Research on Syntactic Knowledge in Language Models9 Jan 2026
- 2025Freeze, Diffuse, Decode: Task-Aware Adaptation of Transformer Embeddings for Antimicrobial Peptide Design28 Nov 2025
- 2025Seed-Induced Uniqueness in Transformer Models: Subspace Alignment Governs Subliminal Transfer2 Nov 2025
- 2025Beyond Linear Probes: Dynamic Safety Monitoring for Language Models30 Sep 2025
- 2025Probing Syntax in Large Language Models: Successes and Remaining Challenges5 Aug 2025
- 2025The Non-Linear Representation Dilemma: Is Causal Abstraction Enough for Mechanistic Interpretability?11 Jul 2025
- 2025Derivational Probing: Unveiling the Layer-wise Derivation of Syntactic Structures in Neural Language Models27 Jun 2025
19 more
- 2025Linguistic Interpretability of Transformer-based Language Models: a systematic review9 Apr 2025
- 2024A polar coordinate system represents syntax in large language models7 Dec 2024
- 2024Incremental Sentence Processing Mechanisms in Autoregressive Transformer Language Models6 Dec 2024
- 2024Mechanistic?7 Oct 2024
- 2024How Reliable are Causal Probing Interventions?28 Aug 2024
- 2024A Critical Study of What Code-LLMs (Do Not) Learn17 Jun 2024
- 2024Non-Linear Inference Time Intervention: Improving LLM Truthfulness27 Mar 2024
- 2024Hitting "Probe"rty with Non-Linearity, and More25 Feb 2024
- 2023Rethinking the Construction of Effective Metrics for Understanding the Mechanisms of Pretrained Language Models19 Oct 2023
- 2023Disentangling the Linguistic Competence of Privacy-Preserving BERT17 Oct 2023
- 2023Arithmetic with Language Models: from Memorization to Computation2 Aug 2023
- 2023Sociodemographic Bias in Language Models: A Survey and Forward Path13 Jun 2023
- 2022The Architectural Bottleneck Principle11 Nov 2022
- 2022Emergent Linguistic Structures in Neural Networks are Fragile31 Oct 2022
- 2022AST-Probe: Recovering abstract syntax trees from hidden representations of pre-trained language models23 Jun 2022
- 2022Kernelized Concept Erasure28 Jan 2022
- 2021The Dangers of Underclaiming: Reasons for Caution When Reporting How NLP Systems Fail15 Oct 2021
- 2021Structural Persistence in Language Models: Priming as a Window into Abstract Language Representations30 Sep 2021
- 2021Conditional probing: measuring usable information beyond a baseline19 Sep 2021
- 2021Conditional probing: measuring usable information beyond a baselinecited by 61
- 2022Kernelized Concept Erasurecited by 47
- 2021Structural Persistence in Language Models: Priming as a Window into Abstract Language Representationscited by 44
- 2021The Dangers of Underclaiming: Reasons for Caution When Reporting How NLP Systems Failcited by 38
- 2025The Non-Linear Representation Dilemma: Is Causal Abstraction Enough for Mechanistic Interpretability?cited by 34
- 2023Sociodemographic Bias in Language Models: A Survey and Forward Pathcited by 27
- 2024Mechanistic?cited by 25
- 2024A polar coordinate system represents syntax in large language modelscited by 24
- 2022AST-Probe: Recovering abstract syntax trees from hidden representations of pre-trained language modelscited by 16
- 2023Arithmetic with Language Models: from Memorization to Computationcited by 12
- 2024Non-Linear Inference Time Intervention: Improving LLM Truthfulnesscited by 9
- 2024How Reliable are Causal Probing Interventions?cited by 9
12 most cited of 29 All 29 (JSON)
- 2026A Survey on the Linear Representation Hypothesis19 Sep 2026
- 2026D2-Monitor: Dynamic Safety Monitoring for Diffusion LLMs via Hesitation Signals25 May 2026
- 2026Implicit Representations of Grammaticality in Language Models6 May 2026
- 2026Probing for Reading Times20 Apr 2026
- 2026Perturbation: A simple and efficient adversarial tracer for representation learning in language models25 Mar 2026
- 2026The Grammar of Transformers: A Systematic Review of Interpretability Research on Syntactic Knowledge in Language Models9 Jan 2026
- 2025Freeze, Diffuse, Decode: Task-Aware Adaptation of Transformer Embeddings for Antimicrobial Peptide Design28 Nov 2025
- 2025Seed-Induced Uniqueness in Transformer Models: Subspace Alignment Governs Subliminal Transfer2 Nov 2025
- 2025Probing Syntax in Large Language Models: Successes and Remaining Challenges5 Aug 2025
- 2025The Non-Linear Representation Dilemma: Is Causal Abstraction Enough for Mechanistic Interpretability?11 Jul 2025
- 2025Derivational Probing: Unveiling the Layer-wise Derivation of Syntactic Structures in Neural Language Models27 Jun 2025
- 2025Linguistic Interpretability of Transformer-based Language Models: a systematic review9 Apr 2025
12 newest of 29 All 29 (JSON)