arXiv reaDer
HQG-Net: 高品質のガイダンスによる非対医用画像強化
HQG-Net: Unpaired Medical Image Enhancement with High-Quality Guidance
Unpaired Medical Image Enhancement (UMIE) は、トレーニング用のペア画像に依存せずに、低品質 (LQ) の医療画像を高品質 (HQ) の医療画像に変換することを目的としています。既存のアプローチのほとんどは Pix2Pix/CycleGAN に基づいており、ある程度は効果的ですが、強化プロセスをガイドするために HQ 情報を明示的に使用できていないため、望ましくないアーティファクトや構造の歪みが生じる可能性があります。この論文では、HQ キューを変分方式で LQ エンハンスメントプロセスに直接エンコードすることで既存の方法の上記の制限を回避し、LQ ドメインと HQ ドメイン間の結合分布の下で UMIE タスクをモデル化する新しい UMIE アプローチを提案します。具体的には、HQ 画像から特徴を抽出し、HQ キューをエンコードすると予想される特徴を拡張ネットワークに明示的に挿入して、変分正規化モジュールによる LQ 拡張をガイドします。生成された HQ 画像が HQ ドメインに確実に収まるように、識別子を使用して拡張ネットワークを敵対的にトレーニングします。さらに、ウェーブレット ベースのピクセル レベルおよびマルチ エンコーダ ベースの特徴レベルの制約を使用してエンハンスメント プロセスをガイドする、コンテンツを意識した損失を提案します。さらに、画像強調を実行する主な動機は、強調された画像を下流のタスクにより適したものにすることであるため、UMIE タスクと下流のタスクを協調的に最適化し、視覚的に魅力的で有利な HQ 画像の生成を支援する 2 レベル学習スキームを提案します。下流のタスク。新たに収集された 2 つのデータセットを含む 3 つの医療データセットでの実験により、提案された方法が拡張品質と下流タスクのパフォーマンスの両方の点で既存の技術よりも優れていることが確認されました。コードと新しく収集したデータセットは、コミュニティの研究のために公開する予定です。
Unpaired Medical Image Enhancement (UMIE) aims to transform a low-quality (LQ) medical image into a high-quality (HQ) one without relying on paired images for training. While most existing approaches are based on Pix2Pix/CycleGAN and are effective to some extent, they fail to explicitly use HQ information to guide the enhancement process, which can lead to undesired artifacts and structural distortions. In this paper, we propose a novel UMIE approach that avoids the above limitation of existing methods by directly encoding HQ cues into the LQ enhancement process in a variational fashion and thus model the UMIE task under the joint distribution between the LQ and HQ domains. Specifically, we extract features from an HQ image and explicitly insert the features, which are expected to encode HQ cues, into the enhancement network to guide the LQ enhancement with the variational normalization module. We train the enhancement network adversarially with a discriminator to ensure the generated HQ image falls into the HQ domain. We further propose a content-aware loss to guide the enhancement process with wavelet-based pixel-level and multi-encoder-based feature-level constraints. Additionally, as a key motivation for performing image enhancement is to make the enhanced images serve better for downstream tasks, we propose a bi-level learning scheme to optimize the UMIE task and downstream tasks cooperatively, helping generate HQ images both visually appealing and favorable for downstream tasks. Experiments on three medical datasets, including two newly collected datasets, verify that the proposed method outperforms existing techniques in terms of both enhancement quality and downstream task performance. We will make the code and the newly collected datasets publicly available for community study.
updated: Sat Jul 15 2023 15:26:25 GMT+0000 (UTC)
published: Sat Jul 15 2023 15:26:25 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト