多中心CT影像中更稳健的新冠检测,通过联合预测来源提升泛化能力。
Robust Multi-Source Covid-19 Detection in CT Images
- 同时学习新冠诊断与扫描来源,缓解数据分布不均带来的偏差。
- 在308张验证图像上实现F1 0.91、AUC 0.96,显著优于单一任务方法。
- 适合医疗多中心数据融合场景,尤其关注模型鲁棒性与公平性。
从胸部CT扫描中进行新冠检测的深度学习模型在同机构数据上表现良好,但在多中心数据(设备、成像协议、人群不同)下性能下降明显。现有方法仅以新冠分类为训练目标,未考虑扫描来源,导致模型偏向训练数据多的中心。为此,我们提出一种多任务学习方法,让模型同时预测新冠诊断和扫描来源,共享EfficientNet-B7主干网络,促使特征提取器学习跨四个参与中心的通用表示。由于训练数据在各中心分布不均,我们在来源分类头采用对数调整交叉熵损失,防止低样本中心被忽略。预处理遵循SSFL框架与KDS方法,每例选取8个代表性切片。在包含308张扫描的验证集上,该方法达到F1分数0.9098和AUC-ROC 0.9647。代码已公开于https://github.com/Purdue-M2/-multisource-covid-ct。
原文摘要 · Abstract (English)
Deep learning models for COVID-19 detection from chest CT scans generally perform well when the training and test data originate from the same institution, but they often struggle when scans are drawn from multiple centres with differing scanners, imaging protocols, and patient populations. One key reason is that existing methods treat COVID-19 classification as the sole training objective, without accounting for the data source of each scan. As a result, the learned representations tend to be biased toward centres that contribute more training data. To address this, we propose a multi-task learning approach in which the model is trained to predict both the COVID-19 diagnosis and the originating data centre. The two tasks share an EfficientNet-B7 backbone, which encourages the feature extractor to learn representations that hold across all four participating centres. Since the training data is not evenly distributed across sources, we apply a logit-adjusted cross-entropy loss [1] to the source classification head to prevent underrepresented centres from being overlooked. Our pre-processing follows the SSFL framework with KDS [2], selecting eight representative slices per scan. Our method achieves an F1 score of 0.9098 and an AUC-ROC of 0.9647 on a validation set of 308 scans. The code is publicly available at https://github.com/Purdue-M2/-multisource-covid-ct.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。