Home

Awesome

On the hidden treasure of dialog in video question answering

Deniz Engin, François Schnitzler, Ngoc Q. K. Duong and Yannis Avrithis, On the hidden treasure of dialog in video question answering, ICCV 2021.

Project page | arXiv


Model Overview

Model

Our VideoQA system converts dialog and video inputs to episode dialog summaries and video descriptions, respectively. Converted inputs and dialog are processed independently in streams, along with the question and each answer, producing a score per answer. Finally, stream embeddings are fused separately per answer and a prediction is made.

Environment Setup

To create a conda environment:

conda env create -f environment.yml

or

conda create --name  dialog-videoqa python=3.6
conda activate dialog-videoqa
conda install -c anaconda numpy pandas scikit-learn
conda install -c conda-forge tqdm
conda install pytorch==1.0.1 torchvision==0.2.2 cudatoolkit=10.0 -c pytorch
pip install pytorch-transformers

Data preparation

Training Models

This section explains single-stream QA and multi-stream QA trainings.

Single-Stream QA

Our main streams are video, scene dialog summary, episode dialog summary. Dialog and plot streams are used for comparison. All stream trainings as follows:

Training video stream:

python stream_main_train.py --train_name video --max_seq_length 512

Training scene dialog summary stream:

python stream_main_train.py --train_name scene_dialog_summary --max_seq_length 512

Training episode dialog summary stream:

python stream_main_train.py --train_name episode_dialog_summary --max_seq_length 300 --seq_stride 200 --mini_batch_size 2 --eval_batch_size 16 

Training dialog stream:

python stream_main_train.py --train_name dialog --max_seq_length 512

Training plot stream:

python stream_main_train.py --train_name plot --max_seq_length 200 --seq_stride 100 --mini_batch_size 2

All single stream models trained on 2 Tesla V100 GPUs (32 GB) except plot trained on 1 Tesla V100. Gradient accumulation is used to fit into memory when training parameters have a "mini_batch_size".

Multi-Stream QA

Our main proposed model uses video, scene dialog summary and episode dialog summary streams for multi-stream attention method.

python fusion_main_train.py --fuse_stream_list video scene_dialog_summary episode_dialog_summary --fusion_method multi-stream-attention

License

Please check the license file for more information.

Acknowledgments

The code is written based on <a href="https://github.com/noagarcia/ROLL-VideoQA" target="_blank">ROLL-VideoQA</a>.

Citation

If this code is helpful for you, please cite the following:

@inproceedings{engin2021hidden,
  title={On the hidden treasure of dialog in video question answering},
  author={Engin, Deniz and Schnitzler, Fran{\c{c}}ois and Duong, Ngoc QK and Avrithis, Yannis},
  journal={ICCV},
  year={2021}
}