Controllable Generation of Diverse Dermatological Imagery for Fair and Efficient Malignancy Classification
Paper • 2607.12987 • Published
How to use hcarrion/epidermal_nevus with Diffusers:
pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1-base", dtype=torch.bfloat16, device_map="cuda")
pipe.load_textual_inversion("hcarrion/epidermal_nevus")These are textual inversion adaptation weights for stabilityai/stable-diffusion-2-1-base trained to generate images of the epidermal nevus dermatological condition.
This model is part of the work presented in the paper Controllable Generation of Diverse Dermatological Imagery for Fair and Efficient Malignancy Classification (MICCAI 2026).
cgDDI (Controllable Generation of Diverse Dermatological Imagery) is a hybrid framework that:
For more details on the dataset, training pipelines, and evaluation, please refer to the official resources:
@inproceedings{carrion2026cgddi,
title = {Controllable Generation of Diverse Dermatological Imagery for Fair and Efficient Malignancy Classification},
author = {Carri{\'o}n, H{\'e}ctor and Norouzi, Narges},
booktitle = {Medical Image Computing and Computer-Assisted Intervention (MICCAI)},
year = {2026},
publisher = {Springer},
series = {Lecture Notes in Computer Science}
}
Base model
stabilityai/stable-diffusion-2-1-base