Generative Adversarial Networks for Video-to-Video Domain Adaptation
arXiv:2004.08058
Abstract
Endoscopic videos from multicentres often have different imaging conditions, e.g., color and illumination, which make the models trained on one domain usually fail to generalize well to another. Domain adaptation is one of the potential solutions to address the problem. However, few of existing works focused on the translation of video-based data. In this work, we propose a novel generative adversarial network (GAN), namely VideoGAN, to transfer the video-based data across different domains. As the frames of a video may have similar content and imaging conditions, the proposed VideoGAN has an X-shape generator to preserve the intra-video consistency during translation. Furthermore, a loss function, namely color histogram loss, is proposed to tune the color distribution of each translated frame. Two colonoscopic datasets from different centres, i.e., CVC-Clinic and ETIS-Larib, are adopted to evaluate the performance of domain adaptation of our VideoGAN. Experimental results demonstrate that the adapted colonoscopic video generated by our VideoGAN can significantly boost the segmentation accuracy, i.e., an improvement of 5%, of colorectal polyps on multicentre datasets. As our VideoGAN is a general network architecture, we also evaluate its performance with the CamVid driving video dataset on the cloudy-to-sunny translation task. Comprehensive experiments show that the domain gap could be substantially narrowed down by our VideoGAN.
arXiv commentAccepted by AAAI 2020
References in corpus (9)
- Adam: A Method for Stochastic Optimization
- Unpaired Image-to-Image Translation using Cycle-Consistent Adversarial Networks
- Unsupervised Image-to-Image Translation Networks
- DualGAN: Unsupervised Dual Learning for Image-to-Image Translation
- Video-to-Video Synthesis
- Domain-Adversarial Training of Neural Networks
- Translating and Segmenting Multimodal Medical Volumes with Cycle- and Shape-Consistency Generative Adversarial Network
- Coherent Online Video Style Transfer
- Temporal Attentive Alignment for Large-Scale Video Domain Adaptation