Author Login Chief Editor Login Reviewer Login Editor Login Remote Office

Computer Engineering

   

Lightweight DiT-Based Video Virtual Try-On via First-Frame Guidance and Temporal Priors

  

  • Published:2026-06-16

基于首帧引导与时序先验的轻量化DiT视频虚拟试穿方法

Abstract: Video virtual try-on technology aims to accurately transfer target garments onto human subjects in videos while maintaining high consistency between body motion and garment appearance, serving as a core technology in fields such as e-commerce, virtual reality, and short-video creation. However, existing technical frameworks still face significant challenges in balancing generation quality and computational efficiency. Traditional Generative Adversarial Network (GAN)-based methods often rely on optical flow estimation for garment warping, which are highly prone to texture distortion and visual artifacts when handling complex motions. In recent years, U-Net-based diffusion models have achieved high-fidelity generation by introducing a garment reference branch. However, when such dual-branch architectures are migrated to larger and more expressive Diffusion Transformer (DiT) backbones, they introduce substantial parameter redundancy and VRAM overhead. Furthermore, existing methods typically inject static garment features repeatedly during the denoising process of each frame. This not only significantly exacerbates the computational burden but also, due to the lack of natural temporal correlation in static features, makes it difficult for models to maintain spatiotemporal coherence during non-rigid deformations, resulting in severe flickering artifacts. To address the aforementioned challenges regarding the adaptability, training efficiency, and resource consumption of DiT architectures in video virtual try-on tasks, this study proposes a lightweight framework named OIE (Once is Enough). The OIE framework adopts a novel single-branch strategy featuring first-frame guidance and one-time injection, effectively decoupling garment editing from temporal generation tasks. First, during the garment appearance injection stage, a pre-trained high-fidelity image virtual try-on model, FiT-DiT, is utilized to precisely edit the video's initial frame, yielding results integrated with fine-grained garment textures. Second, to maximally preserve the temporal priors of the DiT model, only the edited first frame is embedded as the starting token into the latent feature sequence of the backbone network. This avoids the dense cross-branch feature interaction modules typical of traditional dual-branch architectures, achieving zero structural modification to the backbone. Additionally, to address the loss of background layout information caused by human motion, this method designs a lightweight background encoder that smoothly accumulates background information into the backbone features via a mask guider. Finally, during the fine-tuning stage, Low-Rank Adaptation (LoRA) is applied to all self-attention, cross-attention, and feed-forward network (FFN) modules of the DiT, enabling dynamic regulation of the large-scale parameter model with an extremely low number of trainable parameters. Experiments conducted on the ViViD and VVT datasets yield quantitative evaluation results demonstrating that, in terms of efficiency, OIE introduces only a 0.50% additional parameter overhead, with FLOPs and FPS remaining virtually unchanged. Its performance significantly surpasses dual-branch methods such as MagicTryOn (15.11% parameter increase) and ViViD (157.10% parameter increase). Regarding quality metrics, OIE achieves competitive video quality scores under both paired and unpaired settings on the ViViD dataset, attaining a VFIDp of 9.3983 and a VFIDu of 17.0831, significantly leading existing mainstream methods. Ablation studies confirm that high-quality first-frame guidance effectively suppresses error generation in the early stages of synthesis, improving the SSIM metric to 0.8466. Through its decoupling strategy, the OIE framework effectively resolves the computational burden of DiT architectures in video generation, achieving an excellent balance among garment fidelity, temporal coherence, and computational efficiency. This method demonstrates that leveraging strong temporal priors within a single-branch architecture can replace high-frequency feature injection, offering a highly valuable lightweight pathway for high-resolution and real-time video editing tasks.

摘要: 视频虚拟试穿技术旨在将目标服装精准地迁移至视频中的人物主体,同时保持人物动作与服装外观的高度一致性,是电子商务、虚拟现实及短视频创作等领域的核心技术。然而,现有的技术框架在生成质量与计算效率的权衡上仍面临巨大挑战。传统的基于生成对抗网络(GAN)的方法往往依赖光流估计进行服装变形,在处理复杂运动时极易产生纹理失真和视觉伪影。近年来,基于U-Net的扩散模型通过引入服装参考分支实现了高保真生成,但此类双分支架构在迁移至参数量更庞大、表征能力更强的扩散Transformer(DiT)主干网络时,会引入巨大的参数冗余与显存开销。此外,现有方法通常在每一帧去噪过程中重复注入静态服装特征,这不仅显著加剧了计算负担,还因为静态特征缺乏天然的时序关联,导致模型在处理非刚性形变时难以维持时空连贯性,产生严重的闪烁现象。针对上述DiT架构在视频虚拟试穿任务中的适应性、训练效率及资源消耗挑战,本研究提出了一种名为OIE(Once is Enough)的轻量化视频虚拟试穿框架。OIE框架采用了首帧引导、单次注入的新型单分支策略,将服装编辑与时序生成任务解耦。首先,在服装外观注入阶段,利用预训练的高保真图像虚拟试穿模型Fitdit对视频初始帧进行精确编辑,获取集成细粒度服装纹理的结果。其次,为了最大限度保留DiT模型的时序先验,仅将编辑后的首帧作为潜在空间特征序列的起始token嵌入到主干网络中,避免了传统双分支架构中密集的跨分支特征交互模块,实现了主干网络架构的零修改。此外,为解决人体运动导致的背景布局信息丢失,本方法设计了一个轻量级背景编码器,通过掩码引导器平滑地将背景信息累加至主干特征中。最后,在微调阶段,将低秩自适应(LoRA)技术应用于DiT的所有自注意力、交叉注意力及前馈网络模块,以极低的可训练参数量实现了对大规模参数模型的动态调节。实验在ViViD和VVT数据集上进行,定量评估结果显示:在效率方面,OIE仅引入了0.50%的额外参数开销,且FLOPs和FPS几乎无显著变化,表现大幅优于MagicTryOn(参数增幅15.11%)和ViViD(参数增幅157.10%)等双分支方法。在质量指标方面,OIE在ViViD数据集的配对(paired)和非配对(unpaired)设置下均取得了较好的视频质量评分,其中VFIDp达到9.3983,VFIDu达到17.0831,显著领先于现有主流方法。消融实验证实,高质量的首帧引导能够有效抑制生成初期的错误产生,SSIM指标可提升至0.8466。OIE框架通过解耦策略有效解决了DiT架构在视频生成中的计算负担问题,实现了服装保真度、时序连贯性与计算效率的卓越平衡。该方法证明了在单分支架构下利用强大的时序先验知识可替代高频次的特征注入,为高分辨率、实时化的视频编辑任务提供了极具参考价值的轻量化路径。