arXiv:2504.05758cs.LG2025-04被引 14

用概率图模型和变分推断提升少数类识别能力

Addressing Class Imbalance with Probabilistic Graphical Models and Variational Inference

  • 基于深度概率图模型与变分推断自适应增强少数类表征
  • 在信用卡欺诈数据集上各项指标均优于现有方法
  • 适合金融反欺诈、医疗诊断等少数类识别场景

本研究提出一种基于深度概率图模型(DPGM)的不平衡数据分类方法,以解决传统方法对少数类样本学习能力不足的问题。为缓解类别不平衡导致的分类偏差,引入变分推断优化概率建模,使模型能自适应调整少数类表征能力,并结合类别感知权重调节策略,提升分类器对少数类的敏感性。同时,融合对抗学习机制,在隐空间生成少数类样本,使模型更好地刻画高维特征空间中的类别边界。实验在Kaggle“信用卡欺诈检测”数据集上进行,与多种先进不平衡分类方法(如GAN采样、BRF、XGBoost-Cost Sensitive、SAAD、HAN)对比,结果表明该方法在AUC、精确率、召回率和F1分数上均表现最佳,显著提升了少数类识别率并降低了误报率。该方法可广泛应用于金融欺诈检测、医疗诊断和异常检测等不平衡分类任务,为相关研究提供新思路。

原文摘要 · Abstract (English)

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.

不平衡分类概率图模型变分推断欺诈检测

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。