SlotMem: Character-Addressable Internal Memory for Narrative Long Video Generation

· Source: Computer Vision and Pattern Recognition · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Computer Vision · Depth: Expert, quick

Summary

SlotMem is a novel character-addressable internal memory framework designed for multi-character narrative long video generation, addressing the challenge of maintaining recurring character identities across scene transitions and extended temporal gaps. Existing methods often struggle with global consistency due to retrieval cues not aligned with character identity or rely on coarse frame-level memory that entangles identity with incidental visual factors. SlotMem tackles this by employing a Character-Semantic Probe to localize character-relevant visual tokens from cross-attention responses. It then uses a Memory Encoder to compress DiT tokens into compact role-wise slot memory. A Memory Writer continuously updates each character's memory with new observations, while Character-Wise Cross-Attention injects the retrieved role memory exclusively into localized tokens of the corresponding character. Experiments on multiple narrative long video generation benchmarks demonstrate that SlotMem significantly improves long-range character consistency compared to existing baselines, all while preserving comparable video quality. The code for SlotMem is available on GitHub.

Key takeaway

For Machine Learning Engineers developing narrative long video generation models, SlotMem offers a robust solution for maintaining character identity. If your current models struggle with character consistency across long sequences or scene changes, you should consider integrating character-addressable memory. This approach, which localizes and selectively updates character-specific visual tokens, can significantly improve identity preservation without compromising video quality, streamlining your development of more coherent multi-character narratives.

Key insights

SlotMem uses character-addressable memory and selective attention to maintain consistent character identities in long video generation, outperforming prior methods.

Principles

Method

SlotMem localizes character tokens via a Character-Semantic Probe, compresses DiT tokens into role-wise slot memory using a Memory Encoder, updates memory with a Memory Writer, and injects role memory via Character-Wise Cross-Attention.

In practice

Topics

Code references

Best for: Research Scientist, AI Scientist, Machine Learning Engineer, Computer Vision Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Computer Vision and Pattern Recognition.