Title
Feature Consistency Training With JPEG Compressed Images
Abstract
AbstractDeep neural networks (DNNs) are recently found to be vulnerable to JPEG compression artifacts, which distort the feature representations of DNNs leading to serious accuracy degradation. Most existing training methods which aim to address this problem add compressed images to the training data to enhance the robustness of DNNs. However, their improvements are limited since these methods usually regard the compressed images as new training samples instead of distorted samples. The feature distortions between the raw images and the compressed images are not investigated. In this work, we propose a new training method, called Feature Consistency Training, that is designed to minimize the feature distortions caused by JPEG artifacts. At each training iteration, we simultaneously input a raw image and its compressed version with a randomly sampled quality into a DNN model and extract the features from the internal layers. By adding feature consistency constraint to the objective function, the feature distortions in the representation space are minimized in order to learn robust filters. Besides, we present a residual mapping block which takes the quality factor of the compressed image as an additional information to further reduce the feature distortion. Extensive experiments demonstrate that our method outperforms several existed training methods on JPEG compressed images. Furthermore, DNN models trained by our method are found to be more robust to unseen distortions.
Year
DOI
Venue
2020
10.1109/TCSVT.2019.2959815
Periodicals
Keywords
DocType
Volume
Image coding, Distortion, Training, Transform coding, Robustness, Quantization (signal), Feature extraction, Compression artifacts, deep neural network, JPEG compression, classification robustness
Journal
30
Issue
ISSN
Citations 
12
1051-8215
1
PageRank 
References 
Authors
0.38
7
5
Name
Order
Citations
PageRank
Sheng Wan140.76
Tung-Yu Wu2275.23
Heng-Wei Hsu340.76
Wing Hung Wong460796.45
Chen-Yi Lee51211152.40