Deo, Y., Lin, F., Dou, H. et al. (4 more authors) (Accepted: 2024) Few-shot learning in diffusion models for generating cerebral aneurysm geometries. In: Lecture Notes in Computer Science. 21st IEEE International Symposium on Biomedical Imaging (ISBI 2024), 27-30 May 2024, Athens, Greece. Springer Nature . (In Press)
Abstract
The study of brain vessel pathologies is critical for the advancement of neurovascular medicine, yet researchers often face significant hurdles due to the scarcity of imaging data for certain uncommon types of aneurysms. Generative deep learning models have been proposed to address the lack of high-quality labeled medical images - however, the shortage of data also presents a unique challenge in training generative models. To address this issue, our work explores the efficacy of training latent diffusion models (LDMs) with fewshot learning, enabling the generation of detailed vessel segmentations from as few as five images per class. By incorporating set-based vision transformers for class embeddings and leveraging signed distance functions (SDFs) as a novel form of conditioning, our method reduces the need for extensive datasets for training. Comparative studies with established generative models, including variational autoencoders (VAEs) and generative adversarial networks (GANs), highlight the robustness of our approach. Our model not only successfully generates high-quality segmentations of brain vessels with aneurysms but also significantly outperforms the standard generative models.
Metadata
Item Type: | Proceedings Paper | ||||
---|---|---|---|---|---|
Authors/Creators: |
|
||||
Copyright, Publisher and Additional Information: | © 2024 Springer Nature Switzerland AG. This is an author produced version of a conference paper originally presented at 21st IEEE International Symposium on Biomedical Imaging (ISBI 2024), Athens, 27-30 May 2024. |
||||
Keywords: | Diffusion Models, Image Synthesis, Brain Vessel Synthesis, Transformers | ||||
Dates: |
|
||||
Institution: | The University of Leeds | ||||
Academic Units: | The University of Leeds > Faculty of Engineering & Physical Sciences (Leeds) > School of Computing (Leeds) | ||||
Funding Information: |
|
||||
Depositing User: | Symplectic Publications | ||||
Date Deposited: | 21 Feb 2024 15:02 | ||||
Last Modified: | 19 Apr 2024 14:07 | ||||
Status: | In Press | ||||
Publisher: | Springer Nature |
Download
Filename: YISBI4.pdf