| Literature DB >> 31319556 |
Yuantao Chen1, Jiajun Tao1, Jin Wang2,3, Xi Chen1, Jingbo Xie4, Jie Xiong5, Kai Yang6.
Abstract
To address the problem of unstable training and poor accuracy in image classification algorithms based on generative adversarial networks (GAN), a novel sensor network structure for classification processing using auxiliary classifier generative adversarial networks (ACGAN) is proposed in this paper. Firstly, the real/fake discrimination of sensor samples in the network has been canceled at the output layer of the discriminative network and only the posterior probability estimation of the sample tag is outputted. Secondly, by regarding the real sensor samples as supervised data and the generative sensor samples as labeled fake data, we have reconstructed the loss function of the generator and discriminator by using the real/fake attributes of sensor samples and the cross-entropy loss function of the label. Thirdly, the pooling and caching method has been introduced into the discriminator to enable more effective extraction of the classification features. Finally, feature matching has been added to the discriminative network to ensure the diversity of the generative sensor samples. Experimental results have shown that the proposed algorithm (CP-ACGAN) achieves better classification accuracy on the MNIST dataset, CIFAR10 dataset and CIFAR100 dataset than other solutions. Moreover, when compared with the ACGAN and CNN classification algorithms, which have the same deep network structure as CP-ACGAN, the proposed method continues to achieve better classification effects and stability than other main existing sensor solutions.Entities:
Keywords: CP-ACGAN; auxiliary classifier generative adversarial networks (ACGAN); feature matching; generative adversarial networks (GAN); image classification
Year: 2019 PMID: 31319556 PMCID: PMC6679324 DOI: 10.3390/s19143145
Source DB: PubMed Journal: Sensors (Basel) ISSN: 1424-8220 Impact factor: 3.576
Figure 1The original GAN network structure [9].
Figure 2The ACGAN network structure [16].
Figure 3The generator structure of ACGAN [16].
Figure 4Discriminator structure using an ACGAN.
Figure 5The CP-ACGAN network structure.
Figure 6The discriminator structure of the CP-ACGAN.
Figure 7Classification effect of different methods on the MNIST training dataset.
Figure 8Testing accuracy comparison of different methods on the MNIST testing dataset.
Best prediction accuracy of different methods on MNIST, CIFAR10, and CIFAR100.
| Model |
|
|
|
|---|---|---|---|
| Mean Pooling | 0.9951 | 0.7796 | 0.4594 |
| Maximum Pooling | 0.9943 | 0.7639 | 0.4283 |
|
| 0.9950 | 0.7306 | 0.3989 |
|
| 0.9962 | 0.7907 | 0.4803 |
MNIST prediction accuracy mean value, maximum value, minimum value and variance of different methods in 1000 testing epochs.
| Model | Mean Value | Maximum Value | Minimum Value | Variance |
|---|---|---|---|---|
| Mean Pooling | 0.9949 | 0.9949 | 0.9865 | 6.0 × 10−9 |
| Maximum Pooling | 0.9941 | 0.9941 | 0.9865 | 3.3 × 10−9 |
|
| 0.9940 | 0.9945 | 0.9835 | 4.1 × 10−7 |
|
| 0.9956 | 0.9961 | 0.9890 | 1.9 × 10−7 |
Figure 9Comparison of different methods on the MNIST testing dataset.
Figure 10Average classification accuracy of different methods on the CIFAR10 training dataset.
Figure 11Average testing accuracy comparison on the CIFAR10 testing dataset.
CIFAR10 prediction accuracy. Mean value, maximum value, minimum value and variance of different methods in 1000 tests.
| Model | The Number of Testing Samples | Mean Value | Maximum Value | Minimum Value | Variance |
|---|---|---|---|---|---|
| Mean Pooling | 200 | 0.7654 | 0.7765 | 0.5705 | 3.96 × 10−6 |
| 600 | 0.7665 | 0.7775 | 0.5800 | 3.88 × 10−6 | |
| 1000 | 0.7670 | 0.7795 | 0.5883 | 3.79 × 10−6 | |
| Maximum Pooling | 200 | 0.7517 | 0.7605 | 0.5800 | 3.88 × 10−6 |
| 600 | 0.7535 | 0.7645 | 0.5890 | 3.76 × 10−6 | |
| 1000 | 0.7550 | 0.7690 | 0.5960 | 3.67 × 10−6 | |
|
| 200 | 0.7196 | 0.7305 | 0.5700 | 5.88 × 10−5 |
| 600 | 0.7203 | 0.7335 | 0.5780 | 5.76 × 10−5 | |
| 1000 | 0.7220 | 0.7365 | 0.5890 | 5.69 × 10−5 | |
|
| 200 | 0.7682 | 0.7850 | 0.6700 | 3.28 × 10−5 |
| 600 | 0.7699 | 0.7885 | 0.6790 | 3.19 × 10−5 | |
| 1000 | 0.7715 | 0.7905 | 0.6899 | 3.14 × 10−5 |
Figure 12Average classification accuracy of different methods on the CIFAR100 training dataset.
Figure 13Average testing accuracy comparison on the CIFAR100 testing dataset.
CIFAR100 prediction accuracy. Mean value, maximum value, minimum value and variance of different methods in 1000 Tests.
| Model | The Number of Testing Samples | Mean Value | Maximum Value | Minimum Value | Variance |
|---|---|---|---|---|---|
| Mean Pooling | 200 | 0.6752 | 0.7105 | 0.4000 | 4.92 × 10−6 |
| 600 | 0.6792 | 0.7185 | 0.4500 | 4.88 × 10−6 | |
| 1000 | 0.6810 | 0.7252 | 0.4800 | 4.75 × 10−6 | |
| Maximum Pooling | 200 | 0.6514 | 0.6750 | 0.3900 | 5.25 × 10−6 |
| 600 | 0.6560 | 0.6790 | 0.4200 | 5.18 × 10−6 | |
| 1000 | 0.6600 | 0.6830 | 0.4500 | 5.15 × 10−6 | |
|
| 200 | 0.6188 | 0.6480 | 0.4200 | 4.05 × 10−5 |
| 600 | 0.6212 | 0.6505 | 0.4400 | 4.00 × 10−5 | |
| 1000 | 0.6258 | 0.6595 | 0.4500 | 3.96 × 10−5 | |
|
| 200 | 0.7028 | 0.7300 | 0.4700 | 3.75 × 10−5 |
| 600 | 0.7088 | 0.7380 | 0.4900 | 3.70 × 10−5 | |
| 1000 | 0.7122 | 0.7455 | 0.5022 | 3.64 × 10−5 |