Barack's Wife Hillary: Using Knowledge Graphs for Fact-Aware Language Modeling

Total Page:16

File Type:pdf, Size:1020Kb

Barack's Wife Hillary: Using Knowledge Graphs for Fact-Aware Language Modeling Barack’s Wife Hillary: Using Knowledge Graphs for Fact-Aware Language Modeling Robert L. Logan IV∗ Nelson F. Liu†§ Matthew E. Peters§ Matt Gardner§ Sameer Singh∗ ∗ University of California, Irvine, CA, USA † University of Washington, Seattle, WA, USA § Allen Institute for Artificial Intelligence, Seattle, WA, USA {rlogan, sameer}@uci.edu, {mattg, matthewp}@allenai.org, nfl[email protected] Abstract [Super Mario Land] is a [1989] [side-scrolling] [platform video game] developed and published Modeling human language requires the ability by [Nintendo] as a [launch title] for their [Game to not only generate fluent text but also en- Boy] [handheld game console]. code factual knowledge. However, traditional publisher Super Mario Land Nintendo launch game language models are only capable of remem- Q647249 Q8093 Q1425505 Publication platform manufacturer bering facts seen at training time, and often genre have difficulty recalling them. To address this, Date 21 April 1989 platform game Game Boy we introduce the knowledge graph language Date Q828322 Q186437 model (KGLM), a neural language model with instance of mechanisms for selecting and copying facts side-scrolling video game from a knowledge graph that are relevant to Q2281714 handheld game console Q941818 the context. These mechanisms enable the model to render information it has never seen Figure 1: Linked WikiText-2 Example. A localized before, as well as generate out-of-vocabulary knowledge graph containing facts that are (possibly) tokens. We also introduce the Linked WikiText- conveyed in the sentence above. The graph is built by it- 2 dataset,1 a corpus of annotated text aligned to eratively linking each detected entity to Wikidata, then the Wikidata knowledge graph whose contents adding any relations to previously mentioned entities. (roughly) match the popular WikiText-2 bench- Note that not all entities are connected, potentially due mark (Merity et al., 2017). In experiments, we to missing relations in Wikidata. demonstrate that the KGLM achieves signifi- cantly better performance than a strong base- line language model. We additionally com- pare different language models’ ability to com- training. For instance, when conditioned on the text plete sentences requiring factual knowledge, at the top of Figure 1, an AWD-LSTM language and show that the KGLM outperforms even model (Merity et al., 2018) trained on Wikitext-2 very large language models in generating facts. assigns higher probability to the word “PlaySta- tion” than “Game Boy”, even though this sentence 1 Introduction appears verbatim in the training data. This is not surprising—existing models represent the distribu- For language models to generate plausible sen- tion over the entire vocabulary directly, whether tences, they must be both syntactically coherent as they are common words, references to real world well as consistent with the world they describe. Al- entities, or factual information like dates and num- though language models are quite skilled at generat- bers. As a result, language models are unable to ing grammatical sentences, and previous work has generate factually correct sentences, do not gen- shown that language models also possess some de- eralize to rare/unseen entities, and often omit rare gree of common-sense reasoning and basic knowl- tokens from the vocabulary (instead generating UN- edge (Vinyals and Le, 2015; Serban et al., 2016; KNOWN tokens). Trinh and Le, 2019), their ability to generate fac- We introduce the knowledge graph language tually correct text is quite limited. The clearest model (KGLM), a neural language model with limitation of existing language models is that they, mechanisms for selecting and copying information at best, can only memorize facts observed during from an external knowledge graph. The KGLM 1https://rloganiv.github.io/linked-wikitext-2 maintains a dynamically growing local knowledge graph, a subset of the knowledge graph that con- 2 Knowledge Graph Language Model tains entities that have already been mentioned in In this section we introduce a language model that the text, and their related entities. When generating is conditioned on an external, structured knowledge entity tokens, the model either decides to render source, which it uses to generate factual text. a new entity that is absent from the local graph, thereby growing the local knowledge graph, or to 2.1 Problem Setup and Notation render a fact from the local graph. When render- A language model defines a probability distribution ing, the model combines the standard vocabulary over each token within a sequence, conditioned on with tokens available in the knowledge graph, thus the sequence of tokens observed so far. We denote supporting numbers, dates, and other rare tokens. the random variable representing the next token as Figure 1 illustrates how the KGLM works. Ini- xt and the sequence of the tokens before t as x<t, tially, the graph is empty and the model uses the i.e. language models compute p(xt|x<t). RNN lan- entity Super Mario Land to render the first three guage models (Mikolov et al., 2010) parameterize tokens, thus adding it and its relations to the local this distribution using a recurrent structure: knowledge graph. After generating the next two to- W h b kens (“is”, “a”) using the standard language model, p(xt|x<t) = softmax( h t + ), (1) the model selects Super Mario Land as the parent ht = RNN(ht−1, xt−1). entity, Publication Date as the relation to render, and copies one of the tokens of the date entity as We use LSTMs (Hochreiter and Schmidhuber, the token (“1989” in this case). 1997) as the recurrent module in this paper. A knowledge graph (KG) is a directed, labeled To facilitate research on knowledge graph-based graph consisting of entities E as nodes, with edges language modeling, we collect the distantly su- defined over a set of relations R, i.e. KG = pervised Linked WikiText-2 dataset. The underly- {(p, r, e) | p ∈ E, r ∈ R, e ∈ E}, where p is a par- ing text closely matches WikiText-2 (Merity et al., ent entity with relation r to another entity e. Prac- 2017), a popular benchmark for language model- tical KGs have other aspects that make this for- ing, allowing comparisons against existing mod- mulation somewhat inexact: some relations are to els. The tokens in the text are linked to entities in literal values, such as numbers and dates, facts Wikidata (Vrandeciˇ c´ and Krötzsch, 2014) using a may be expressed as properties on relations, and combination of human-provided links and off-the- entities have aliases as the set of strings that can shelf linking and coreference models. We also use refer to the entity. We also define a local knowl- relations between these entities in Wikidata to con- edge graph for a subset of entities E<t as KG<t = struct plausible reasons for why an entity may have {(p, r, e) | p ∈ E<t, r ∈ R, e ∈ E}, i.e. contains been mentioned: it could either be related to an entities E<t and all facts they participate in. entity that is already mentioned (including itself) or a brand new, unrelated entity for the document. 2.2 Generative KG Language Model We train and evaluate the KGLM on Linked The primary goal of the knowledge graph lan- WikiText-2. When compared against AWD-LSTM, guage model (KGLM) is to enable a neural lan- a recent and performant language model, KGLM guage model to generate entities and facts from obtains not only a lower overall perplexity, but also a knowledge graph. To encourage the model to a substantially lower unknown-penalized perplex- generate facts that have appeared in the context ity (Ueberla, 1994; Ahn et al., 2016), a metric that already, KGLM will maintain a local knowledge allows fair comparisons between models that accu- graph containing all facts involving entities that rately model rare tokens and ones that predict them have appeared in the context. As the model decides to be unknown. We also compare factual com- to refer to entities that have not been referred to pletion capabilities of these models, where they yet, it will grow the local knowledge graph with predict the next word after a factual sentence (e.g., additional entities and facts to reflect the new entity. “Barack is married to ”) and show that KGLM Formally, we will compute p(xt, Et|x<t, E<t) is significantly more accurate. Lastly, we show that where x<t is the sequence of observed tokens, E<t the model is able to generate accurate facts for rare is the set of entities mentioned in x<t, and KG<t is entities, and can be controlled via modifications the local knowledge graph determined by E<t, as the knowledge graph. described above. The generative process is: knowledge graph, denoted by ve for entity e and graph, the text is made up of disjoint sentences that vr for relation r. To select et from all entities in do not provide sufficient context to train a pow- case tt = new, we use: erful language model. Our goals are much more aligned to the data-to-text task (Ahn et al., 2016; v h h p(et) = softmax( e · ( t,p + t,r)) Lebret et al., 2016; Wiseman et al., 2017; Yang et al., 2017; Gardent et al., 2017; Ferreira et al., over all e ∈ E. The reason we add h and h is t,p t,r 2018), where a small table-sized KB is provided to to mimic the structure of TransE, which we use to generate a short piece of text; we are interested in obtain entity and relation embeddings. Details on language models that dynamically decide the facts TransE will be provided in Section 4. For mention to incorporate from the knowledge graph, guided of a related entity, t = related, we pick a parent t by the discourse.
Recommended publications
  • Machine Learning: Unsupervised Methods Sepp Hochreiter Other Courses
    Machine Learning Unsupervised Methods Part 1 Sepp Hochreiter Institute of Bioinformatics Johannes Kepler University, Linz, Austria Course 3 ECTS 2 SWS VO (class) 1.5 ECTS 1 SWS UE (exercise) Basic Course of Master Bioinformatics Basic Course of Master Computer Science: Computational Engineering / Int. Syst. Class: Mo 15:30-17:00 (HS 18) Exercise: Mon 13:45-14:30 (S2 053) – group 1 Mon 14:30-15:15 (S2 053) – group 2+group 3 EXAMS: VO: 3 part exams UE: weekly homework (evaluated) Machine Learning: Unsupervised Methods Sepp Hochreiter Other Courses Lecture Lecturer 365,077 Machine Learning: Unsupervised Techniques VL Hochreiter Mon 15:30-17:00/HS 18 365,078 Machine Learning: Unsupervised Techniques – G1 UE Hochreiter Mon 13:45-14:30/S2 053 365,095 Machine Learning: Unsupervised Techniques – G2+G3 UE Hochreiter Mon 14:30-15:15/S2 053 365,041 Theoretical Concepts of Machine Learning VL Hochreiter Thu 15:30-17:00/S3 055 365,042 Theoretical Concepts of Machine Learning UE Hochreiter Thu 14:30-15:15/S3 055 365,081 Genome Analysis & Transcriptomics KV Regl Fri 8:30-11:00/S2 053 365,082 Structural Bioinformatics KV Regl Tue 8:30-11:00/HS 11 365,093 Deep Learning and Neural Networks KV Unterthiner Thu 10:15-11:45/MT 226 365,090 Special Topics on Bioinformatics (B/C/P/M): KV Klambauer block Population genetics 365,096 Special Topics on Bioinformatics (B/C/P/M): KV Klambauer block Artificial Intelligence in Life Sciences 365,079 Introduction to R KV Bodenhofer Wed 15:30-17:00/MT 127 365,067 Master's Seminar SE Hochreiter Mon 10:15-11:45/S3 318 365,080 Master's Thesis Seminar SS SE Hochreiter Mon 10:15-11:45/S3 318 365,091 Bachelor's Seminar SE Hochreiter - 365,019 Dissertantenseminar Informatik 3 SE Hochreiter Mon 10:15-11:45/S3 318 347,337 Bachelor Seminar Biological Chemistry JKU (incl.
    [Show full text]
  • Prof. Dr. Sepp Hochreiter
    Speaker: Univ.-Prof. Dr. Sepp Hochreiter Institute for Machine Learning & LIT AI Lab, Johannes Kepler University Linz, Austria Title: Deep Learning -- the Key to Enable Artificial Intelligence Abstract: Deep Learning has emerged as one of the most successful fields of machine learning and artificial intelligence with overwhelming success in industrial speech, language and vision benchmarks. Consequently it became the central field of research for IT giants like Google, Facebook, Microsoft, Baidu, and Amazon. Deep Learning is founded on novel neural network techniques, the recent availability of very fast computers, and massive data sets. In its core, Deep Learning discovers multiple levels of abstract representations of the input. The main obstacle to learning deep neural networks is the vanishing gradient problem. The vanishing gradient impedes credit assignment to the first layers of a deep network or early elements of a sequence, therefore limits model selection. Most major advances in Deep Learning can be related to avoiding the vanishing gradient like unsupervised stacking, ReLUs, residual networks, highway networks, and LSTM networks. Currently, LSTM recurrent neural networks exhibit overwhelmingly successes in different AI fields like speech, language, and text analysis. LSTM is used in Google’s translate and speech recognizer, Apple’s iOS 10, Facebooks translate, and Amazon’s Alexa. We use LSTM in collaboration with Zalando and Bayer, e.g. to analyze blogs and twitter news related to fashion and health. In the AUDI Deep Learning Center, which I am heading, and with NVIDIA we apply Deep Learning to advance autonomous driving. In collaboration with Infineon we use Deep Learning for perception tasks, e.g.
    [Show full text]
  • Evaluating Recurrent Neural Network Explanations
    Evaluating Recurrent Neural Network Explanations Leila Arras1, Ahmed Osman1, Klaus-Robert Muller¨ 2;3;4, and Wojciech Samek1 1Machine Learning Group, Fraunhofer Heinrich Hertz Institute, Berlin, Germany 2Machine Learning Group, Technische Universitat¨ Berlin, Berlin, Germany 3Department of Brain and Cognitive Engineering, Korea University, Seoul, Korea 4Max Planck Institute for Informatics, Saarbrucken,¨ Germany fleila.arras, [email protected] Abstract a particular decision, i.e., when the input is a se- quence of words: which words are determinant Recently, several methods have been proposed for the final decision? This information is crucial to explain the predictions of recurrent neu- ral networks (RNNs), in particular of LSTMs. to unmask “Clever Hans” predictors (Lapuschkin The goal of these methods is to understand the et al., 2019), and to allow for transparency of the network’s decisions by assigning to each in- decision-making process (EU-GDPR, 2016). put variable, e.g., a word, a relevance indicat- Early works on explaining neural network pre- ing to which extent it contributed to a partic- dictions include Baehrens et al.(2010); Zeiler and ular prediction. In previous works, some of Fergus(2014); Simonyan et al.(2014); Springen- these methods were not yet compared to one berg et al.(2015); Bach et al.(2015); Alain and another, or were evaluated only qualitatively. We close this gap by systematically and quan- Bengio(2017), with several works focusing on ex- titatively comparing these methods in differ- plaining the decisions of convolutional neural net- ent settings, namely (1) a toy arithmetic task works (CNNs) for image recognition. More re- which we use as a sanity check, (2) a five-class cently, this topic found a growing interest within sentiment prediction of movie reviews, and be- NLP, amongst others to explain the decisions of sides (3) we explore the usefulness of word general CNN classifiers (Arras et al., 2017a; Ja- relevances to build sentence-level representa- covi et al., 2018), and more particularly to explain tions.
    [Show full text]
  • Employing a Transformer Language Model for Information Retrieval and Document Classification
    DEGREE PROJECT IN THE FIELD OF TECHNOLOGY ENGINEERING PHYSICS AND THE MAIN FIELD OF STUDY COMPUTER SCIENCE AND ENGINEERING, SECOND CYCLE, 30 CREDITS STOCKHOLM, SWEDEN 2020 Employing a Transformer Language Model for Information Retrieval and Document Classification Using OpenAI's generative pre-trained transformer, GPT-2 ANTON BJÖÖRN KTH ROYAL INSTITUTE OF TECHNOLOGY SCHOOL OF ELECTRICAL ENGINEERING AND COMPUTER SCIENCE Employing a Transformer Language Model for Information Retrieval and Document Classification ANTON BJÖÖRN Master in Machine Learning Date: July 11, 2020 Supervisor: Håkan Lane Examiner: Viggo Kann School of Electrical Engineering and Computer Science Host company: SSC - Swedish Space Corporation Company Supervisor: Jacob Ask Swedish title: Transformermodellers användbarhet inom informationssökning och dokumentklassificering iii Abstract As the information flow on the Internet keeps growing it becomes increasingly easy to miss important news which does not have a mass appeal. Combating this problem calls for increasingly sophisticated information retrieval meth- ods. Pre-trained transformer based language models have shown great gener- alization performance on many natural language processing tasks. This work investigates how well such a language model, Open AI’s General Pre-trained Transformer 2 model (GPT-2), generalizes to information retrieval and classi- fication of online news articles, written in English, with the purpose of compar- ing this approach with the more traditional method of Term Frequency-Inverse Document Frequency (TF-IDF) vectorization. The aim is to shed light on how useful state-of-the-art transformer based language models are for the construc- tion of personalized information retrieval systems. Using transfer learning the smallest version of GPT-2 is trained to rank and classify news articles achiev- ing similar results to the purely TF-IDF based approach.
    [Show full text]
  • An SMO Algorithm for the Potential Support Vector Machine
    LETTER Communicated by Terrence Sejnowski An SMO Algorithm for the Potential Support Vector Machine Tilman Knebel [email protected] Downloaded from http://direct.mit.edu/neco/article-pdf/20/1/271/817177/neco.2008.20.1.271.pdf by guest on 02 October 2021 Neural Information Processing Group, Fakultat¨ IV, Technische Universitat¨ Berlin, 10587 Berlin, Germany Sepp Hochreiter [email protected] Institute of Bioinformatics, Johannes Kepler University Linz, 4040 Linz, Austria Klaus Obermayer [email protected] Neural Information Processing Group, Fakultat¨ IV and Bernstein Center for Computational Neuroscience, Technische Universitat¨ Berlin, 10587 Berlin, Germany We describe a fast sequential minimal optimization (SMO) procedure for solving the dual optimization problem of the recently proposed potential support vector machine (P-SVM). The new SMO consists of a sequence of iteration steps in which the Lagrangian is optimized with respect to either one (single SMO) or two (dual SMO) of the Lagrange multipliers while keeping the other variables fixed. An efficient selection procedure for Lagrange multipliers is given, and two heuristics for improving the SMO procedure are described: block optimization and annealing of the regularization parameter . A comparison of the variants shows that the dual SMO, including block optimization and annealing, performs ef- ficiently in terms of computation time. In contrast to standard support vector machines (SVMs), the P-SVM is applicable to arbitrary dyadic data sets, but benchmarks are provided against libSVM’s -SVR and C-SVC implementations for problems that are also solvable by standard SVM methods. For those problems, computation time of the P-SVM is comparable to or somewhat higher than the standard SVM.
    [Show full text]
  • A Deep Learning Architecture for Conservative Dynamical Systems: Application to Rainfall-Runoff Modeling
    A Deep Learning Architecture for Conservative Dynamical Systems: Application to Rainfall-Runoff Modeling Grey Nearing Google Research [email protected] Frederik Kratzert LIT AI Lb & Institute for Machine Learning, Johannes Kepler University [email protected] Daniel Klotz LIT AI Lab & Institute for Machine Learning, Johannes Kepler University [email protected] Pieter-Jan Hoedt LIT AI Lab & Institute for Machine Learning, Johannes Kepler University [email protected] Günter Klambauer LIT AI Lab & Institute for Machine Learning, Johannes Kepler University [email protected] Sepp Hochreiter LIT AI Lab & Institute for Machine Learning, Johannes Kepler University [email protected] Hoshin Gupta Department of Hydrology and Atmospheric Sciences, University of Arizona [email protected] Sella Nevo Yossi Matias Google Research Google Research [email protected] [email protected] Abstract The most accurate and generalizable rainfall-runoff models produced by the hydro- logical sciences community to-date are based on deep learning, and in particular, on Long Short Term Memory networks (LSTMs). Although LSTMs have an explicit state space and gates that mimic input-state-output relationships, these models are not based on physical principles. We propose a deep learning architecture that is based on the LSTM and obeys conservation principles. The model is benchmarked on the mass-conservation problem of simulating streamflow. AI for Earth Sciences Workshop at NeurIPS 2020. 1 Introduction Due to computational challenges and to the increasing volume and variety of hydrologically-relevant Earth observation data, machine learning is particularly well suited to help provide efficient and reliable flood forecasts over large scales [12]. Hydrologists have known since the 1990’s that machine learning generally produces better streamflow estimates than either calibrated conceptual models or process-based models [7].
    [Show full text]
  • LONG SHORT TERM MEMORY NEURAL NETWORK for KEYBOARD GESTURE DECODING Ouais Alsharif, Tom Ouyang, Françoise Beaufays, Shumin Zhai
    LONG SHORT TERM MEMORY NEURAL NETWORK FOR KEYBOARD GESTURE DECODING Ouais Alsharif, Tom Ouyang, Franc¸oise Beaufays, Shumin Zhai, Thomas Breuel, Johan Schalkwyk Google ABSTRACT Gesture typing is an efficient input method for phones and tablets using continuous traces created by a pointed object (e.g., finger or stylus). Translating such continuous gestures into textual input is a challenging task as gesture inputs ex- hibit many features found in speech and handwriting such as high variability, co-articulation and elision. In this work, we address these challenges with a hybrid approach, combining a variant of recurrent networks, namely Long Short Term Memories [1] with conventional Finite State Transducer de- coding [2]. Results using our approach show considerable improvement relative to a baseline shape-matching-based system, amounting to 4% and 22% absolute improvement respectively for small and large lexicon decoding on real datasets and 2% on a synthetic large scale dataset. Index Terms— Long-short term memory, LSTM, gesture Fig. 1. An example keyboard with the word Today gestured. typing, keyboard decoder efficiency, accuracy and robustness. Efficiency is im- 1. INTRODUCTION portant because such solutions tend to run on mobile devices which permit a smaller footprint in terms of computation and The Word-Gesture Keyboard (WGK) [3], first published in memory. Accuracy and robustness are also crucial, as users early 2000s [4, 5] has become a popular text input method on are not expected to gesture type words with high accuracy. touchscreen mobile devices. In the last few years this new Since most users would be using a mobile device for input, keyboard input paradigm has appeared in many products in- it is highly likely that gestures would be offset, noisy or even cluding Google’s Android keyboard.
    [Show full text]
  • Are Pre-Trained Language Models Aware of Phrases?Simplebut Strong Baselinesfor Grammar Induction
    Published as a conference paper at ICLR 2020 ARE PRE-TRAINED LANGUAGE MODELS AWARE OF PHRASES?SIMPLE BUT STRONG BASELINES FOR GRAMMAR INDUCTION Taeuk Kim1, Jihun Choi1, Daniel Edmiston2 & Sang-goo Lee1 1Dept. of Computer Science and Engineering, Seoul National University, Seoul, Korea 2Dept. of Linguistics, University of Chicago, Chicago, IL, USA ftaeuk,jhchoi,[email protected], [email protected] ABSTRACT With the recent success and popularity of pre-trained language models (LMs) in natural language processing, there has been a rise in efforts to understand their in- ner workings. In line with such interest, we propose a novel method that assists us in investigating the extent to which pre-trained LMs capture the syntactic notion of constituency. Our method provides an effective way of extracting constituency trees from the pre-trained LMs without training. In addition, we report intriguing findings in the induced trees, including the fact that some pre-trained LMs outper- form other approaches in correctly demarcating adverb phrases in sentences. 1 INTRODUCTION Grammar induction, which is closely related to unsupervised parsing and latent tree learning, allows one to associate syntactic trees, i.e., constituency and dependency trees, with sentences. As gram- mar induction essentially assumes no supervision from gold-standard syntactic trees, the existing approaches for this task mainly rely on unsupervised objectives, such as language modeling (Shen et al., 2018b; 2019; Kim et al., 2019a;b) and cloze-style word prediction (Drozdov et al., 2019) to train their task-oriented models. On the other hand, there is a trend in the natural language pro- cessing (NLP) community of leveraging pre-trained language models (LMs), e.g., ELMo (Peters et al., 2018) and BERT (Devlin et al., 2019), as a means of acquiring contextualized word represen- tations.
    [Show full text]
  • Which Neural Net Architectures Give Rise to Exploding and Vanishing Gradients?
    Which Neural Net Architectures Give Rise to Exploding and Vanishing Gradients? Boris Hanin Department of Mathematics Texas A& M University College Station, TX, USA [email protected] Abstract We give a rigorous analysis of the statistical behavior of gradients in a randomly initialized fully connected network with ReLU activations. Our results show that the empirical variance of the squaresN of the entries in the input-output Jacobian of is exponential in a simple architecture-dependent constant β, given by the sumN of the reciprocals of the hidden layer widths. When β is large, the gradients computed by at initialization vary wildly. Our approach complements the mean field theory analysisN of random networks. From this point of view, we rigorously compute finite width corrections to the statistics of gradients at the edge of chaos. 1 Introduction A fundamental obstacle in training deep neural nets using gradient based optimization is the exploding and vanishing gradient problem (EVGP), which has attracted much attention (e.g. [BSF94, HBF+01, MM15, XXP17, PSG17, PSG18]) after first being studied by Hochreiter [Hoc91]. The EVGP occurs when the derivative of the loss in the SGD update @ W W λ L , (1) − @W is very large for some trainable parameters W and very small for others: @ L 0 or . @W ⇡ 1 This makes the increment in (1) either too small to be meaningful or too large to be precise. In practice, a number of ways of overcoming the EVGP have been proposed (see e.g. [Sch]). Let us mention three general approaches: (i) using architectures such as LSTMs [HS97], highway networks [SGS15], or ResNets [HZRS16] that are designed specifically to control gradients; (ii) precisely initializing weights (e.g.
    [Show full text]
  • Rectified Factor Networks for Biclustering
    Workshop track - ICLR 2016 RECTIFIED FACTOR NETWORKS FOR BICLUSTERING Djork-Arne´ Clevert, Thomas Unterthiner & Sepp Hochreiter Institute of Bioinformatics Johannes Kepler University, Linz, Austria fokko,unterthiner,[email protected] ABSTRACT Biclustering is evolving into one of the major tools for analyzing large datasets given as matrix of samples times features. Biclustering has been successfully ap- plied in life sciences, e.g. for drug design, in e-commerce, e.g. for internet retailing or recommender systems. FABIA is one of the most successful biclustering methods which excelled in dif- ferent projects and is used by companies like Janssen, Bayer, or Zalando. FABIA is a generative model that represents each bicluster by two sparse membership vec- tors: one for the samples and one for the features. However, FABIA is restricted to about 20 code units because of the high computational complexity of computing the posterior. Furthermore, code units are sometimes insufficiently decorrelated. Sample membership is difficult to determine because vectors do not have exact zero entries and can have both large positive and large negative values. We propose to use the recently introduced unsupervised Deep Learning approach Rectified Factor Networks (RFNs) to overcome the drawbacks of FABIA. RFNs efficiently construct very sparse, non-linear, high-dimensional representations of the input via their posterior means. RFN learning is a generalized alternating min- imization algorithm based on the posterior regularization method which enforces non-negative and normalized posterior means. Each code unit represents a biclus- ter, where samples for which the code unit is active belong to the bicluster and features that have activating weights to the code unit belong to the bicluster.
    [Show full text]
  • Blimp: the Benchmark of Linguistic Minimal Pairs for English
    BLiMP: The Benchmark of Linguistic Minimal Pairs for English Alex Warstadt1, Alicia Parrish1, Haokun Liu2, Anhad Mohananey2, Wei Peng2, Sheng-FuWang1, Samuel R. Bowman1,2,3 1Department of Linguistics 2Department of Computer Science 3Center for Data Science New York University New York University New York University {warstadt,alicia.v.parrish,haokunliu,anhad, weipeng,shengfu.wang,bowman}@nyu.edu Abstract of these studies uses a different set of metrics, and focuses on a small set of linguistic We introduce The Benchmark of Linguistic paradigms, severely limiting any possible big- Minimal Pairs (BLiMP),1 a challenge set for picture conclusions. evaluating the linguistic knowledge of lan- guage models (LMs) on major grammatical (1) a. ThecatsannoyTim.(grammatical) phenomena in English. BLiMP consists of b. *ThecatsannoysTim.(ungrammatical) 67 individual datasets, each containing 1,000 minimal pairs—that is, pairs of minimally dif- We introduce the Benchmark of Linguistic ferent sentences that contrast in grammatical Minimal Pairs (shortened to BLiMP), a linguis- acceptability and isolate specific phenomenon tically motivated benchmark for assessing the in syntax, morphology, or semantics. We gen- sensitivity of LMs to acceptability contrasts across erate the data according to linguist-crafted a wide range of English phenomena,covering both grammar templates, and human aggregate previously studied and novel contrasts. BLiMP agreement with the labels is 96.4%. We consists of 67 datasets automatically generated evaluate n-gram, LSTM, and Transformer from linguist-crafted grammar templates, each (GPT-2 and Transformer-XL) LMs by observ- containing 1,000 minimal pairs and organized ing whether they assign a higher probability to by phenomenon into 12 categories.
    [Show full text]
  • Revisit Long Short-Term Memory: an Optimization Perspective
    Revisit Long Short-Term Memory: An Optimization Perspective Qi Lyu, Jun Zhu State Key Laboratory of Intelligent Technology and Systems (LITS) Tsinghua National Laboratory for Information Science and Technology (TNList) Department of Computer Science and Technology, Tsinghua University, Beijing 100084, China [email protected], [email protected] Abstract Long Short-Term Memory (LSTM) is a deep recurrent neural network archi- tecture with high computational complexity. Contrary to the standard practice to train LSTM online with stochastic gradient descent (SGD) methods, we pro- pose a matrix-based batch learning method for LSTM with full Backpropagation Through Time (BPTT). We further solve the state drifting issues as well as im- proving the overall performance for LSTM using revised activation functions for gates. With these changes, advanced optimization algorithms are applied to LSTM with long time dependency for the first time and show great advantages over SGD methods. We further demonstrate that large-scale LSTM training can be greatly accelerated with parallel computation architectures like CUDA and MapReduce. 1 Introduction Long Short-Term Memory (LSTM) [1] is a deep recurrent neural network (RNN) well-suited to learn from experiences to classify, process and predict time series when there are very long time lags of unknown size between important events. LSTM consists of LSTM blocks instead of (or in addition to) regular network units. A LSTM block may be described as a “smart” network unit that can remember a value for an arbitrary length of time. This is one of the main reasons why LSTM outperforms other sequence learning methods in numerous applications such as achieving the best known performance in speech recognition [2], and winning the ICDAR handwriting competition in 2009.
    [Show full text]