Learning beyond datasets: Knowledge Graph Augmented Neural Networks for Natural language Processing
This work addresses the limitation of data dependency in NLP models by integrating external knowledge, offering a scalable solution for various tasks, though it is incremental as it builds on existing attention and convolution methods.
The authors tackled the problem of machine learning models' heavy dependence on specific training data by enhancing them with organized world knowledge from knowledge graphs for NLP tasks, resulting in significant performance improvements on text classification and natural language inference datasets and enabling training with substantially less labeled data.
Machine Learning has been the quintessential solution for many AI problems, but learning is still heavily dependent on the specific training data. Some learning models can be incorporated with a prior knowledge in the Bayesian set up, but these learning models do not have the ability to access any organised world knowledge on demand. In this work, we propose to enhance learning models with world knowledge in the form of Knowledge Graph (KG) fact triples for Natural Language Processing (NLP) tasks. Our aim is to develop a deep learning model that can extract relevant prior support facts from knowledge graphs depending on the task using attention mechanism. We introduce a convolution-based model for learning representations of knowledge graph entity and relation clusters in order to reduce the attention space. We show that the proposed method is highly scalable to the amount of prior information that has to be processed and can be applied to any generic NLP task. Using this method we show significant improvement in performance for text classification with News20, DBPedia datasets and natural language inference with Stanford Natural Language Inference (SNLI) dataset. We also demonstrate that a deep learning model can be trained well with substantially less amount of labeled training data, when it has access to organised world knowledge in the form of knowledge graph.