Make-A-Story: Visual Memory Conditioned Consistent Story Generation

TL;DR

This paper introduces Make-A-Story, a diffusion-based model with visual memory for coherent multi-frame story visualization, effectively handling references and maintaining scene consistency across complex storylines.

Contribution

It presents a novel autoregressive diffusion framework with a visual memory module and reference-aware attention for improved story visualization.

Findings

Outperforms prior methods in visual quality and story consistency.

Effectively models actor and background references across frames.

Validated on extended MUGEN, PororoSV, and FlintstonesSV datasets.

Abstract

There has been a recent explosion of impressive generative models that can produce high quality images (or videos) conditioned on text descriptions. However, all such approaches rely on conditional sentences that contain unambiguous descriptions of scenes and main actors in them. Therefore employing such models for more complex task of story visualization, where naturally references and co-references exist, and one requires to reason about when to maintain consistency of actors and backgrounds across frames/scenes, and when not to, based on story progression, remains a challenge. In this work, we address the aforementioned challenges and propose a novel autoregressive diffusion-based framework with a visual memory module that implicitly captures the actor and background context across the generated frames. Sentence-conditioned soft attention over the memories enables effective reference resolution and learns to maintain scene and actor consistency when needed. To validate the effectiveness of our approach, we extend the MUGEN dataset and introduce additional characters, backgrounds and referencing in multi-sentence storylines. Our experiments for story generation on the MUGEN, the PororoSV and the FlintstonesSV dataset show that our method not only outperforms prior state-of-the-art in generating frames with high visual quality, which are consistent with the story, but also models appropriate correspondences between the characters and the background.