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...

Ausführliche Beschreibung

Gespeichert in:
Bibliographische Detailangaben
Veröffentlicht in:Information sciences Jg. 629; S. 109 - 122
Hauptverfasser: Luo, Changfan, Xu, Yiping, Shao, Yongkang, Wang, Zihan, Hu, Jianzheng, Yuan, Jiawei, Liu, Yuchen, Duan, Meiyu, Huang, Lan, Zhou, Fengfeng
Format: Journal Article
Sprache:Englisch
Veröffentlicht: Elsevier Inc 01.06.2023
Schlagworte:
bioOMIC data
EvaGoNet
Feature engineering
Gaussian mixture VAE
VAE
WGANgp
WGANgp
VAE
Gaussian mixture VAE
Feature engineering
bioOMIC data
EvaGoNet
ISSN:0020-0255, 1872-6291
Online-Zugang:Volltext
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
Beschreibung
Zusammenfassung: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