深層生成モデルは、データの生成と操作のための影響力のある手段として登場しました。データ属性を選択的に変更することで、これらのモデルの制御性を向上させることが最近の焦点となっています。変分オートエンコーダ (VAE) は、隠れた属性の捕捉に有望であることが示されていますが、多くの場合、不鮮明な再構成が生成されます。医用画像処理では、さまざまな画像処理ドメインを通じてこれらの属性を制御することは困難です。最近、ソフト イントロスペクティブ VAE は、敵対的損失を VAE トレーニングに組み込むことで、VAE と敵対的生成ネットワーク (GAN) の両方の利点を活用し、優れた画像合成機能を実証しました。この研究では、Soft-Intro VAE フレームワークに属性の正規化損失を組み込むことにより、Attributed Soft Introspective VAE (Attri-SIVAE) を提案します。私たちは、さまざまなスキャナーベンダーや収集センターなど、さまざまな分野からの心臓 MRI データに対して提案された方法を実験的に評価します。提案された方法は、最先端の属性付き正規化 VAE と比較して、再構築と正則化の点で同様のパフォーマンスを達成しますが、さらに、比較された方法とは異なり、異なるデータセットでテストされた場合でも同じ正則化レベルを維持することにも成功します。
Deep generative models have emerged as influential instruments for data generation and manipulation. Enhancing the controllability of these models by selectively modifying data attributes has been a recent focus. Variational Autoencoders (VAEs) have shown promise in capturing hidden attributes but often produce blurry reconstructions. Controlling these attributes through different imaging domains is difficult in medical imaging. Recently, Soft Introspective VAE leverage the benefits of both VAEs and Generative Adversarial Networks (GANs), which have demonstrated impressive image synthesis capabilities, by incorporating an adversarial loss into VAE training. In this work, we propose the Attributed Soft Introspective VAE (Attri-SIVAE) by incorporating an attribute regularized loss, into the Soft-Intro VAE framework. We evaluate experimentally the proposed method on cardiac MRI data from different domains, such as various scanner vendors and acquisition centers. The proposed method achieves similar performance in terms of reconstruction and regularization compared to the state-of-the-art Attributed regularized VAE but additionally also succeeds in keeping the same regularization level when tested on a different dataset, unlike the compared method.