Skip to content

yxjdarren/MvDSCN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

23 Commits
 
 
 
 
 
 

Repository files navigation

MvDSCN

🎲 Tensorflow Repo for "Multi-view Deep Subspace Clustering Networks"

The code repository for "Multi-View Deep Subspace Clustering Networks" (the paper has been accepted by T-CYB) in Tensorflow.

Overview

In this work, we propose a novel multi-view deep subspace clustering network (MvDSCN) by learning a multi-view self-representation matrix in an end to end manner. MvDSCN consists of two sub-networks, i.e., diversity network (Dnet) and universality network (Unet). A latent space is built upon deep convolutional auto-encoders and a self-representation matrix is learned in the latent space using a fully connected layer. Dnet learns view-specific self-representation matrices while Unet learns a common self-representation matrix for all views. To exploit the complementarity of multi-view representations, Hilbert Schmidt Independence Criterion (HSIC) is introduced as a diversity regularization, which can capture the non-linear and high-order inter-view relations. As different views share the same label space, the self-representation matrices of each view are aligned to the common one by a universality regularization.

MvDSCN

Requirements

  • Tensorflow
  • scipy
  • numpy
  • sklearn
  • munkres

Usage

  • Test by Released Result:
python main.py --test
  • Train Network with Finetune.

We have released the pretrain model in /pretrain folder, you can train it with finetune:

python main.py --ft
  • Pretrain Auoencoder From Scratch:

You re-train autoencoder from scarath:

python main.py

Releases

No releases published

Packages

No packages published

Languages