谷歌浏览器插件
订阅小程序
在清言上使用

Finetuning of GLIDE Stable Diffusion Model for AI-based Text-Conditional Image Synthesis of Dermoscopic Images

Veronika Shavlokhova,Andreas Vollmer, Christos C. Zouboulis, Michael Vollmer, Jakob Wollborn, Gernot Lang, Alexander Kuebler,Stefan Hartmann, Christian Stoll,Elisabeth Roider,Babak Saravi

FRONTIERS IN MEDICINE(2023)

引用 0|浏览0
暂无评分
摘要
BackgroundThe development of artificial intelligence (AI)-based algorithms and advances in medical domains rely on large datasets. A recent advancement in text-to-image generative AI is GLIDE (Guided Language to Image Diffusion for Generation and Editing). There are a number of representations available in the GLIDE model, but it has not been refined for medical applications.MethodsFor text-conditional image synthesis with classifier-free guidance, we have fine-tuned GLIDE using 10,015 dermoscopic images of seven diagnostic entities, including melanoma and melanocytic nevi. Photorealistic synthetic samples of each diagnostic entity were created by the algorithm. Following this, an experienced dermatologist reviewed 140 images (20 of each entity), with 10 samples originating from artificial intelligence and 10 from original images from the dataset. The dermatologist classified the provided images according to the seven diagnostic entities. Additionally, the dermatologist was asked to indicate whether or not a particular image was created by AI. Further, we trained a deep learning model to compare the diagnostic results of dermatologist versus machine for entity classification.ResultsThe results indicate that the generated images possess varying degrees of quality and realism, with melanocytic nevi and melanoma having higher similarity to real images than other classes. The integration of synthetic images improved the classification performance of the model, resulting in higher accuracy and precision. The AI assessment showed superior classification performance compared to dermatologist.ConclusionOverall, the results highlight the potential of synthetic images for training and improving AI models in dermatology to overcome data scarcity.
更多
查看译文
关键词
GLIDE,text-to-image,stable diffusion,dermoscopy,cancer,dermatology
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要