生成モデルは、さまざまな画像認識および合成タスクのために医用画像領域に適用されています。ただし、医学教育の支援などの重要なアプリケーションには、より制御可能で解釈可能な画像合成モデルがまだ不足しています。この作業では、効率的な自己注意と対照的な学習モジュールを活用し、最先端の生成的敵対的ネットワーク(GAN)に基づいて、高品質を生成できるAttributeGANと呼ばれる属性認識画像合成モデルを実現します。多属性入力に基づく組織病理学画像。既存の単一属性条件付き生成モデルと比較して、提案されたモデルは入力属性をより適切に反映し、属性値間のよりスムーズな補間を可能にします。尿路上皮癌の染色されたH&E画像を含む組織病理学データセットで実験を行い、最先端のモデルやモデルのさまざまなバリアントとの包括的な定量的および定性的な比較を通じて、提案されたモデルの有効性を示します。コードはhttps://github.com/karenyyy/MICCAI2021AttributeGANで入手できます。
Generative models have been applied in the medical imaging domain for various image recognition and synthesis tasks. However, a more controllable and interpretable image synthesis model is still lacking yet necessary for important applications such as assisting in medical training. In this work, we leverage the efficient self-attention and contrastive learning modules and build upon state-of-the-art generative adversarial networks (GANs) to achieve an attribute-aware image synthesis model, termed AttributeGAN, which can generate high-quality histopathology images based on multi-attribute inputs. In comparison to existing single-attribute conditional generative models, our proposed model better reflects input attributes and enables smoother interpolation among attribute values. We conduct experiments on a histopathology dataset containing stained H&E images of urothelial carcinoma and demonstrate the effectiveness of our proposed model via comprehensive quantitative and qualitative comparisons with state-of-the-art models as well as different variants of our model. Code is available at https://github.com/karenyyy/MICCAI2021AttributeGAN.