用统计距离引导的无监督域适应模型,自动评估心脏核磁图像质量。
Statistical Distance-Guided Unsupervised Domain Adaptation for Automated Multi-Class Cardiovascular Magnetic Resonance Image Quality Assessment
- 基于注意力机制,通过近似Wasserstein距离实现跨域特征对齐
- 在4个数据集上识别4类常见伪影,准确率优于以往方法
- 无需人工标注,适合大规模临床图像质量控制
本研究提出一种基于注意力的统计距离引导无监督域适应模型,用于多类别心血管磁共振(CMR)图像质量评估。模型包含特征提取器、标签预测器和统计距离估计器。以带标注的数据集为源域,分布不同的无标注数据集为目标域。统计距离估计器在小批量内近似源与目标数据特征向量间的Wasserstein距离。标签预测器对源数据进行标签预测,训练时采用交叉熵、中心损失及距离估计值组成的组合损失函数。在4个数据集(含成像与k空间数据)上评估了该模型对呼吸/心搏运动、吉布斯振铃和混叠四种常见伪影的识别能力。大量实验表明,该模型在图像与k空间分析中均能有效覆盖源域与目标域之间的分布偏移。可解释性分析与消融实验验证了各模块的有效性。相比先前研究,该模型在性能和检测伪影种类数量上均有提升。由于在无繁琐标注下具备良好域偏移适应能力,适用于CMR后处理质量控制或大规模队列研究中的图像与k空间质量评估。
原文摘要 · Abstract (English)
This study proposes an attention-based statistical distance-guided unsupervised domain adaptation model for multi-class cardiovascular magnetic resonance (CMR) image quality assessment. The proposed model consists of a feature extractor, a label predictor and a statistical distance estimator. An annotated dataset as the source set and an unlabeled dataset as the target set with different statistical distributions are considered inputs. The statistical distance estimator approximates the Wasserstein distance between the extracted feature vectors from the source and target data in a mini-batch. The label predictor predicts data labels of source data and uses a combinational loss function for training, which includes cross entropy and centre loss functions plus the estimated value of the distance estimator. Four datasets, including imaging and k-space data, were used to evaluate the proposed model in identifying four common CMR imaging artefacts: respiratory and cardiac motions, Gibbs ringing and Aliasing. The results of the extensive experiments showed that the proposed model, both in image and k-space analysis, has an acceptable performance in covering the domain shift between the source and target sets. The model explainability evaluations and the ablation studies confirmed the proper functioning and effectiveness of all the model's modules. The proposed model outperformed the previous studies regarding performance and the number of examined artefacts. The proposed model can be used for CMR post-imaging quality control or in large-scale cohort studies for image and k-space quality assessment due to the appropriate performance in domain shift coverage without a tedious data-labelling process.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。