arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.02482cs.CV
收藏

利用丰富上下文条件扩散模型提升故事可视化的连贯性

Boosting Consistency in Story Visualization with Rich-Contextual Conditional Diffusion Models

Tencent AI Lab(腾讯AI实验室)

Fei Shen, Hu Ye, Sibo Liu, Jun Zhang, Cong Wang, Xiao Han, Wei Yang

AI总结:

本文提出丰富上下文条件扩散模型RCDMs,通过两阶段框架联合注入参考图像、预测帧语义嵌入和文本嵌入,以单次前向推理实现语义与时间连贯的故事可视化生成。

摘要:

近期研究展示了条件扩散模型在生成连贯故事方面的巨大潜力。然而,当前方法主要以自回归且过度依赖字幕的方式生成故事,往往低估了序列生成过程中帧的上下文连贯性和相关性。为解决这一问题,我们提出了一种新颖的丰富上下文条件扩散模型(RCDMs),这是一种两阶段方法,旨在增强故事生成的语义连贯性和时间连贯性。具体而言,在第一阶段,提出了帧先验Transformer扩散模型,通过对齐已知片段字幕与帧之间的语义相关性,预测未知片段的帧语义嵌入。第二阶段建立了一个具有丰富上下文条件的鲁棒模型,这些条件包括已知片段的参考图像、预测的未知片段帧语义嵌入以及所有字幕的文本嵌入。通过在图像和特征层面联合注入这些丰富的上下文条件,RCDMs能够生成语义和时间上连贯的故事。此外,与自回归模型相比,RCDMs可以通过单次前向推理生成连贯故事。我们的定性和定量结果表明,所提出的RCDMs在具有挑战性的场景中表现更优。代码和模型将在 https://github.com/muzishen/RCDMs 上公开。

英文摘要:

Recent research showcases the considerable potential of conditional diffusion models for generating consistent stories. However, current methods, which predominantly generate stories in an autoregressive and excessively caption-dependent manner, often underrate the contextual consistency and relevance of frames during sequential generation. To address this, we propose a novel Rich-contextual Conditional Diffusion Models (RCDMs), a two-stage approach designed to enhance story generation's semantic consistency and temporal consistency. Specifically, in the first stage, the frame-prior transformer diffusion model is presented to predict the frame semantic embedding of the unknown clip by aligning the semantic correlations between the captions and frames of the known clip. The second stage establishes a robust model with rich contextual conditions, including reference images of the known clip, the predicted frame semantic embedding of the unknown clip, and text embeddings of all captions. By jointly injecting these rich contextual conditions at the image and feature levels, RCDMs can generate semantic and temporal consistency stories. Moreover, RCDMs can generate consistent stories with a single forward inference compared to autoregressive models. Our qualitative and quantitative results demonstrate that our proposed RCDMs outperform in challenging scenarios. The code and model will be available at https://github.com/muzishen/RCDMs.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑