用关键帧掩码提升Mamba长序列动作生成能力,解决记忆衰减与文本对齐难题。
KMM: Key Frame Mask Mamba for Extended Motion Generation
- 设计关键帧掩码机制,增强Mamba对动作关键帧的关注度。
- 在BABEL数据集上FID降低57%,参数量减少70%。
- 适合需要高效高精度动作生成的研究者与开发者。
人类动作生成是生成式计算机视觉的前沿领域,广泛应用于视频创作、游戏开发和机器人操控。近期的Mamba架构在建模长而复杂的序列方面表现优异,但仍面临两大挑战:其一,直接将Mamba用于长时动作生成效果不佳,因隐式记忆容量有限导致记忆衰减;其二,相比Transformer,Mamba在多模态融合与文本查询对齐方面表现较弱,常混淆方向(左/右)或遗漏长文本查询内容。为此,本文提出KMM,一种新型架构,包含关键帧掩码建模,可增强Mamba对动作片段中关键动作的关注,有效缓解记忆衰减问题,是首个在状态空间模型中实现帧级战略掩码的方法。同时,设计对比学习范式以改善多模态融合与运动-文本对齐。在主流数据集BABEL上进行大量实验,结果表明,KMM达到当前最优性能,相较之前最优方法FID降低超过57%,参数量减少70%。
原文摘要 · Abstract (English)
Human motion generation is a cut-edge area of research in generative computer vision, with promising applications in video creation, game development, and robotic manipulation. The recent Mamba architecture shows promising results in efficiently modeling long and complex sequences, yet two significant challenges remain: Firstly, directly applying Mamba to extended motion generation is ineffective, as the limited capacity of the implicit memory leads to memory decay. Secondly, Mamba struggles with multimodal fusion compared to Transformers, and lack alignment with textual queries, often confusing directions (left or right) or omitting parts of longer text queries. To address these challenges, our paper presents three key contributions: Firstly, we introduce KMM, a novel architecture featuring Key frame Masking Modeling, designed to enhance Mamba's focus on key actions in motion segments. This approach addresses the memory decay problem and represents a pioneering method in customizing strategic frame-level masking in SSMs. Additionally, we designed a contrastive learning paradigm for addressing the multimodal fusion problem in Mamba and improving the motion-text alignment. Finally, we conducted extensive experiments on the go-to dataset, BABEL, achieving state-of-the-art performance with a reduction of more than 57% in FID and 70% parameters compared to previous state-of-the-art methods. See project website: https://steve-zeyu-zhang.github.io/KMM
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。