arXiv:2507.02979cs.CVcs.LG2025-07

IMET通过迭代识别错分类样本,提升医学图像分类的准确性和鲁棒性。

Iterative Misclassification Error Training (IMET): An Optimized Neural Network Training Technique for Image Classification

  • 基于课程学习与核心集思想,动态筛选难例样本进行训练。
  • 在多个医学图像数据集上显著提升模型准确率与抗噪能力。
  • 特别适合小样本、噪声多的医疗图像场景,如罕见病诊断。

深度学习模型在医学图像诊断中表现优异,但医学数据常含噪声、误标或难以泛化的异常样本,且高质量数据集样本量小,易导致过拟合,模型记忆噪声而非学习通用规律,这在医疗诊断中可能带来生命风险。现有数据高效训练策略如核心集选择可选取代表性子集以逼近全数据性能并降低计算开销;课程学习则通过逐步增加训练难度加速收敛。然而,如何设计跨领域、跨数据集、跨模型通用且低计算成本的难度排序机制仍具挑战。本文提出迭代错分类误差训练(IMET),融合课程学习与核心集思想,旨在识别错分类样本,引导模型重点关注边缘案例与罕见结果,优化训练过程。在基准医学图像分类数据集上对主流ResNet架构的评估表明,IMET能有效提升模型在医疗图像分析中的鲁棒性与准确性。

原文摘要 · Abstract (English)

Deep learning models have proven to be effective on medical datasets for accurate diagnostic predictions from images. However, medical datasets often contain noisy, mislabeled, or poorly generalizable images, particularly for edge cases and anomalous outcomes. Additionally, high quality datasets are often small in sample size that can result in overfitting, where models memorize noise rather than learn generalizable patterns. This in particular, could pose serious risks in medical diagnostics where the risk associated with mis-classification can impact human life. Several data-efficient training strategies have emerged to address these constraints. In particular, coreset selection identifies compact subsets of the most representative samples, enabling training that approximates full-dataset performance while reducing computational overhead. On the other hand, curriculum learning relies on gradually increasing training difficulty and accelerating convergence. However, developing a generalizable difficulty ranking mechanism that works across diverse domains, datasets, and models while reducing the computational tasks and remains challenging. In this paper, we introduce Iterative Misclassification Error Training (IMET), a novel framework inspired by curriculum learning and coreset selection. The IMET approach is aimed to identify misclassified samples in order to streamline the training process, while prioritizing the model's attention to edge case senarious and rare outcomes. The paper evaluates IMET's performance on benchmark medical image classification datasets against state-of-the-art ResNet architectures. The results demonstrating IMET's potential for enhancing model robustness and accuracy in medical image analysis are also presented in the paper.

医学图像训练优化错误修正小样本

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