Video Annotator: A framework for efficiently building video classifiers using vision-language models and active learning
Amir Ziai, Aneesh Vartakavi
TL;DR
Video Annotator (VA) addresses the high cost and inefficiency of traditional video annotation by integrating active learning with vision-language model backbones and direct domain-expert involvement in a continuous annotation-deployment loop. The framework uses pre-trained video-text embeddings, text-to-video search for bootstrap labeling, and lightweight binary classifiers per label within a four-feed active-learning interface to rapidly refine models. VA introduces Model quality and Data diversity scores, and demonstrates substantial improvements in $AP$ (up to about $8.3$ points) over competitive baselines across 56 tasks, supported by a dataset of 153k annotations and accompanying code. This approach enables fast, trusted, scalable video classification and suggests broad applicability to other media modalities, with potential for practical impact in search, personalization, and content understanding.
Abstract
High-quality and consistent annotations are fundamental to the successful development of robust machine learning models. Traditional data annotation methods are resource-intensive and inefficient, often leading to a reliance on third-party annotators who are not the domain experts. Hard samples, which are usually the most informative for model training, tend to be difficult to label accurately and consistently without business context. These can arise unpredictably during the annotation process, requiring a variable number of iterations and rounds of feedback, leading to unforeseen expenses and time commitments to guarantee quality. We posit that more direct involvement of domain experts, using a human-in-the-loop system, can resolve many of these practical challenges. We propose a novel framework we call Video Annotator (VA) for annotating, managing, and iterating on video classification datasets. Our approach offers a new paradigm for an end-user-centered model development process, enhancing the efficiency, usability, and effectiveness of video classifiers. Uniquely, VA allows for a continuous annotation process, seamlessly integrating data collection and model training. We leverage the zero-shot capabilities of vision-language foundation models combined with active learning techniques, and demonstrate that VA enables the efficient creation of high-quality models. VA achieves a median 6.8 point improvement in Average Precision relative to the most competitive baseline across a wide-ranging assortment of tasks. We release a dataset with 153k labels across 56 video understanding tasks annotated by three professional video editors using VA, and also release code to replicate our experiments at: http://github.com/netflix/videoannotator.
