arXiv:2505.18741cs.CVcs.AI2025-05被引 1

提出混合采样方法,提升多样质量图像的训练效果

MoMBS: Mixed-order minibatch sampling enhances model training from diverse-quality images

  • 基于损失与不确定度双重指标筛选样本
  • 区分低质量样本为标注差或过拟合,针对性优化
  • 适合医学图像等高质量差异大的场景

自然图像在噪声标签分类中存在标签多样性(清晰与噪声),在长尾分类中存在类别分布多样性(丰富与稀疏)。类似地,通用病灶检测中的医学图像在清晰度和标注正确性上也存在显著差异。如何有效利用不同质量的训练图像成为深度模型学习的关键问题。传统方法如自适应课程学习(SCL)和在线难例挖掘(OHEM)通过重加权高损失样本缓解此问题。然而,这些方法仍面临两大挑战:(i) 仅依赖损失衡量样本难度,不够精确;(ii) SCL存在利用不足,OHEM则过度使用难样本。本文重新审视小批量采样(MBS)这一广泛使用但未充分探索用于处理多样质量样本的技术。我们发现小批量内样本会相互影响,因此提出新型混合阶小批量采样(MoMBS)方法。MoMBS结合损失与不确定性,超越单一损失依赖,能更精细地区分高损失样本——是标注差且代表性不足,还是已充分表示且过拟合。该方法优先将代表性不足的样本作为梯度主要来源,并通过混合阶采样设计避免其受劣质或过拟合样本干扰。

原文摘要 · Abstract (English)

Natural images exhibit label diversity (clean vs. noisy) in noisy-labeled image classification and prevalence diversity (abundant vs. sparse) in long-tailed image classification. Similarly, medical images in universal lesion detection (ULD) exhibit substantial variations in image quality, encompassing attributes such as clarity and label correctness. How to effectively leverage training images with diverse qualities becomes a problem in learning deep models. Conventional training mechanisms, such as self-paced curriculum learning (SCL) and online hard example mining (OHEM), relieve this problem by reweighting images with high loss values. Despite their success, these methods still confront two challenges: (i) the loss-based measure of sample hardness is imprecise, preventing optimum handling of different cases, and (ii) there exists under-utilization in SCL or over-utilization OHEM with the identified hard samples. To address these issues, this paper revisits the minibatch sampling (MBS), a technique widely used in deep network training but largely unexplored concerning the handling of diverse-quality training samples. We discover that the samples within a minibatch influence each other during training; thus, we propose a novel Mixed-order Minibatch Sampling (MoMBS) method to optimize the use of training samples with diverse qualities. MoMBS introduces a measure that takes both loss and uncertainty into account to surpass a sole reliance on loss and allows for a more refined categorization of high-loss samples by distinguishing them as either poorly labeled and under represented or well represented and overfitted. We prioritize under represented samples as the main gradient contributors in a minibatch and keep them from the negative influences of poorly labeled or overfitted samples with a mixed-order minibatch sampling design.

图像质量采样策略医学影像模型训练

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