mlsanityData to 14 Oct 2026

Addressing Class Imbalance with Probabilistic Graphical Models and Variational Inference

arXiv 2504.05758 ↗First posted 8 Apr 2025cs.LGcited by 9

Not on the map. Why?

Abstract (arXiv)

This study proposes a method for imbalanced data classification based on deep probabilistic graphical models (DPGMs) to solve the problem that traditional methods have insufficient learning ability for minority class samples. To address the classification bias caused by class imbalance, we introduce variational inference optimization probability modeling, which enables the model to adaptively adjust the representation ability of minority classes and combines the class-aware weight adjustment strategy to enhance the classifier's sensitivity to minority classes. In addition, we combine the adversarial learning mechanism to generate minority class samples in the latent space so that the model can better characterize the category boundary in the high-dimensional feature space. The experiment is evaluated on the Kaggle "Credit Card Fraud Detection" dataset and compared with a variety of advanced imbalanced classification methods (such as GAN-based sampling, BRF, XGBoost-Cost Sensitive, SAAD, HAN). The results show that the method in this study has achieved the best performance in AUC, Precision, Recall and F1-score indicators, effectively improving the recognition rate of minority classes and reducing the false alarm rate. This method can be widely used in imbalanced classification tasks such as financial fraud detection, medical diagnosis, and anomaly detection, providing a new solution for related research.

Key facts. Cited by 9 papers in the map (data through 14 Oct 2026). Builds on 5 of its 27 references that are in the map. First posted on arXiv 8 Apr 2025.

Builds on

5 of 27 references.

References · 5

  1. 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,166
  2. 2025Contrastive Learning for Cold Start Recommendation with Adaptive Feature Fusioncited by 11
  3. 2025A Deep Learning Framework Integrating CNN and BiLSTM for Financial Systemic Risk Analysis and Predictioncited by 9
  4. 2025A Hybrid Model for Few-Shot Text Classification Using Transfer and Meta-Learningcited by 8
  5. 2024Learning to Solve the Constrained Most Probable Explanation Task in Probabilistic Graphical Modelscited by 3

Built on by

Cited by year

Citing papers per year: 2025: 9.

9 papers.