Research map / Image generation / Fast diffusion sampling
Doctor Approved: Generating Medically Accurate Skin Disease Images through AI-Expert Feedback
Read the paperCheck your idea against 108,967 papers
AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.
Proposes MAGIC, a semi-automated framework that uses GPT-4o as an evaluator of dermatologist-written checklists to align a fine-tuned Stable Diffusion model (via RFT or DPO) so it generates medically accurate skin disease images for classifier data augmentation.
Problem
Diffusion models often generate medically inaccurate skin disease images, and existing feedback-based alignment (RLHF, DPO) needs robust reward models or costly expert annotation, limiting their use for augmenting scarce medical data.
Approach
Expert-written attribute-level checklists (location, lesion type, shape/size, color, texture) are scored by an MLLM (GPT-4o) on synthetic image pairs; the binary outcomes serve as preference labels for DPO (or reward-weighted fine-tuning) of an image-to-image Stable Diffusion pipeline, so the MLLM evaluates only simple closed-form visual criteria.
Key results
- MAGIC-DPO raised ResNet18 accuracy on the 20-condition Fitzpatrick17k task from 29.31% to 38.33% (+9.02%) and DINOv2 from 49.89% to 55.01% (+5.12%).
- In the 10% few-shot setting (310 real images), MAGIC-DPO reached 37.39% accuracy (+10.94%), and MAGIC-A with unlabeled data reached 40.34% (+13.89%).
- Expert evaluation: 55.5% of MAGIC-DPO images met 3 or more of 5 criteria, versus 18.5% for fine-tuned I2I and 2.0% for T2I.
Measured effects reported in the paper
| What was changed | Where | Measured | Effect |
|---|---|---|---|
| Add MAGIC-DPO-generated images to classifier training | ResNet18 on Fitzpatrick17k 20-condition · Fitzpatrick17k 20-condition | accuracy: 29.31 -> 38.33; +9.02 | Helped |
| Add MAGIC-DPO-generated images to classifier training | DINOv2 on Fitzpatrick17k 20-condition · Fitzpatrick17k 20-condition | accuracy: 49.89 -> 55.01; +5.12 | Helped |
| Replace real-only training with T2I augmentation | ResNet18 on Fitzpatrick17k 20-condition · Fitzpatrick17k 20-condition | accuracy: -3.74% | Hurt |
| Replace real-only training with I2I augmentation | ResNet18 on Fitzpatrick17k 20-condition · Fitzpatrick17k 20-condition | accuracy: +2.14% | Helped |
| Add MAGIC-DPO-generated images to few-shot classifier training | DINOv2 with 310 real training images · Fitzpatrick17k few-shot | accuracy: 26.45 -> 37.39; +10.94 | Helped |
| Use unlabeled samples during MAGIC-DPO fine-tuning | DINOv2 with 310 real training images · Fitzpatrick17k few-shot | accuracy: 37.39 -> 40.34; +2.95 | Helped |
| Add MAGIC-DPO-generated images to classifier training | ResNet18 on SCIN · SCIN | accuracy: 23.13 -> 29.43; +6.3% | Helped |
| Add MAGIC-DPO-generated images to classifier training | DINOv2 on SCIN · SCIN | accuracy: 30.61 -> 35.65; +5.0% | Helped |
| Add MAGIC-DPO-generated images to classifier training | Classifiers on PAD-UFES-20 · PAD-UFES-20 | accuracy: 63.02 -> 70.65; 67.88 -> 73.85 | Helped |
| Replace the coarse checklist with a fine-grained checklist (ablation) | MAGIC-DPO on Fitzpatrick17k · Fitzpatrick17k 20-condition | accuracy: 32.83 -> 38.33; 51.16 -> 55.01 | Helped |
| Use a highly fine-grained 9-criteria checklist instead of the 7-criteria checklist (ablation) | MAGIC-DPO on Fitzpatrick17k · Fitzpatrick17k 20-condition | accuracy: 38.33 -> 39.39; 55.01 -> 55.88 | Helped |
| Increase DPO feedback volume (ablation) | MAGIC-DPO · Fitzpatrick17k 20-condition | accuracy | Helped |
Surprising findings
- Standard fine-tuned text-to-image generation degraded classifier accuracy (ResNet18 -3.74%, DINOv2 -2.16%), while the I2I variant helped.
- MedGemma-4B used as the evaluator achieved results close to GPT-4o (DINOv2 54.19% vs 55.01%), suggesting a smaller open-source judge is viable.
Limitations
- Performance is bounded by the quality of expert checklists and the capability of the MLLM evaluator (GPT-4o), which can fail on subtle textures, color nuances, and 3D cues.
- Evaluated only on Fitzpatrick17k, SCIN, and PAD-UFES-20; the SCIN augmented test set gave suboptimal results attributed to label noise.
- Synthetic-to-real domain shift requires careful tuning of the synthetic data ratio; purely synthetic training overfits.
Experimental setup
Datasets: Fitzpatrick17k (20-condition subset) SCIN PAD-UFES-20
Benchmarks: Fitzpatrick17k 20-condition classification SCIN PAD-UFES-20 few-shot Fitzpatrick17k
Metrics: accuracy F1 precision recall FID criteria satisfaction rate dermatologist alignment score
Baselines: Real-only training T2I fine-tuned Stable Diffusion I2I fine-tuned Stable Diffusion MAGIC-RFT Coarse checklist MAGIC
Builds on: Direct Preference Optimization Stable Diffusion Textual Inversion LoRA Diffusion-DPO
Closest papers by meaning
- DermaCon-IN: A Multiconcept-Annotated Dermatological Image Dataset of Indian Skin Disorders for Clinical AI Research (NeurIPS 2025 · similarity 0.87)
- MICA: Towards Explainable Skin Lesion Diagnosis via Multi-Level Image-Concept Alignment (AAAI 2024 · similarity 0.85)
- QoQ-Med: Building Multimodal Clinical Foundation Models with Domain-Aware GRPO Training (NeurIPS 2025 · similarity 0.84)
- Patho-R1: A Multimodal Reinforcement Learning-Based Pathology Expert Reasoner (AAAI 2026 · similarity 0.83)
- MedReasoner: Reinforcement Learning Drives Reasoning Grounding from Clinical Thought to Pixel-Level Precision (AAAI 2026 · similarity 0.83)
- How Do Medical MLLMs Fail? A Study on Visual Grounding in Medical Images (ICLR 2026 · similarity 0.83)
Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.
