arXiv:2508.13452cs.LGcs.CV2025-08中稿 · CIKM 2025被引 2

解决层次化多标签分类中的结构一致性和损失平衡问题

Hierarchy-Consistent Learning and Adaptive Loss Balancing for Hierarchical Multi-Label Classification

  • 融合原型对比学习与自适应任务加权,保持标签语义一致性
  • 在三个数据集上准确率更高,层次违规率降低至12.3%以下
  • 适合需要严格层次关系的医疗、知识图谱等场景

层次化多标签分类(HMC)在保持结构一致性和多任务学习(MTL)中的损失权重平衡方面面临重大挑战。为此,我们提出一种基于MTL的分类器HCAL,集成原型对比学习与自适应任务加权机制。其核心优势在于语义一致性:显式建模标签并实现子类到父类的特征聚合;另一重要优势是自适应损失加权机制,通过监测各任务收敛速度动态分配优化资源,有效缓解传统MTL中“一强多弱”的优化偏差。为增强鲁棒性,引入原型扰动机制,在原型中注入可控噪声以扩展决策边界。此外,我们提出了量化指标层次违规率(HVR)以评估层次一致性与泛化能力。在三个数据集上的大量实验表明,所提分类器在分类准确率和层次违规率方面均优于基线模型。

原文摘要 · Abstract (English)

Hierarchical Multi-Label Classification (HMC) faces critical challenges in maintaining structural consistency and balancing loss weighting in Multi-Task Learning (MTL). In order to address these issues, we propose a classifier called HCAL based on MTL integrated with prototype contrastive learning and adaptive task-weighting mechanisms. The most significant advantage of our classifier is semantic consistency including both prototype with explicitly modeling label and feature aggregation from child classes to parent classes. The other important advantage is an adaptive loss-weighting mechanism that dynamically allocates optimization resources by monitoring task-specific convergence rates. It effectively resolves the "one-strong-many-weak" optimization bias inherent in traditional MTL approaches. To further enhance robustness, a prototype perturbation mechanism is formulated by injecting controlled noise into prototype to expand decision boundaries. Additionally, we formalize a quantitative metric called Hierarchical Violation Rate (HVR) as to evaluate hierarchical consistency and generalization. Extensive experiments across three datasets demonstrate both the higher classification accuracy and reduced hierarchical violation rate of the proposed classifier over baseline models.

多标签分类层次结构自适应损失原型学习

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