KMM: Key Frame Mask Mamba for Extended Motion Generation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Zhang, Zeyu, Gao, Hang, Liu, Akide, Chen, Qi, Chen, Feng, Wang, Yiran, Li, Danning, Zhao, Rui, Li, Zhenming, Zhou, Zhongwen, Tang, Hao, Zhuang, Bohan
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866912330207461376
author Zhang, Zeyu
Gao, Hang
Liu, Akide
Chen, Qi
Chen, Feng
Wang, Yiran
Li, Danning
Zhao, Rui
Li, Zhenming
Zhou, Zhongwen
Tang, Hao
Zhuang, Bohan
author_facet Zhang, Zeyu
Gao, Hang
Liu, Akide
Chen, Qi
Chen, Feng
Wang, Yiran
Li, Danning
Zhao, Rui
Li, Zhenming
Zhou, Zhongwen
Tang, Hao
Zhuang, Bohan
contents 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
format Preprint
id arxiv_https___arxiv_org_abs_2411_06481
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle KMM: Key Frame Mask Mamba for Extended Motion Generation
Zhang, Zeyu
Gao, Hang
Liu, Akide
Chen, Qi
Chen, Feng
Wang, Yiran
Li, Danning
Zhao, Rui
Li, Zhenming
Zhou, Zhongwen
Tang, Hao
Zhuang, Bohan
Computer Vision and Pattern Recognition
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
title KMM: Key Frame Mask Mamba for Extended Motion Generation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2411.06481