Residential College | false |
Status | 已發表Published |
Effective interpretable learning for large-scale categorical data | |
Zhang, Yishuo1; Zaidi, Nayyar1; Zhou, Jiahui2; Wang, Tao3; Li, Gang1,4 | |
2024-07 | |
Source Publication | Data Mining and Knowledge Discovery |
ISSN | 1384-5810 |
Volume | 38Issue:4Pages:2223-2251 |
Abstract | Large scale categorical datasets are ubiquitous in machine learning and the success of most deployed machine learning models rely on how effectively the features are engineered. For large-scale datasets, parametric methods are generally used, among which three strategies for feature engineering are quite common. The first strategy focuses on managing the breadth (or width) of a network, e.g., generalized linear models (aka. wide learning). The second strategy focuses on the depth of a network, e.g., Artificial Neural networks or ANN (aka. deep learning). The third strategy relies on factorizing the interaction terms, e.g., Factorization Machines (aka. factorized learning). Each of these strategies brings its own advantages and disadvantages. Recently, it has been shown that for categorical data, combination of various strategies leads to excellent results. For example, WD-Learning, xdeepFM, etc., leads to state-of-the-art results. Following the trend, in this work, we have proposed another learning framework—WBDF-Learning, based on the combination of wide, deep, factorization, and a newly introduced component named Broad Interaction network (BIN). BIN is in the form of a Bayesian network classifier whose structure is learned apriori, and parameters are learned by optimizing a joint objective function along with wide, deep and factorized parts. We denote the learning of BIN parameters as broad learning. Additionally, the parameters of BIN are constrained to be actual probabilities—therefore, it is extremely interpretable. Furthermore, one can sample or generate data from BIN, which can facilitate learning and provides a framework for knowledge-guided machine learning. We demonstrate that our proposed framework possesses the resilience to maintain excellent classification performance when confronted with biased datasets. We evaluate the efficacy of our framework in terms of classification performance on various benchmark large-scale categorical datasets and compare against state-of-the-art methods. It is shown that, WBDF framework (a) exhibits superior performance on classification tasks, (b) boasts outstanding interpretability and (c) demonstrates exceptional resilience and effectiveness in scenarios involving skewed distributions. |
Keyword | Discriminative Bayesian Network Models Feature Engineering Interpretable Models Large Categorical Datasets Low-bias Models |
DOI | 10.1007/s10618-024-01030-1 |
URL | View the original |
Indexed By | SCIE |
Language | 英語English |
WOS Research Area | Computer Science |
WOS Subject | Computer Science, Artificial Intelligence ; Computer Science, Information Systems |
WOS ID | WOS:001232069900001 |
Publisher | SPRINGER, VAN GODEWIJCKSTRAAT 30, 3311 GZ DORDRECHT, NETHERLANDS |
Scopus ID | 2-s2.0-85194544818 |
Fulltext Access | |
Citation statistics | |
Document Type | Journal article |
Collection | ASIA-PACIFIC ACADEMY OF ECONOMICS AND MANAGEMENT |
Corresponding Author | Zaidi, Nayyar |
Affiliation | 1.School of Information Technology, Deakin University, Melbourne, 3125, Australia 2.Asia-Pacific Academy of Economics and Management, University of Macau, Zhuhai, Macao 3.Institute of Information Engineering, Chinese Academy of Sciences, Beijing, China 4.Deakin Cyber Research and Innovation Centre, Geelong, 3216, Australia |
Recommended Citation GB/T 7714 | Zhang, Yishuo,Zaidi, Nayyar,Zhou, Jiahui,et al. Effective interpretable learning for large-scale categorical data[J]. Data Mining and Knowledge Discovery, 2024, 38(4), 2223-2251. |
APA | Zhang, Yishuo., Zaidi, Nayyar., Zhou, Jiahui., Wang, Tao., & Li, Gang (2024). Effective interpretable learning for large-scale categorical data. Data Mining and Knowledge Discovery, 38(4), 2223-2251. |
MLA | Zhang, Yishuo,et al."Effective interpretable learning for large-scale categorical data".Data Mining and Knowledge Discovery 38.4(2024):2223-2251. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment