arXiv:2511.07801cs.CV2025-11被引 1

通过稀疏标签耦合提升胸部X光多标签诊断准确率

Learning Sparse Label Couplings for Multilabel Chest X-Ray Diagnosis

  • 在分类器后加入轻量级标签图精修模块,学习标签间稀疏耦合关系
  • 在14种胸腔病变上实现92.64%的宏AUC,提升显著且计算开销极小
  • 无需额外标注,适合医疗影像实际部署,可复现性强

我们研究胸部X光的多标签分类任务,提出基于SE-ResNeXt101 (32×4d) 的简单高效流水线。主干网络在14种胸腔病变上微调,采用Sigmoid输出头,使用多标签迭代分层(MIS)进行稳健交叉验证,保持标签共现结构。为应对极端类别不平衡和不对称错误代价,采用对称损失、混合精度训练、带预热的余弦学习率衰减、梯度裁剪及权重指数移动平均(EMA)。提出一个轻量级标签图精修模块,置于分类器后:基于各标签概率,学习稀疏可训练的标签间耦合矩阵,通过单步消息传递修正logits,仅增加一个L1正则化参数头。推理时应用水平翻转测试增强,并对MIS各折预测取平均(紧凑深度集成)。评估采用宏AUC,按类别计算ROC-AUC并跳过单类标签以反映平衡性能。在自建数据集上,强基线模型达92.64%宏AUC;加入标签图精修模块后,各折验证宏AUC持续提升,计算开销几乎可忽略。该方法可复现、硬件友好,无需额外标注,为更强的多标签胸片分类提供实用路径。

原文摘要 · Abstract (English)

We study multilabel classification of chest X-rays and present a simple, strong pipeline built on SE-ResNeXt101 $(32 \times 4d)$. The backbone is finetuned for 14 thoracic findings with a sigmoid head, trained using Multilabel Iterative Stratification (MIS) for robust cross-validation splits that preserve label co-occurrence. To address extreme class imbalance and asymmetric error costs, we optimize with Asymmetric Loss, employ mixed-precision (AMP), cosine learning-rate decay with warm-up, gradient clipping, and an exponential moving average (EMA) of weights. We propose a lightweight Label-Graph Refinement module placed after the classifier: given per-label probabilities, it learns a sparse, trainable inter-label coupling matrix that refines logits via a single message-passing step while adding only an L1-regularized parameter head. At inference, we apply horizontal flip test-time augmentation (TTA) and average predictions across MIS folds (a compact deep ensemble). Evaluation uses macro AUC averaging classwise ROC-AUC and skipping single-class labels in a fold to reflect balanced performance across conditions. On our dataset, a strong SE-ResNeXt101 baseline attains competitive macro AUC (e.g., 92.64% in our runs). Adding the Label-Graph Refinement consistently improves validation macro AUC across folds with negligible compute. The resulting method is reproducible, hardware-friendly, and requires no extra annotations, offering a practical route to stronger multilabel CXR classifiers.

多标签分类医学影像胸部X光标签耦合

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。