arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1908.02172cs.LGstat.ML
收藏

基于贝叶斯网络的多标签分类器链标签相关性分析

Bayesian Network Based Label Correlation Analysis For Multi-label Classifier Chain

Shenzhen University(深圳大学);University of Exeter(埃克塞特大学);City University of Hong Kong(香港城市大学)

Ran Wang, Suhe Ye, Ke Li, Sam Kwong

摘要:

分类器链(CC)是一种多标签学习方法,它根据标签顺序构造一系列二进制分类器。序列中的每个分类器负责预测一个标签的相关性。当训练分类器时,将继续进行的标签作为扩展特征。如果扩展特征与标签高度相关,则性能将得到改善,否则,性能不会受到影响甚至下降。如何发现标签相关性和确定标签顺序是CC方法的关键。本文采用贝叶斯网络(BN)对标签相关性进行建模,提出了一种新的基于贝叶斯网络的CC方法(BNCC)。首先,使用条件熵来描述标签之间的依赖关系。然后,以节点为标签,边的权值为依赖关系,建立BN。提出了一种新的评分函数来评价BN结构,并引入了一种启发式算法来优化BN。最后,通过对优化后的BN的节点进行拓扑排序,得到了构造CC模型的标号顺序。实验比较证明了该方法的可行性和有效性。

英文摘要:

Classifier chain (CC) is a multi-label learning approach that constructs a sequence of binary classifiers according to a label order. Each classifier in the sequence is responsible for predicting the relevance of one label. When training the classifier for a label, proceeding labels will be taken as extended features. If the extended features are highly correlated to the label, the performance will be improved, otherwise, the performance will not be influenced or even degraded. How to discover label correlation and determine the label order is critical for CC approach. This paper employs Bayesian network (BN) to model the label correlations and proposes a new BN-based CC method (BNCC). First, conditional entropy is used to describe the dependency relations among labels. Then, a BN is built up by taking nodes as labels and weights of edges as their dependency relations. A new scoring function is proposed to evaluate a BN structure, and a heuristic algorithm is introduced to optimize the BN. At last, by applying topological sorting on the nodes of the optimized BN, the label order for constructing CC model is derived. Experimental comparisons demonstrate the feasibility and effectiveness of the proposed method.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑