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 *

Full-text access for editors Full-text access for subscribers Purchase this article Comment on this article