Generating Realistic X-ray Scattering Images Using Stable Diffusion and Human-in-the-loop Annotations
Fuente:
arXiv
Saved in:
| Main Authors: | , , , |
|---|---|
| Format: | Preprint |
| Published: |
2024
|
| Subjects: | |
| Online Access: | |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866913477203853312 |
|---|---|
| author | Zhao, Zhuowen Chong, Xiaoya Chavez, Tanny Hexemer, Alexander |
| author_facet | Zhao, Zhuowen Chong, Xiaoya Chavez, Tanny Hexemer, Alexander |
| contents | We fine-tuned a foundational stable diffusion model using X-ray scattering images and their corresponding descriptions to generate new scientific images from given prompts. However, some of the generated images exhibit significant unrealistic artifacts, commonly known as "hallucinations". To address this issue, we trained various computer vision models on a dataset composed of 60% human-approved generated images and 40% experimental images to detect unrealistic images. The classified images were then reviewed and corrected by human experts, and subsequently used to further refine the classifiers in next rounds of training and inference. Our evaluations demonstrate the feasibility of generating high-fidelity, domain-specific images using a fine-tuned diffusion model. We anticipate that generative AI will play a crucial role in enhancing data augmentation and driving the development of digital twins in scientific research facilities. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2408_12720 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Generating Realistic X-ray Scattering Images Using Stable Diffusion and Human-in-the-loop Annotations Zhao, Zhuowen Chong, Xiaoya Chavez, Tanny Hexemer, Alexander Image and Video Processing Artificial Intelligence Machine Learning We fine-tuned a foundational stable diffusion model using X-ray scattering images and their corresponding descriptions to generate new scientific images from given prompts. However, some of the generated images exhibit significant unrealistic artifacts, commonly known as "hallucinations". To address this issue, we trained various computer vision models on a dataset composed of 60% human-approved generated images and 40% experimental images to detect unrealistic images. The classified images were then reviewed and corrected by human experts, and subsequently used to further refine the classifiers in next rounds of training and inference. Our evaluations demonstrate the feasibility of generating high-fidelity, domain-specific images using a fine-tuned diffusion model. We anticipate that generative AI will play a crucial role in enhancing data augmentation and driving the development of digital twins in scientific research facilities. |
| title | Generating Realistic X-ray Scattering Images Using Stable Diffusion and Human-in-the-loop Annotations |
| topic | Image and Video Processing Artificial Intelligence Machine Learning |
| url | https://arxiv.org/abs/2408.12720 |