Latent Diffusion Models with Image-Derived Annotations for Enhanced AI-Assisted Cancer Diagnosis in Histopathology

Diagnostics (Basel). 2024 Jul 5;14(13):1442. doi: 10.3390/diagnostics14131442.

Abstract

Artificial Intelligence (AI)-based image analysis has immense potential to support diagnostic histopathology, including cancer diagnostics. However, developing supervised AI methods requires large-scale annotated datasets. A potentially powerful solution is to augment training data with synthetic data. Latent diffusion models, which can generate high-quality, diverse synthetic images, are promising. However, the most common implementations rely on detailed textual descriptions, which are not generally available in this domain. This work proposes a method that constructs structured textual prompts from automatically extracted image features. We experiment with the PCam dataset, composed of tissue patches only loosely annotated as healthy or cancerous. We show that including image-derived features in the prompt, as opposed to only healthy and cancerous labels, improves the Fréchet Inception Distance (FID) by 88.6. We also show that pathologists find it challenging to detect synthetic images, with a median sensitivity/specificity of 0.55/0.55. Finally, we show that synthetic data effectively train AI models.

Keywords: AI in health care; cancer detection; diffusion models; digital health; generative AI; histopathology; medical imaging; synthetic data.

Grants and funding

P.O., G.J.-P., J.M.T., J.H., G.D.-B., S.S., M.R., U.B., S.S., K.S., J.V., B.L. and S.M. are employed by Bayer AG.