Source-linked AI summary
Neural Architectures for Fine-grained Entity Type Classification
Sonse Shimaoka, Pontus Stenetorp, Kentaro Inui, Sebastian Riedel
TL;DR
Fine-grained entity classification benefits from models that use informative context, but prior attentive architectures did not combine learned and hand-crafted features, and comparisons often used different training data. This paper evaluates attentive variants with feature integration and hierarchical label encoding, finding that training-data choice substantially affects performance and that its best model achieves 75.36% loose micro F1 on FIGER (GOLD).
Problem
Prior attentive architectures omit hand-crafted features, while published comparisons often evaluate models trained on different data despite using the same evaluation datasets.
Source: §1 · p. 2, §4.1 · p. 5Method
The paper investigates attentive neural variants for multi-class, multi-label fine-grained classification, combining learned and hand-crafted features and using hierarchical label encoding for parameter sharing.
Source: §1 · p. 2, §3 · p. 3, §3.4 · p. 5Results
Takeaways & Limitations
Attention can learn to focus on expressions important for fine-grained type classification, and evaluation results depend substantially on the training data used.
Source: §1 · p. 2, §5 · p. 8The authors identify re-implementing more literature methods with common training data and finding more consistent benefits from alternative hierarchical encodings as future work.
Source: §5 · p. 8
Abstract
from arXiv · showhide
In this work, we investigate several neural network architectures for fine-grained entity type classification. Particularly, we consider extensions to a recently proposed attentive neural architecture and make three key contributions. Previous work on attentive neural architectures do not consider hand-crafted features, we combine learnt and hand-crafted features and observe that they complement each other. Additionally, through quantitative analysis we establish that the attention mechanism is capable of learning to attend over syntactic heads and the phrase containing the mention, where both are known strong hand-crafted features for our task. We enable parameter sharing through a hierarchical label encoding method, that in low-dimensional projections show clear clusters for each type hierarchy. Lastly, despite using the same evaluation dataset, the literature frequently compare models trained using different data. We establish that the choice of training data has a drastic impact on performance, with decreases by as much as 9.85% loose micro F1 score for a previously proposed method. Despite this, our best model achieves state-of-the-art results with 75.36% loose micro F1 score on the well- established FIGER (GOLD) dataset.
1 Introduction
The paper investigates attentive neural architectures for fine-grained entity classification, combining learned representations with hand-crafted features and hierarchical labels. It also shows that training-data choice strongly affects performance, while the best model reaches 75.36% loose micro F1 on FIGER (GOLD).
- Fine-grained classification expands coarse entity types into more specific categories, supporting applications including question answering, knowledge base population, and co-reference resolution.Source: §1 · p. 1
- The authors combine learned and hand-crafted features because previous attentive architectures did not consider their direct integration, observing that the features complement each other.Source: §1 · p. 2
- Attention analysis shows that the model attends to syntactic heads and tokens surrounding mentions, which are relevant signals for fine-grained classification.Source: §1 · p. 2
- Hierarchical label encoding enables parameter sharing and produces coherent clusters in low-dimensional label projections.Source: §1 · p. 2
- 75.36% loose micro F1 on FIGER (GOLD) establishes the best model's state-of-the-art result despite comparisons with models trained on different data.Source: §1 · p. 2
- Training-data choice has a drastic effect on performance, with decreases of up to 9.85% loose Micro F1 for a previously proposed method.Source: §1 · p. 2
2 Related Work
Prior work develops fine-grained type inventories, context-dependent classification, label-noise reduction, and attention-based neural models. This paper distinguishes itself by combining hand-crafted features with attention and comparing models trained on the same publicly available data.
- Fine-grained entity classification evolved from manually annotated datasets and distant supervision toward larger hierarchical taxonomies and context-dependent evaluation.Source: §2 · p. 2
- Gillick et al. (2014) introduced context-dependent classification and label-cleaning heuristics, while Ren et al. (2016) pursued further noise reduction.Source: §2 · p. 2
- Earlier entity-classification neural models used feature embeddings, recurrent mention representations, fixed context windows, or attentive context composition.Source: §2 · pp. 2–3
- Attention mechanisms support alignment or focus over informative text in translation, question answering, textual entailment, and entity classification.Source: §2 · p. 3
- This work uses the same publicly available training data for model comparisons and directly combines hand-crafted features with an attentive neural model.Source: §2 · p. 3
- The experiments' hand-crafted features are based on those of Gillick et al. (2014) and are extracted for each entity mention.Source: Table 1 · p. 3
3 Models
The paper defines a common multi-class, multi-label framework for fine-grained entity classification and compares sparse, neural, hybrid, attentive, and hierarchy-aware variants. Its models represent mentions and context in different ways, optionally combine learned and hand-crafted features, and share label parameters through type hierarchies.
3 Models
- The models predict multiple entity types with logistic regression, while differing in how they construct the mention-context input.At inference, the system assigns the highest-probability type and adds types whose probabilities exceed 0.5.Source: §3 · p. 3
3.1 Sparse Feature Model
- The sparse feature baseline feeds binary indicators for hand-crafted mention features into the logistic regression layer.Its feature design is intended to provide a comparable baseline despite differences in clustering and topic-model resources.Source: §3.1 · pp. 3–4
3.2 Neural Models
- The neural models separately represent the mention and its context before concatenating them for prediction.Mention words are composed from embeddings, with the simple composition chosen to reduce overfitting.Source: §3.2 · p. 4
3.2.1 Averaging Encoder
- The Averaging encoder represents left and right context by averaging their word embeddings and concatenating the resulting vectors.The encoder uses a fixed context window on both sides of the mention.Source: §3.2 · p. 4, §3.2.1 · p. 4
3.2.2 LSTM Encoder
- The LSTM encoder processes left and right context sequences with LSTMs and uses their outputs as the context representation.The left context is processed left-to-right, while the right context is processed right-to-left.Source: §3.2.2 · p. 4
3.2.3 Attentive Encoder
- The Attentive Encoder applies bidirectional LSTMs and normalized attention weights to focus the context representation on salient local information.The weighted sum of bidirectional-LSTM outputs forms the context vector used for classification.Source: §3.2.3 · pp. 4–5
3.3 Hybrid Models
- Hybrid variants project sparse features and combine them with learned mention and context representations, allowing both feature types to inform classification.The paper presents this as a direct combination of hand-crafted features with attentive neural modeling.Source: §3.3 · p. 5
3.4 Hierarchical Label Encoding
- Hierarchical label encoding factors each label weight matrix as a learned matrix times a constant sparse binary matrix, sharing parameters across related types.Each type receives a hierarchy-coded column, and shared parameters may help infrequent types use annotations from related labels.Source: §3.4 · p. 5, Figure 2 · p. 5
4 Experiments
The experiments evaluate models on FIGER (GOLD) and OntoNotes while addressing substantial variation in the training data used for comparison. The authors use publicly available automatically induced training data and manually annotated evaluation sets.
4.1 Datasets
- Different training datasets are used to evaluate models on the same benchmarks, making fair comparison difficult.Some training data is not publicly available.Source: §4.1 · p. 5
- The experiments evaluate models on the manually annotated FIGER (GOLD) and OntoNotes datasets, with pronominal mentions discarded from OntoNotes.The resulting evaluation set contains 8,963 mentions.Source: §4.1 · p. 6
- The experiments train models on publicly available automatically induced datasets from Ren et al. (2016), using their raw noisy data.The authors selected the raw data because they could not replicate the reported benefits of the denoising pipeline.Source: §4.1 · p. 6
4.2 Pre-trained Word Embeddings
The models use fixed pre-trained word embeddings to improve generalization to words absent from the training data.
4.2 Pre-trained Word Embeddings
- The models use freely available 300-dimensional cased Common Crawl embeddings trained on 840 billion tokens.Embeddings are not updated during training, and unseen words use the “unk” embedding.Source: §4.2 · p. 6
4.3 Evaluation Criteria
Model performance is evaluated using strict accuracy, loose macro scores, and loose micro scores, following prior criteria.
4.3 Evaluation Criteria
- Performance is measured by strict accuracy, loose macro, and loose micro scores.The evaluation follows the criteria of Ling and Weld (2012).Source: §4.3 · p. 6
4.4 Hyperparameter Settings
The experiments use fixed neural-model hyperparameters and select the final model by development-set performance. They also note that the raw training data’s denoising benefits could not be replicated.
4.4 Hyperparameter Settings
- Neural and hybrid models use 300-dimensional embeddings, 100-unit LSTM and attention layers, and 50-dimensional sparse-feature projections.These hyperparameters were selected through preliminary development-set experiments.Source: §4.4 · p. 6
- Training uses Adam with a 0.001 learning rate, batch size 1,000, and five epochs, with dropout applied to mention and sparse-feature representations.The context window is C = 10, with padding beyond sentence boundaries.Source: §4.4 · p. 6
- The denoising-performance benefits reported by Ren et al. (2016) could not be replicated after running their pipeline.The authors contacted the original researchers but had not received a response at the time of writing.Source: §4.4 · p. 6
- The final model is selected as the best-performing model on the development set before test-set evaluation.The implementation uses Python and TensorFlow.Source: §4.4 · p. 6
4.5 Results
Results are reported separately for models trained on different datasets: hand-crafted features generally improve neural models, while training-data choice substantially affects comparability and performance.
4.5 Results
- Hand-crafted features consistently improve neural models, complementing learned features across the FIGER (GOLD) and OntoNotes evaluations.On OntoNotes, adding them substantially improves the averaging encoder, making the three neural models much more similar in performance.Source: §4.5.1 · p. 6, §4.5.2 · p. 7
4.5.1 FIGER (GOLD)
- The attentive encoder with hand-crafted features achieves the best performance among the investigated FIGER models, while hierarchical encoding has inconsistent effects there.The FIGER comparison uses models trained on the same W2M data in Table 5.Source: §4.5.1 · p. 7, Table 5 · p. 7
- Compared with earlier models trained on different data, the best model achieves state-of-the-art loose macro and micro scores without noise-reduction strategies.The closest competitor has higher accuracy but lower F1 scores, while a hand-crafted-feature variant outperforms a feature-learning variant despite 600,000 fewer training examples.Source: §4.5.1 · p. 7, Table 6 · p. 7
4.5.2 OntoNotes
- Hierarchical label encoding consistently improves OntoNotes performance, although the authors do not obtain state-of-the-art results across models using different training data.The authors report performance within 1 point of the state of the art in F1-score despite using different non-proprietary noisy data.Source: §4.5.2 · p. 7, Figure 3 · p. 8
4.5 Results
- The authors describe a dramatic 9.85-point loose micro score drop and stress comparing models on an equal footing with the same training data.The Shimaoka et al. (2016) model drops as much as 3.36 points of loose micro score when using a smaller dataset.Source: §4.5.1 · p. 7, §4.5.2 · p. 8
4.6 PCA visualisation of label embeddings
PCA visualisation shows that hierarchical label encoding produces clear, distinct clusters in the learned label embeddings.
4.6 PCA visualisation of label embeddings
- Hierarchical encoding forms clear distinct clusters in PCA projections of the learned label embeddings.The comparison contrasts non-hierarchical and hierarchical encodings, with subtype colors shared by their parent type.Source: §4.6 · p. 8
4.7 Attention Analysis
The attention analysis examines which contextual words the attentive model prioritizes when classifying fine-grained entity types. It finds that attention identifies syntactic heads and words surrounding the mention without explicit supervision.
4.7 Attention Analysis
- The attentive model without hand-crafted features learns to focus on syntactic heads and words in the phrase surrounding a mention without explicit supervision.The analysis quantifies how often the strongest-attended word is the syntactic head or lies before or after the mention.Source: §4.7 · p. 8, Table 7 · p. 8
5 Conclusions and Future Work
The paper concludes that training data strongly affects fine-grained entity classification results, while its best model and attention analysis provide strong reported outcomes. Future work targets broader equal-data comparisons and more consistent hierarchical encodings.
5 Conclusions and Future Work
- 75.36% loose micro F1 on FIGER (GOLD) is state of the art, while 64.93% is the best reported result for publicly trained OntoNotes models.The FIGER result is achieved despite comparisons with models trained on larger datasets.Source: §5 · p. 8
- The study finds that training-data choice significantly affects performance and that attention can learn expressions important for fine-grained type classification.The authors suggest these observations can motivate further research on the linguistic phenomena attentive models learn and how to improve them.Source: §5 · p. 8
- Future work includes re-implementing more literature methods under the same training data and exploring alternative hierarchical label encodings for more consistent gains.The code used for the experiments is publicly available to support reproducibility.Source: §5 · p. 8