UM  > ASIA-PACIFIC ACADEMY OF ECONOMICS AND MANAGEMENT
Residential Collegefalse
Status已發表Published
Effective interpretable learning for large-scale categorical data
Zhang, Yishuo1; Zaidi, Nayyar1; Zhou, Jiahui2; Wang, Tao3; Li, Gang1,4
2024-07
Source PublicationData Mining and Knowledge Discovery
ISSN1384-5810
Volume38Issue: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.

KeywordDiscriminative Bayesian Network Models Feature Engineering Interpretable Models Large Categorical Datasets Low-bias Models
DOI10.1007/s10618-024-01030-1
URLView the original
Indexed BySCIE
Language英語English
WOS Research AreaComputer Science
WOS SubjectComputer Science, Artificial Intelligence ; Computer Science, Information Systems
WOS IDWOS:001232069900001
PublisherSPRINGER, VAN GODEWIJCKSTRAAT 30, 3311 GZ DORDRECHT, NETHERLANDS
Scopus ID2-s2.0-85194544818
Fulltext Access
Citation statistics
Document TypeJournal article
CollectionASIA-PACIFIC ACADEMY OF ECONOMICS AND MANAGEMENT
Corresponding AuthorZaidi, Nayyar
Affiliation1.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.
Related Services
Recommend this item
Bookmark
Usage statistics
Export to Endnote
Google Scholar
Similar articles in Google Scholar
[Zhang, Yishuo]'s Articles
[Zaidi, Nayyar]'s Articles
[Zhou, Jiahui]'s Articles
Baidu academic
Similar articles in Baidu academic
[Zhang, Yishuo]'s Articles
[Zaidi, Nayyar]'s Articles
[Zhou, Jiahui]'s Articles
Bing Scholar
Similar articles in Bing Scholar
[Zhang, Yishuo]'s Articles
[Zaidi, Nayyar]'s Articles
[Zhou, Jiahui]'s Articles
Terms of Use
No data!
Social Bookmark/Share
All comments (0)
No comment.
 

Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.