LM-IGTD: a 2D image generator for low-dimensional and mixed-type tabular data to leverage the potential of convolutional neural networks

TL;DR

This paper introduces LM-IGTD, a novel method for transforming low-dimensional and mixed-type tabular data into images to leverage CNNs, improving predictive performance over traditional models in multiple datasets.

Contribution

The paper presents LM-IGTD, an innovative, automatic pipeline for converting tabular data into images with interpretability, enhancing CNN-based prediction accuracy on diverse datasets.

Findings

LM-IGTD outperformed traditional ML models on 5 out of 12 datasets.

CNNs trained on LM-IGTD images achieved comparable or better results than traditional models.

The approach enhances interpretability through feature-image mapping and post hoc analysis.

Abstract

Tabular data have been extensively used in different knowledge domains. Convolutional neural networks (CNNs) have been successfully used in many applications where important information about data is embedded in the order of features (images), outperforming predictive results of traditional models. Recently, several researchers have proposed transforming tabular data into images to leverage the potential of CNNs and obtain high results in predictive tasks such as classification and regression. In this paper, we present a novel and effective approach for transforming tabular data into images, addressing the inherent limitations associated with low-dimensional and mixed-type datasets. Our method, named Low Mixed-Image Generator for Tabular Data (LM-IGTD), integrates a stochastic feature generation process and a modified version of the IGTD. We introduce an automatic and interpretable end-to-end pipeline, enabling the creation of images from tabular data. A mapping between original features and the generated images is established, and post hoc interpretability methods are employed to identify crucial areas of these images, enhancing interpretability for predictive tasks. An extensive evaluation of the tabular-to-image generation approach proposed on 12 low-dimensional and mixed-type datasets, including binary and multi-class classification scenarios. In particular, our method outperformed all traditional ML models trained on tabular data in five out of twelve datasets when using images generated with LM-IGTD and CNN. In the remaining datasets, LM-IGTD images and CNN consistently surpassed three out of four traditional ML models, achieving similar results to the fourth model.