Inductive Topic Variational Graph Auto-Encoder for Text Classification
| Autor: | Jian-Yun Nie, Qianqian Xie, Min Peng, Jimin Huang, Pan Du |
|---|---|
| Rok vydání: | 2021 |
| Předmět: |
Structure (mathematical logic)
Topic model Computer science business.industry 05 social sciences Probabilistic logic 010501 environmental sciences computer.software_genre 01 natural sciences Autoencoder 0502 economics and business Bipartite graph Graph (abstract data type) Artificial intelligence 050207 economics business computer Feature learning Natural language processing 0105 earth and related environmental sciences Interpretability |
| Zdroj: | NAACL-HLT |
| Popis: | Graph convolutional networks (GCNs) have been applied recently to text classification and produced an excellent performance. However, existing GCN-based methods do not assume an explicit latent semantic structure of documents, making learned representations less effective and difficult to interpret. They are also transductive in nature, thus cannot handle out-of-graph documents. To address these issues, we propose a novel model named inductive Topic Variational Graph Auto-Encoder (T-VGAE), which incorporates a topic model into variational graph-auto-encoder (VGAE) to capture the hidden semantic information between documents and words. T-VGAE inherits the interpretability of the topic model and the efficient information propagation mechanism of VGAE. It learns probabilistic representations of words and documents by jointly encoding and reconstructing the global word-level graph and bipartite graphs of documents, where each document is considered individually and decoupled from the global correlation graph so as to enable inductive learning. Our experiments on several benchmark datasets show that our method outperforms the existing competitive models on supervised and semi-supervised text classification, as well as unsupervised text representation learning. In addition, it has higher interpretability and is able to deal with unseen documents. |
| Databáze: | OpenAIRE |
| Externí odkaz: |
|