arXiv:2511.03891cs.CVcs.AI2025-11

用同类图像合成新图片,提升小样本疾病诊断准确率

Improving Diagnostic Performance on Small and Imbalanced Datasets Using Class-Based Input Image Composition

  • 将同类别图像拼合成复合图,增强类内差异和信息密度
  • 在2064张眼底OCT图上实现99.6%准确率与0.995的F1分数
  • 适合处理数据少且类别不均衡的医学影像诊断场景

小样本、类别不平衡的数据集以及低质量输入图像会导致深度学习模型误判率升高。本文提出类别级图像合成方法,通过融合同一类别的多张图像生成复合输入图像(CoImg),提升类内差异性和每张训练样本的信息密度,增强模型区分细微病变模式的能力。在包含2064张高分辨率人眼视网膜OCT扫描的OCTDL数据集(Kulyabin等,2024)上进行评估,该数据集涵盖七种疾病且存在显著类别不平衡。我们构建了类平衡版本Co-OCTDL,每张图像以3×1布局的复合形式呈现。采用VGG16模型进行对比实验,保持相同架构与超参数确保公平性。结果表明,新表示方法显著提升诊断性能:准确率达99.6%,F1分数为0.995,AUC达0.9996,远超原始数据集基线;假阳性率大幅降低,证明该方法可有效提升弱数据下的预测质量。

原文摘要 · Abstract (English)

Small, imbalanced datasets and poor input image quality can lead to high false predictions rates with deep learning models. This paper introduces Class-Based Image Composition, an approach that allows us to reformulate training inputs through a fusion of multiple images of the same class into combined visual composites, named Composite Input Images (CoImg). That enhances the intra-class variance and improves the valuable information density per training sample and increases the ability of the model to distinguish between subtle disease patterns. Our method was evaluated on the Optical Coherence Tomography Dataset for Image-Based Deep Learning Methods (OCTDL) (Kulyabin et al., 2024), which contains 2,064 high-resolution optical coherence tomography (OCT) scans of the human retina, representing seven distinct diseases with a significant class imbalance. We constructed a perfectly class-balanced version of this dataset, named Co-OCTDL, where each scan is resented as a 3x1 layout composite image. To assess the effectiveness of this new representation, we conducted a comparative analysis between the original dataset and its variant using a VGG16 model. A fair comparison was ensured by utilizing the identical model architecture and hyperparameters for all experiments. The proposed approach markedly improved diagnostic results.The enhanced Dataset achieved near-perfect accuracy (99.6%) with F1-score (0.995) and AUC (0.9996), compared to a baseline model trained on raw dataset. The false prediction rate was also significantly lower, this demonstrates that the method can producehigh-quality predictions even for weak datasets affected by class imbalance or small sample size.

医学影像数据增强分类优化小样本学习

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