返回
A Bayesian network learning method for sparse and unbalanced data with GNN-based multilabel classification application
DOI:10.1016/j.asoc.2024.111393.png)
摘要
En 中文
Creating a well-defined Bayesian network (BN) is helpful for developing effective graph neural networks (GNNs) that exploit annotated labels in multilabel classification (MLC) tasks. Obtaining correct BNs can be challenging when the labels are highly unbalanced and sparse. This study proposes a novel scoring method to address data imbalance and sparsity by introducing an imbalanced weight term. Comparison results show that the BNs learned from the proposed scoring method are simpler and more accurate than the well-known methods. The obtained BNs based on the proposed method are used to construct GNNs for MLC tasks based on three datasets. The findings suggest that the adjacency matrices based on moral graphs of BNs exhibit more excellent stability than those derived from mathematical approaches. By selecting the parameter thoughtfully, GNNs based on the proposed BDsp scores showcase up to 12% classification performance improvement compared to GNNs based on existing Bayesian scores.
Keyword:
Bayesian network
Structure learning
Imbalance data
Sparse data
Graph neural network
期刊
IF:
6.6
论文数:
1.4W
被引数:
4.8W
机构
引用论文
Towards a sustainable nature reserve management: Using Bayesian network to quantify the threat of disturbance to ecosystem services迈向可持续的自然保护区管理: 使用贝叶斯网络量化干扰对生态系统服务的威胁
ECOSYSTEM SERVICES
IF6.6

