arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2502.03386cs.LG

基于概率模型的不平衡数据分类结构化推理框架

A Structured Reasoning Framework for Unbalanced Data Classification Using Probabilistic Models

Shanghai Jiao Tong University(上海交通大学);Yale University(耶鲁大学);University of California San Diego(加利福尼亚大学圣迭戈分校);Tulane University(杜兰大学);University of Rochester(罗切斯特大学)

Junliang Du, Shiyu Dou, Bohuan Yang, Jiacheng Hu, Tai An

AI总结:

本文提出一种基于马尔可夫网络的不平衡数据分类框架,通过联合概率建模、边际概率估计和加权损失优化提升少数类识别能力,并在信用卡欺诈检测中显著优于传统模型。

摘要:

本文研究了一种用于不平衡数据的马尔可夫网络模型,旨在解决传统机器学习模型在类别分布不均环境中存在的分类偏差和少数类识别能力不足的问题。通过构建联合概率分布和条件依赖关系,该模型能够实现样本类别的全局建模与推理优化。研究引入了边际概率估计和加权损失优化策略,并结合正则化约束与结构化推理方法,有效提升了模型的泛化能力和鲁棒性。在实验阶段,选取了一个真实的信用卡欺诈检测数据集,并与逻辑回归、支持向量机、随机森林和XGBoost等模型进行了比较。实验结果表明,马尔可夫网络在加权准确率、F1分数和AUC-ROC等指标上表现良好,显著优于传统分类模型,展示了其在不平衡数据场景中强大的决策能力和适用性。未来研究可聚焦于大规模不平衡数据环境下的高效模型训练、结构优化以及深度学习集成,并推动其在金融风控、医疗诊断和智能监控等实际应用中的广泛应用。

英文摘要:

This paper studies a Markov network model for unbalanced data, aiming to solve the problems of classification bias and insufficient minority class recognition ability of traditional machine learning models in environments with uneven class distribution. By constructing joint probability distribution and conditional dependency, the model can achieve global modeling and reasoning optimization of sample categories. The study introduced marginal probability estimation and weighted loss optimization strategies, combined with regularization constraints and structured reasoning methods, effectively improving the generalization ability and robustness of the model. In the experimental stage, a real credit card fraud detection dataset was selected and compared with models such as logistic regression, support vector machine, random forest and XGBoost. The experimental results show that the Markov network performs well in indicators such as weighted accuracy, F1 score, and AUC-ROC, significantly outperforming traditional classification models, demonstrating its strong decision-making ability and applicability in unbalanced data scenarios. Future research can focus on efficient model training, structural optimization, and deep learning integration in large-scale unbalanced data environments and promote its wide application in practical applications such as financial risk control, medical diagnosis, and intelligent monitoring.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑