• Spectroscopy and Spectral Analysis
  • Vol. 41, Issue 2, 425 (2021)
Rong-sheng ZHU1、1、*, Tao SHEN1、1, Ying-li LIU1、1, Yan ZHU1、1, and Xiang-wei CUI1、1
Author Affiliations
  • 1[in Chinese]
  • 11. Faculty of Information Engineering and Automation, Kunming University of Science and Technology, Kunming 650504, China
  • show less
    DOI: 10.3964/j.issn.1000-0593(2021)02-0425-05 Cite this Article
    Rong-sheng ZHU, Tao SHEN, Ying-li LIU, Yan ZHU, Xiang-wei CUI. Wasserstein GAN for the Classification of Unbalanced THz Database[J]. Spectroscopy and Spectral Analysis, 2021, 41(2): 425 Copy Citation Text show less
    Terahertz spectra of three substances
    Fig. 1. Terahertz spectra of three substances
    WGAN generated data graphs under different iterations(a) is the original data; (b), (c), (d), (e), and (f) respectively represent the generated data after 1 000 iterations, 5 000 rounds, 10 000 rounds, 100 000 rounds, and 200 000 rounds
    Fig. 2. WGAN generated data graphs under different iterations
    (a) is the original data; (b), (c), (d), (e), and (f) respectively represent the generated data after 1 000 iterations, 5 000 rounds, 10 000 rounds, 100 000 rounds, and 200 000 rounds
    OriginalPredicted
    MaltotrioseMalthexaoseMaltoheptaose
    Maltotriose25192 440
    Malthexaose42 373323
    Maltoheptaose302 697
    Table 1. Confusion matrix of test database after training model with Database1
    OriginalPredicted
    MaltotrioseMalthexaoseMaltoheptaose
    Maltotriose2 35513332
    Malthexaose72 6894
    Maltoheptaose73222 371
    Table 2. Confusion matrix of test database after training model with Database2
    train/%validation/%test/%
    SVM91.6391.5765.69
    SVM-COPY89.1189.5484.04
    FWSVM83.6882.2385.77
    SVM-WGAN92.4591.1391.54
    Table 3. Comparison of the accuracy of the dataset under different algorithms
    train/%validation/%test/%
    Imbalance194.2292.0684.28
    Imbalance1_WGAN94.1391.2592.08
    Imbalance291.6391.5865.69
    Imbalance2_WGAN92.4591.1491.54
    Imbalance394.0793.9257.70
    Imbalance3_WGAN92.2191.8790.27
    Table 4. Compares the accuracy of the training set and test set of the dataset under different unbalance
    Rong-sheng ZHU, Tao SHEN, Ying-li LIU, Yan ZHU, Xiang-wei CUI. Wasserstein GAN for the Classification of Unbalanced THz Database[J]. Spectroscopy and Spectral Analysis, 2021, 41(2): 425
    Download Citation