期刊
INFORMATION SCIENCES
卷 546, 期 -, 页码 1135-1147出版社
ELSEVIER SCIENCE INC
DOI: 10.1016/j.ins.2020.09.067
关键词
Machine learning; Factorization machine; Classification; Domain knowledge; Fuzzy set theory
资金
- National Natural Science Foundation of China [71972164, 71672163, 71722007, 71931001]
Rational and accurate classification requires considering historical information and domain knowledge. The proposed fuzzy factorization machine (fuzzy FM) integrates fuzzy set theory and factorization machine techniques, assigning each instance a membership through expert estimations. Differentiated weighting strategies in UFFM and BFFM variants improve the classification of imbalanced data, with experiments showing better performance compared to standard FM.
Rational and accurate classification cannot be achieved without considering both the historical information and domain knowledge. We propose fuzzy factorization machine (fuzzy FM) to integrate fuzzy set theory and factorization machine techniques for knowledge-enhanced classification. Each instance is assigned a membership through experts' estimations, and the instance's contribution to the objective function is weighted by its membership instead of the equal penalty in the standard FM. By adopting differentiated weighting strategies, we propose two variants of fuzzy FM: unilaterally weighted fuzzy FM (UFFM) and bilaterally weighted fuzzy FM (BFFM). In BFFM, each instance may not be fully assigned to one of two classes for better classification of imbalanced data, while in UFFM, each instance can only be assigned to one class. A set of membership generation approaches is summarized to quantify experts' prior estimations. We introduce solving methods based on stochastic gradient descent for UFFM and BFFM. Experiments on real credit datasets demonstrate that the proposed fuzzy FM models can yield better rational classification than previous baselines (including the standard FM). The proposed fuzzy FM is a generic machine learning framework that can be applied to various rational classification tasks. (C) 2020 Elsevier Inc. All rights reserved.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据