Residential College | false |
Status | 已發表Published |
Self-supervised Gaussian Restricted Boltzmann Machine via joint contrastive representation and contrastive divergence | |
Wang, Xinlei1; Chu, Jielei1; Yu, Hua2; Gong, Zhiguo3; Li, Tianrui1 | |
2024-09-05 | |
Source Publication | Knowledge-Based Systems |
ISSN | 0950-7051 |
Volume | 299Pages:112121 |
Abstract | In this paper, we propose a novel self-supervised Gaussian Restricted Boltzmann Machine with contrastive learning (CL-GRBM), which fuses contrastive representation learning and contrastive divergence to optimize and enhance the representation of GRBM. Built upon the concept of contrastive representation learning, CL-GRBM aims to enhance the representation capacity of the GRBM. Firstly, a pair of positive samples are constructed by one times Gibbs sampling with the original data. Then, the contrastive loss is used to pull the positive samples closer and push other samples further apart, making similar representations closer and different representations farther apart. In summary, during the training process of CL-GRBM using the CD algorithm, the objective is to align the sampling probability distribution of the visible layer in CL-GRBM as closely as possible with the empirical distribution of the original data. In the feature space of the sampling probabilities in the hidden layer, the distance between positive samples is minimized to capture the intrinsic structure of the data. According to the experimental verification, the proposed CL-GRBM shows better performance than contrastive models. As a shallow self-supervised model, it has even better performance than some excellent deep self-supervised models. |
Keyword | Clustering Contrastive Divergence Contrastive Representation Learning Representation Probability Distribution |
DOI | 10.1016/j.knosys.2024.112121 |
URL | View the original |
Language | 英語English |
Publisher | Elsevier B.V. |
Scopus ID | 2-s2.0-85196389364 |
Fulltext Access | |
Citation statistics | |
Document Type | Journal article |
Collection | Faculty of Science and Technology DEPARTMENT OF COMPUTER AND INFORMATION SCIENCE |
Corresponding Author | Chu, Jielei |
Affiliation | 1.School of Computing and Artificial Intelligence, Southwest Jiaotong University, Chengdu, 611756, China 2.China E-port Data Center Chengdu Branch, Chengdu, 610041, China 3.Department of Computer and Information Science, University of Macau, Macao |
Recommended Citation GB/T 7714 | Wang, Xinlei,Chu, Jielei,Yu, Hua,et al. Self-supervised Gaussian Restricted Boltzmann Machine via joint contrastive representation and contrastive divergence[J]. Knowledge-Based Systems, 2024, 299, 112121. |
APA | Wang, Xinlei., Chu, Jielei., Yu, Hua., Gong, Zhiguo., & Li, Tianrui (2024). Self-supervised Gaussian Restricted Boltzmann Machine via joint contrastive representation and contrastive divergence. Knowledge-Based Systems, 299, 112121. |
MLA | Wang, Xinlei,et al."Self-supervised Gaussian Restricted Boltzmann Machine via joint contrastive representation and contrastive divergence".Knowledge-Based Systems 299(2024):112121. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment