COPD Multi-Task Diagnosis on Chest X-Ray Using CNN-Based Slot Attention

초록

This study proposes a unified deep-learning framework for the concurrent classification of Chronic Obstructive Pulmonary Disease (COPD) severity and regression of the FEV1/FVC ratio from chest X-ray (CXR) images. We integrated a ConvNeXt-Large backbone with a Slot Attention mechanism to effectively disentangle and refine disease-relevant features for multi-task learning. Evaluation on a clinical dataset demonstrated that the proposed model with a 5-slot configuration achieved superior performance compared to standard CNN and Vision Transformer baselines. On the independent test set, the model attained an Accuracy of 0.9107, Sensitivity of 0.8603, and Specificity of 0.9324 for three-class severity stratification. Simultaneously, it achieved a Mean Absolute Error (MAE) of 8.2649 and a Mean Squared Error (MSE) of 151.4704, and an R2 of 0.7591 for FEV1/FVC ratio estimation. Qualitative analysis using saliency maps also suggested that the slot-based approach contributes to attention patterns that are more constrained to clinically relevant pulmonary structures. These results suggest that our slot-attention-based multi-task model offers a robust solution for automated COPD assessment from standard radiographs.

키워드