用大规模胸片数据提升少样本病理识别能力
MetaChest: Generalized few-shot learning of pathologies from chest X-rays
- 构建47万张胸片数据集,支持多标签少样本学习
- 多类别每轮训练+更多样本提升识别准确率
- 高效模型可媲美大模型,适合医疗场景部署
标注数据稀缺是深度学习在医学图像分析中的主要挑战。少样本学习旨在仅凭少量标注样例识别新类别,但现有方法多基于标准设置(任务中所有类别均为新类)。而胸片病理分类等实际应用常需在识别新病种的同时利用已知知识,更符合广义少样本学习场景。本文提出MetaChest,一个包含479,215张胸片的大型数据集,来自四个公开数据库。该数据集包含用于标准少样本分类的元训练集,以及生成多标签任务的算法。我们评估了标准迁移学习与ProtoNet扩展在多种少样本多标签分类任务上的表现。结果表明,增加每轮任务的类别数和每类训练样本数可显著提升性能;尽管未针对少样本优化,迁移学习方法始终优于ProtoNet变体。此外,更高分辨率图像提升精度但增加计算开销,而高效模型在资源消耗大幅降低的前提下达到与大模型相当的性能。
原文摘要 · Abstract (English)
The limited availability of annotated data presents a major challenge for applying deep learning methods to medical image analysis. Few-shot learning methods aim to recognize new classes from only a small number of labeled examples. These methods are typically studied under the standard few-shot learning setting, where all classes in a task are new. However, medical applications such as pathology classification from chest X-rays often require learning new classes while simultaneously leveraging knowledge of previously known ones, a scenario more closely aligned with generalized few-shot classification. Despite its practical relevance, few-shot learning has been scarcely studied in this context. In this work, we present MetaChest, a large-scale dataset of 479,215 chest X-rays collected from four public databases. MetaChest includes a meta-set partition specifically designed for standard few-shot classification, as well as an algorithm for generating multi-label episodes. We conduct extensive experiments evaluating both a standard transfer learning approach and an extension of ProtoNet across a wide range of few-shot multi-label classification tasks. Our results demonstrate that increasing the number of classes per episode and the number of training examples per class improves classification performance. Notably, the transfer learning approach consistently outperforms the ProtoNet extension, despite not being tailored for few-shot learning. We also show that higher-resolution images improve accuracy at the cost of additional computation, while efficient model architectures achieve comparable performance to larger models with significantly reduced resource requirements.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。