Learning multi-way relations via tensor decomposition with neural networks

K Maruhashi, M Todoriki, T Ohwa, K Goto… - Proceedings of the …, 2018 - ojs.aaai.org
K Maruhashi, M Todoriki, T Ohwa, K Goto, Y Hasegawa, H Inakoshi, H Anai
Proceedings of the AAAI Conference on Artificial Intelligence, 2018ojs.aaai.org
How can we classify multi-way data such as network traffic logs with multi-way relations
between source IPs, destination IPs, and ports? Multi-way data can be represented as a
tensor, and there have been several studies on classification of tensors to date. One critical
issue in the classification of multi-way relations is how to extract important features for
classification when objects in different multi-way data, ie, in different tensors, are not
necessarily in correspondence. In such situations, we aim to extract features that do not …
Abstract
How can we classify multi-way data such as network traffic logs with multi-way relations between source IPs, destination IPs, and ports? Multi-way data can be represented as a tensor, and there have been several studies on classification of tensors to date. One critical issue in the classification of multi-way relations is how to extract important features for classification when objects in different multi-way data, ie, in different tensors, are not necessarily in correspondence. In such situations, we aim to extract features that do not depend on how we allocate indices to an object such as a specific source IP; we are interested in only the structures of the multi-way relations. However, this issue has not been considered in previous studies on classification of multi-way data. We propose a novel method which can learn and classify multi-way data using neural networks. Our method leverages a novel type of tensor decomposition that utilizes a target core tensor expressing the important features whose indices are independent of those of the multi-way data. The target core tensor guides the tensor decomposition into more effective results and is optimized in a supervised manner. Our experiments on three different domains show that our method is highly accurate, especially on higher order data. It also enables us to interpret the classification results along with the matrices calculated with the novel tensor decomposition.
ojs.aaai.org
以上显示的是最相近的搜索结果。 查看全部搜索结果