arXiv:2502.03238cs.CVcs.AI2025-02中稿 · Computers in Biolo…被引 7

解决医学影像诊断中的长尾分布问题,提升罕见病识别准确率。

Long-tailed Medical Diagnosis with Relation-aware Representation Learning and Iterative Classifier Calibration

  • 通过关系感知表征学习增强编码器对语义特征的捕捉能力。
  • 迭代分类器校准使少数类性能显著提升,整体准确率超现有方法。
  • 适合医疗影像长尾数据场景,尤其关注罕见病诊断的研究者。

近年来,计算机辅助诊断展现出良好性能,有效减轻临床医生负担。然而,不同疾病间样本分布不均导致算法偏向多数类别,稀有类别表现差。现有方法将此问题视为长尾问题,通过解耦特征表示与分类进行处理。但由于尾部类别样本少、分布不平衡,仍易出现表征偏差和分类器校准不足。为此,我们提出一种长尾医学诊断(LMD)框架,用于在长尾数据上实现均衡的医学图像分类。初始阶段,设计关系感知表征学习(RRL)方案,通过多种数据增强促使编码器捕捉内在语义特征。后续阶段,提出迭代分类器校准(ICC)方案,生成大量平衡的虚拟特征,并以期望-最大化方式微调编码器。该方法补偿少数类别,促进无偏分类器优化,同时保留多数类诊断知识。在三个公开长尾医学数据集上的实验表明,所提LMD框架显著优于当前最优方法。源代码可在https://github.com/peterlipan/LMD获取。

原文摘要 · Abstract (English)

Recently computer-aided diagnosis has demonstrated promising performance, effectively alleviating the workload of clinicians. However, the inherent sample imbalance among different diseases leads algorithms biased to the majority categories, leading to poor performance for rare categories. Existing works formulated this challenge as a long-tailed problem and attempted to tackle it by decoupling the feature representation and classification. Yet, due to the imbalanced distribution and limited samples from tail classes, these works are prone to biased representation learning and insufficient classifier calibration. To tackle these problems, we propose a new Long-tailed Medical Diagnosis (LMD) framework for balanced medical image classification on long-tailed datasets. In the initial stage, we develop a Relation-aware Representation Learning (RRL) scheme to boost the representation ability by encouraging the encoder to capture intrinsic semantic features through different data augmentations. In the subsequent stage, we propose an Iterative Classifier Calibration (ICC) scheme to calibrate the classifier iteratively. This is achieved by generating a large number of balanced virtual features and fine-tuning the encoder using an Expectation-Maximization manner. The proposed ICC compensates for minority categories to facilitate unbiased classifier optimization while maintaining the diagnostic knowledge in majority classes. Comprehensive experiments on three public long-tailed medical datasets demonstrate that our LMD framework significantly surpasses state-of-the-art approaches. The source code can be accessed at https://github.com/peterlipan/LMD.

医学影像长尾学习分类校准

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