Tree Transformer: Integrating Tree Structures into Self-Attention
Not on the map. Why?
Abstract (arXiv)
Pre-training Transformer from large-scale raw texts and fine-tuning on the desired task have achieved state-of-the-art results on diverse NLP tasks. However, it is unclear what the learned attention captures. The attention computed by attention heads seems not to match human intuitions about hierarchical structures. This paper proposes Tree Transformer, which adds an extra constraint to attention heads of the bidirectional Transformer encoder in order to encourage the attention heads to follow tree structures. The tree structures can be automatically induced from raw texts by our proposed "Constituent Attention" module, which is simply implemented by self-attention between two adjacent words. With the same training procedure identical to BERT, the experiments demonstrate the effectiveness of Tree Transformer in terms of inducing tree structures, better language modeling, and further learning more explainable attention scores.
Key facts. Cited by 68 papers in the map (data through 7 Oct 2026). First posted on arXiv 14 Sep 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 15; 2022: 10; 2023: 20; 2024: 10; 2025: 8; 2026: 5.
68 papers.
Sort:
By area:
- 2021SynCoBERT: Syntax-Guided Multi-Modal Contrastive Pre-Training for Code Representationcited by 63
- 2023Uncertainty in Natural Language Generation: From Theory to Applicationscited by 61
- 2021Integrating Visuospatial, Linguistic and Commonsense Structure into Story Visualizationcited by 51
- 2023HiCLIP: Contrastive Language-Image Pretraining with Hierarchy-aware Attentioncited by 44
- 2023LasUIE: Unifying Information Extraction with Latent Adaptive Structure-aware Generative Language Modelcited by 43
- 2022Transformer Grammars: Augmenting Transformer Language Models with Syntactic Inductive Biases at Scalecited by 40
- 2021Link Prediction on N-ary Relational Facts: A Graph-based Approachcited by 29
- 2021Structural Guidance for Transformer Language Modelscited by 25
- 2023Pushdown Layers: Encoding Recursive Structure in Transformer Language Modelscited by 18
- 2023Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patternscited by 17
- 2021Auto-Parsing Network for Image Captioning and Visual Question Answeringcited by 16
- 2021Modeling Hierarchical Structures with Continuous Recursive Neural Networkscited by 15
38 more
- 2023Misalign, Contrast then Distill: Rethinking Misalignments in Language-Image Pretrainingcited by 10
- 2023Learning Trajectory-Word Alignments for Video-Language Taskscited by 7
- 2023Contextual Distortion Reveals Constituency: Masked Language Models are Implicit Parserscited by 7
- 2024Dependency Transformer Grammars: Integrating Dependency Structures into Transformer Language Modelscited by 7
- 2022An Attention Matrix for Every Decision: Faithfulness-based Arbitration Among Multiple Attention-Based Interpretations of Transformers in Text Classificationcited by 6
- 2023Beam Tree Recursive Cellscited by 6
- 2024HDT: Hierarchical Document Transformercited by 6
- 2023Differentiable Tree Operations Promote Compositional Generalizationcited by 5
- 2024Compositional Generalization Across Distributional Shifts with Sparse Tree Operationscited by 5
- 2021Improving BERT Pretraining with Syntactic Supervisioncited by 4
- 2022An Interpretability Evaluation Benchmark for Pre-trained Language Modelscited by 4
- 2023Efficient Beam Tree Recursioncited by 4
- 2023Probabilistic Transformer: A Probabilistic Dependency Model for Contextual Word Representationcited by 4
- 2024Tree-Planted Transformers: Unidirectional Transformer Language Models with Implicit Syntactic Supervisioncited by 4
- 2021Co-training an Unsupervised Constituency Parser with Weak Supervisioncited by 3
- 2022Image Captioning In the Transformer Agecited by 3
- 2022RSTGen: Imbuing Fine-Grained Interpretable Control into Long-FormText Generatorscited by 3
- 2022Transformer with Tree-order Encoding for Neural Program Generationcited by 3
- 2022Forming Trees with Treeformerscited by 3
- 2024Transformer Mechanisms Mimic Frontostriatal Gating Operations When Trained on Human Working Memory Taskscited by 3
- 2024Tree Transformers are an Ineffective Model of Syntactic Constituencycited by 3
- 2024Sneaking Syntax into Transformer Language Models with Tree Regularizationcited by 3
- 2021RL-GRIT: Reinforcement Learning for Grammar Inferencecited by 2
- 2023StrAE: Autoencoding for Pre-Trained Embeddings using Explicit Structurecited by 2
- 2023Towards Early Prediction of Human iPSC Reprogramming Successcited by 2
- 2023Tree Cross Attentioncited by 2
- 2023Mixture-of-Linguistic-Experts Adapters for Improving and Interpreting Pre-trained Language Modelscited by 2
- 2021Enhancing the Transformer Decoder with Transition-based Syntaxcited by 1
- 2021ORCHARD: A Benchmark For Measuring Systematic Generalization of Multi-Hierarchical Reasoningcited by 1
- 2022Dependency-based Mixture Language Modelscited by 1
- 2023An Effective Method using Phrase Mechanism in Neural Machine Translationcited by 1
- 2024Structural Optimization Ambiguity and Simplicity Bias in Unsupervised Neural Grammar Inductioncited by 1
- 2024Representation Learning of Geometric Treescited by 1
- 2025Tabular Embeddings for Tables with Bi-Dimensional Hierarchical Metadata and Nestingcited by 1
- 2025Bringing Structure to Naturalness: On the Naturalness of ASTscited by 1
- 2025A Systematic Study of Compositional Syntactic Transformer Language Modelscited by 1
- 2025Tree-Based Deep Learning for Ranking Symbolic Integration Algorithmscited by 1
- 2025Cross-modal Full-mode Fine-grained Alignment for Text-to-Image Person Retrievalcited by 1
50 most cited of 68 All 68 (JSON)
- 2026Learning Sparse Decision Trees via Transformer Variational Auto-Encoders1 Sep 2026
- 2026Mahalanobis-Based Multi-Head Attention for Complex State Propagation25 Aug 2026
- 2026GiLT: Augmenting Transformer Language Models with Dependency Graphs15 May 2026
- 2026Dynamic sparsity in tree-structured feed-forward layers at scale18 Mar 2026
- 2026SymPlex: A Structure-Aware Transformer for Symbolic PDE Solving3 Feb 2026
- 2025Probability Distribution Collapse: A Critical Bottleneck to Compact Unsupervised Neural Grammar Induction25 Sep 2025
- 2025Cross-modal Full-mode Fine-grained Alignment for Text-to-Image Person Retrieval17 Sep 2025
- 2025Understanding Syntactic Generalization in Structure-inducing Language Models11 Aug 2025
- 2025Tree-Based Deep Learning for Ranking Symbolic Integration Algorithms8 Aug 2025
- 2025A Systematic Study of Compositional Syntactic Transformer Language Models28 Jun 2025
- 2025Sparsified State-Space Models are Efficient Highway Networks27 May 2025
- 2025Bringing Structure to Naturalness: On the Naturalness of ASTs11 Apr 2025
38 more
- 2025Tabular Embeddings for Tables with Bi-Dimensional Hierarchical Metadata and Nesting20 Feb 2025
- 2024Compositional Generalization Across Distributional Shifts with Sparse Tree Operations18 Dec 2024
- 2024Sneaking Syntax into Transformer Language Models with Tree Regularization28 Nov 2024
- 2024Tree Transformers are an Ineffective Model of Syntactic Constituency25 Nov 2024
- 2024On Eliciting Syntax from Language Models via Hashing5 Oct 2024
- 2024Representation Learning of Geometric Trees16 Aug 2024
- 2024Dependency Transformer Grammars: Integrating Dependency Structures into Transformer Language Models24 Jul 2024
- 2024Structural Optimization Ambiguity and Simplicity Bias in Unsupervised Neural Grammar Induction23 Jul 2024
- 2024HDT: Hierarchical Document Transformer11 Jul 2024
- 2024Tree-Planted Transformers: Unidirectional Transformer Language Models with Implicit Syntactic Supervision20 Feb 2024
- 2024Transformer Mechanisms Mimic Frontostriatal Gating Operations When Trained on Human Working Memory Tasks13 Feb 2024
- 2023Misalign, Contrast then Distill: Rethinking Misalignments in Language-Image Pretraining19 Dec 2023
- 2023Picking the Underused Heads: A Network Pruning Perspective of Attention Head Selection for Fusing Dialogue Coreference Information15 Dec 2023
- 2023Probabilistic Transformer: A Probabilistic Dependency Model for Contextual Word Representation26 Nov 2023
- 2023Sparse Attention-Based Neural Networks for Code Classification11 Nov 2023
- 2023Pushdown Layers: Encoding Recursive Structure in Transformer Language Models29 Oct 2023
- 2023Mixture-of-Linguistic-Experts Adapters for Improving and Interpreting Pre-trained Language Models24 Oct 2023
- 2023Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns3 Oct 2023
- 2023Tree Cross Attention29 Sep 2023
- 2023An Effective Method using Phrase Mechanism in Neural Machine Translation21 Aug 2023
- 2023Uncertainty in Natural Language Generation: From Theory to Applications28 Jul 2023
- 2023Efficient Beam Tree Recursion20 Jul 2023
- 2023Differentiable Tree Operations Promote Compositional Generalization1 Jun 2023
- 2023Contextual Distortion Reveals Constituency: Masked Language Models are Implicit Parsers1 Jun 2023
- 2023Beam Tree Recursive Cells31 May 2023
- 2023Towards Early Prediction of Human iPSC Reprogramming Success23 May 2023
- 2023StrAE: Autoencoding for Pre-Trained Embeddings using Explicit Structure9 May 2023
- 2023LasUIE: Unifying Information Extraction with Latent Adaptive Structure-aware Generative Language Model13 Apr 2023
- 2023HiCLIP: Contrastive Language-Image Pretraining with Hierarchy-aware Attention6 Mar 2023
- 2023Adaptively Clustering Neighbor Elements for Image-Text Generation5 Jan 2023
- 2023Learning Trajectory-Word Alignments for Video-Language Tasks5 Jan 2023
- 2022Bird-Eye Transformers for Text Generation Models8 Oct 2022
- 2022An Attention Matrix for Every Decision: Faithfulness-based Arbitration Among Multiple Attention-Based Interpretations of Transformers in Text Classification22 Sep 2022
- 2022An Interpretability Evaluation Benchmark for Pre-trained Language Models28 Jul 2022
- 2022Forming Trees with Treeformers14 Jul 2022
- 2022Unsupervised and Few-shot Parsing from Pretrained Language Models10 Jun 2022
- 2022Transformer with Tree-order Encoding for Neural Program Generation30 May 2022
- 2022RSTGen: Imbuing Fine-Grained Interpretable Control into Long-FormText Generators25 May 2022
50 newest of 68 All 68 (JSON)
- 2025Sparsified State-Space Models are Efficient Highway Networks27 May 2025
- 2021SynCoBERT: Syntax-Guided Multi-Modal Contrastive Pre-Training for Code Representationcited by 63
- 2023Uncertainty in Natural Language Generation: From Theory to Applicationscited by 61
- 2021Integrating Visuospatial, Linguistic and Commonsense Structure into Story Visualizationcited by 51
- 2023LasUIE: Unifying Information Extraction with Latent Adaptive Structure-aware Generative Language Modelcited by 43
- 2022Transformer Grammars: Augmenting Transformer Language Models with Syntactic Inductive Biases at Scalecited by 40
- 2021Link Prediction on N-ary Relational Facts: A Graph-based Approachcited by 29
- 2021Structural Guidance for Transformer Language Modelscited by 25
- 2023Pushdown Layers: Encoding Recursive Structure in Transformer Language Modelscited by 18
- 2023Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patternscited by 17
- 2021Auto-Parsing Network for Image Captioning and Visual Question Answeringcited by 16
- 2021Modeling Hierarchical Structures with Continuous Recursive Neural Networkscited by 15
- 2023Misalign, Contrast then Distill: Rethinking Misalignments in Language-Image Pretrainingcited by 10
12 most cited of 66 All 66 (JSON)
- 2026Learning Sparse Decision Trees via Transformer Variational Auto-Encoders1 Sep 2026
- 2026Mahalanobis-Based Multi-Head Attention for Complex State Propagation25 Aug 2026
- 2026GiLT: Augmenting Transformer Language Models with Dependency Graphs15 May 2026
- 2026Dynamic sparsity in tree-structured feed-forward layers at scale18 Mar 2026
- 2026SymPlex: A Structure-Aware Transformer for Symbolic PDE Solving3 Feb 2026
- 2025Probability Distribution Collapse: A Critical Bottleneck to Compact Unsupervised Neural Grammar Induction25 Sep 2025
- 2025Cross-modal Full-mode Fine-grained Alignment for Text-to-Image Person Retrieval17 Sep 2025
- 2025Understanding Syntactic Generalization in Structure-inducing Language Models11 Aug 2025
- 2025Tree-Based Deep Learning for Ranking Symbolic Integration Algorithms8 Aug 2025
- 2025A Systematic Study of Compositional Syntactic Transformer Language Models28 Jun 2025
- 2025Bringing Structure to Naturalness: On the Naturalness of ASTs11 Apr 2025
- 2025Tabular Embeddings for Tables with Bi-Dimensional Hierarchical Metadata and Nesting20 Feb 2025
12 newest of 66 All 66 (JSON)