Search papers, labs, and topics across Lattice.
This paper introduces TRCGL-Net, a novel framework designed to enhance multi-label chest X-ray classification by addressing the challenges posed by long-tailed distributions in clinical data. The approach integrates a text-guided conditional diffusion model for generating realistic tail-class images, a channel reweighting mechanism for feature recalibration, and a graph convolution network for modeling label co-occurrence relationships. Experimental results on the PadChest dataset demonstrate significant improvements in tail-class mean average precision (mAP) and overall performance, highlighting TRCGL-Net's effectiveness in recognizing rare diseases amidst class imbalance.
TRCGL-Net achieves a remarkable tail-class mAP of 0.4904, setting a new benchmark for rare disease recognition in chest X-ray classification.
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.
