Unsupervised Temporal Video Segmentation as an Auxiliary Task for Predicting the Remaining Surgery Duration

• Published in: arXiv.org 2020-02
• Main Authors: Rivoir, Dominik, Bodenstedt, Sebastian, Felix von Bechtolsheim, Distler, Marius, Weitz, Jürgen, Speidel, Stefanie
• Format: Article
• Language: English
• Subjects: Anesthesia; Annotations; Cognitive tasks; Computer vision; Feature extraction; Ground truth; Machine learning; Neural networks; Regularization; Segmentation; Surgery; Training
• ISSN: 2331-8422

Estimating the remaining surgery duration (RSD) during surgical procedures can be useful for OR planning and anesthesia dose estimation. With the recent success of deep learning-based methods in computer vision, several neural network approaches have been proposed for fully automatic RSD prediction based solely on visual data from the endoscopic camera. We investigate whether RSD prediction can be improved using unsupervised temporal video segmentation as an auxiliary learning task. As opposed to previous work, which presented supervised surgical phase recognition as auxiliary task, we avoid the need for manual annotations by proposing a similar but unsupervised learning objective which clusters video sequences into temporally coherent segments. In multiple experimental setups, results obtained by learning the auxiliary task are incorporated into a deep RSD model through feature extraction, pretraining or regularization. Further, we propose a novel loss function for RSD training which attempts to counteract unfavorable characteristics of the RSD ground truth. Using our unsupervised method as an auxiliary task for RSD training, we outperform other self-supervised methods and are comparable to the supervised state-of-the-art. Combined with the novel RSD loss, we slightly outperform the supervised approach.