EvaGoNet: An integrated network of variational autoencoder and Wasserstein generative adversarial network with gradient penalty for binary classification tasks

Feature engineering is an effective method for solving classification problems. Many existing feature engineering studies have focused on image or video data and not on structured data. This study proposes EvaGoNet, which refines the decoder module of the Gaussian mixture variational autoencoder usi...

Celý popis

Uloženo v:
Podrobná bibliografie
Vydáno v:Information sciences Ročník 629; s. 109 - 122
Hlavní autoři: Luo, Changfan, Xu, Yiping, Shao, Yongkang, Wang, Zihan, Hu, Jianzheng, Yuan, Jiawei, Liu, Yuchen, Duan, Meiyu, Huang, Lan, Zhou, Fengfeng
Médium: Journal Article
Jazyk:angličtina
Vydáno: Elsevier Inc 01.06.2023
Témata:
ISSN:0020-0255, 1872-6291
On-line přístup:Získat plný text
Tagy: Přidat tag
Žádné tagy, Buďte první, kdo vytvoří štítek k tomuto záznamu!
Popis
Shrnutí:Feature engineering is an effective method for solving classification problems. Many existing feature engineering studies have focused on image or video data and not on structured data. This study proposes EvaGoNet, which refines the decoder module of the Gaussian mixture variational autoencoder using the Wasserstein generative adversarial network with gradient penalty (WGANgp) and embeds the top-ranked original features to update the latent features based on their discriminative powers. Comprehensive experiments show that EvaGoNet-encoded features outperform existing classifiers on 12 benchmark datasets, particularly on the small, imbalanced datasets col (accuracy = 0.8581), spe (accuracy = 1.0000), and leu (accuracy = 0.8021). EvaGoNet-engineered features improve binary classification task outcomes on six high-dimensional, imbalanced bioOMIC datasets. EvaGoNet achieves a medium-ranked training speed among the compared algorithms and considerably fast prediction speeds in the predictions of the testing samples. Therefore, EvaGoNet can be a candidate feature engineering framework for many practical applications that require one training procedure and many prediction tasks of the testing samples. EvaGoNet is implemented in Python TensorFlow and is available at https://healthinformaticslab.org/supp/resources.php.
ISSN:0020-0255
1872-6291
DOI:10.1016/j.ins.2023.01.133