针对胸部X光罕见病识别难问题,提出生成增强与共现建模新框架。
TRCGL-Net: A Long-Tailed Multi-Label Chest X-Ray Classification Framework with Generative Data Augmentation and Label Co-Occurrence Modeling

- 用文本引导的扩散模型生成罕见病图像,提升数据多样性和语义一致性。
- 在PadChest数据集上,尾部类别mAP达0.4904,整体mAP为0.4408。
- 适合医学影像分析、长尾分布处理等场景,尤其关注罕见病检测。
胸部X光多标签分类是智能医学影像诊断的核心任务。然而,真实临床数据常呈现极端长尾分布,导致尾部罕见疾病性能下降。这一问题不仅源于数据稀缺,还受两大内在因素影响:1)复杂解剖背景下尾类病灶表征被抑制;2)头部类别主导标签共现关系建模。为此,本文提出TRCGL-Net。首先,采用可学习的文本引导条件扩散模型,在疾病语义约束下生成高质量尾类胸片样本,提升罕见病模式的数据多样性与真实性,缓解类别不平衡并保持病理语义一致。其次,引入通道重加权机制,通过强调疾病相关特征通道实现特征再校准,提升长尾分布下的特征判别力;进一步采用类别感知注意力机制生成类别特异性注意力图,使模型能定位病灶区域并聚焦细粒度病变。最后,基于标签共现构建图卷积网络,建立类别间信息传播机制。在PadChest数据集上的实验表明,该方法达到尾部类别mAP 0.4904,整体mAP 0.4408,mAUC 0.8989,优于现有最先进方法。TRCGL-Net有效提升了长尾分布下罕见病的识别性能,缓解了胸部X光多标签分类中的极端类别不平衡问题。
原文摘要 · Abstract (English)
Chest X-ray multi-label classification is a core task in intelligent medical imaging diagnosis. However, real clinical data often exhibit extreme long-tailed distributions, leading to degraded performance on rare diseases in tail classes. This issue is not only driven by data scarcity but also by two intrinsic factors:1) attenuation of tail-class lesion representations under complex anatomical backgrounds, and 2) dominance of head classes in modeling label co-occurrence relationships. To address these challenges, we propose TRCGL-Net. First, a learnable text-guided conditional diffusion model is employed to generate high-quality tail-class chest X-ray image samples under disease semantic constraints, improving data diversity and realism of rare disease patterns while alleviating class imbalance and preserving pathology-consistent semantics.Second, a channel reweighting mechanism is introduced to perform feature recalibration by emphasizing disease-relevant feature channels, thereby improving feature discriminability under long-tailed distributions.A class-aware attention mechanism is further applied to generate class-specific attention maps, enabling the model to localize disease-relevant regions and focus on fine-grained lesion areas.Finally, a graph convolution network based on label co occurrence is introduced to establish an information propagation mechanism among categories. Experiments on the PadChest dataset show that the proposed method achieves a tail-class mAP of 0.4904, an overall mAP of 0.4408, and an mAUC of 0.8989, outperforming state-of-the-art methods. TRCGL-Net effectively improves recognition performance for rare diseases under long-tailed distributions and mitigates the impact of extreme class imbalance in chest X-ray multi-label classification.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。