Skip to main navigation Skip to search Skip to main content

The Structure Transfer Machine Theory and Applications

  • Baochang Zhang*
  • , Wankou Yang
  • , Ze Wang
  • , Lian Zhuo
  • , Jungong Han
  • , Xiantong Zhen
  • *Corresponding author for this work
  • Southeast University, Nanjing
  • University of Warwick
  • Inception Institute of Artificial Intelligence

Research output: Contribution to journalArticlepeer-review

Abstract

Representation learning is a fundamental but challenging problem, especially when the distribution of data is unknown. In this paper, we propose a new representation learning method, named Structure Transfer Machine (STM), which enables feature learning process to converge at the representation expectation in a probabilistic way. We theoretically show that such an expected value of the representation (mean) is achievable if the manifold structure can be transferred from the data space to the feature space. The resulting structure regularization term, named manifold loss, is incorporated into the loss function of the typical deep learning pipeline. The STM architecture is constructed to enforce the learned deep representation to satisfy the intrinsic manifold structure from the data, which results in robust features that suit various application scenarios, such as digit recognition, image classification and object tracking. Compared with state-of-the-art CNN architectures, we achieve better results on several commonly used public benchmarks.

Original languageEnglish
Article number8911371
Pages (from-to)2889-2902
Number of pages14
JournalIEEE Transactions on Image Processing
Volume29
DOIs
StatePublished - 2020

Keywords

  • Transfer learning
  • convolutional neural networks
  • learning theory
  • manifold loss

Fingerprint

Dive into the research topics of 'The Structure Transfer Machine Theory and Applications'. Together they form a unique fingerprint.

Cite this