Uncertainty-Aware Deep Video Compression with Ensembles

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Ma, Wufei, Li, Jiahao, Li, Bin, Lu, Yan
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866914731792531456
author Ma, Wufei
Li, Jiahao
Li, Bin
Lu, Yan
author_facet Ma, Wufei
Li, Jiahao
Li, Bin
Lu, Yan
contents Deep learning-based video compression is a challenging task, and many previous state-of-the-art learning-based video codecs use optical flows to exploit the temporal correlation between successive frames and then compress the residual error. Although these two-stage models are end-to-end optimized, the epistemic uncertainty in the motion estimation and the aleatoric uncertainty from the quantization operation lead to errors in the intermediate representations and introduce artifacts in the reconstructed frames. This inherent flaw limits the potential for higher bit rate savings. To address this issue, we propose an uncertainty-aware video compression model that can effectively capture the predictive uncertainty with deep ensembles. Additionally, we introduce an ensemble-aware loss to encourage the diversity among ensemble members and investigate the benefits of incorporating adversarial training in the video compression task. Experimental results on 1080p sequences show that our model can effectively save bits by more than 20% compared to DVC Pro.
format Preprint
id arxiv_https___arxiv_org_abs_2403_19158
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Uncertainty-Aware Deep Video Compression with Ensembles
Ma, Wufei
Li, Jiahao
Li, Bin
Lu, Yan
Computer Vision and Pattern Recognition
Image and Video Processing
Deep learning-based video compression is a challenging task, and many previous state-of-the-art learning-based video codecs use optical flows to exploit the temporal correlation between successive frames and then compress the residual error. Although these two-stage models are end-to-end optimized, the epistemic uncertainty in the motion estimation and the aleatoric uncertainty from the quantization operation lead to errors in the intermediate representations and introduce artifacts in the reconstructed frames. This inherent flaw limits the potential for higher bit rate savings. To address this issue, we propose an uncertainty-aware video compression model that can effectively capture the predictive uncertainty with deep ensembles. Additionally, we introduce an ensemble-aware loss to encourage the diversity among ensemble members and investigate the benefits of incorporating adversarial training in the video compression task. Experimental results on 1080p sequences show that our model can effectively save bits by more than 20% compared to DVC Pro.
title Uncertainty-Aware Deep Video Compression with Ensembles
topic Computer Vision and Pattern Recognition
Image and Video Processing
url https://arxiv.org/abs/2403.19158