让模型学会视频动作的时间顺序,提升手术和烹饪视频理解能力
A Stitch in Time: Learning Procedural Workflow via Self-Supervised Plackett-Luce Ranking
- 用排列概率模型引导模型按时间排序视频帧,学习全局流程
- 在五个基准上表现优异,手术阶段识别准确率提升11.4个百分点
- 适合需要理解动作时序的场景,如医疗手术分析、智能厨房
程序性活动(如日常烹饪或复杂外科手术)是按特定时间顺序执行的高度结构化动作序列。尽管当前自监督学习方法在静态图像和短片段上表现良好,但这些模型常忽略活动的内在时序结构。我们通过实验发现:在正向与逆向视频上预训练的模型产生高度相似的特征,表明其对时序顺序不敏感。为此,我们提出PL-Stitch框架,利用视频帧的自然时序作为强监督信号。该方法融合基于Plackett-Luce(PL)模型的两个新概率目标:主目标训练模型对采样帧进行时间排序,强制学习全局流程推进;辅目标为时空拼图损失,捕捉跨帧物体对应关系。在五个外科与烹饪基准上,PL-Stitch持续取得更优性能,例如在Cholec80数据集上手术阶段识别的k-NN准确率提升11.4个百分点,在Breakfast数据集上烹饪动作分割线性探测准确率提升5.7个百分点,验证了其在程序性视频表征学习中的有效性。代码与模型已开源。
原文摘要 · Abstract (English)
Procedural activities, ranging from routine cooking to complex surgical operations, are highly structured sequences of actions performed in a specific temporal order. Despite the success of current self-supervised learning (SSL) methods on static images and short clips, these models often overlook the underlying sequential structure of such activities. We expose this lack of procedural awareness with a motivating experiment: models pretrained on forward and time-reversed sequences produce highly similar features, confirming that their representations are blind to the underlying procedural order. To address this shortcoming, we propose PL-Stitch, a self-supervised framework that harnesses the inherent temporal order of video frames as a powerful supervisory signal. Our approach integrates two novel probabilistic objectives based on the Plackett-Luce (PL) model. The primary PL objective trains the model to sort sampled frames chronologically, compelling it to learn the global workflow progression. The secondary objective, a spatio-temporal jigsaw loss, complements the learning by capturing fine-grained, cross-frame object correspondences. Our approach consistently achieves superior performance across five surgical and cooking benchmarks. Specifically, PL-Stitch yields significant gains in surgical phase recognition (e.g., +11.4 pp in k-NN accuracy on Cholec80) and cooking action segmentation (e.g., +5.7 pp in linear probing accuracy on Breakfast), demonstrating its effectiveness for procedural video representation learning. Code and models are available at https://github.com/visurg-ai/PL-Stitch.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。