论文标题

使用补丁不变网络的无监督联合图像传输和不确定性定量

Unsupervised Joint Image Transfer and Uncertainty Quantification Using Patch Invariant Networks

论文作者

Angermann, Christoph, Haltmeier, Markus, Siyal, Ahsan Raza

论文摘要

无监督的图像传输可以在大量配对训练数据不丰富的应用中内部和模式间图像翻译。为了确保从输入到目标域的结构映射,现有的未配对图像传输的方法通常基于周期矛盾,由于学习了反向映射而导致了其他计算资源和不稳定。本文提出了一种新的单向域映射方法,该方法不依赖任何配对的训练数据。通过使用GAN架构和基于贴片不变性的新发电机损失来实现适当的转移。更具体地说,在不同的尺度上评估和比较发电机的输出,也导致人们对高频细节以及隐式数据扩展的关注增加。这种新颖的贴片损失还提供了通过对贴片残差的输入依赖性比例映射进行建模,可以准确预测不确定性。该方法在三个公认的医疗数据库上进行了全面评估。与四种最先进的方法相比,我们观察到这些数据集的精度明显更高,这表明所提出的未造成图像转移的方法的潜力很大,并且考虑了不确定性。所提出的框架的实现在此处发布:\ url {https://github.com/anger-man/unsupervise-image-image-ransfer-ander-and-uq}。

Unsupervised image transfer enables intra- and inter-modality image translation in applications where a large amount of paired training data is not abundant. To ensure a structure-preserving mapping from the input to the target domain, existing methods for unpaired image transfer are commonly based on cycle-consistency, causing additional computational resources and instability due to the learning of an inverse mapping. This paper presents a novel method for uni-directional domain mapping that does not rely on any paired training data. A proper transfer is achieved by using a GAN architecture and a novel generator loss based on patch invariance. To be more specific, the generator outputs are evaluated and compared at different scales, also leading to an increased focus on high-frequency details as well as an implicit data augmentation. This novel patch loss also offers the possibility to accurately predict aleatoric uncertainty by modeling an input-dependent scale map for the patch residuals. The proposed method is comprehensively evaluated on three well-established medical databases. As compared to four state-of-the-art methods, we observe significantly higher accuracy on these datasets, indicating great potential of the proposed method for unpaired image transfer with uncertainty taken into account. Implementation of the proposed framework is released here: \url{https://github.com/anger-man/unsupervised-image-transfer-and-uq}.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源