Effective interpretable learning for large-scale categorical data
Yishuo Zhang,Nayyar Zaidi,Jiahui Zhou,Tao Wang,Gang Li
DOI: https://doi.org/10.1007/s10618-024-01030-1
IF: 5.406
2024-05-28
Data Mining and Knowledge Discovery
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.
computer science, information systems, artificial intelligence