解决多视图聚类中类别不平衡问题,提升少数类样本表现
PROTOCOL: Partial Optimal Transport-enhanced Contrastive Learning for Imbalanced Multi-view Clustering
- 用部分最优传输建模类别不平衡,动态感知各类别分布
- 通过加权对比学习增强少数类样本特征表示,提升聚类精度
- 首次系统研究不平衡多视图聚类,适合处理真实数据偏差场景
尽管对比多视图聚类已取得显著进展,但其隐含假设为类别分布均衡。而现实多视图数据普遍存在类别不平衡现象,导致现有方法因无法感知和建模此类不平衡而性能下降。为此,我们首次系统研究不平衡多视图聚类,聚焦两大核心问题:一是感知类别不平衡分布,二是缓解少数类样本的表征退化。提出PROTOCOL框架,首先将多视图特征映射至共识空间,将不平衡聚类重构为部分最优传输(POT)问题,引入渐进质量约束与加权KL散度以建模类别分布。其次,在特征与类别层面设计基于POT的重平衡对比学习,结合对数调整与类别敏感学习,强化少数类样本表示。大量实验表明,PROTOCOL在不平衡多视图数据上显著提升聚类性能,填补了该领域的关键研究空白。
原文摘要 · Abstract (English)
While contrastive multi-view clustering has achieved remarkable success, it implicitly assumes balanced class distribution. However, real-world multi-view data primarily exhibits class imbalance distribution. Consequently, existing methods suffer performance degradation due to their inability to perceive and model such imbalance. To address this challenge, we present the first systematic study of imbalanced multi-view clustering, focusing on two fundamental problems: i. perceiving class imbalance distribution, and ii. mitigating representation degradation of minority samples. We propose PROTOCOL, a novel PaRtial Optimal TranspOrt-enhanced COntrastive Learning framework for imbalanced multi-view clustering. First, for class imbalance perception, we map multi-view features into a consensus space and reformulate the imbalanced clustering as a partial optimal transport (POT) problem, augmented with progressive mass constraints and weighted KL divergence for class distributions. Second, we develop a POT-enhanced class-rebalanced contrastive learning at both feature and class levels, incorporating logit adjustment and class-sensitive learning to enhance minority sample representations. Extensive experiments demonstrate that PROTOCOL significantly improves clustering performance on imbalanced multi-view data, filling a critical research gap in this field.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。