|国家预印本平台
首页|T-LoRA: Single Image Diffusion Model Customization Without Overfitting

T-LoRA: Single Image Diffusion Model Customization Without Overfitting

T-LoRA: Single Image Diffusion Model Customization Without Overfitting

来源:Arxiv_logoArxiv
英文摘要

While diffusion model fine-tuning offers a powerful approach for customizing pre-trained models to generate specific objects, it frequently suffers from overfitting when training samples are limited, compromising both generalization capability and output diversity. This paper tackles the challenging yet most impactful task of adapting a diffusion model using just a single concept image, as single-image customization holds the greatest practical potential. We introduce T-LoRA, a Timestep-Dependent Low-Rank Adaptation framework specifically designed for diffusion model personalization. In our work we show that higher diffusion timesteps are more prone to overfitting than lower ones, necessitating a timestep-sensitive fine-tuning strategy. T-LoRA incorporates two key innovations: (1) a dynamic fine-tuning strategy that adjusts rank-constrained updates based on diffusion timesteps, and (2) a weight parametrization technique that ensures independence between adapter components through orthogonal initialization. Extensive experiments show that T-LoRA and its individual components outperform standard LoRA and other diffusion model personalization techniques. They achieve a superior balance between concept fidelity and text alignment, highlighting the potential of T-LoRA in data-limited and resource-constrained scenarios. Code is available at https://github.com/ControlGenAI/T-LoRA.

Vera Soboleva、Aibek Alanov、Andrey Kuznetsov、Konstantin Sobolev

计算技术、计算机技术

Vera Soboleva,Aibek Alanov,Andrey Kuznetsov,Konstantin Sobolev.T-LoRA: Single Image Diffusion Model Customization Without Overfitting[EB/OL].(2025-07-08)[2025-07-19].https://arxiv.org/abs/2507.05964.点此复制

评论