.png){kind=icon}
.png){kind=icon}
Return
A Bayesian network learning method for sparse and unbalanced data with GNN-based multilabel classification application
.png){kind=icon}
PRE
AI
DOI：10.1016/j.asoc.2024.111393.png)
.png){kind=icon}
Original request for help.png){kind=icon}
Share.png){kind=icon}
Save
Abstract
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.
Keywords:
Bayesian network
Structure learning
Imbalance data
Sparse data
Graph neural network
Journal
IF:
6.6
Papers:
1.4W
Citations:
4.8W