教師なし画像転送により、大量のペアトレーニングデータが豊富でない医療アプリケーションのモダリティ内およびモダリティ間転送が可能になります。入力からターゲットドメインへの構造を維持するマッピングを確実にするために、対になっていない医用画像転送の既存の方法は、一般にサイクルの一貫性に基づいており、逆マッピングの学習により追加の計算リソースと不安定性を引き起こします。このホワイトペーパーでは、トレーニングプロセス全体を通じて、ペアのデータが不要な単方向ドメインマッピングの新しい方法を紹介します。 GANアーキテクチャと、パッチの不変性に基づく新しいジェネレータ損失を採用することで、合理的な転送が保証されます。より正確には、ジェネレーターの出力はさまざまなスケールで評価および比較されます。これにより、高周波の詳細と暗黙的なデータ拡張への注目が高まります。この新しい用語は、パッチ残差の入力依存スケールマップをモデル化することにより、偶然性の不確実性を予測する機会も提供します。提案された方法は、3つの有名な医療データベースで包括的に評価されます。対になっていない画像転送のための4つの異なる最先端の方法と比較して、これらのデータセットの優れた精度は、不確実性を意識した医用画像変換のためのこのアプローチの大きな可能性を示唆しています。提案されたフレームワークの実装は、https://github.com/anger-man/unsupervised-image-transfer-and-uqでリリースされています。
Unsupervised image transfer enables intra- and inter-modality transfer for medical 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 medical image transfer are commonly based on cycle-consistency, causing additional computation resources and instability due to the learning of an inverse mapping. This paper presents a novel method for uni-directional domain mapping where no paired data is needed throughout the entire training process. A reasonable transfer is ensured by employing the GAN architecture and a novel generator loss based on patch invariance. To be more precise, generator outputs are evaluated and compared on different scales, which brings increased attention to high-frequency details as well as implicit data augmentation. This novel term also gives the opportunity to predict aleatoric uncertainty by modeling an input-dependent scale map for the patch residuals. The proposed method is comprehensively evaluated on three renowned medical databases. Superior accuracy on these datasets compared to four different state-of-the-art methods for unpaired image transfer suggests the great potential of this approach for uncertainty-aware medical image translation. Implementation of the proposed framework is released here: https://github.com/anger-man/unsupervised-image-transfer-and-uq.