arXiv:2508.06429cs.CVcs.AI2025-08被引 2

用少量标注数据实现医学图像精准分类,靠图像翻译提升学习效果。

SPARSE Data, Rich Results: Few-Shot Semi-Supervised Learning via Class-Conditioned Image Translation

  • 通过条件图像翻译生成真实样本,替代噪声生成,增强未标注数据利用。
  • 在每类仅5个样本时准确率达87.6%,优于现有方法,且所有设置下均领先。
  • 适合标注成本高的医疗场景,尤其适用于极少量标注数据的实战应用。

深度学习在医学影像中表现卓越,但受限于标注数据不足。本文提出一种基于GAN的半监督学习框架,专为低标注数据环境设计,在每类5至50个样本的条件下评估。方法包含三个网络:用于类别条件图像翻译的生成器、判别真实性的判别器和专用分类器,采用三阶段训练流程。模型交替进行有监督训练与无监督学习,通过图像到图像的翻译而非噪声生成来利用大量未标注数据。采用基于集成的伪标签策略,结合判别器与分类器的置信度预测,并引入指数移动平均保持时间一致性,实现对未标注数据的可靠标签估计。在11个MedMNIST数据集上的全面评估显示,本方法显著优于六种前沿的GAN-based半监督方法,在极端5样本设置下表现尤为突出。无论5、10、20还是50样本/类,均保持优势。该框架为标注成本高昂的医学影像应用提供了实用解决方案,即使在极小标注数据下也能实现稳健分类。代码已公开于https://github.com/GuidoManni/SPARSE。

原文摘要 · Abstract (English)

Deep learning has revolutionized medical imaging, but its effectiveness is severely limited by insufficient labeled training data. This paper introduces a novel GAN-based semi-supervised learning framework specifically designed for low labeled-data regimes, evaluated across settings with 5 to 50 labeled samples per class. Our approach integrates three specialized neural networks -- a generator for class-conditioned image translation, a discriminator for authenticity assessment and classification, and a dedicated classifier -- within a three-phase training framework. The method alternates between supervised training on limited labeled data and unsupervised learning that leverages abundant unlabeled images through image-to-image translation rather than generation from noise. We employ ensemble-based pseudo-labeling that combines confidence-weighted predictions from the discriminator and classifier with temporal consistency through exponential moving averaging, enabling reliable label estimation for unlabeled data. Comprehensive evaluation across eleven MedMNIST datasets demonstrates that our approach achieves statistically significant improvements over six state-of-the-art GAN-based semi-supervised methods, with particularly strong performance in the extreme 5-shot setting where the scarcity of labeled data is most challenging. The framework maintains its superiority across all evaluated settings (5, 10, 20, and 50 shots per class). Our approach offers a practical solution for medical imaging applications where annotation costs are prohibitive, enabling robust classification performance even with minimal labeled data. Code is available at https://github.com/GuidoManni/SPARSE.

半监督学习医学影像少样本GAN

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