An End-to-End Compression Framework Based on Convolutional Neural Networks

An End-to-End Compression Framework Based on Convolutional Neural Networks

Feng Jiang, Wen Tao, Shaohui Liu, Jie Ren, Xun Guo, Debin Zhao,  F. Jiang, W. Tao, S. Liu and D. Zhao are with the School of Computer Science and Technology, Harbin Institute of Technology, Harbin 150001, China (e-mail: shliu@hit.edu.cn).
Abstract

Deep learning, e.g., convolutional neural networks (CNNs), has achieved great success in image processing and computer vision especially in high level vision applications such as recognition and understanding. However, it is rarely used to solve low-level vision problems such as image compression studied in this paper. Here, we move forward a step and propose a novel compression framework based on CNNs. To achieve high-quality image compression at low bit rates, two CNNs are seamlessly integrated into an end-to-end compression framework. The first CNN, named compact convolutional neural network (ComCNN), learns an optimal compact representation from an input image, which preserves the structural information and is then encoded using an image codec (e.g., JPEG, JPEG2000 or BPG). The second CNN, named reconstruction convolutional neural network (RecCNN), is used to reconstruct the decoded image with high-quality in the decoding end. To make two CNNs effectively collaborate, we develop a unified end-to-end learning algorithm to simultaneously learn ComCNN and RecCNN, which facilitates the accurate reconstruction of the decoded image using RecCNN. Such a design also makes the proposed compression framework compatible with existing image coding standards. Experimental results validate that the proposed compression framework greatly outperforms several compression frameworks that use existing image coding standards with state-of-the-art deblocking or denoising post-processing methods.

Deep learning, compression framework, compact representation, convolutional neural networks (CNNs).

I Introduction

In recent years, image compression attracts increasing interest in image processing and computer vision due to its potential applications in many vision systems. The aim of image compression is to reduce irrelevance and redundancy of an image in order to store or transmit the image at low bit rates [wallace1992jpeg]. Traditional image coding standards[ghanbari2003standard] (such as JPEG and JPEG2000) attempt to distribute the available bits for every nonzero quantized transform coefficient in the whole image. While the compression ratio increases, the bits per pixel (BPP) decreases as a result of the use of bigger quantization steps, which will cause the decoded image to have blocking artifacts or noises. To overcome this problem, a lot of efforts have been devoted to improving the quality of the decoded image using a post-processing deblocking or denoising method. Zhai et al. [zhai2008efficient] propose an effective deblocking method for JPEG images through post-filtering in shifted windows of image blocks. Foi et al. [foi2007pointwise] develop an image deblocking filtering based on shape-adaptive DCT, in conjunction with the anisotropic local polynomial approximation-intersection of confidence intervals technique. Inspired by the success of nonlocal filters and bilateral filters for image debolcking, several nonlocal filters have been proposed for image deblocking [zhang2011image, francisco2012generic, wang2013adaptive]. Recently, Zhang et al. [zhang2016concolor] propose a constrained non-convex low-rank model for image deblocking. Although desired performance is achieved, these post-processing methods are very time-consuming because solving the optimal solutions involves computationally expensive iterative processes. Therefore, it is difficult to apply them to practical applications.


Fig. 1: Left: the JPEG-coded image (PSNR = 27.33 dB) with QF = 5 , where we could see blocking artifacts,ring effects and blurring on the eyes, abrupt intensity changes on the face. Right: the decoded image (PSNR = 31.14 dB) by our proposed compression framework at the same bit rate with left, where the compression artifacts vanished and generate more details.

Motivated by the excellent performance of convolutional neural networks (CNNs) in low level computer vision [dong2015compression, guo2016building, wang2016d3] in recent years and the fact that existing image codecs are extensively used across the world, we propose an end-to-end compression framework, which consists of two CNNs and an image codec as shown in Fig. 2. The first CNN, named compact convolutional neural network (ComCNN), learns an optimal compact representation from an input image, which is then encoded using an image codec (e.g., JPEG, JPEG2000 or BPG). The second CNN, named reconstruction convolutional neural network (RecCNN), is used to reconstruct the decoded image with high quality in the decoding end. Existing image coding standards usually consists of transformation, quantization and entropy coding. Unfortunately, the rounding function in quantization is not differentiable, which brings great challenges to train deep neural networks when performing the backpropagation algorithm. To address this problem, we present a simple but effective learning algorithm to train the proposed end-to-end compression framework by simultaneously learning ComCNN and RecCNN to facilitate the accurately reconstruction of the decoded image using RecCNN. An example of image compression is shown in Fig. 1, from which we can see that the proposed framework achieves much better quality with more visual details. In addition, as shown in Fig. 2, the compact representation obtained by ComCNN preserves the structural information of the image, therefore, an image codec can be effectively utilzed to compress the compact representation.


Fig. 2: The proposed novel compression framework. The ComCNN preserves more useful information for reconstruction. Meanwhile, the RecCNN handle the task of reconstructing the decoded image. represents an image codec, encoding and decoding the compact representation produced by the ComCNN, which consists of a transform coding stage followed by quantization and entropy coding. In this work, JPEG, JPEG2000 and BPG are adopted.

The contributions of this work are summarized as follows:

  1. We propose an end-to-end compression framework using two CNNs and an image codec. ComCNN produces a compact representation for encoding using an image codec. RecCNN reconstructs the decoded image, respectively. To our best knowledge, it is the first time to connect the existing image coding standards with CNNs using a compact intermediate representation.

  2. We further propose an effective learning algorithm to simultaneously learn two CNNs, which addresses the problem that the gradient can not be passed in the backpropagation algorithm since the rounding function in quantization is not differentiable.

  3. The proposed compression framework is compatible with existing image codecs (e.g. JPEG, JPEG2000 or BPG), which makes our method more applicable to other tasks.

The remainder of this paper is organized as follows. Section 2 presents a brief review of related work. Section 3 elaborates the proposed compression framework, including the architectures of ComCNN and RecCNN. Section 4 illustrates the training parameters setting and the solutions to train the ComCNN and RecCNN. Experimental results are also reported in Section 4. In Section 5, we conclude this paper.

Ii Related Work

Ii-a Image Deblocking and Artifacts Reduction

In the literature, there have been some methods proposed to improve the quality of decoded images using post-processing techniques, which can be roughly categorized into deblocking oriented and restoration oriented methods. The deblocking oriented methods focus on removing blocking and ringing artifacts of the decoded images. Yeh et al.[yeh2014self] propose a self-learning based post-processing method for image/video deblocking by formulating deblocking as a morphological component analysis based image decomposition problem. Yoo et al.[yoo2014post] propose a two-step framework for reducing blocking artifacts in different regions based on increment of inter-block correlation, which classifies the coded image into flat regions and edge regions. Liu et al.[liu2016data] learned sparse representations within the dual DCT-pixel domain, and achieved very promising results. Recently, Dong et al.[dong2015compression] propose a compact and efficient network (ARCNN) for seamless attenuation of different compression artifacts. Innovatively, D3[wang2016d3] and DDCN[guo2016building] integrate dual-domain sparse coding and the prior knowledge of JPEG compression, which achieve impressive results.

The restoration oriented methods regard the compression operation as a distortion process and reduce artifacts by restoring the clear images. Sun et al. [sun2007postprocessing] model the quantization distortion as Gaussian noises and use field of experts as image priors to restore the images. Zhang et al. [zhang2012reducing, zhang2013compression] propose to utilize similarity priors of image blocks to reduce compression artifacts by estimating the transform coefficients of overlapped blocks from non-local blocks. Recently, Zhang et al.[zhang2016concolor] develop a novel algorithm for image deblocking using a constrained non-convex low-rank model, which formulates image deblocking as an optimization problem within maximum a posteriori framework.

In the aforementioned methods, image prior models play important roles in both the deblocking oriented and restoration oriented methods. However, these methods involve computationally expensive iterative processes when solving the optimal solutions with complex formula derivations. Therefore, they may be not suitable for practical applications. In short, all the related methods reviewed above attempt to improve image quality only from the perspective of image post-processing. In other words, the connection between the encoder front-end processing and the decoder back-end processing is ignored. We attempt to jointly optimize the encoder and decoder joint optimization to improve the compression performance.

Ii-B Image Super-Resolution Based on Deep Learning

Recently, CNNs have been used successfully for image super-resolution (SR) especially when residual learning[he2015deep] and gradients-based optimization algorithms[duchi2011adaptive, zeiler2012adadelta, kingma2014adam] are proposed to train deeper network efficiently. Dong et al. propose a CNN based SR method[dong2016image] named SRCNN, which consists of three layers: patch extraction, non-linear mapping and reconstruction. Although Dong et al. conclude in their paper that deeper networks do not result in better performance in some cases, other researchers argue that increasing depth significantly boosts performance. For example, VDSR[kim2015accurate] shows a significant improvement in accuracy, which uses 20 weight layers. DRCN[kim2015deeply] has a very deep recursive layer (up to 16 recursions) and outperforms previous methods by a large margin.

Ii-C Image Compression Based on Deep Learning

Recently, deep learning has been used both for lossy and lossless image compression and achieved competitive performance. For the lossy image compression, Toderici et al.[toderici2015variable] propose a general framework for variable-rate image compression and a novel architecture based on convolutional and deconvolutional LSTM recurrent networks. Further, Toderici et al.[toderici2016full] proposed a neural network which is competitive across compression rates on images of arbitrary size. For a given compression rate, both methods learn the compression models by minimizing the distortion. Theis et al.[theis2017lossy] propose compressive autoencoders, which uses a smooth approximation of the discrete of the rounding function and upper-bound the discrete entropy rate loss for continuous relaxation. Ballé et al.[balle2016end] make use a generalized divisive normalization (GDN) for joint nonlinearity and replace rounding quantization with additive uniform noise for continuous relaxation. Li et al.[li2017learning] proposed a content-weighted compression method with the importance map of image. For the lossless image compression, the methods proposed by Theis et al.[theis2015generative] and van den Oord et al.[oord2016pixel] achieves state-of-the-art results.

Overall, although the image compression methods based on deep learning achieve competitive performance, they ignored the compatibility with existing image codecs, which limits their use in some existing systems. In contrast, the proposed compression framework takes into account both compression performance and compatibility with existing image codecs.

Iii The Proposed Compression Framework

In this section, we first introduce the architecture of the proposed compression framework and then present the detailed learning algorithm.

Iii-a Architecture of End-to-End Compression Framework

As shown in Fig. 2, the proposed compression framework consists of two CNNs and an image codec. The compact representation CNN (ComCNN) is used to generate a compact representation of the input image for the encoding, which preserves structural information of the image and therefore facilitates the accurate reconstruction of high-quality images. The reconstruction CNN (RecCNN) is used to enhance the quality of the decoded image. These two CNNs collaborate with each other and are optimized simultaneously to achieve high-quality image compression at low bit rates.

Iii-A1 Compact Representation Convolutional Neural Network (ComCNN)

As shown in Fig. 3, ComCNN has 3 weight layers, which maintain the spatial structure of the original image and therefore facilitate the accurate reconstruction of the decoded image using RecCNN111We have tried deeper networks to obtain better performance, but only negligible improvements at the expense of a lot of training time and costs. . The combination of convolution and ReLU [krizhevsky2012imagenet] is used in ComCNN. The first layer is used to perform patch extraction and representation which extracts overlapping patches from the input image. Let represents the number of image channels. A total of 64 filters of size are used to generate 64 feature maps and the ReLU nonlinearity is utilized as an activation function. The second layer has two significant intentions: downscaling and enhancing the features, which are implemented by convolutional operations with setting the stride to 2. The sizes of 64 filters are and ReLU is also applied. For the last layer, c filters of size are utilized to construct the compact representation.


Fig. 3: The architecture of the proposed ComCNN and feature maps of different layers.Upscaled image is obtained by Bicubic interpolation on Decoded image.

Iii-A2 Reconstruction Convolutional Neural Network (RecCNN)

As shown in Fig.4, RecCNN is composed of 20 weight layers, which have three types of layer combinations: Convolution + ReLU, Convolution + Batch Normalization[ioffe2015batch] + ReLU and Convolution. For the first layer, 64 filters of size are used to generate 64 feature maps, followed by ReLU. For the 2nd to 19th layers, 64 filters of size are used, and batch normalization is added between convolution and ReLU. For the last layer, c filters of size are used to reconstruct the output. Residual learning and batch normalization are applied to speed up the training process and boost the performance. The compressed image is upsampled to the original image size using bicubic interpolation.


Fig. 4: The architecture of the proposed RecCNN and feature maps of different layers. Upscaled image is obtained by bicubic interpolation on decoded image. The network predicts a residual image, then the sum of the input and the residual gives the high quality output.

Iii-B Learning Algorithm

According to the proposed architecture, both ComCNN and RecCNN try to make the reconstructed image as similar as possible to the original image. Therefore, the end-to-end optimization goal can be formulated as

(1)

where represents the original image, and are the parameters of ComCNN and RecCNN, respectively. and represent the ComCNN and RecCNN, respectively. represents an image codec (e.g. JPEG, JPEG2000 or BPG). From this objective function, we can see that an original image passes through the compression pipeline, including ComCNN, image codec and RecCNN, and finally outputs the reconstructed image . Such a process is an end-to-end compression.

Unfortunately, the in Eq.(1) involves a rounding function, which is not differentiable when performing the back propagation algorithm. To solve this problem, we designed an iterative optimization learning algorithm. By fixing , we can get

(2)

and by fixing , we can obtain

(3)

Iii-B1 Updating the Parameters of RecCNN

According to the network topology, an auxiliary variables is introduced and defined as the decoded compact representation of , which can be formulated as

(4)

After combining Eq.(4) and Eq.(3), we can obtain

(5)

Iii-B2 Updating the Parameters of ComCNN

From Eq.(2), we can see that it is not a trivial task to obtain the optimal since the is an inherently non-differentiable operation when performing back propagation. To solve this problem, we define an auxiliary variable as the optimal input of RecCNN:

(6)

Here we make a reasonable and general assumption that is monotonic with respect to , which can be expressed as

, if and only if

(7)

Let be the solution of , i.e., for any possible , it satisfies that

(8)

Following assumption (7), we can obtain that

(9)

Accordingly,

(10)

Combining with Eq.(2), we can get , which is

(11)

Since is an codec, a reasonable solution of Eq.(12) is

(12)

Combine Eq.(13) and the assumption (7) above, it arrives

(13)

Here, we get Eq.(13) with a reasonable assumption and rigorous derivations, which is the approximation of Eq.(2). In this paper, we use Eq.(13) to train ComCNN instead of Eq.(2).

We can obtain the optimal and by iteratively optimizing Eq.(5) and Eq.(13), respectively. In light of all derivations above, the complete description of the proposed algorithm is given in Algorithm 1.

1:Input: The original image x
2:Initialization:Random initial and
3:for  do
4:     Update by computing Eq.(2)
5:     for  do
6:         Update by training RecCNN to compute Eq.(5)
7:     end for
8:     for  do
9:         Update by training ComCNN to compute
10:         Eq.(13)
11:     end for
12:end for
13:Return: , and
Algorithm 1 The Proposed Compression Framework for Training Sub-Networks

Iii-C Loss Functions

Iii-C1 For ComCNN training

Given a set of original images and trained parameters , we use mean squared error (MSE) as the loss function

(14)

where and represents the batch size and the trainable parameter, respectively.

Iii-C2 For RecCNN training

Having obtained a set of compact representation from ComCNN and original images , we use MSE as the loss function:

(15)

where represents the trainable parameter. represents the residual learned by RecCNN. Clearly, it looks somewhat different from Eq.(5), but they are not contradictory. Actually, they are essentially identical, and Eq.(15) is just expresses Eq.(5) as the form of the residual.

Iv Experiments

To evaluate the performance of the proposed compression framework, we conduct experimental comparisons against standard compression methods (e.g., JPEG, JPEG 2000 and BPG) with a post-processing deblocking or denoising method. Five representative image deblocking methods, i.e. Sun’sš[sun2007postprocessing], DicTV[chang2014reducing], Zhang’s[zhang2013compression], Ren’sš[ren2013image], Zhang’s[zhang2016concolor] and two representative image denoising methods, i.e. BM3D[dabov2007image] and WNNM[gu2014weighted] are chosen due to their state-of-the-art performance. Moreover, ARCNN[dong2015compression] is also chosen since it is a landmark deblocking method based on deep learning and achieves the state-of-the-art performance. Meanwhile, in order to demonstrate the effectiveness of ComCNN, we remove ComCNN in the framework and just using RecCNN to reconstruct the decoded image. Similarly, we remove the RecCNN to examine the effect of ComCNN using bicubic interpolation to obtain the reconstructed image of the same size as the original image. The results of all the compared methods are obtained by running the source codes of the original authors with the optimal parameters. Through this section, we use the name of the post-processing method to denote a compared method.

We use the MatConvNet package[vedaldi2015matconvnet] to train the proposed networks. All experiments are carried out in the Matlab (R2015b) environment running on a computer with Inter(R) Xeon(R) CPU E5-2670 2.60GHz and an Nvidia Tesla K40c GPU. It takes about 28 hours and one day to train the ComCNN and RecCNN on GPU, respectively.

Iv-a Datasets for Training and Testing

Following[chen2015trainable], we use 400 images of size for training. A total of 204800 ()222 For each image, there are 8 augmentations and 64 patches of size extracted. patches are sampled with a stride of 20 on the training images as well as their augmentations (flip and rotate with different angles). Our experiments indicate that using a larger training set can only bring negligible performance improvements. For testing, we use 7 images as shown in Fig.5 , which are widely used in the literature. Please note that all the test images are not included in the training dataset.


Fig. 5: The used test images.

Iv-B Model Initialization

We initialize the weights of ComCNN using the method in [he2015delving] and use Adam algorithm [kingma2014adam] with and . We train ComCNN for 50 epochs using a batch size of 128. The learning rate is decayed exponentially from 0.01 to 0.0001 for 50 epochs. The weights initialization and gradient updating of RecCNN is the same as ComCNN. RecCNN is also trained for 50 epochs using the same batch size with ComCNN. The learning rate is decayed exponentially from 0.1 to 0.0001 for 50 epochs.

Iv-C Experimental Results

In our experiments, we set different quality factors (QF) to achieve the same bits per pixel (bpp) for both the proposed and compared methods. For the proposed method, we first manually adjust the QF for the compression of compact representation by JPEG to achieve almost the same bpp with the compared image enhancement methods. Then we compare the PSNR and SSIM of the proposed method with the compared methods. The comparison results for all test images with QF = 5 and QF = 10 are provided in Table I and Table II, respectively, where the best results are highlighted in bold.

Test Images Butterfly Carmerman House Lena Peppers Leaves Parrots Average
PSNR
JPEG 22.58 24.45 27.77 27.33 27.17 22.49 26.19 25.43
Sun’s[sun2007postprocessing] 23.83 25.25 29.09 28.87 29.05 23.47 27.45 26.72
Zhang’s[zhang2013compression] 24.20 25.39 29.24 29.00 29.07 24.13 27.78 26.97
Ren’s[ren2013image] 24.58 25.46 29.66 29.07 29.07 24.56 27.87 27.18
BM3D[dabov2007image] 24.05 25.27 29.21 28.63 28.52 24.02 27.33 26.72
DicTV[chang2014reducing] 23.10 24.54 28.45 28.07 27.95 23.01 26.83 25.99
WNNM[gu2014weighted] 24.75 25.49 29.62 28.95 28.99 24.68 27.80 27.18
Zhang’s[zhang2016concolor] 25.30 25.61 30.12 29.51 29.61 24.99 28.27 27.63
ARCNN[dong2015compression] 25.64 25.27 29.68 29.31 29.02 25.07 28.13 27.45
ComCNN 23.05 24.93 29.17 29.46 29.33 23.19 27.67 26.69
RecCNN 25.99 26.33 30.13 29.63 29.81 25.21 28.52 28.02
Proposed 26.23 26.53 31.45 31.14 30.84 25.52 30.12 28.83
SSIM
JPEG 0.7378 0.7283 0.7733 0.7367 0.7087 0.7775 0.7581 0.7456
Sun’s[sun2007postprocessing] 0.8321 0.7687 0.8113 0.8061 0.7931 0.8380 0.8323 0.8104
Zhang’s[zhang2013compression] 0.8313 0.7672 0.8141 0.8035 0.7895 0.8548 0.8308 0.8130
Ren’s[ren2013image] 0.8419 0.7666 0.8197 0.8010 0.7876 0.8720 0.8310 0.8171
BM3D[dabov2007image] 0.8184 0.7607 0.8082 0.7837 0.7639 0.8510 0.8118 0.7997
DicTV[chang2014reducing] 0.7769 0.6658 0.7963 0.7744 0.7456 0.8104 0.8005 0.7671
WNNM[gu2014weighted] 0.8445 0.7674 0.8178 0.7947 0.7827 0.8749 0.8287 0.8158
Zhang’s[zhang2016concolor] 0.8667 0.7666 0.8285 0.8169 0.8031 0.8882 0.8460 0.8308
ARCNN[dong2015compression] 0.8741 0.7674 0.8209 0.8142 0.7961 0.8983 0.8446 0.8308
ComCNN 0.7488 0.7662 0.8119 0.8042 0.7966 0.7885 0.8377 0.7934
RecCNN 0.8760 0.7945 0.8251 0.8195 0.8004 0.8803 0.8497 0.8351
Proposed 0.8847 0.8167 0.8456 0.8486 0.8328 0.8912 0.8951 0.8535
TABLE I: JPEG: QF = 5, PSNR (dB) and SSIM results of all competitive algorithms grayscale image deblocking and denoising.
Test Images Butterfly Carmerman House Lena Peppers Leaves Parrots Average
PSNR
JPEG 25.24 26.47 30.56 30.41 30.14 25.40 28.96 28.17
Sun’s[sun2007postprocessing] 26.52 27.26 32.00 31.72 31.62 26.60 30.04 29.39
Zhang’s[zhang2013compression] 26.83 27.45 32.11 31.92 31.68 27.26 30.50 29.68
Ren’s[ren2013image] 27.17 27.43 32.41 31.92 31.63 27.59 30.34 29.78
BM3D[dabov2007image] 26.64 27.25 32.07 31.77 31.42 26.98 30.05 29.45
DicTV[chang2014reducing] 26.09 26.92 31.77 31.55 31.29 26.33 29.82 29.11
WNNM[gu2014weighted] 27.22 27.40 32.42 31.93 31.64 27.66 30.33 29.80
Zhang’s[zhang2016concolor] 27.09 27.72 33.04 32.19 31.94 28.20 30.66 30.12
ARCNN[dong2015compression] 28.54 27.62 32.53 32.05 31.50 28.31 30.62 30.16
ComCNN 26.32 27.05 31.82 31.63 31.04 26.51 29.97 29.12
RecCNN 28.04 27.33 32.76 32.35 31.34 28.53 30.85 30.17
Proposed 28.60 27.44 33.25 33.11 31.83 28.77 31.11 30.59
SSIM
JPEG 0.8325 0.7965 0.8183 0.8183 0.7839 0.8609 0.8336 0.8206
Sun’s[sun2007postprocessing] 0.8871 0.8358 0.8504 0.8590 0.8322 0.9138 0.8783 0.8652
Zhang’s[zhang2013compression] 0.8923 0.8329 0.8513 0.8597 0.8317 0.9212 0.8804 0.8671
Ren’s[ren2013image] 0.9010 0.8259 0.8526 0.8571 0.8300 0.9309 0.8775 0.8679
BM3D[dabov2007image] 0.8896 0.8240 0.8492 0.8549 0.8250 0.9207 0.8749 0.8626
DicTV[chang2014reducing] 0.8699 0.8046 0.8484 0.8559 0.8244 0.9032 0.8741 0.8544
WNNM[gu2014weighted] 0.9019 0.8248 0.8531 0.8571 0.8303 0.9325 0.8775 0.8681
Zhang’s[zhang2016concolor] 0.9142 0.8401 0.8609 0.8661 0.8358 0.9406 0.8842 0.8774
ARCNN[dong2015compression] 0.9237 0.8389 0.8591 0.8711 0.8434 0.9495 0.8942 0.8828
ComCNN 0.8796 0.8154 0.8497 0.8573 0.8296 0.9095 0.8766 0.8597
RecCNN 0.9192 0.8429 0.8584 0.8679 0.8396 0.9443 0.8884 0.8801
Proposed 0.9245 0.8448 0.8678 0.8838 0.8532 0.9475 0.9047 0.8895
TABLE II: JPEG: QF = 10, PSNR (dB) and SSIM results of all competitive algorithms grayscale image deblocking and denoising.

As seen from Table I and Table II, in the case of QF = 5, the proposed compression framework achieves 1.20dB gains in PSNR and 0.0227 gains in SSIM compared against Zhang’s [zhang2016concolor], which is state-of-the-art in the compared methods. It is worth mentioning that the proposed framework outperforms all the compared image enhancement methods including ARCNN[dong2015compression], which is a milestone based on CNN. Meanwhile, in the case of QF = 10, the proposed compression framework achieves 0.43dB and 0.0067 gains in PSNR and SSIM, respectively, compared against ARCNN[dong2015compression]. The visual quality comparisons in the case of QF = 5 for Lena is provided in Fig. 6. We can see that the blocking artifacts are obvious in the image decoded directly by the standard JPEG. DicTV[chang2014reducing], Sun’s[sun2007postprocessing], WNNM[gu2014weighted] and BM3D[dabov2007image] remove the artifacts partially, but there are still some artifacts visible in the reconstructed image. Zhang’s [zhang2013compression] and Ren’s [ren2013image] generate better results than Sun’s[sun2007postprocessing] and BM3D[dabov2007image]. However, the blur effects along the edges are generated at the same time. Zhang’s [zhang2016concolor] achieves a better PSNR and SSIM, but it makes the image over-smoothing and discards some details in image edges. ARCNN[dong2015compression] and ReCNN achieve better visual quality than other compared methods. The proposed compression framework not only removes most of the artifacts significantly, but also preserves more details on both edges and textures than all the compared methods. In order to verify the effect of ComCNN, we remove ComCNN and use RecCNN alone to reconstruct the decoded image. Similarly, we remove RecCNN and only use ComCNN and bicubic interpolation to examine the effect of RecCNN. As shown in Table I and Table II, worse performances are obtained only with ComCNN or RecCNN. In addition, we show examples of the compact representation produced by ComCNN in Fig.7. It can be seen that the compact representation maintains the structural information of the original image, but it is different from traditional down sampling methods. In a nutshell, both ComCNN and RecCNN play key roles in the proposed compression framework. Due to the collaboration of ComCNN and RecCNN, the compact representation preserves more useful information for the final image reconstruction. Our testing codes are available in GitHub 333https://github.com/compression-framework/compression_framwork_for_tesing.


Fig. 6: Visual quality comparison of image deblocking for Lena in the case of QF = 5. From left to right and top to bottom: original image, JPEG(PSNR = 27.33 dB, SSIM = 0.7367), the deblocking results by Sun’s (PSNR = 28.87 dB, SSIM = 0.8061), Zhang’s (PSNR = 29.00 dB, SSIM = 0.8035), Ren’s (PSNR = 29.07 dB, SSIM = 0.8010), BM3D (PSNR = 28.63 dB, SSIM = 0.7837), DicTV (PSNR = 28.07 dB, SSIM = 0.7744), WNNM (PSNR = 28.95 dB, SSIM = 0.7947), Zhang’s (PSNR = 29.31 dB, SSIM = 0.8169), ARCNN (PSNR = 29.31 dB, SSIM = 0.8142), RecCNN (PSNR = 29.63 dB, SSIM = 0.8195) and the proposed framework (PSNR = 31.14 dB, SSIM = 0.8486)

.


Fig. 7: Visual quality comparison of the down sampling using bicubic interpolation and the compact representation produced by ComCNN.

We also evaluate our framework on JPEG 2000 and BPG (Better Portable Graphics)444F. Bellard, The BPG Image Format, http://bellard.org/bpg/ and achieve excellent performance. BPG compression is based on the High Efficiency Video Coding (HEVC), which is considered as a major advance in compression techniques. For JPEG2000, we test the proposed compression framework at different bit rates (from 0.1bpp to 0.4bpp) and compare it with JPEG 2000. Table III presents the comparison results with JPEG 2000. It can seen that our framework significantly outperforms JPEG2000 on all test bit-rates of all test images in terms of both PSNR and SSIM. For bpp from 0.1 to 0.4, the proposed framework achieves on average 3.06dB, 2.45dB, 1.34dB, 1.09dB and 0.1047, 0.0709, 0.0525, 0.0435 gains in PSNR and SSIM compared against JPEG2000. In Fig. 8, one can see that the proposed compression framework achieves much better subjective performance than JPEG2000, especially at very low bit rate. Our framework preserves more high-frequency information and recovers sharp edges and pure textures in the reconstructed image. For BPG, we test the BPG codec at QP (quality parameter) = 43 and 47. Further, we keep the bit-rates of the proposed compression framework almost the same for each image. The results are shown in Table IV. One can see that, if we treat RecCNN as a post-processing method, RecCNN achieves on average 0.81dB and 0.0168 gains in PSNR and SSIM. And the proposed compression framework achieves on average 0.99dB and 0.0218 gains in PSNR and SSIM while saving 5.22% bit-rates. It is worth noting that the performance of our proposed compression framework on BPG is not so obvious on JPEG and JPEG2000 when compared, because BPG is already a very good compression method, which might not be significantly improved further.


Fig. 8: Subjective performance comparison of JPEG2000 when the bit rate from 0.1bpp to 0.4bpp for Parrot. From left to right and top to bottom, the corresponding PSNR(in dB) and SSIM values are (26.05, 0.7853), (31.22, 0.8895), (29.96, 0.8589), (32.72, 0.9242), (32.42, 0.8897), (33.48, 0.9382), (34.42, 0.9150) and (35.09, 0.9480).
  Test Images
Rate(bpp)
\@killglue
0.1 0.2 0.3 0.4
JPEG2000 Proposed JPEG2000 Proposed JPEG2000 Proposed JPEG2000 Proposed
PSNR
Butterfly 18.92 23.66 21.99 26.73 24.01 27.47 25.44 29.43
Cameraman 23.33 26.66 26.23 28.44 28.30 29.35 29.80 30.41
House 28.02 30.54 31.92 34.34 34.04 35.30 35.35 35.76
Lena 29.88 31.07 32.98 34.68 34.81 35.65 36.13 36.53
Parrots 26.05 31.22 29.96 32.72 32.42 33.48 34.42 35.09
Peppers 29.66 31.07 32.56 33.43 34.11 34.49 35.02 35.46
Average 25.98 27.37 29.27 31.72 31.28 32.62 32.69 33.78
SSIM
Butterfly 0.5888 0.8308 0.7257 0.8866 0.7968 0.8894 0.8419 0.9266
Cameraman 0.6764 0.8238 0.7659 0.8667 0.8146 0.8955 0.8482 0.9026
House 0.7702 0.8310 0.8375 0.8777 0.8662 0.8989 0.8823 0.9122
Lena 0.8176 0.8492 0.8763 0.9002 0.8973 0.9183 0.9143 0.9320
Parrots 0.7853 0.8895 0.8589 0.9242 0.8897 0.9382 0.9150 0.9480
Peppers 0.7851 0.8273 0.8368 0.8733 0.8591 0.8988 0.8719 0.9127
Average 0.7273 0.8419 0.8169 0.8878 0.8540 0.9065 0.8789 0.9224
TABLE III: JPEG2000: PSNR (dB) and SSIM results of all competitive algorithms grayscale image deblocking and denoising.
Test Images BPG BPG + RecCNN ComCNN + BPG + RecCNN
Size(Bytes) PSNR/SSIM Size(Bytes) PSNR/SSIM Size(Bytes) PSNR/SSIM Bits Saving
Cameraman 752 25.95/0.7723 752 26.55/0.7867 642 26.70/0.7887 14.63%
1251 28.12/0.8188 1251 28.77/0.8288 1105 28.85/0.8325 11.67%
House 389 29.06/0.8023 389 29.61/0.8121 384 30.21/0.8274 1.29%
592 31.35/0.8349 592 32.02/0.8433 524 32.31/0.8489 11.49%
Lena 1616 28.79/0.7896 1616 29.20/0.8016 1585 29.47/0.8066 1.92%
2737 30.97/0.8351 2737 31.46/0.8456 2702 31.74/0.8529 1.28%
Butterfly 1277 24.43/0.8335 1277 25.79/0.8691 1203 25.65/0.8648 5.79%
1999 26.86/0.8875 1999 28.03/0.9069 1934 28.09/0.9076 3.25%
Peppers 1797 28.93/0.7759 1797 29.59/0.7944 1722 29.75/0.7978 4.17%
2747 30.81/0.8108 2747 31.38/0.8220 2699 31.43/0.8287 1.75%
Leaves 1651 24.55/0.8670 1651 25.88/0.9046 1564 25.97/0.9090 5.27%
2492 27.17/0.9195 2492 28.71/0.9405 2423 28.83/0.9476 2.77%
Parrots 621 27.94/0.8235 621 28.61/0.8389 595 28.85/0.8483 4.19%
1017 30.25/0.8562 1017 30.95/0.8680 980 31.19/0.8716 3.64%
Average - 28.23/0.8305 - 29.04/0.8473 - 29.22/0.8523 5.22%
TABLE IV: BPG: PSNR (dB) and SSIM Results of BPG, BPG + RecCNN and the Proposed Method .

In order to show the effectiveness of the proposed compression framework, we test our method on Set5[kim2015accurate], Set14[kim2015accurate], LIVE1[sheikh2005live] and General-100[dong2016accelerating] datasets. It is worth mentioning that the General-100 dataset contains 100 bmp-format images with no compression, which are very suitable for compression task. Results are shown in Table VI, from which we can see that the performance of our proposed compression exceeds JPEG and JPEG2000 by a larger margin for all four testing datasets.

Iv-D Running Time

The running time of all compared methods when dealing with a grayscale image in CPU or GPU are shown in Table V. It should be noted that it is not possible to test the running time in GPU for all other compared methods. As we can see from Table V, the proposed framework needs only 1.56s and 0.017s in CPU and GPU, respectively. Our compression framework is faster than other post-processing methods. In addition, we also calculate the running time of our sub-network RecCNN, which almost takes the entire running time of our method. Because RecCNN has 20 layers, which is much deeper than ComCNN with only 3 layers.

Sun’s[sun2007postprocessing] Zhang’s[zhang2013compression] Ren’s[ren2013image] BM3D[dabov2007image] DicTV[chang2014reducing] WNNM[gu2014weighted] Zhang’s[zhang2016concolor] RecCNN Proposed
132/- 251/- 36/- 3.40/- 53/- 230/- 442/- 1.50/0.015 1.56/0.017
TABLE V: Running time (s) of compared methods in CPU (/GPU) tested on a grayscale image.
JPEG
DataSets Quality JPEG Proposed
Factors PSNR/SSIM PSNR/SSIM
Set5 5 26.13/0.7206 29.20/0.8387
10 28.99/0.8109 31.40/0.8854
Set14 5 24.90/0.6686 26.89/0.7914
10 27.49/0.7762 28.91/0.8336
LIVE1 5 24.60/0.6666 26.78/0.7934
10 27.02/0.7720 28.84/0.8411
Genreal-100 5 25.93/0.7228 27.29/0.8447
10 28.92/0.8199 30.16/0.8767
JPEG2000
DataSets Rate(bpp) JPEG2000 Proposed
PSNR/SSIM PSNR/SSIM
Set5 0.1 26.09/0.7176 29.00/0.8163
0.2 29.11/0.8157 31.94/0.8859
0.3 31.06/0.8629 33.32/0.9086
Set14 0.1 25.23/0.6554 27.88/0.7911
0.2 27.80/0.7534 28.58/0.8273
0.3 29.57/0.8063 30.42/0.8867
LIVE1 0.1 25.39/0.6612 27.14/0.7478
0.2 27.58/0.7478 28.54/0.8183
0.3 29.19/0.7991 30.07/0.8544
General-100 0.1 26.45/0.7179 27.82/0.8048
0.2 29.88/0.8153 30.86/0.8761
0.3 32.00/0.8639 32.88/0.9083
TABLE VI: Average PSNR(dB)/SSIM results of JPEG and the proposed method for quality factors 5 and 10, JPEG2000 and the proposed method for bpp 0.1, 0.2, 0.3 for Set5, Set14, LIVE1 and General-100

V Conclusion

In this paper, we propose an effective end-to-end compression framework based on two CNNs, one of which is used to produce compact intermediate representation for encoding using an image encoder. The other CNN is used to reconstruct the decoded image with high quality. These two CNNs collaborate each other and are trained using a unified optimization method. Experimental results demonstrate that the proposed compression framework achieves state-of-the-art performance and is much faster than most post-processing algorithms. Our work indicates that the performance of the proposed compression framework can be significantly improved by applying the proposed framework, which will inspire other researchers to design better deep neural networks for image compression along this orientation.

Acknowledgment

This work is partially funded by the Major State Basic Research Development Program of China (973 Program 2015CB351804), the Science and Technology Commission of China No.17-H863-03-ZT-003-010-01 and the Natural Science Foundation of China under Grant No. 61572155 and 61672188.

References

Feng Jiang received the B.S., M.S., and Ph.D. degrees in computer science from Harbin Institute of Technology (HIT), Harbin, China, in 2001, 2003, and 2008, respectively. He is now an Associated Professor in the Department of Computer Science, HIT and a visiting scholar in the School of Electrical Engineering, Princeton University. His research interests include computer vision, image and video processing and pattern recognition.

Wen Tao received the B.S. degrees in computer science from Harbin Institute of Technology(HIT), Harbin, China, in 2016. He is now working towards the M.S. degree at School of Computer Science and Technology, HIT. His current research interests are in image processing and computer vision.

Shaohui Liu , Associate professor, has got a Bachelor of Science Degree in Computational Mathematics and Its application Software Development, a Master of Science Degree in Computational Mathematics of Science and a Doctor of Philosophy Degree in Computer Application Technology from Harbin Institute of Technology, Harbin, P. R. China respectively in 1999, 2001 and 2007. He is a senior member of CCF. Dr. Liu was a visiting professor and post-doctor at Sejong University in South Korea, and a Visiting Scholar at University of Missouri Columbia in America. His research mainly includes image and video processing and analysis, computer vision, multimedia security. In the related fields, he has coauthored more than 80 papers which are cited more than 1000 times totally.

Jie Ren received his B.S. degree in 2015 from University of Electronic Science and Technology of China; now, he is a master in Harbin Institute of Technology. His main research interest includes image processing and multimedia compression technology.

Xun Guo received his Ph.D. degree in computer science from Harbin Institute of Technology, China, in 2007. From 2007 to 2012, he was with MediaTek Inc. as group manager, where he led a research team and worked mostly on video compression, especially on technology development for HEVC standard. He joined Microsoft Research Asia in 2012, where he is now a lead researcher. His research interests include video coding and processing, multimedia system and computer vision.

Debin Zhao (M’11) received the B.S., M.S., and Ph.D. degrees in computer science from Harbin Institute of Technology (HIT), Harbin, China, in 1985, 1988, and 1998, respectively. He is now a Professor in the Department of Computer Science, HIT. He has published over 200 technical articles in refereed journals and conference proceedings in the areas of image and video coding, video processing, video streaming and transmission, and pattern recognition.

Comments 0
Request Comment
You are adding the first comment!
How to quickly get a good reply:
  • Give credit where it’s due by listing out the positive aspects of a paper before getting into which changes should be made.
  • Be specific in your critique, and provide supporting evidence with appropriate references to substantiate general statements.
  • Your comment should inspire ideas to flow and help the author improves the paper.

The better we are at sharing our knowledge with each other, the faster we move forward.
""
The feedback must be of minimum 40 characters and the title a minimum of 5 characters
   
Add comment
Cancel
Loading ...
49506
This is a comment super asjknd jkasnjk adsnkj
Upvote
Downvote
""
The feedback must be of minumum 40 characters
The feedback must be of minumum 40 characters
Submit
Cancel

You are asking your first question!
How to quickly get a good answer:
  • Keep your question short and to the point
  • Check for grammar or spelling errors.
  • Phrase it like a question
Test
Test description