Skip to content
forked from HITS-AIN/PINK

Parallelized rotation and flipping INvariant Kohonen maps

License

Notifications You must be signed in to change notification settings

RafaelMostert/PINK

 
 

Repository files navigation

Build Status Join the chat at https://gitter.im/HITS-AIN-PINK/Lobby

Parallelized rotation and flipping INvariant Kohonen maps (PINK)

SOM of radio-synthesis data taken from the Radio Galaxy Zoo

Requirements

  • C++ with ISO 14 standard (GCC 6.1 and above)
  • CMake >= 3.0
  • CUDA >= 9.1 (highly recommended because of the speed-up)
  • conan.io (optional for C++ dependencies) or
  • doxygen 1.8.13 (optional for developer documentation)

Conan.io will install automatically the C++ dependencies (PyBind11 and google-test). Otherwise you can also install these libraries yourself.

Installation

We provide deb- and rpm-packages at https://github.com/HITS-AIN/PINK/releases

or you can install PINK from the sources:

cmake -DCMAKE_INSTALL_PREFIX=<INSTALL_PATH> .
make install

Usage

To train a the self-organizing map (SOM) please execute

Pink --train <image-file> <result-file>

where image-file is the input file of images for the training and result-file is the output file for the trained SOM. All files are in binary mode described here.

To map an image to the trained SOM please execute

Pink --map <image-file> <result-file> <SOM-file>

where image-file is the input file of images for the mapping, SOM-file is the input file for the trained SOM, and result-file is the output file for the resulting heatmap.

Please use also the command Pink -h to get more informations about the usage and the options.

Python scripts

For conversion and visualization of images and SOM some python scripts are available.

  • convert_data_binary_file.py Convert binary data file from PINK version 1 to 2
  • data_precondition.py: Convert numpy into binary files and scale data to be within the range [0.0, 1.0]
  • show_heatmap.py: Visualize the mapping result
  • show_images.py: Visualize binary images file format
  • show_som.py: Visualize binary SOM file format
  • train.py: SOM training using the PINK Python interface

Benchmarks

The input data for the SOM training are radio-synthesis images of Radio Galaxy Zoo containing 176750 images of the dimension 124x124. The SOM layout is hexagonal of the dimension 21x21 which has 331 neurons (see image above). The size of the neurons is 64x64. The accuracy for the rotational invariance is 1 degree and the flip invariance is used.

CPU: 2x Intel Gold 5118 (12 cores in total)

GPU: 4x NVIDIA Tesla P40

PINK 1.1 Pink 2.0
CPU
CPU + 1x GPU 3069 909
CPU + 2x GPU 2069 636
CPU + 4x GPU 1891 858

Publication

Kai Lars Polsterer, Fabian Gieseke, Christian Igel, Bernd Doser, and Nikos Gianniotis. Parallelized rotation and flipping INvariant Kohonen maps (PINK) on GPUs. 24th European Symposium on Artificial Neural Networks, Computational Intelligence and Machine Learning (ESANN), pp. 405-410, 2016. pdf

License

Distributed under the GNU GPLv3 License. See accompanying file LICENSE or copy at http://www.gnu.org/licenses/gpl-3.0.html.

About

Parallelized rotation and flipping INvariant Kohonen maps

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Jupyter Notebook 82.9%
  • C++ 12.6%
  • Python 2.9%
  • Other 1.6%