PROTOCOL: Partial Optimal Transport-enhanced Contrastive Learning for Imbalanced Multi-view Clustering
Xuqian Xue, Yiming Lei, Qi Cai, Hongming Shan, Junping Zhang
Abstract
While contrastive multi-view clustering has achieved remarkable success, it implicitly assumes balanced class distribution. However, realworld 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.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext c56a0a51-c57d-4f96-a931-09e73dacd0d8Cited by top-tier papers2
- GRPO-based Cluster Decision Agent for Unknown- Multi-view ClusteringXuqian Xue, Jun Zhang, Qi Cai, Zhizhong Huang et al.ICML 2026
- OPTION: Optimal Transport–Guided Flow Matching for Incomplete and Unaligned Multi-View ClusteringSiyuan Zhou, Zhibin GuICML 2026
Builds on19
- Self-labelling via simultaneous clustering and representation learningYuki Markus Asano, Christian Rupprecht, Andrea VedaldiICLR 2020 · 873 citations
- Parametric Contrastive LearningJiequan Cui, Zhisheng Zhong, Shu Liu, Bei Yu et al.ICCV 2021 · 375 citations
- Multi-level Feature Learning for Contrastive Multi-view ClusteringJie Xu, Huayi Tang, Yazhou Ren, Liang Peng et al.CVPR 2022 · 335 citations
- Self-supervised Label Augmentation via Input TransformationsHankook Lee, Sung Ju Hwang, Jinwoo ShinICML 2020 · 218 citations
- Deep Safe Incomplete Multi-view Clustering: Theorem and AlgorithmHuayi Tang, Yong LiuICML 2022 · 118 citations
Related papers
- Dual-Level Distribution Alignment for Deep Incomplete Multi-View ClusteringFujian Ren, Wenlan Chen, Lu Gao, Fei Guo et al.ACM MM 2025
- COMPLETER: Incomplete Multi-View Clustering via Contrastive PredictionYijie Lin, Yuanbiao Gou, Zitao Liu, Boyun Li et al.CVPR 2021
- P2OT: Progressive Partial Optimal Transport for Deep Imbalanced ClusteringChuyu Zhang, Hui Ren, Xuming HeICLR 2024 · 12 citations
- Learning to Re-weight Examples with Optimal Transport for Imbalanced ClassificationDandan Guo, Zhuo Li, Meixi Zheng, He Zhao et al.NeurIPS 2022 · 46 citations
- Dual-stage Contrastive Learning-enhanced Multi-view Variational ClusteringYanxi Liu, Yipin Hu, Fangxi Liu, Yanwei Yu et al.ICML 2026
