arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.20168eess.IVcs.CV

用于多模态脑图像翻译的无监督肿瘤感知蒸馏

Unsupervised Tumor-Aware Distillation for Multi-Modal Brain Image Translation

School of Computer Science and Engineering, South China University of Technology(华南理工大学计算机科学与工程学院);Golisano College of Computing and Information Sciences, Rochester Institute of Technology(罗切斯特理工大学戈利萨诺计算与信息科学学院)

Chuan Huang, Jia Wei, Rui Li

AI总结:

本文提出无监督肿瘤感知蒸馏网络UTAD-Net,通过教师-学生架构利用未配对图像和肿瘤掩码学习端到端映射并蒸馏知识,解决了脑图像翻译中肿瘤变形问题,提升了图像生成质量及下游分割效果。

摘要:

来自MRI扫描的多模态脑图像在临床诊断中被广泛使用,以提供来自不同模态的互补信息。然而,由于时间、成本和伪影等各种因素,在实践中获取完全配对的多模态图像具有挑战性,导致出现模态缺失的脑图像。为了解决这个问题,无监督多模态脑图像翻译已被广泛研究。现有方法在翻译过程中存在脑肿瘤变形的问题,因为它们在翻译整幅图像时未能关注肿瘤区域。在本文中,我们提出了一种称为UTAD-Net的无监督肿瘤感知蒸馏教师-学生网络,它能够精确地感知和翻译肿瘤区域。具体而言,我们的模型由两部分组成:教师网络和学生网络。教师网络首先使用未配对图像和相应的肿瘤掩码学习从源模态到目标模态的端到端映射。然后,翻译知识被蒸馏到学生网络中,使其能够在没有掩码的情况下生成更真实的肿瘤区域和完整图像。实验表明,与最先进的方法相比,我们的模型在图像质量的定量和定性评估上均取得了具有竞争力的性能。此外,我们展示了生成的图像在下游分割任务上的有效性。我们的代码可在https://github.com/scut-HC/UTAD-Net获取。

英文摘要:

Multi-modal brain images from MRI scans are widely used in clinical diagnosis to provide complementary information from different modalities. However, obtaining fully paired multi-modal images in practice is challenging due to various factors, such as time, cost, and artifacts, resulting in modality-missing brain images. To address this problem, unsupervised multi-modal brain image translation has been extensively studied. Existing methods suffer from the problem of brain tumor deformation during translation, as they fail to focus on the tumor areas when translating the whole images. In this paper, we propose an unsupervised tumor-aware distillation teacher-student network called UTAD-Net, which is capable of perceiving and translating tumor areas precisely. Specifically, our model consists of two parts: a teacher network and a student network. The teacher network learns an end-to-end mapping from source to target modality using unpaired images and corresponding tumor masks first. Then, the translation knowledge is distilled into the student network, enabling it to generate more realistic tumor areas and whole images without masks. Experiments show that our model achieves competitive performance on both quantitative and qualitative evaluations of image quality compared with state-of-the-art methods. Furthermore, we demonstrate the effectiveness of the generated images on downstream segmentation tasks. Our code is available at https://github.com/scut-HC/UTAD-Net.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑