针对等级分类中不同相邻类别的误判后果差异,提出新型对比学习方法
CLOC: Contrastive Learning for Ordinal Classification with Multi-Margin N-pair Loss
- 基于多边界n对损失函数,灵活优化关键类间的决策边界
- 在五个真实图像数据集上超越现有方法,尤其在临床敏感边界表现更优
- 可解释性强,适合医疗等需谨慎决策的场景
在等级分类任务中,相邻类别误判虽常见,但后果各异。例如,良性肿瘤分类错误影响较小,而从癌前到癌症的临界误判可能严重影响治疗方案。然而,现有方法未区分这些边界的权重,均视为同等重要。为此,本文提出CLOC,一种基于多边界n对损失(MMNP)的对比学习新方法,通过优化多个边际来学习有序表示。该方法能灵活调整关键相邻类别间的决策边界,实现类间平滑过渡,降低对训练数据偏见的过拟合风险。我们在五个真实世界图像数据集(Adience、Historical Colour Image Dating、Knee Osteoarthritis、Indian Diabetic Retinopathy Image、Breast Carcinoma Subtyping)及一个模拟临床决策偏倚的合成数据集上进行了实验。结果表明,CLOC优于现有方法,并展现出可解释性与可控性,所学有序表示更符合临床与实际需求。
原文摘要 · Abstract (English)
In ordinal classification, misclassifying neighboring ranks is common, yet the consequences of these errors are not the same. For example, misclassifying benign tumor categories is less consequential, compared to an error at the pre-cancerous to cancerous threshold, which could profoundly influence treatment choices. Despite this, existing ordinal classification methods do not account for the varying importance of these margins, treating all neighboring classes as equally significant. To address this limitation, we propose CLOC, a new margin-based contrastive learning method for ordinal classification that learns an ordered representation based on the optimization of multiple margins with a novel multi-margin n-pair loss (MMNP). CLOC enables flexible decision boundaries across key adjacent categories, facilitating smooth transitions between classes and reducing the risk of overfitting to biases present in the training data. We provide empirical discussion regarding the properties of MMNP and show experimental results on five real-world image datasets (Adience, Historical Colour Image Dating, Knee Osteoarthritis, Indian Diabetic Retinopathy Image, and Breast Carcinoma Subtyping) and one synthetic dataset simulating clinical decision bias. Our results demonstrate that CLOC outperforms existing ordinal classification methods and show the interpretability and controllability of CLOC in learning meaningful, ordered representations that align with clinical and practical needs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。