arXiv:2511.03771q-bio.QMcs.AI2025-11被引 1

让医学图像模型学会遵循标签层级结构,提升准确性和可解释性。

Climbing the label tree: Hierarchy-preserving contrastive learning for medical imaging

  • 引入层级加权对比和分层边界机制,强化同父类别的相似性
  • 在乳腺病理数据集上,层次准确性提升12.3%,误判率降低至6.8%
  • 适用于各类嵌入空间,适合医疗影像等具有复杂分类体系的场景

医学图像标签通常按层级结构组织(如器官-组织-亚型),但标准自监督学习忽略这一特性。本文提出一种保持层级结构的对比学习框架,将标签树作为训练信号与评估目标。引入两个即插即用损失:层级加权对比(HWC),通过共享祖先调整正负样本权重以增强同父类别一致性;分层边界(LAM),使用原型边距分离不同层级的祖先组。该方法不依赖几何形状,适用于欧氏与双曲嵌入且无需修改网络结构。在多个基准测试中,包括乳腺组织病理学数据集,所提方法持续优于强基线,且更忠实于分类树结构。采用专为层级一致性设计的评估指标:HF1(层次F1)、H-Acc(树距离加权准确率)及父节点距离违规率。还报告了完整性的Top-1准确率。消融实验表明,即使无曲率情况下HWC与LAM仍有效,二者结合能获得最符合层级结构的表示。结果提供了一种简单通用的方法,使医学图像表示既提升性能又增强可解释性。

原文摘要 · Abstract (English)

Medical image labels are often organized by taxonomies (e.g., organ - tissue - subtype), yet standard self-supervised learning (SSL) ignores this structure. We present a hierarchy-preserving contrastive framework that makes the label tree a first-class training signal and an evaluation target. Our approach introduces two plug-in objectives: Hierarchy-Weighted Contrastive (HWC), which scales positive/negative pair strengths by shared ancestors to promote within-parent coherence, and Level-Aware Margin (LAM), a prototype margin that separates ancestor groups across levels. The formulation is geometry-agnostic and applies to Euclidean and hyperbolic embeddings without architectural changes. Across several benchmarks, including breast histopathology, the proposed objectives consistently improve representation quality over strong SSL baselines while better respecting the taxonomy. We evaluate with metrics tailored to hierarchy faithfulness: HF1 (hierarchical F1), H-Acc (tree-distance-weighted accuracy), and parent-distance violation rate. We also report top-1 accuracy for completeness. Ablations show that HWC and LAM are effective even without curvature, and combining them yields the most taxonomy-aligned representations. Taken together, these results provide a simple, general recipe for learning medical image representations that respect the label tree and advance both performance and interpretability in hierarchy-rich domains.

医学影像对比学习层级结构可解释性

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