Discriminative Auto-Encoder With Local and Global Graph Embedding

IEEE Access

Jie Lai¹  Lei Lei¹  Rui Li¹  Xiaodan Wang²  Yafei Song² 

[1] College of Air and Missile Defense, Air Force Engineering University, Xi&x2019;an, China;

关键词: Manifold learning;    graph embedding;    discriminative auto-encoder;    deep learning;   

DOI  :  10.1109/ACCESS.2020.2972132

来源: DOAJ

【 摘 要 】

In order to exploit the potential intrinsic low-dimensional structure of the high-dimensional data from the manifold learning perspective, we propose a global graph embedding with globality-preserving property, which requires that samples should be mapped close to their low-dimensional class representation data distribution centers in the embedding space. Then we propose a novel local and global graph embedding auto-encoder(LGAE) to capture the geometric structure of data, its cost function have three terms, a reconstruction loss to reproduce the input data based on the learned representation, a local graph embedding regularization to enforce mapping the neighboring samples close together in the embedding space, a global embedding regularization to enforce mapping samples close to their low-dimensional class representation distribution centers. Thus in the learning process, our LGAE can map samples from same class close together in the embedding space, as well as reduce the scatter within-class and increase the margin between-class, it will also detect the local and global intrinsic geometric structure of data and discover the latent discriminant information in the embedding space. We build stacked LGAE for classification tasks and conduct comprehensive experiments on several benchmark datasets, the results confirm that our proposed framework can learn discriminative representation, speed up the network convergence process, and significantly improve the classification performance.

【 授权许可】

Unknown