arXiv:2604.09359cs.LG2026-04

提出新方法解决多标签学习中的噪声标签问题,提升模型鲁棒性。

Bringing Clustering to MLL: Weakly-Supervised Clustering for Partial Multi-Label Learning

  • 通过分解隶属矩阵,融合聚类与多标签特性
  • 在24个数据集上超越6种前沿方法,性能全面领先
  • 适合处理带噪声标签的多标签学习任务

多标签学习(MLL)中的标签噪声对模型训练构成重大挑战,尤其在部分多标签学习(PML)中,候选标签包含相关与无关标签。聚类虽可自然用于识别噪声,但传统方法因隶属度和为1的约束,无法直接适用于多标签场景。本文提出一种弱监督聚类方法WSC-PML,通过将聚类隶属矩阵分解为两个部分:Π保持聚类结构,F保留多标签特性,实现聚类与多标签学习的无缝结合。该方法采用三阶段流程:从噪声标签中学习初始原型,构建基于置信度的弱监督信号,迭代优化聚类与标签预测。在24个数据集上的实验表明,该方法在所有评估指标上均优于6种现有先进方法。

原文摘要 · Abstract (English)

Label noise in multi-label learning (MLL) poses significant challenges for model training, particularly in partial multi-label learning (PML) where candidate labels contain both relevant and irrelevant labels. While clustering offers a natural approach to exploit data structure for noise identification, traditional clustering methods cannot be directly applied to multi-label scenarios due to a fundamental incompatibility: clustering produces membership values that sum to one per instance, whereas multi-label assignments require binary values that can sum to any number. We propose a novel weakly-supervised clustering approach for PML (WSC-PML) that bridges clustering and multi-label learning through membership matrix decomposition. Our key innovation decomposes the clustering membership matrix $\mathbf{A}$ into two components: $\mathbf{A} = \mathbfΠ \odot \mathbf{F}$, where $\mathbfΠ$ maintains clustering constraints while $\mathbf{F}$ preserves multi-label characteristics. This decomposition enables seamless integration of unsupervised clustering with multi-label supervision for effective label noise handling. WSC-PML employs a three-stage process: initial prototype learning from noisy labels, adaptive confidence-based weak supervision construction, and joint optimization via iterative clustering refinement. Extensive experiments on 24 datasets demonstrate that our approach outperforms six state-of-the-art methods across all evaluation metrics.

多标签学习聚类弱监督噪声标签

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