SMAE: A Split Masked Graph Autoencoder

Ruiting Wang,Enguang Zuo,Chen,Cheng Chen,Junyi Yan,Jie Zhong,Ziwei Yan,Xiaoyi Lv
DOI: https://doi.org/10.1109/icme57554.2024.10688216
2024-01-01
Abstract:Autoencoders, as a generative self-supervised learning, have received more and more attention in recent years in image, video, and other media-related information processing. However, Graph AutoEncoder (GAE) has yet to achieve the capability demonstrated by contrastive learning in the task-centered on attribute networks. The main limitation lies in the fact that traditional autoencoder architectures require pretext tasks that align with downstream tasks, resulting in limited expressive power of the encoder. In this paper, we propose a novel separable-task generative self-supervised learning framework capable of providing high-quality representations, Split Masked AutoEncoder (SMAE), which unleashes the encoder’s ability to extract representations through an intelligent design. Our approach focuses on unlocking the potential of the encoder by introducing encoding transfer and feature replacement strategies, thereby enabling self-supervised pretext tasks to achieve atomic separation and fully unleash the encoder’s feature representation potential. We conducted extensive experiments on widely-used graph classification datasets, and the results demonstrate that SMAE outperforms state-of-the-art baselines in terms of graph classification accuracy and generation quality. Furthermore, our experimental findings show that prediction at the representation layer is more effective than original graph layer reconstruction in the field of masked graph autoencoders.
What problem does this paper attempt to address?