从真实视频学习鱼群集体行为,无需轨迹数据。
CBIL: Collective Behavior Imitation Learning for Fish from Real Videos
- 用自监督视频编码器提取隐状态,替代人工设计规则。
- 通过对抗性模仿学习捕捉复杂鱼群运动模式,提升真实性。
- 可应用于鱼群动画生成与异常行为检测,跨物种通用。
重现真实的集体行为是一项充满挑战的任务。传统基于规则的方法依赖人工设计的准则,限制了生成行为的多样性和真实性。近期的模仿学习方法虽从数据中学习,但通常需要真实运动轨迹,且在高密度、随机运动群体中难以保证真实性。本文提出一种可扩展的方法——集体行为模仿学习(CBIL),直接从视频中学习鱼群集体行为,无需依赖采集的运动轨迹。该方法首先利用掩码视频自编码器(MVAE)进行视频表征学习,在自监督下从视频输入中提取隐状态,有效将二维观测映射为紧凑且表达力强的隐空间表示。随后提出一种新颖的对抗性模仿学习方法,高效捕捉鱼群复杂运动模式,实现对运动模式分布的拟合。同时引入生物启发式奖励和先验知识以正则化并稳定训练过程。模型训练完成后,可用于多种动画任务,具备跨物种适用性。我们进一步展示了其在野外视频中检测异常鱼群行为的应用效果。
原文摘要 · Abstract (English)
Reproducing realistic collective behaviors presents a captivating yet formidable challenge. Traditional rule-based methods rely on hand-crafted principles, limiting motion diversity and realism in generated collective behaviors. Recent imitation learning methods learn from data but often require ground truth motion trajectories and struggle with authenticity, especially in high-density groups with erratic movements. In this paper, we present a scalable approach, Collective Behavior Imitation Learning (CBIL), for learning fish schooling behavior directly from videos, without relying on captured motion trajectories. Our method first leverages Video Representation Learning, where a Masked Video AutoEncoder (MVAE) extracts implicit states from video inputs in a self-supervised manner. The MVAE effectively maps 2D observations to implicit states that are compact and expressive for following the imitation learning stage. Then, we propose a novel adversarial imitation learning method to effectively capture complex movements of the schools of fish, allowing for efficient imitation of the distribution for motion patterns measured in the latent space. It also incorporates bio-inspired rewards alongside priors to regularize and stabilize training. Once trained, CBIL can be used for various animation tasks with the learned collective motion priors. We further show its effectiveness across different species. Finally, we demonstrate the application of our system in detecting abnormal fish behavior from in-the-wild videos.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。