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

Full description

Saved in:
Bibliographic Details
Published in:Information sciences Vol. 629; pp. 109 - 122
Main Authors: Luo, Changfan, Xu, Yiping, Shao, Yongkang, Wang, Zihan, Hu, Jianzheng, Yuan, Jiawei, Liu, Yuchen, Duan, Meiyu, Huang, Lan, Zhou, Fengfeng
Format: Journal Article
Language:English
Published: Elsevier Inc 01.06.2023
Subjects:
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 Access:Get full text
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary: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