轻量微调扩散模型,用小模块适配医学影像,数据少也高效。
LiteDiff
- 冻结主模型,只训练小模块适配器,降低计算开销。
- 在三个胸片数据集上优于全量微调,最小数据下仍稳定。
- 适合医疗等数据稀缺场景的扩散模型快速部署。
近年来,扩散模型在高保真图像生成方面取得显著进展。然而,由于领域特定数据有限且全模型适配计算成本高,将其用于医疗影像等专业领域仍具挑战。本文提出 LiteDiff(轻量级扩散模型适配)方法,在冻结的扩散U-Net中引入轻量级适配层,并结合潜在形态自编码器(增强领域特定潜在一致性)与像素级判别器(实现对抗性对齐),以优化训练。通过仅微调小型残差适配模块,大幅降低计算开销并缓解过拟合问题,即使在极少量数据条件下亦表现良好。我们在三个胸片数据集(Kaggle Chest X-Ray Pneumonia、NIH Chest X-ray14、VinBigData Chest X-ray)上进行实验,结果表明 LiteDiff 相较于朴素全量微调展现出更优的适应效率。该框架为扩散模型的迁移学习提供了新方向,推动其在低数据领域的应用。
原文摘要 · Abstract (English)
In recent years, diffusion models have demonstrated remarkable success in high-fidelity image synthesis. However, fine-tuning these models for specialized domains, such as medical imaging, remains challenging due to limited domain-specific data and the high computational cost of full model adaptation. In this paper, we introduce Lite-Diff (Lightweight Diffusion Model Adaptation), a novel finetuning approach that integrates lightweight adaptation layers into a frozen diffusion U-Net while enhancing training with a latent morphological autoencoder (for domain-specific latent consistency) and a pixel level discriminator(for adversarial alignment). By freezing weights of the base model and optimizing only small residual adapter modules, LiteDiff significantly reduces the computational overhead and mitigates overfitting, even in minimal-data settings. Additionally, we conduct ablation studies to analyze the effects of selectively integrating adaptation layers in different U-Net blocks, revealing an optimal balance between efficiency and performance. Experiments on three chest X-ray datasets - (1) Kaggle Chest X-Ray Pneumonia, (2) NIH Chest X-ray14 and (3) VinBigData Chest X_ray demonstrate that LiteDiff achieves superior adaptation efficiency compared to naive full fine-tuning. Our framework provides a promising direction for transfer learning in diffusion models, facilitating their deployment in diverse low data domains.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。