Overview of VideoTag Large-scale Video Classification Model
developVideoTag is a large-scale video classification model based on tens of millions of short video data points from Baidu's business. It supports 3,000 practical labels derived from industrial practice and is designed for large-scale short video classification scenarios (millions/billions of videos).
Core Architecture
The model uses a two-stage modeling approach:
- Image Modeling (Extractor): A large-scale video feature extraction model is trained using a smaller set of video samples (around 100,000).
- Sequence Learning (Predictor): A predictor is trained using tens of millions of data points to learn from the feature sequences extracted in the first stage.
Technical Workflow
- Data Processing: Videos are decoded into a sequence of image frames, which are then fed into VideoTag for training or prediction.
- Image Modeling: Uses a TSN (Temporal Segment Networks) architecture to extract features from video frames. Each frame is converted into a feature vector, and a video is represented as a sequence of these vectors.
- Sequence Learning: Models the feature sequences using techniques like Attention clusters, LSTM, and Nextvlad to learn combinations of features. The provided example code specifically utilizes the Attention_LSTM network for sequence feature prediction.
- Prediction: Results from multiple models can be fused to improve classification accuracy.