Active learning for efficient annotation of surgical videos with weak supervision
Related Experiment Videos
Manasa Dendukuri1,2, Matjaž Jogan3,4, Daniel A Hashimoto3,4,5
1GRASP Laboratory, University of Pennsylvania, 3330 Walnut Street, Philadelphia, PA, 19104, USA. maniden@seas.upenn.edu.
Purpose:
Precise spatial-temporal annotation of laparoscopic videos is time-consuming and requires expert knowledge. We propose a human-in-the-loop knowledge acquisition framework that combines active learning with dual-loss optimization to significantly reduce the annotation effort needed for automatic localization and segmentation of objects in the surgical field.
Methods:
Our method employs a foundation model to generate temporally consistent class activation maps (CAMs) from video using two complementary training objectives: a weak supervision loss on video-level tool presence labels for weakly annotated data, and an image-level mask loss on human-corrected annotations obtained through active learning. Rather than requiring dense pixel-level annotation upfront, our pipeline iteratively proposes pseudo-masks that guide the expert annotator to refine the knowledge previously captured by the model.
Results:
We demonstrate that our framework reduces the effort of surgical video annotation by approximately 50% by the end of training in comparison to fully manual annotation.
Conclusion:
Through eliminating the need for large, fully annotated datasets from the start, this framework enables scalability to the development of surgical tool segmentation models. This iterative human-in-the-loop refinement supports efficient knowledge acquisition with minimal expert input, providing a practical and deployable strategy for expanding tool segmentation to larger, more diverse datasets and real-world clinical settings. Code shared at: https://github.com/PCASOlab/AL-WSL .