用迁移学习检测儿童肺炎,准确率接近完美。
Pediatric Pneumonia Detection from Chest X-Rays:A Comparative Study of Transfer Learning and Custom CNNs
- 用预训练模型微调,比从零训练效果更好。
- 最佳模型准确率达99.43%,误判仅3例。
- 可视化验证了预测区域与临床肺部一致。
儿童肺炎是五岁以下儿童死亡的主要原因,每年超过70万例死亡。胸部X光的准确诊断受限于放射科医生数量和判断差异。本研究对比了从零训练的自定义CNN与迁移学习(ResNet50、DenseNet121、EfficientNet-B0)在儿科肺炎检测中的表现,评估了冻结主干和微调两种方式。使用包含5,216张儿科胸片的数据集,按80/10/10划分训练、验证和测试集。七个模型通过准确率、F1分数和AUC进行评估,并采用Grad-CAM提供可解释性。结果表明,微调后的ResNet50表现最佳:准确率99.43%,F1分数99.61%,AUC 99.93%,仅3例误判。微调平均比冻结主干提升5.5个百分点。Grad-CAM确认预测聚焦于临床相关的肺部区域。结论:微调迁移学习显著优于从零训练的CNN,展现出近乎完美的准确性,具有在资源匮乏地区作为筛查工具的潜力。未来工作应将结果扩展至多中心及成人数据集验证。
原文摘要 · Abstract (English)
Pneumonia is a leading cause of mortality in children under five, with over 700,000 deaths annually. Accurate diagnosis from chest X-rays is limited by radiologist availability and variability. Objective: This study compares custom CNNs trained from scratch with transfer learning (ResNet50, DenseNet121, EfficientNet-B0) for pediatric pneumonia detection, evaluating frozen-backbone and fine-tuning regimes. Methods: A dataset of 5,216 pediatric chest X-rays was split 80/10/10 for training, validation, and testing. Seven models were trained and assessed using accuracy, F1-score, and AUC. Grad-CAM visualizations provided explainability. Results: Fine-tuned ResNet50 achieved the best performance: 99.43\% accuracy, 99.61\% F1-score, and 99.93\% AUC, with only 3 misclassifications. Fine-tuning outperformed frozen-backbone models by 5.5 percentage points on average. Grad-CAM confirmed clinically relevant lung regions guided predictions. Conclusions: Transfer learning with fine-tuning substantially outperforms CNNs trained from scratch for pediatric pneumonia detection, showing near-perfect accuracy. This system has strong potential as a screening tool in resource-limited settings. Future work should validate these findings on multi-center and adult datasets. Keywords: Pneumonia detection, deep learning, transfer learning, CNN, chest X-ray, pediatric diagnosis, ResNet, DenseNet, EfficientNet, Grad-CAM.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。