BRAIN with Alzheimer's MRI generation (LoRA text2image fine-tuning)
These are LoRA adaption weights for stabilityai/stable-diffusion-xl-base-1.0.
Original dataset: https://www.kaggle.com/datasets/tourist55/alzheimers-dataset-4-class-of-images.
With this model, you can generate an MRI of the brain of 4 classes:
- NonDemented
- VeryMildDemented
- MildDemented
- ModerateDemeneted
For generation use the following corresponding prompts (or any other modification)
- MRI of NonDemented brain
- MRI of VeryMildDemented brain
- MRI of MildDemented brain
- MRI of ModerateDemeneted brain
![img_5](/BogdanTurbal/synthetic_mri_generation_sd_xl/resolve/main/image_5.png)
LoRA for the text encoder was enabled: False.
Special VAE used for training: madebyollin/sdxl-vae-fp16-fix.
Intended uses & limitations
How to use
Limitations and bias
[TODO: provide examples of latent issues and potential remediations]
Training details
[TODO: describe the data used to train the model]