arrow
Return

A conditional variational autoencoder based self-transferred algorithm for imbalanced classification

delete2021-04-01
delete20
PRE
AI
Y
Yudi Zhao
K
Kuangrong Hao *
唐雪嵩 (Xue‐song Tang)
L
Lei Chen
B
Bing Wei
DOI:10.1016/j.knosys.2021.106756delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
In this paper, we propose a conditional variational autoencoder-based self-transferred (CVAE_SeTred) algorithm to solve the highly imbalanced classification problem, where the training instances of the minority classes are rare. Our method belongs to an over-sampling technique that utilizes variational autoencoders (VAEs) to generate training samples for the minority classes. Traditional over-sampling methods mainly rely on minority classes themselves, our approach exploits the information from both the majority and minority classes and aims to transfer instructional knowledge from the majority classes to the minority classes, where the majority and minority classes are analogized as the self-transferred (SeTred) source and target domain, respectively. Specifically, our model comprises two encoders, one decoder, and one domain classifier and can simultaneously conduct distribution learning, SeTred learning, image generation, and dataset rebalancing in a joint and unified framework. The proposed method can not only learn domain-invariant and multivariate Gaussian distributed latent variables but also generate discriminative samples for the minority class according to designated labels. We verify the effectiveness of the CVAE_SeTred model on both imbalanced datasets constructed from benchmark datasets and a more challenging real-world industrial application, such as imbalanced classification for fabric defects. Experimental results indicate that our method outperforms other comparative methods and can generate samples with better diversity. (C) 2021 Elsevier B.V. All rights reserved.
Keywords:
Highly imbalanced classification
Over-sampling
Variational autoencoders
Knowledge transfer
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

K
Knowledge-Based Systems
IF:
7.6
Papers:
1.2W
Citations:
4.5W

Organization

D
Donghua University
Scholars:
2.0W
Papers: 1.4W
Citations: 2.9W
Cited Papers

Cited Papers

Altered expression of synaptotagmin 13 mRNA in adult mouse brain after contextual fear conditioning
err2012-09-01
err0
PREAI
errSeungrie Han; Soontaek Hong; Dongmin Lee; Myeong-hoe Lee; June-seek Choi; Min Jung Koh; Woong Sun; Hyun Kim; Hyun Woo Lee
errShare
errSave
errShare
errSave
Arsenic removal from aqueous solutions by adsorption using novel MIL-53(Fe) as a highly efficient adsorbent
err2015-01-01
err0
PREAI
errTuan. A. Vu; Giang. H. Le; Canh. D. Dao; Lan. Q. Dang; Kien. T. Nguyen; Quang. K. Nguyen; Phuong. T. Dang; Hoa. T. K. Tran; Quang. T. Duong; Tuyen. V. Nguyen; Gun. D. Lee
errShare
errSave
Coherent Two-Electron Spin Qubits in an Optically Active Pair of Coupled InGaAs Quantum Dots
err2012-09-06
err0
PREAI
errK. M. Weiss; J. M. Elzerman; Y. L. Delley; J. Miguel-Sanchez; A. Imamoğlu
errShare
errSave
errShare
errSave
Transfer Boosting With Synthetic Instances for Class Imbalanced Object Recognition
err2018-01-01
err59
PREAI
errZhang, Xuesong; Zhuang, Yan; Wang, Wei; Pedrycz, Witold
errShare
errSave
Transendocardial Delivery of AAV6 Results in Highly Efficient and Global Cardiac Gene Transfer in Rhesus Macaques
err2011-08-01
err0
PREAI
errGuangping Gao; Lawrence T. Bish; Meg M. Sleeper; Xin Mu; Lan Sun; You Lou; Jiachuan Duan; Chunyan Hu; Li Wang; H. Lee Sweeney
errShare
errSave
researcher View more