Visualization transforms of non-spatial data for convolutional neural networks

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

1 Scopus citations

Abstract

Many datasets in important fields like healthcare and finance are often in a tabular format, where each observation is expressed as a vector of various feature values. While there exist several competitive algorithms such as random forests and gradient boosting, convolutional neural networks (CNNs) are making tremendous strides in terms of new research and applications. In order to exploit the power of convolution neural networks for these tabular datasets, we propose two vector-to-image transformations. One is a direct transformation, while the other is an indirect mechanism to first modulate the latent space of a trained generative adversarial network (GAN) with the observation vectors and then generate the images using the generator. On both simulated and real datasets, we show that CNNs trained on images based on our proposed transforms lead to better predictive performance compared to random forests and neural networks that are trained on the raw tabular datasets.

Original languageEnglish
Title of host publicationApplications of Machine Learning 2020
EditorsMichael E. Zelinski, Tarek M. Taha, Jonathan Howe, Abdul A. Awwal, Khan M. Iftekharuddin
PublisherSPIE
ISBN (Electronic)9781510638280
DOIs
StatePublished - 2020
Externally publishedYes
EventApplications of Machine Learning 2020 - Virtual, Online, United States
Duration: Aug 24 2020Sep 4 2020

Publication series

NameProceedings of SPIE - The International Society for Optical Engineering
Volume11511
ISSN (Print)0277-786X
ISSN (Electronic)1996-756X

Conference

ConferenceApplications of Machine Learning 2020
Country/TerritoryUnited States
CityVirtual, Online
Period08/24/2009/4/20

Keywords

  • Convolutional neural networks
  • GAN
  • Vector-to-image transform
  • Visualization

Fingerprint

Dive into the research topics of 'Visualization transforms of non-spatial data for convolutional neural networks'. Together they form a unique fingerprint.

Cite this