Title: A Bayesian network correlation-based classifier chain algorithm for multilabel learning
Authors: Hao Zhang; Kai-Biao Lin; Wei Weng; Juan Wen; Chin-Ling Chen
Addresses: School of Computer and Information Engineering, Xiamen University of Technology, Xiamen 361024, China ' School of Computer and Information Engineering, Xiamen University of Technology, Xiamen 361024, China ' School of Computer and Information Engineering, Xiamen University of Technology, Xiamen 361024, China ' Department of Statistics, School of Economics, Xiamen University, Xiamen 361005, China ' School of Computer and Information Engineering, Xiamen University of Technology, Xiamen 361024, China; School of Information Engineering, Changchun Sci-Tech University, Changchun 130600, China
Abstract: This article introduces a new multilabel learning method, the Bayesian network correlation-based CC (BNCC) algorithm, to decrease the uncertainty in the label order from the classifier chain (CC) algorithm. It uses a neural network constructed in TensorFlow as the classifier of all labels and calculates the corresponding error function, which is used to eliminate the influence of the feature set on all labels. A directed acyclic graph (DAG) Bayesian network is constructed by using the error function to identify the correlations between the labels. The optimal correlation label is identified via topological sorting. Finally, the sorted sequence is used as the chain order of the CC. The experimental results demonstrate that the proposed method is superior to the unordered CC model and other multilabel learning algorithms on several benchmark datasets.
Keywords: multilabel learning; Bayesian network; classifier chain; label relationship.
DOI: 10.1504/IJCSE.2022.124551
International Journal of Computational Science and Engineering, 2022 Vol.25 No.4, pp.437 - 447
Received: 23 Nov 2020
Accepted: 08 Jul 2021
Published online: 28 Jul 2022 *