改进检测模型,提升胸部X光肺炎识别准确率。
CGF-DETR: Cross-Gated Fusion DETR for Enhanced Pneumonia Detection in Chest X-rays
- 引入跨门控融合机制,增强多尺度特征提取
- 达到82.2% [email protected],比基线高3.7个百分点
- 适合需要快速精准诊断的医疗影像场景
肺炎仍是全球发病率和死亡率的主要原因,亟需高效精准的自动化检测系统。尽管基于Transformer的检测器如RT-DETR在目标检测中表现良好,但在肺部X光片肺炎检测中的应用仍不充分。本文提出CGF-DETR,一种专为肺炎检测优化的实时检测变压器。在主干网络中引入XFABlock,通过融合卷积注意力与CSP结构提升多尺度特征提取能力;为实现高效特征聚合,提出SPGA模块,以动态门控机制和单头自注意力替代标准多头注意力;在颈部设计GCFC3模块,通过多路径卷积融合增强特征表示,同时借助结构重参数化保持实时性能。在RSNA肺炎检测数据集上的实验表明,CGF-DETR达到82.2% [email protected],较基线RT-DETR-l提升3.7%,推理速度保持在48.1 FPS。消融实验验证各模块均有效贡献,完整模型在mAP@[0.5:0.95]上达50.4%。
原文摘要 · Abstract (English)
Pneumonia remains a leading cause of morbidity and mortality worldwide, necessitating accurate and efficient automated detection systems. While recent transformer-based detectors like RT-DETR have shown promise in object detection tasks, their application to medical imaging, particularly pneumonia detection in chest X-rays, remains underexplored. This paper presents CGF-DETR, an enhanced real-time detection transformer specifically designed for pneumonia detection. We introduce XFABlock in the backbone to improve multi-scale feature extraction through convolutional attention mechanisms integrated with CSP architecture. To achieve efficient feature aggregation, we propose SPGA module that replaces standard multi-head attention with dynamic gating mechanisms and single-head self-attention. Additionally, GCFC3 is designed for the neck to enhance feature representation through multi-path convolution fusion while maintaining real-time performance via structural re-parameterization. Extensive experiments on the RSNA Pneumonia Detection dataset demonstrate that CGF-DETR achieves 82.2% [email protected], outperforming the baseline RT-DETR-l by 3.7% while maintaining comparable inference speed at 48.1 FPS. Our ablation studies confirm that each proposed module contributes meaningfully to the overall performance improvement, with the complete model achieving 50.4% mAP@[0.5:0.95]
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。