Title: Factorisation machines considering the latent characteristics behind target data

Authors: Tomoya Sugisaki; Kenta Mikawa; Masayuki Goto

Addresses: Graduate School of Creative Science and Engineering, Waseda University, 3-4-1 Okubo, Shinjuku-ku, 169-8555, Japan ' Faculty of Engineering, Shonan Institute of Technology, 1-1-25, Tsujido-nishikaigan, Fujisawa, 251-8511, Japan ' School of Creative Science and Engineering, Waseda University, 3-4-1 Okubo, Shinjuku-ku, 169-8555, Japan

Abstract: A factorisation machine (FM) is often used as a regression-type prediction model that considers the interactions between features. In recent years, many researchers have focused on the differences between interactions from the perspective of prediction accuracy. In some cases, it can be assumed that there are several groups with different characteristics in terms of the relationships between input and output data for a target population. Therefore, there is a possibility that the effects of interactions will differ between input data groups, which are used to construct latent clusters. However, there has been little research on FMs considering the latent characteristics behind target data. Therefore, this paper proposes a novel type of FM that considers latent data characteristics. The proposed model can express the differences between the interaction effects of latent groups and it is expected to improve the prediction accuracy of FMs for complex problems. Based on a demonstration experiment using the MovieLens dataset, the effectiveness of our proposed model is verified.

Keywords: machine learning; factorisation machines; latent characteristics; interaction; regression model.

DOI: 10.1504/AJMSA.2020.112735

Asian Journal of Management Science and Applications, 2020 Vol.5 No.2, pp.111 - 128

Received: 13 Mar 2020
Accepted: 24 Jul 2020

Published online: 01 Feb 2021 *

Full-text access for editors Full-text access for subscribers Purchase this article Comment on this article