Stable Diffusion Model-Based Scintigraphy Image Synthesis: Data Augmentation Toward Enhanced Multiclass Thyroid Diagnosis
The objective of this study is to assess the efficacy of advanced augmentation techniques, such as stable diffusion, in improving the performance of deep learning models in the classification of scintigraphic thyroid images. In this retrospective study, 2983 anterior view scintigraphic images were collected and subsequently categorized into four thyroid conditions. Both stable diffusion and conventional augmentation techniques were utilized. The generated images, alongside real images, were used to train a ResNet101V2 architecture under six different training strategies. The strategies were assessed against external datasets to evaluate model performance in terms of accuracy, precision, recall, and F1-score. The use of synthetic data in training led to consistently superior performances compared to training with only real data. Specifically, the models trained with synthetic data augmentation demonstrated higher precision and recall. The incorporation of synthetic images generated via stable diffusion significantly enhanced the diagnostic capabilities of AI models in thyroid scintigraphy interpretation. This approach not only improves the classification accuracy but also provides a viable solution to the challenge of data scarcity in medical imaging.
Authors
- Yazdan Salimi (ORCID: https://orcid.org/0000-0002-1233-9576)
- Isaac Shiri (ORCID: https://orcid.org/0000-0002-5735-0736)
- Habib Zaidi (ORCID: https://orcid.org/0000-0001-7559-5297)
- Ali Rasouli (ORCID: https://orcid.org/0000-0002-0313-0236)
- Abdollah Saberi (ORCID: https://orcid.org/0000-0001-7327-2558)
- Arman Rahmin
- Seyed Rasoul Zakavi (ORCID: https://orcid.org/0000-0002-8276-2134)
- Emran Askari (ORCID: https://orcid.org/0000-0002-2973-2251)
- Mostafa Haghir Chehreghani (ORCID: https://orcid.org/0000-0003-3436-0541)
- Ghasem Hajianfar (ORCID: https://orcid.org/0000-0001-5359-2407)
- Mohsen Arabi
- Atena Aghaee (ORCID: https://orcid.org/0000-0002-1351-3142)
- Kourosh Fattahi
- Soroush Bagheri (ORCID: https://orcid.org/0009-0007-6227-2166)
- Maziar Sabouri (ORCID: https://orcid.org/0000-0001-7525-8952)
- Ehsan Bayat
- Amirhossein Sanaat (ORCID: https://orcid.org/0000-0001-8437-2060)
- Mehdi Mogharrabi
- Azin Asadzadeh
Institutions
- University of British Columbia (CA)
- Amirkabir University of Technology (IR)
- Mashhad University of Medical Sciences (IR)
- Golestan University (IR)
- Kashan University of Medical Sciences (IR)
- Kermanshah University of Medical Sciences (IR)
- University Hospital of Geneva (CH)
- Jahrom University of Medical Sciences (IR)
- Golestan University of Medical Sciences (IR)
- Shaheed Rajaei Cardiovascular Medical and Research Center (IR)
- Alborz University of Medical Sciences (IR)
Publication Details
- Published
- 2024-09-08
- DOI
- https://doi.org/10.1109/euvip61797.2024.10772863
- Citations
- 1
- Primary Topic
- Medical Imaging and Analysis
- Type
- article
- Field-Weighted Citation Impact
- 0.18
Funders
- Schweizerischer Nationalfonds zur Förderung der Wissenschaftlichen Forschung