The quest to learn image data distributions efficiently, particularly in environments constrained by computational and data resources, is a significant challenge in the field of computer vision. This paper introduces a modified version of the traditional Restricted Boltzmann Machine (RBM), termed the complex-valued RBM (cvRBM), which incorporates complex-valued inputs by utilizing a two-channel approach where the second channel is generated using a Sobel filter. The cvRBM aims to enhance the model’s capacity to handle image data by processing additional edge-detected information alongside original images. We evaluate the efficacy of cvRBM in comparison to the standard RBM on the MNIST dataset, focusing on their ability to reconstruct input data as measured by reconstruction error. The findings suggest that while cvRBM provides a richer representation of image content, it requires careful tuning to match the performance of traditional RBM models.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Complex-Valued Restricted Boltzmann Machines for Learning Image Data Distributions

  • Maciej Jaworski,
  • Agnieszka Niemczynowicz,
  • Radosław Kycia

摘要

The quest to learn image data distributions efficiently, particularly in environments constrained by computational and data resources, is a significant challenge in the field of computer vision. This paper introduces a modified version of the traditional Restricted Boltzmann Machine (RBM), termed the complex-valued RBM (cvRBM), which incorporates complex-valued inputs by utilizing a two-channel approach where the second channel is generated using a Sobel filter. The cvRBM aims to enhance the model’s capacity to handle image data by processing additional edge-detected information alongside original images. We evaluate the efficacy of cvRBM in comparison to the standard RBM on the MNIST dataset, focusing on their ability to reconstruct input data as measured by reconstruction error. The findings suggest that while cvRBM provides a richer representation of image content, it requires careful tuning to match the performance of traditional RBM models.