Controllable Generation of Diverse Dermatological Imagery for Fair and Efficient Malignancy Classification
Paper • 2607.12987 • Published
How to use hcarrion/hematoma 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/hematoma")This repository contains the Textual Inversion adaptation weights trained on the hematoma disease class as part of the cgDDI framework. It was introduced in the paper Controllable Generation of Diverse Dermatological Imagery for Fair and Efficient Malignancy Classification.
These weights are learned on top of the stabilityai/stable-diffusion-2-1-base backbone to capture disease-specific dermatological concepts, allowing the controllable generation of diverse and skin-tone-balanced synthetic training datasets.
To invoke this concept in your prompts, use the token:
<hematoma-class>
@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