A Stitch in Time: Learning Procedural Workflow via Self-Supervised Plackett-Luce Ranking
About
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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Action Segmentation | Breakfast | Acc21.6 | 116 | |
| Surgical Phase Recognition | Cholec80 | -- | 65 | |
| Action Segmentation | GTEA | Accuracy62.4 | 49 | |
| Surgical workflow recognition | M2CAI 2016 | Accuracy77.1 | 39 | |
| Surgical Phase Recognition | Autolaparo | Average F169 | 36 | |
| Surgical Phase Recognition | AutoLaparo (five-fold cross-val) | Accuracy80.1 | 4 | |
| Surgical Phase Recognition | Cholec80 (five-fold cross-validation) | Accuracy82.6 | 4 | |
| Surgical Phase Recognition | M2CAI 16 (five-fold cross-val) | Accuracy75.2 | 4 |