arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2309.03386cs.LGcs.AI
收藏

用于慢性病预测的基于社区的分层正未标记(PU)模型融合

Community-Based Hierarchical Positive-Unlabeled (PU) Model Fusion for Chronic Disease Prediction

Worcester Polytechnic Institute(伍斯特理工学院);Alibaba Group(阿里巴巴集团);Indiana University, Bloomington(印第安纳大学伯明顿分校)

Yang Wu, Xurui Li, Xuhong Zhang, Yangyang Kang, Changlong Sun, Xiaozhong Liu

AI总结:

针对慢性病预测中PU学习忽略人群差异的问题,提出PUtree算法,通过分层构建社区PU模型、数据增强、对抗风险估计与模型融合,显著提升了二分类性能。

摘要:

正未标记(PU)学习是由二分类问题带来的一项挑战,其中存在大量未标记数据以及少量正数据实例,可用于解决慢性病筛查问题。最先进的PU学习方法已促成各种风险估计器的开发,但它们忽略了不同人群之间的差异。为了解决这个问题,我们提出了一种新颖的正未标记学习树(PUtree)算法。PUtree旨在慢性病预测任务中考虑诸如不同年龄或收入阶层等社区因素。我们提出了一种新颖的二分类决策方法,该方法分层构建基于社区的PU模型,然后聚合它们的输出。我们的方法能够解释树上的每个PU模型,以实现优化的非叶PU节点分裂。此外,掩码恢复数据增强策略使得模型能够在各个社区中进行充分训练。另外,所提方法包含一个对抗性PU风险估计器以捕获分层PU关系,以及一个整合来自每条树路径数据的模型融合网络,从而产生稳健的二分类结果。我们在两个基准测试和一个新的糖尿病预测数据集上证明了PUtree及其变体的优越性能。

英文摘要:

Positive-Unlabeled (PU) Learning is a challenge presented by binary classification problems where there is an abundance of unlabeled data along with a small number of positive data instances, which can be used to address chronic disease screening problem. State-of-the-art PU learning methods have resulted in the development of various risk estimators, yet they neglect the differences among distinct populations. To address this issue, we present a novel Positive-Unlabeled Learning Tree (PUtree) algorithm. PUtree is designed to take into account communities such as different age or income brackets, in tasks of chronic disease prediction. We propose a novel approach for binary decision-making, which hierarchically builds community-based PU models and then aggregates their deliverables. Our method can explicate each PU model on the tree for the optimized non-leaf PU node splitting. Furthermore, a mask-recovery data augmentation strategy enables sufficient training of the model in individual communities. Additionally, the proposed approach includes an adversarial PU risk estimator to capture hierarchical PU-relationships, and a model fusion network that integrates data from each tree path, resulting in robust binary classification results. We demonstrate the superior performance of PUtree as well as its variants on two benchmarks and a new diabetes-prediction dataset.

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

↑