Constrained-size Tensorflow Models for YouTube-8M Video Understanding Challenge

August 21, 2018 · Entered Twilight · 🏛 ECCV Workshops

"Last commit was 7.0 years ago (≥5 year threshold)"

Evidence collected by the PWNC Scanner

Repo contents: .gitignore, CONTRIBUTING.md, LICENSE, README.md, __init__.py, _config.yml, average_precision_calculator.py, avg_checkpoints.py, cloudml-gpu.yaml, convert_prediction_from_json_to_csv.py, core.py, eval.py, eval_util.py, export_model.py, feature_extractor, frame_level_models.py, inference.py, layers.py, local_cloud_matched.csv, losses.py, mean_average_precision_calculator.py, model_utils.py, models.py, overwrite_float16_ckpt.py, readers.py, train.py, utils.py, video_level_models.py

Authors Tianqi Liu, Bo Liu arXiv ID 1808.06739 Category cs.CV: Computer Vision Citations 5 Venue ECCV Workshops Repository https://github.com/boliu61/youtube-8m ⭐ 6 Last Checked 1 month ago

Abstract

This paper presents our 7th place solution to the second YouTube-8M video understanding competition which challenges participates to build a constrained-size model to classify millions of YouTube videos into thousands of classes. Our final model consists of four single models aggregated into one tensorflow graph. For each single model, we use the same network architecture as in the winning solution of the first YouTube-8M video understanding competition, namely Gated NetVLAD. We train the single models separately in tensorflow's default float32 precision, then replace weights with float16 precision and ensemble them in the evaluation and inference stages., achieving 48.5% compression rate without loss of precision. Our best model achieved 88.324% GAP on private leaderboard. The code is publicly available at https://github.com/boliu61/youtube-8m