Arxiv:1805.02262V1 [Cs.CL] 6 May 2018
Total Page:16
File Type:pdf, Size:1020Kb
Construction of the Literature Graph in Semantic Scholar Waleed Ammar, Dirk Groeneveld, Chandra Bhagavatula, Iz Beltagy, Miles Crawford, Doug Downey,y Jason Dunkelberger, Ahmed Elgohary, Sergey Feldman, Vu Ha, Rodney Kinney, Sebastian Kohlmeier, Kyle Lo, Tyler Murray, Hsu-Han Ooi, Matthew Peters, Joanna Power, Sam Skjonsberg, Lucy Lu Wang, Chris Wilhelm, Zheng Yuan,y Madeleine van Zuylen, and Oren Etzioni [email protected] Allen Institute for Artificial Intelligence, Seattle WA 98103, USA yNorthwestern University, Evanston IL 60208, USA Abstract We describe a deployed scalable system for organizing published scientific litera- ture into a heterogeneous graph to facili- tate algorithmic manipulation and discov- ery. The resulting literature graph consists of more than 280M nodes, representing pa- pers, authors, entities and various interac- tions between them (e.g., authorships, cita- tions, entity mentions). We reduce litera- ture graph construction into familiar NLP tasks (e.g., entity extraction and linking), point out research challenges due to differ- ences from standard formulations of these tasks, and report empirical results for each task. The methods described in this pa- Figure 1: Part of the literature graph. per are used to enable semantic features in www.semanticscholar.org. 2017). We describe methods used in a scalable de- 1 Introduction ployed production system for extracting structured The goal of this work is to facilitate algorithmic information from scientific documents into the lit- discovery in the scientific literature. Despite no- erature graph (see Fig. 1). The literature graph is table advances in scientific search engines, data a directed property graph which summarizes key mining and digital libraries (e.g., Wu et al., 2014), information in the literature and can be used to an- researchers remain unable to answer simple ques- swer the queries mentioned earlier as well as more tions such as: complex queries. For example, in order to com- pute the Erdos˝ number of an author X, the graph arXiv:1805.02262v1 [cs.CL] 6 May 2018 • What is the percentage of female subjects in can be queried to find the number of nodes on the depression clinical trials? shortest undirected path between author X and Paul • Which of my co-authors published one or more Erdos˝ such that all edges on the path are labeled papers on coreference resolution? “authored”. • Which papers discuss the effects of Ranibizumab We reduce literature graph construction into fa- on the Retina? miliar NLP tasks such as sequence labeling, entity linking and relation extraction, and address some In this paper, we focus on the problem of ex- of the impractical assumptions commonly made in tracting structured data from scientific documents, the standard formulations of these tasks. For ex- which can later be used in natural language inter- ample, most research on named entity recognition faces (e.g., Iyer et al., 2017) or to improve ranking tasks report results on large labeled datasets such of results in academic search (e.g., Xiong et al., as CoNLL-2003 and ACE-2005 (e.g., Lample et al., 2016), and assume that entity types in the test set Entities. Each node of this type represents a match those labeled in the training set (including unique scientific concept discussed in the literature, work on domain adaptation, e.g., Daumé, 2007). with attributes such as ‘canonical name’, ‘aliases’ These assumptions, while useful for developing and ‘description’. Our literature graph has 0.4M and benchmarking new methods, are unrealistic for nodes of this type. We describe how we populate many domains and applications. The paper also entity nodes in §4.3. serves as an overview of the approach we adopt Entity mentions. Each node of this type rep- at www.semanticscholar.org in a step towards resents a textual reference of an entity in one of more intelligent academic search engines (Etzioni, the papers, with attributes such as ‘mention text’, 2011). ‘context’, and ‘confidence’. We describe how we In the next section, we start by describing our populate the 237M mentions in the literature graph symbolic representation of the literature. Then, we in §4.1. discuss how we extract metadata associated with a paper such as authors and references, then how we 2.2 Edge Types extract the entities mentioned in paper text. Before Citations. We instantiate a directed citation we conclude, we briefly describe other research edge from paper nodes p1 −! p2 for each p2 refer- challenges we are actively working on in order to enced in p1. Citation edges have attributes such as improve the quality of the literature graph. ‘from paper id’, ‘to paper id’ and ‘contexts’ (the tex- tual contexts where p2 is referenced in p1). While 2 Structure of The Literature Graph some of the paper sources provide these attributes as metadata, it is often necessary to extract them The literature graph is a property graph with di- from the paper PDF as detailed in §3. rected edges. Unlike Resource Description Frame- Authorship. We instantiate a directed author- work (RDF) graphs, nodes and edges in property ship edge between an author node and a paper node graphs have an internal structure which is more a −! p for each author of that paper. suitable for representing complex data types such Entity linking edges. We instantiate a directed as papers and entities. In this section, we describe edge from an extracted entity mention node to the the attributes associated with nodes and edges of entity it refers to. different types in the literature graph. Mention–mention relations. We instantiate a directed edge between a pair of mentions in the 2.1 Node Types same sentential context if the textual relation ex- Papers. We obtain metadata and PDF files traction model predicts one of a predefined list of of papers via partnerships with publishers (e.g., relation types between them in a sentential con- Springer, Nature), catalogs (e.g., DBLP, MED- text.1 We encode a symmetric relation between LINE), pre-publishing services (e.g., arXiv, bioRx- m1 and m2 as two directed edges m1 −! m2 and ive), as well as web-crawling. Paper nodes are m2 −! m1. associated with a set of attributes such as ‘title’, ‘ab- Entity–entity relations. While mention– stract’, ‘full text’, ‘venues’ and ‘publication year’. mention edges represent relations between men- While some of the paper sources provide these at- tions in a particular context, entity–entity edges tributes as metadata, it is often necessary to extract represent relations between abstract entities. These them from the paper PDF (details in §3). We de- relations may be imported from an existing knowl- terministically remove duplicate papers based on edge base (KB) or inferred from other edges in the string similarity of their metadata, resulting in 37M graph. unique paper nodes. Papers in the literature graph cover a variety of scientific disciplines, including 3 Extracting Metadata computer science, molecular biology, microbiology In the previous section, we described the overall and neuroscience. structure of the literature graph. Next, we discuss Authors. Each node of this type represents a how we populate paper nodes, author nodes, au- unique author, with attributes such as ‘first name’ thorship edges, and citation edges. and ‘last name’. The literature graph has 12M 1Due to space constraints, we opted not to discuss our nodes of this type. relation extraction models in this draft. Although some publishers provide sufficient Field Precision Recall F1 metadata about their papers, many papers are pro- title 85.5 85.5 85.5 vided with incomplete metadata. Also, papers ob- authors 92.1 92.1 92.1 tained via web-crawling are not associated with bibliography titles 89.3 89.4 89.3 any metadata. To fill in this gap, we built the Sci- bibliography authors 97.1 97.0 97.0 enceParse system to predict structured data from bibliography venues 91.7 89.7 90.7 the raw PDFs using recurrent neural networks bibliography years 98.0 98.0 98.0 (RNNs).2 For each paper, the system extracts the paper title, list of authors, and list of references; Table 1: Results of the ScienceParse system. each reference consists of a title, a list of authors, a venue, and a year. Preparing the input layer. We split each Short-Term Memory, Hochreiter and Schmidhuber, PDF into individual pages, and feed each page to 1997), i.e., Apache’s PDFBox library3 to convert it into a se- quence of tokens, where each token has features, ! ! ! gk = LSTM(Wik; gk−1); gk = [gk ; gk ]; e.g., ‘text’, ‘font size’, ‘space width’, ‘position on ! ! ! h = LSTM(gk; h ); hk = [h ; g ] the page’. k k−1 k k We normalize the token-level features before where W is a weight matrix, gk and hk are de- feeding them as inputs to the model. For each of the ! ! ‘font size’ and ‘space width’ features, we compute fined similarly to gk and hk but process token three normalized values (with respect to current sequences in the opposite direction. page, current document, and the whole training Following Collobert et al. (2011), we feed the corpus), each value ranging between -0.5 to +0.5. output of the second layer hk into a dense layer to The token’s ‘position on the page’ is given in XY predict unnormalized label weights for each token coordinate points. We scale the values linearly to and learn label bigram feature weights (often de- range from (−0:5; −0:5) at the top-left corner of the scribed as a conditional random field layer when page to (0:5; 0:5) at the bottom-right corner. used in neural architectures) to account for depen- In order to capture case information, we add dencies between labels. seven numeric features to the input representa- Training.