Skip to content
/ RFA Public

Extracts hidden correlations between high-dimensional datasets

Notifications You must be signed in to change notification settings

cbeny/RFA

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

52 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Relevant Feature Analysis

This is an example of implementation of the approach presented in https://arxiv.org/abs/1904.10387v3

example.jl is a minimal working example of RFA, which learns the correlations between two datasets. In this example it is applied to supervised classification on MNIST, but the code is designed so that it can be straighforwardly extended to the general unsupervised setting. This can be used to reproduce the top row of Fig. 1.

example.py is a tensorflow implementation of the above.

mnist.py does the same, but is slightly streamlined for supervised classification. In this setting, RFA boils down to a new loss function, coupled with an extra post-processing operation.

cifar.jl can be used to reproduce the bottow row of Fig. 1.

occlusion.ipynb applies RFA to reconstruct the occluded part of an image (ipython notebook using tensorflow). This can also be forked and run directly in Kaggle: https://www.kaggle.com/cedben/uncovering-correlations-in-mnist-with-rfa