Multi-Task Identification of Entities, Relations, and Coreference for Scientific Knowledge Graph Construction
Yi Luan, Luheng He, Mari Ostendorf, Hannaneh Hajishirzi
Introduction
As scientific communities grow and evolve, new tasks, methods, and datasets are introduced and different methods are compared with each other. Despite advances in search engines, it is still hard to identify new technologies and their relationships with what existed before. To help researchers more quickly identify opportunities for new combinations of tasks, methods and data, it is important to design intelligent algorithms that can extract and organize scientific information from a large collection of documents.
Organizing scientific information into structured knowledge bases requires information extraction (IE) about scientific entities and their relationships. However, the challenges associated with scientific IE are greater than for a general domain. First, annotation of scientific text requires domain expertise which makes annotation costly and limits resources. In addition, most relation extraction systems are designed for within-sentence relations. However, extracting information from scientific articles requires extracting relations across sentences. Figure 1 illustrates this problem. The cross-sentence relations between some entities can only be connected by entities that refer to the same scientific concept, including generic terms (such as the pronoun it, or phrases like our method) that are not informative by themselves. With co-reference, context-free grammar can be connected to MORPA through the intermediate co-referred pronoun it. Applying existing IE systems to this data, without co-reference, will result in much lower relation coverage (and a sparse knowledge base).
In this paper, we develop a unified learning model for extracting scientific entities, relations, and coreference resolution. This is different from previous work Luan et al. (2017b); Gupta and Manning (2011); Tsai et al. (2013); Gábor et al. (2018) which often addresses these tasks as independent components of a pipeline. Our unified model is a multi-task setup that shares parameters across low-level tasks, making predictions by leveraging context across the document through coreference links. Specifically, we extend prior work for learning span representations and coreference resolution Lee et al. (2017); He et al. (2018). Different from a standard tagging system, our system enumerates all possible spans during decoding and can effectively detect overlapped spans. It avoids cascading errors between tasks by jointly modeling all spans and span-span relations.
To explore this problem, we create a dataset SciERC for scientific information extraction, which includes annotations of scientific terms, relation categories and co-reference links. Our experiments show that the unified model is better at predicting span boundaries, and it outperforms previous state-of-the-art scientific IE systems on entity and relation extraction Luan et al. (2017b); Augenstein et al. (2017). In addition, we build a scientific knowledge graph integrating terms and relations extracted from each article. Human evaluation shows that propagating coreference can significantly improve the quality of the automatic constructed knowledge graph.
In summary we make the following contributions. We create a dataset for scientific information extraction by jointly annotating scientific entities, relations, and coreference links. Extending a previous end-to-end coreference resolution system, we develop a multi-task learning framework that can detect scientific entities, relations, and coreference clusters without hand-engineered features. We use our unified framework to build a scientific knowledge graph from a large collection of documents and analyze information in scientific literature.
Related Work
There has been growing interest in research on automatic methods for information extraction from scientific articles. Past research in scientific IE addressed analyzing citations Athar and Teufel (2012b, a); Kas (2011); Gabor et al. (2016); Sim et al. (2012); Do et al. (2013); Jaidka et al. (2014); Abu-Jbara and Radev (2011), analyzing research community Vogel and Jurafsky (2012); Anderson et al. (2012), and unsupervised methods for extracting scientific entities and relations Gupta and Manning (2011); Tsai et al. (2013); Gábor et al. (2016).
More recently, two datasets in SemEval 2017 and 2018 have been introduced, which facilitate research on supervised and semi-supervised learning for scientific information extraction. SemEval 17 Augenstein et al. (2017) includes 500 paragraphs from articles in the domains of computer science, physics, and material science. It includes three types of entities (called keyphrases): Tasks, Methods, and Materials and two relation types: hyponym-of and synonym-of. SemEval 18 Gábor et al. (2018) is focused on predicting relations between entities within a sentence. It consists of six relation types. Using these datasets, neural models Ammar et al. (2017, 2018); Luan et al. (2017b); Augenstein and Søgaard (2017) are introduced for extracting scientific information. We extend these datasets by increasing relation coverage, adding cross-sentence coreference linking, and removing some annotation constraints. Different from most previous IE systems for scientific literature and general domains Miwa and Bansal (2016); Xu et al. (2016); Peng et al. (2017); Quirk and Poon (2017); Luan et al. (2018); Adel and Schütze (2017), which use preprocessed syntactic, discourse or coreference features as input, our unified framework does not rely on any pipeline processing and is able to model overlapping spans.
While Singh et al. (2013) show improvements by jointly modeling entities, relations, and coreference links, most recent neural models for these tasks focus on single tasks Clark and Manning (2016); Wiseman et al. (2016); Lee et al. (2017); Lample et al. (2016); Peng et al. (2017) or joint entity and relation extraction Katiyar and Cardie (2017); Zhang et al. (2017); Adel and Schütze (2017); Zheng et al. (2017). Among those studies, many papers assume the entity boundaries are given, such as Clark and Manning (2016), Adel and Schütze (2017) and Peng et al. (2017). Our work relaxes this constraint and predicts entity boundaries by optimizing over all possible spans. Our model draws from recent end-to-end span-based models for coreference resolution Lee et al. (2017, 2018) and semantic role labeling He et al. (2018) and extends them for the multi-task framework involving the three tasks of identification of entity, relation and coreference.
Neural multi-task learning has been applied to a range of NLP tasks. Most of these models share word-level representations Collobert and Weston (2008); Klerke et al. (2016); Luan et al. (2016, 2017a); Rei (2017), while Peng et al. (2017) uses high-order cross-task factors. Our model instead propagates cross-task information via span representations, which is related to Swayamdipta et al. (2017).
Dataset
Our dataset (called SciERC) includes annotations for scientific entities, their relations, and coreference clusters for 500 scientific abstracts. These abstracts are taken from 12 AI conference/workshop proceedings in four AI communities from the Semantic Scholar CorpusThese conferences include general AI (AAAI, IJCAI), NLP (ACL, EMNLP, IJCNLP), speech (ICASSP, Interspeech), machine learning (NIPS, ICML), and computer vision (CVPR, ICCV, ECCV) at http://labs.semanticscholar.org/corpus/. SciERC extends previous datasets in scientific articles SemEval 2017 Task 10 (SemEval 17) Augenstein et al. (2017) and SemEval 2018 Task 7 (SemEval 18) Gábor et al. (2018) by extending entity types, relation types, relation coverage, and adding cross-sentence relations using coreference links. Our dataset is publicly available at: http://nlp.cs.washington.edu/sciIE/. Table 1 shows the statistics of SciERC.
We define six types for annotating scientific entities (Task, Method, Metric, Material, Other-ScientificTerm and Generic) and seven relation types (Compare, Part-of, Conjunction, Evaluate-for, Feature-of, Used-for, Hyponym-Of). Directionality is taken into account except for the two symmetric relation types (Conjunction and Compare). Coreference links are annotated between identical scientific entities. A Generic entity is annotated only when the entity is involved in a relation or is coreferred with another entity. Annotation guidelines can be found in Appendix A. Figure 1 shows an annotated example.
Following annotation guidelines from QasemiZadeh and Schumann (2016) and using the BRAT interface Stenetorp et al. (2012), our annotators perform a greedy annotation for spans and always prefer the longer span whenever ambiguity occurs. Nested spans are allowed when a subspan has a relation/coreference link with another term outside the span.
Human Agreements
One domain expert annotated all the documents in the dataset; 12% of the data is dually annotated by 4 other domain experts to evaluate the user agreements. The kappa score for annotating entities is 76.9%, relation extraction is 67.8% and coreference is 63.8%.
Comparison with previous datasets
SciERC is focused on annotating cross-sentence relations and has more relation coverage than SemEval 17 and SemEval 18, as shown in Table 1. SemEval 17 is mostly designed for entity recognition and only covers two relation types. The task in SemEval 18 is to classify a relation between a pair of entities given entity boundaries, but only intra-sentence relations are annotated and each entity only appears in one relation, resulting in sparser relation coverage than our dataset (3.2 vs. 9.4 relations per abstract). SciERC extends these datasets by adding more relation types and coreference clusters, which allows representing cross-sentence relations, and removing annotation constraints. Table 1 gives a comparison of statistics among the three datasets. In addition, SciERC aims at including broader coverage of general AI communities.
Model
We develop a unified framework (called SciIE) to identify and classify scientific entities, relations, and coreference resolution across sentences. SciIE is a multi-task learning setup that extends previous span-based models for coreference resolution Lee et al. (2017) and semantic role labeling He et al. (2018). All three tasks of entity recognition, relation extraction, and coreference resolution are treated as multinomial classification problems with shared span representations. SciIE benefits from expressive contextualized span representations as classifier features. By sharing span representations, sentence-level tasks can benefit from information propagated from coreference resolution across sentences, without increasing the complexity of inference. Figure 2 shows a high-level overview of the SciIE multi-task framework.
The input is a document represented as a sequence of words , from which we derive , the set of all possible within-sentence word sequence spans (up to a reasonable length) in the document. The output contains three structures: the entity types for all spans , the relations for all pair of spans , and the coreference links for all spans in . The output structures are represented with a set of discrete random variables indexed by spans or pairs of spans. Specifically, the output structures are defined as follows.
Entity recognition is to predict the best entity type for every candidate span. Let represent the set of all possible entity types including the null-type . The output structure is a set of random variables indexed by spans: for .
Relation extraction is to predict the best relation type given an ordered pair of spans . Let be the set of all possible relation types including the null-type . The output structure is a set of random variables indexed over pairs of spans that belong to the same sentence: for .
Coreference resolution is to predict the best antecedent (including a special null antecedent) given a span, which is the same mention-ranking model used in Lee et al. (2017). The output structure is a set of random variables defined as: for .
2 Model Definition
We formulate the multi-task learning setup as learning the conditional probability distribution . For efficient training and inference, we decompose assuming spans are conditionally independent given :
where the conditional probabilities of each random variable are independently normalized:
where denotes the unnormalized model score for an entity type and a span , denotes the score for a relation type and span pairs , and denotes the score for a binary coreference link between and . These scores are further decomposed into span and pairwise span scores computed from feed-forward networks, as will be explained in Section 4.3.
For simplicity, we omit from the functions and from the observation.
Given a set of all documents , the model loss function is defined as a weighted sum of the negative log-likelihood loss of all three tasks:
where , , and are gold structures of the entity types, relations, and coreference, respectively. The task weights , , and are introduced as hyper-parameters to control the importance of each task.
For entity recognition and relation extraction, and are computed with the definition in Equation (4.2). For coreference resolution, we use the marginalized loss following Lee et al. (2017) since each mention can have multiple correct antecedents. Let be the set of all correct antecedents for span , we have: .
3 Scoring Architecture
We use feedforward neural networks (FFNNs) over shared span representations to compute a set of span and pairwise span scores. For the span scores, measures how likely a span has an entity type , and and measure how likely a span is a mention in a relation or a coreference link, respectively. The pairwise scores and measure how likely two spans are associated in a relation or a coreference link, respectively. Let be the fixed-length vector representation for span . For different tasks, the span scores for and pairwise span scores for are computed as follows:
where is element-wise multiplication, and are neural network parameters to be learned.
We use these scores to compute the different :
The scores in Equation (4) are defined for entity types, relations, and antecedents that are not the null-type . Scores involving the null label are set to a constant 0: .
We use the same span representations from Lee et al. (2017) and share them across the three tasks. We start by building bi-directional LSTMs Hochreiter and Schmidhuber (1997) from word, character and ELMo Peters et al. (2018) embeddings.
For a span , its vector representation is constructed by concatenating ’s left and right end points from the BiLSTM outputs, an attention-based soft “headword,” and embedded span width features. Hyperparameters and other implementation details will be described in Section 6.
4 Inference and Pruning
Following previous work, we use beam pruning to reduce the number of pairwise span factors from to at both training and test time, where is the number of words in the document. We define two separate beams: to prune spans for the coreference resolution task, and for relation extraction. The spans in the beams are sorted by their span scores and respectively, and the sizes of the beams are limited by and . We also limit the maximum width of spans to a fixed number , which further reduces the number of span factors to .
Knowledge Graph Construction
We construct a scientific knowledge graph from a large corpus of scientific articles. The corpus includes all abstracts (110k in total) from 12 AI conference proceedings from the Semantic Scholar Corpus. Nodes in the knowledge graph correspond to scientific entities. Edges correspond to scientific relations between pairs of entities. The edges are typed according to the relation types defined in Section 3. Figure 4 shows a part of a knowledge graph created by our method. For example, Statistical Machine Translation (SMT) and grammatical error correction are nodes in the graph, and they are connected through a Used-for relation type. In order to construct the knowledge graph for the whole corpus, we first apply the SciIE model over single documents and then integrate the entities and relations across multiple documents (Figure 3).
The SciIE model extracts entities, their relations, and coreference clusters within one document. Phrases are heuristically normalized (described in Section 6) using entities and coreference links. In particular, we link all entities that belong to the same coreference cluster to replace generic terms with any other non-generic term in the cluster. Moreover, we replace all the entities in the cluster with the entity that has the longest string. Our qualitative analysis shows that there are fewer ambiguous phrases using coreference links (Figure 5). We calculate the frequency counts of all entities that appear in the whole corpus. We assign nodes in the knowledge graph by selecting the most frequent entities (with counts ) in the corpus, and merge in any remaining entities for which a frequent entity is a substring.
Assigning edges (relations)
A pair of entities may appear in different contexts, resulting in different relation types between those entities (Figure 6). For every pair of entities in the graph, we calculate the frequency of different relation types across the whole corpus.We assign edges between entities by selecting the most frequent relation type.
Experimental Setup
We evaluate our unified framework SciIE on SciERC and SemEval 17. The knowledge graph for scientific community analysis is built using the Semantic Scholar Corpus (110k abstracts in total).
We compare our model with the following baselines on SciERCdataset:
LSTM+CRF The state-of-the-art NER system Lample et al. (2016), which applies CRF on top of LSTM for named entity tagging, the approach has also been used in scientific term extraction Luan et al. (2017b).
LSTM+CRF+ELMo LSTM+CRF with ELMo as an additional input feature.
E2E Rel State-of-the-art joint entity and relation extraction system Miwa and Bansal (2016) that has also been used in scientific literature Peters et al. (2017); Augenstein et al. (2017). This system uses syntactic features such as part-of-speech tagging and dependency parsing.
E2E Rel(Pipeline) Pipeline setting of E2E Rel. Extract entities first and use entity results as input to relation extraction task.
E2E Rel+ELMo E2E Rel with ELMo as an additional input feature.
E2E Coref State-of-the-art coreference system Lee et al. (2017) combined with ELMo. Our system SciIE extends E2E Coref with multi-task learning.
In the SemEval task, we compare our model SciIE with the best reported system in the SemEval leaderboard Peters et al. (2017), which extends E2E Rel with several in-domain features such as gazetteers extracted from existing knowledge bases and model ensembles. We also compare with the state of the art on keyphrase extraction Luan et al. (2017b), which applies semi-supervised methods to a neural tagging model.We compare with the inductive setting results.
2 Implementation details
Our system extends the implementation and hyper-parameters from Lee et al. (2017) with the following adjustments. We use a 1 layer BiLSTM with 200-dimensional hidden layers. All the FFNNs have 2 hidden layers of 150 dimensions each. We use 0.4 variational dropout Gal and Ghahramani (2016) for the LSTMs, 0.4 dropout for the FFNNs, and 0.5 dropout for the input embeddings. We model spans up to 8 words. For beam pruning, we use for coreference resolution and for relation extraction. For constructing the knowledge graph, we use the following heuristics to normalize the entity phrases. We replace all acronyms with their corresponding full name and normalize all the plural terms with their singular counterparts.
Experimental Results
We evaluate SciIE on SciERC and SemEval 17 datasets. We provide qualitative results and human evaluation of the constructed knowledge graph.
Table 2 compares the result of our model with baselines on the three tasks: entity recognition (Table 2(a)), relation extraction (Table 2(b)), and coreference resolution (Table 2(c)). As evidenced by the table, our unified multi-task setup SciIE outperforms all the baselines. For entity recognition, our model achieves 1.3% and 2.4% relative improvement over LSTM+CRF with and without ELMo, respectively. Moreover, it achieves 1.8% and 2.7% relative improvement over E2E Rel with and without ELMo, respectively. For relation extraction, we observe more significant improvement with 13.1% relative improvement over E2E Rel and 7.4% improvement over E2E Rel with ELMo. For coreference resolution, SciIE outperforms E2E Coref with 4.5% relative improvement. We still observe a large gap between human-level performance and a machine learning system. We invite the community to address this challenging task.
Ablations
We evaluate the effect of multi-task learning in each of the three tasks defined in our dataset. Table 3 reports the results for individual tasks when additional tasks are included in the learning objective function. We observe that performance improves with each added task in the objective. For example, Entity recognition (65.7) benefits from both coreference resolution (67.5) and relation extraction (66.8). Relation extraction (37.9) significantly benefits when multi-tasked with coreference resolution (7.1% relative improvement). Coreference resolution benefits when multi-tasked with relation extraction, with 4.9% relative improvement.
Results on SemEval 17
Table 4 compares the results of our model with the state of the art on the SemEval 17 dataset for tasks of span identification, keyphrase extraction and relation extraction as well as the overall score. Span identification aims at identifying spans of entities. Keyphrase classification and relation extraction has the same setting with the entity and relation extraction in SciERC. Our model outperforms all the previous models that use hand-designed features. We observe more significant improvement in span identification than keyphrase classification. This confirms the benefit of our model in enumerating spans (rather than BIO tagging in state-of-the-art systems). Moreover, we have competitive results compared to the previous state of the art in relation extraction. We observe less gain compared to the SciERC dataset mainly because there are no coference links, and the relation types are not comprehensive.
2 Knowledge Graph Analysis
We provide qualitative analysis and human evaluations on the constructed knowledge graph.
Figure 7 shows the historical trend analysis (from 1996 to 2016) of the most popular applications of the phrase neural network, selected according to the statistics of the extracted relation triples with the ‘Used-for’ relation type from speech, computer vision, and NLP conference papers. We observe that, before 2000, neural network has been applied to a greater percentage of speech applications compared to the NLP and computer vision papers. In NLP, neural networks first gain popularity in language modeling and then extend to other tasks such as POS Tagging and Machine Translation. In computer vision, the application of neural networks gains popularity in object recognition earlier (around 2010) than the other two more complex tasks of object detection and image segmentation (hardest and also the latest).
Knowledge Graph Evaluation
Figure 8 shows the human evaluation of the constructed knowledge graph, comparing the quality of automatically generated knowledge graphs with and without the coreference links. We randomly select 10 frequent scientific entities and extract all the relation triples that include one of the selected entities leading to 1.5k relation triples from both systems. We ask four domain experts to annotate each of these extracted relations to define ground truth labels. Each domain expert is assigned 2 or 3 entities and all of the corresponding relations. Figure 8 shows precision/recall curves for both systems. Since it is not feasible to compute the actual recall of the systems, we compute the pseudo-recall Zhang et al. (2015) based on the output of both systems. We observe that the knowledge graph curve with coreference linking is mostly above the curve without coreference linking. The precision of both systems is high (above 84% for both systems), but the system with coreference links has significantly higher recall.
Conclusion
In this paper, we create a new dataset and develop a multi-task model for identifying entities, relations, and coreference clusters in scientific articles. By sharing span representations and leveraging cross-sentence information, our multi-task setup effectively improves performance across all tasks. Moreover, we show that our multi-task model is better at predicting span boundaries and outperforms previous state-of-the-art scientific IE systems on entity and relation extraction, without using any hand-engineered features or pipeline processing. Using our model, we are able to automatically organize the extracted information from a large collection of scientific articles into a knowledge graph. Our analysis shows the importance of coreference links in making a dense, useful graph.
We still observe a large gap between the performance of our model and human performance, confirming the challenges of scientific IE. Future work includes improving the performance using semi-supervised techniques and providing in-domain features. We also plan to extend our multi-task framework to information extraction tasks in other domains.
This research was supported by the Office of Naval Research under the MURI grant N00014-18-1-2670, NSF (IIS 1616112, III 1703166), Allen Distinguished Investigator Award, and gifts from Allen Institute for AI, Google, Amazon, and Bloomberg. We are grateful to Waleed Ammar and AI2 for sharing the Semantic Scholar Corpus. We also thank the anonymous reviewers, UW-NLP group and Shoou-I Yu for their helpful comments.
References
Appendix A Annotation Guideline
Task: Applications, problems to solve, systems to construct.
E.g. information extraction, machine reading system, image segmentation, etc.
Method: Methods , models, systems to use, or tools, components of a system, frameworks.
E.g. language model, CORENLP, POS parser, kernel method, etc.
Evaluation Metric: Metrics, measures, or entities that can express quality of a system/method.
E.g. F1, BLEU, Precision, Recall, ROC curve, mean reciprocal rank, mean-squared error, robustness, time complexity, etc.
Material: Data, datasets, resources, Corpus, Knowledge base.
E.g. image data, speech data, stereo images, bilingual dictionary, paraphrased questions, CoNLL, Panntreebank, WordNet, Wikipedia, etc.
Evaluation Metric: Metric, measure, or term that can express quality of a system/method.
E.g. F1, BLEU, Precision, Recall, ROC curve, mean reciprocal rank, mean-squared error,robustness, compile time, time complexity…
Generic: General terms or pronouns that may refer to a entity but are not themselves informative, often used as connection words.
E.g model, approach, prior knowledge, them, it…
A.2 Relation Category
Relation link can not go beyond sentence boundary. We define 4 asymmetric relation types (Used-for, Feature-of, Hyponym-of, Part-of), together with 2 symmetric relation types (Compare, Conjunction). B always points to A for asymmetric relations
Used-for: B is used for A, B models A, A is trained on B, B exploits A, A is based on B. E.g.
The TISPER system has been designed to enable many text applications.
Feature-of: B belongs to A, B is a feature of A, B is under A domain. E.g.
genre-specific regularities of discourse structure
Hyponym-of: B is a hyponym of A, B is a type of A. E.g.
NLP applications such as machine translation and language generation
The system includes two models: speech recognition and natural language understanding
Compare: Symmetric relation (use blue to denote entity). Opposite of conjunction, compare two models/methods, or listing two opposing entities. E.g.
Unlike the quantitative prior, the qualitative prior is often ignored…
We compare our system with previous sequential tagging systems…
Conjunction: Symmetric relation (use blue to denote entity). Function as similar role or use/incorporate with. E.g.
obtained from human expert or knowledge base
NLP applications such as machine translation and language generation
A.3 Coreference
Two Entities that points to the same concept.
We introduce a machine reading system… The system…
The prior knowledge include…Such knowledge can be applied to…
We develop a part-of-speech tagging system…The POS tagger…
A.4 Notes
Entity boundary annotation follows the ACL RD-TEC Annotation Guideline QasemiZadeh and Schumann (2016), with the extention that spans can be embedded in longer spans, only if the shorter span is involved in a relation.
Do not include determinators (such as ‘the’, ‘a’), or adjective pronouns (such as ‘this’,’its’, ‘these’, ‘such’) to the span. If generic phrases are not involved in a relation, do not tag them.
We introduce a neural based approach.. Its benefit is…
We introduce a neural based approach, which is a…
…advantage gained from local smoothness which… We present algorithms exploiting local smoothness in more aggressive ways…
Same scientific term but refer to different examples:
We use a data structure, we also use another data structure…
X is not used in Y or X is hard to be applied in Y
Appendix B Annotation and Knowledge Graph Examples
Here we take a screen shot of the BRAT interface for an ACL paper in Figure 9. We also attach the original figure of Figure 3 in Figure 10. More examples can be found in the project websitehttp://nlp.cs.washington.edu/sciIE/.