A Bayesian network learning method for sparse and unbalanced data with GNN-based multilabel classification application

Ling Chen, Xiangming Jiang, Yuhong Wang

Research output: Journal article publicationJournal articleAcademic researchpeer-review

Abstract

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.

Original languageEnglish
Article number111393
JournalApplied Soft Computing
Volume154
DOIs
Publication statusPublished - Mar 2024

Keywords

  • Bayesian network
  • Graph neural network
  • Imbalance data
  • Sparse data
  • Structure learning

ASJC Scopus subject areas

  • Software

Fingerprint

Dive into the research topics of 'A Bayesian network learning method for sparse and unbalanced data with GNN-based multilabel classification application'. Together they form a unique fingerprint.

Cite this