L ANGUAGE M ODELS ARE R EALISTIC T ABULAR D ATA G ENERATORS
Anonymous authors
Paper under double-blind review
A BSTRACT
Tabular data is among the oldest and most ubiquitous forms of data.
However, the generation of synthetic samples with the original data’s characteristics remains a significant challenge for tabular data.
While many generative models from the computer vision domain, such as variational autoencoders or generative adversarial networks, have been adapted for tabular data generation, less research has been directed towards recent transformer-based large language models (LLMs), which are also generative in nature.
To this end, we propose GReaT ( G eneration of Rea listic T abular data), which exploits an auto-regressive generative LLM to sample synthetic and yet highly realistic tabular data.
Furthermore, GReaT can model tabular data distributions by conditioning on any subset of features; the remaining features are sampled without additional overhead.
We demonstrate the effectiveness of the proposed approach in a series of experiments that quantify the validity and quality of the produced data samples from multiple angles.
We find that GReaT maintains state-of-the-art performance across numerous real-world and synthetic data sets with heterogeneous feature types coming in various sizes.
I NTRODUCTION
Tabular data is one of the most common forms of data in machine learning (ML) – over 65% of data sets in the Google Dataset Search platform contain tabular files in either CSV or XLS formats (Benjelloun et al., 2020).
However, due to the expensive nature of data collection processes, tabular data sets (i) are often class imbalanced, (i.e., tabular data sets tend to have long-tailed label distributions (Cao et al., 2019)), (ii) contain critical person-related information and cannot be shared due to privacy protection or socio-ethical principles (Gasc´on et al., 2017), and (iii) often come with impurity issues such as noisy or missing values which impede the application of modern ML algorithms
(Lin & Tsai, 2020).
Synthetically generated data has the potential to alleviate these three important issues.
Therefore, the generation of realistic artificial tabular data has received considerable attention in recent years (Choi et al., 2017; Park et al., 2018;
Xu et al., 2019; Borisov et al., 2021).
Apart from real-world impurity issues, there also exist various technical problems that make the generation of synthetic data difficult.
Typically, tabular data contains various feature types, such as categorical features (e.g., name , countryOfOrigin , jobTitle ) and numerical features (e.g., age , income ).The
categorical variables (i.e., words or clauses) may frequently contain most of the information.
For example, the highly used Adult Income data set consists of seven numerical and eight categorical variables (Dua & Graff, 2017).
This heterogeneity of feature types and values leads to three core challenges in tabular data preprocessing and modeling:
Extensive and lossy preprocessing.
For most of the existing tabular data generation methods, extensive data preprocessing of tabular data is required, which usually includes the following steps: (i) categorical data encoding into numbers, (ii) data scaling or normalization, (iii) replacing missing values, and (iv) removing outliers and smoothing.
These data transformation steps may result in the loss of important information or the introduction of artifacts that are not present in the original data.
As an example, the categorical encoding into numeric values may introduce an artificial ordering into previously unordered values (Borisov et al., 2021).
Therefore, the problem of lossy preprocessing may strongly influence the quality of generated data as a result.
Context knowledge for coherent semantics.
Almost all common synthetic data generation methods transform tabular data into a fully numerical representation.
However, tabular data sets fre-
[Figure]
quently consist of variables that are contextually interconnected.
In the Adult Income data set (Dua & Graff, 2017), the features age , marital-status , and education have a clear coherence relationship: There is a certain minimal legal age for marriage, and it is challenging to get a Ph.D. at a young age.
Such context knowledge should ideally be considered when generating realistic synthetic data samples.
We refer to this common issue as the contextual knowledge problem .
Arbitrary conditioning.
A versatile model that can generate data for a large variety of applications should be able to synthesize data conditioned on an arbitrary set of variables.
This allows imputation of any missingness pattern in the data or oversampling of arbitrary subsets.
Currently, the majority of the methods do not provide extensions to arbitrary conditioning and require the generative model to be re-trained according to each specific set of features to be conditioned on (Mirza & Osindero, 2014).
We refer to generators that allow for conditional generation with any specified feature combination as supporting arbitrary conditioning .
Most modern deep-learning approaches for tabular data generation build on generative models transferred from the computer vision domain (Borisov et al., 2021), such as Variational Autoencoders (VAEs, Kingma & Welling, 2013) or Generative Adversarial Networks (GANs, Goodfellow et al., 2014).
However, deep learning models have equally revolutionized the field of natural language processing (NLP, Radford et al., 2019; ?
; Raffel et al., 2020).
Modern large language models (LLMs) are often constructed in the form of auto-regressive density models over sequences of words (Radford et al., 2019; Bengio et al., 2000).
This begs the question to which extent successful architectures for NLP are apt to the tabular data generation task.
Carrying this thought further, we present a novel method for probabilistic data generation that covers the outlined core challenges and results in state-of-the-art performance (see Fig.
1 for a qualitative example).
We argue that pretrained self-attention-based LLMs (Vaswani et al., 2017) are suitable for the probabilistic modeling of heterogeneous tabular data sets after these data sets have been appropriately transformed into a textual representation .
We do so by constructing syntactically correct sentences based on feature names and row values without losing information or introducing artificial orderings and thus mitigate the issue of lossy preprocessing .
This step, to which we refer as textual encoding , maintains substantially more information than usual transformations.
Since we include the variable names in the encoding, a model trained on this data can directly access contextual information.
To be able to make sense of this information, we suggest using established and pretrained language models to perform the generation task.
This could be a possible path towards tackling the contextualization problem .
Finally, we introduce a feature order permutation step to shuffle the feature order in the textual encodings.
Training a model on such data will result in a versatile generator that supports arbitrary conditioning .
Specifically, our work offers the following contributions relative to the existing literature on the generation of synthetic tabular data:
• Novel paradigm.
We propose the first approach for realistic heterogeneous tabular data modeling and generation utilizing a transformer-decoder network architecture.
Thereby, we connect tabular and textual data modalities via a textual encoding scheme .
• Arbitrary conditioning.
When trained on textual encodings with random feature order permutations, our model inherits its arbitrary conditioning power from the LLM, which can model the data distribution conditioned on any given subset of features and sample the remaining features.
• Extensive experimental results.
We show that our G eneration of Rea listic T abular Data (GReaT) obtains state-of-the-art generative performance on a variety of data sets across several measures.
We have open-sourced our experimental results, making them available as strong benchmarks for the benefit of the community.
• Python package.
Finally, we are also going to provide an easy-to-use Python implementation of the GReaT model, where it takes only three lines of code to generate new synthetic samples.
The code can be found in the supplementary materials.
R ELATED W ORK
While the generation of images and text is extensively explored (Karras et al., 2020; Subramanian et al., 2017), the generation of synthetic tabular is less commonly considered in the recent machine learning literature.
Classical methods for tabular data modeling include Bayesian networks, in particular those based on the Chow-Liu approximation (Chow & Liu, 1968) or statistical tools such as copulas (Kamthe et al., 2021).
More recent methods for probabilistic tabular data modeling utilize generative adversarial networks (Choi et al., 2017; Park et al., 2018; Mottini et al., 2018; Xu et al., 2019; Koivu et al., 2020) or variational autoencoders (Xu et al., 2019; Ma et al., 2020; Vardhan &
Kok, 2020; Darabi & Elor, 2021).
The mixed structure of discrete and continuous features along with their different value distributions still poses a significant challenge.
CTGAN, the current state-of-the-art approach by Xu et al.
(2019), places special focus on the conditional distributions between the features to generate semantically meaningful data.
For non-Gaussian feature distributions, the authors propose a mode-specific normalization technique.
However, as we pointed out in Section 1, such encoding techniques can be lossy and introduce artificial orderings.
Furthermore, they do not profit from contextual information.
We thoroughly compare our method to their approach in the experimental evaluation section.
In a parallel development, the area of natural language processing was dominated by recurrent neural networks (RNNs) before self-attention-based neural networks (Vaswani et al., 2017) revolutionized this field.
Based on their technique, various auto-encoding models (Devlin et al., 2018; Sanh et al., 2019; Lan et al., 2019) for tasks like sentence classification, sequence-to-sequence models (Raffel et al., 2020; Lewis et al., 2020) for translation or summarizing, and auto-regressive models (Radford et al., 2019; ? ) for natural language generation tasks showed the strength of the self-attention mechanism.
In the light of these successes, transformer-based models have also been devised for tabular data classification (Arik & Pfister, 2019; Somepalli et al., 2021; Kossen et al., 2021) and learning joint representations of tabular and textual data (Yin et al., 2020).
Padhi et al.
(2021) investigated the generation of multi-variate time series data, using a transformer architecture based on BERT (Devlin et al., 2018).
However, there are no prior works on generating highly realistic non-sequential tabular data with the help of attention -based LLMs.
Our work is the first to rigorously explore this path, which leads to state-of-the-art performance.
[Figure]
GR EA T: G ENERATION OF R EA LISTIC T ABULAR D ATA
This section presents the GReaT approach for fully-conditional tabular data generation using transformer-based neural networks.
GReaT consists of two major stages: (1) the fine-tuning of a pretrained large language model (LLM) on a textually encoded tabular data set as shown in Fig.and (2) sampling from the fine-tuned LLM to generate synthetic tabular data.
We illustrate the sampling procedure in Fig.
In the following, we describe each component of the fine-tuning and sampling steps in detail.
We conclude this section with a brief summary of our approach.
3. GR EA T FINE - TUNING
Textual encoding.
Standard pretrained generative LLMs expect sequences of words as inputs.
Hence, to apply an LLM on tabular data, we have to convert each row of our data set into a textual representation.
To this end, we propose a textual subject-predicate-subject encoding scheme.
Definition 1 (Textual encoder)
Given a tabular data set of m columns with feature names f 1 , f 2 , . . .
, f m and n rows of samples s 1 , . . .
, s n , we let the entry v i,j , i ∈ { 1 , ..., n } , j ∈ { 1 , ..., m } represent the value of the j -th feature of the i -th data point.
Taking the feature name and value into account, each sample s i of the table is transformed into a textual representation t i using the following subject-predicate-object transformation:
[Equation]
where t i,j , the textually encoded feature, is a clause with information about a single value and its corresponding feature name, and [ · ] denotes the concatenation operator.
Remark 1.
For the scope of this work, we treat the target variable as a regular feature in our formulations.
Fig.
2 (upper panels) shows an illustrative example of the proposed encoding technique.
This result could be a sequence like “Occupation is doctor, Gender is female, Age is 34,” , which requires minimal preprocessing and does not suffer from information loss.
While in standard natural language sequences the word order is crucial, in our case there is no preferred order between the individual features.
Random feature order permutation.
By transforming a tabular feature vector into a sequence using the textual subject-predicate-object encoding scheme, pseudo-positional information is artificially introduced into the transformed tabular data sample.
However, naturally there is no spacial
[Figure]
In comparison to other state-of-the-art approaches, GReaT allows arbitrary conditioning on feature subsets without model retraining, i.e., the can be performed by conditioning on any feature name or combination of ordering relationship between features in tabular data sets (Borisov et al., 2021).
To reconstruct the feature order independence, we randomly permute the encoded short sentences t i,j of the full textual representation t i by permutations k .
Definition 2 (Feature order permutation function)
Formally, the result of applying a random feature order permutation k to t i , where each k j ∈ { 1 , ..., m } is arbitrary and k j ̸ = k j ′ for j ̸ = j ′ , is defined as
[Equation]
When using shuffled orders of the textually encoded features, we fine-tune our generative language model on samples without order dependencies .
Moreover, using permutations of the above form is highly beneficial as they allow for arbitrary conditioning in tabular data generation, which is further discussed in the following Subsection 3.2.
Fine-tuning a pretrained auto-regressive language model.
Finally, we describe the fine-tuning procedure of a pretrained LLM to the encoded tabular data for generation tasks.
We suppose a textually encoded tabular data set T =
{ t i ( k i ) } i =1 ,...,n that was transformed into text by the proposed encoding scheme.
Let k i be randomly drawn permutations and n denote the number of rows.
To be processed with an LLM, the input sentences t ∈ T need to be encoded into a sequence of tokens from a discrete and finite vocabulary W .
These tokens can be character, word or subword encodings such as the Byte-Pair-Encodings (Sennrich et al., 2015).
Thus, t ∈ T is represented by a sequence of tokens ( w 1 , . . . , w j ) =
TOKENIZE ( t ) with tokens w 1 , . .
. , w j ∈ W , where j denotes the number of tokens required to describe the character sequence t .
Commonly, the probability of natural-language sequences is factorized in an auto-regressive manner in LLMs (Jelinek, 1980;
Bengio et al., 2000).
It is represented as a product of output probabilities conditioned on previously observed tokens,
[Equation]
In this formulation, it becomes evident that LLMs formally need to be highly capable predictors for follow-up tokens given an arbitrary-length sequence of preceding tokens.
Indeed, the model is trained to output a probability distribution over possible next tokens w k from an input sequence w 1 , ..., w k − 1 of arbitrary length.
The entire model is usually fitted by optimizing the parameters to maximize the probability (cid:81) t ∈T p ( t ) of the entire training data set.
As a result, an end-user can choose any existing generative language model for tabular data modeling and exploit the vast amount of contextual knowledge present in these models (Roberts et al.,
For instance, in generative transfomer-decoder LLM architectures (e.g., the GPT models (Radford et al., 2018; 2019; ? )) word-embeddings are obtained from large corpus of text (e.g., GPT3 model trained on 45 TB of textual data ( ? )).
By learning from such large collection of data, Transformers are able to build robust contextualized representations of language (Liu et al., 2021).
Fine-tuning enables the model to leverage this contextual information in combination with the feature and category names to boost the models capabilities in a manner that is similar to transfer learning by learning bidirectional representations (Raffel et al., 2020).
3. GR EA T SAMPLING OF SYNTHETIC DATA
Having obtained a fine-tuned auto-regressive model q of the textual training data set that returns a categorical output distribution z
= q ( w 1 , . . .
, w k − 1 ) over possible follow up tokens for an input sequence w 1 , . . .
, w k − 1 , we can apply several sampling strategies.
Usually, the next token ω is sampled by weighted choice sampling with a temperature parameter T > 0 from the output z of the
LLM,
[Equation]
We note that the auto-regressive paradigm offers the possibility of sampling from token distributions p ( w k +1: j | w 1: k ) with arbitrary preconditioning w 1: k .
When we use random feature order permutations at train time, we can also start the textual sequence with any possible combination of features and values at inference time.
As a result, the GReaT method is particularly flexible and could possibly be used in a variety of real-world problems such as missing value imputation (Kachuee et al., 2020) or generation of realistic counterfactual explanations (Pawelczyk et al., 2020).
Moreover, the sampling of the conditional distribution comes at no additional cost.
Sampling and extraction of synthetic tabular data.
Therefore, the setup provides several ways to sample new tabular data points using the GReaT method.
We initialize the model with certain conditions and let the LLM sample the remaining tokens to complete the feature vector (in its textual representation).
In total, we propose three options of preconditioning:
• Feature Name Preconditioning.
Only a feature name, but no value is provided as an initial condition.
This type of conditioning is able to generate samples from the entire joint data distribution p ( V 1 , . .
, V n ) , where V denotes the random variables representing the n features in the data set.
• Name-Value Pair Preconditioning.
In this case, a single feature name and a value are provided.
Starting from this input, GReaT will complete the sample.
This approach will generate samples from the distribution p ( V \{ i } | V i
= v i ) .
Because modeling a single feature distribution is usually tractable (by frequentist estimation for categorical features or by fitting a parametric density estimator for continuous features), we can first sample a value for V i and then apply Name-Value Pair Preconditioning to sample the remaining features.
Thereby, we can also sample the entire data distribution.
• Multiple Name-Value Pair Preconditioning.
We can also provide multiple Name-Value pairs V
i 1 = v
i 1 , . .
, V i k
= v i k as precondition to the LLM to realize arbitrary conditioning.
By providing the textual encoding of this condition, we are able to sample from the distribution of the remaining features effectively p ( V \{ i 1 ,...,i k } | V
i 1 = v i 1 , . .
, V i k
= v i k ) .
We illustrate the GReaT sampling possibilities in Fig.
3 and Fig. 8 that underline the high flexibility of the proposed approach for synthetic data generation.
We apply commonly accepted patternmatching algorithms using regular expressions to convert the generated textual feature representations back to a tabular format (Aho, 1991).
We dismiss the respective samples in the rare case where the required format is violated.
Generation rates of invalid samples were monitored and found to be consistently below 1 %.
The main cause of this rare event lies in the generation procedure of the pre-trained LLM, which chooses the next token based on a probability distribution.
For example, the Occupation variable in the Adult dataset has a specific support, e.g., { “Adm-clerical”, “Execmanagerial”, “Prof-specialty”, ... } .
However, in rare cases, the pre-trained LLM might leave this support and output, for example, ‘Adm clerical’ (without dash), or mix up these categories and return ‘Adm-managerial’.
To lower the chances of this unwanted effect, we suggest reducing the sampling
[Figure]
l
[Figure]
s
[Figure]
e s
[Figure]
s
[Figure]
temperature T (Eq. 5), which sharpens the probability distribution.
In practice, a simple validation function can quickly reject those infrequently invalid samples.
A brief summary of the strengths of the GReaT method.
GReaT comes with several significant advantages over related approaches:
It (i) allows the end-user to have full probabilistic control over the sampling procedure by its arbitrary conditioning power; it (ii) utilizes the knowledge from large text data bases to obtain a better representation that includes context knowledge; (iii) the proposed approach is easy to use, since it does not require a preprocessing of the data values.
There is no need to specify discrete or numerical variables beforehand and the information loss due to preprocessing is therefore kept at its bare minimum.
E XPERIMENTAL E VALUATION
In this section, we empirically demonstrate the performance of the proposed GReaT approach using multiple qualitative and quantitative experiments.
Lastly, for better reproducibility, we provide information on the packages and parameters for the selected LLMs.
Data sets.
For the evaluation of the proposed algorithm, we utilize six real-world data sets that come from various domains.
They also come in different sizes, reaching from less than 1,000 to over 100,000 samples.
We also consider three synthetic data sets with varying numbers of features.
Key characteristics of each data set are presented in Table 7.
We split all data sets into 80% train and 20% test sets to avoid any data leakage.
All models are trained or fine-tuned on the same training data samples.
To demonstrate the power of our synthetic data generation framework to work out-ofthe-box with real-world data, we apply zero data preprocessing, e.g., feature names and values are used as they are presented in original data sets.
In contrast, other baselines require extensive data preprocessing of the data.
Baselines.
As baselines for our experiments, we use three deep learning-based methods.
CTGAN (Xu et al., 2019) is based on a generative adversarial network (GAN) (Goodfellow et al., 2014) for tabular data that allows to condition the generation process on only a single discrete feature.
The same authors proposed TVAE (Xu et al., 2019), a variational autoencoder (VAE) for tabular data.
The CopulaGAN model from the Synthetic Data Vault (SDV) framework (Patki et al., 2016) is applying the Gaussian copulas to simplify the underlying CTGAN.
We analyze the main properties of the chosen baselines in Tab. 10.
We compare those baselines to the proposed method with two different pretrained transformerdecoder LLM models of various sizes.
The smaller, distilled version of GPT-2 (Sanh et al., 2019) has 82 million learned parameters.
We term the corresponding tabular data generator Distill-GReaT.
An original version of GPT-2 by Radford et al.
(2019) has over 355 million trainable parameters – we refer to this version simply as GReaT.
A description of the architecture of the selected generative
LLMs can be found in Table 8.
We apply name-value pair preconditioning to start sampling.
For the evaluation of synthetic tabular data, we select four measures, all of which have been featured in multiple previous studies on synthetic tabular data generation (Borisov et al., 2021).
Machine learning efficiency (MLE).
Since the generated data set should be able to replace the real data in a training process, this measure evaluates the performance of discriminative models trained on synthetic data sets.
Therefore, the models are tested on real test data, and the scores are compared to the original performance when the models were trained on the original, real training data set.
Table 1 and Table 4 present the results for the machine learning efficiency of proposed generative models compared to the baseline models.
To be independent of the concrete discriminative model, we evaluated the synthetic data sets using a Linear/Logistic Regression (LR), a Decision Tree (DT) and a Random Forest (RF) (Ho, 1995).
We observe that either GReaT and Distill-GReaT outperform all competitors and entail considerable performance improvements
[Table] [Table]
Results are averages
Distance to closest records (DCR) histogram.
To verify that the generated data is similar to original samples while not being exact copies, this measure computes the distance to the closest record in original training data set T train .
For each synthetic record s gen , it is given by
DCR ( s gen ) =
min { Distance( s gen , s i ) | s i ∈ T train } .
As a distance measure, we use the L 1 -norm of the differences.
We set the difference to be 0 for equal for categorical features and to 1 otherwise.
In the best case, all DCR scores are non-zero and their distribution is close to that of DCRs computed with points from the original test data set T test .
Visualizations of the distribution of the minimal distances can be found in Fig.
4 and Appendix B. Indeed, the results show that our model generates unseen samples in the expected proximity to the original ones.
The same holds for all used models and we did not observe large differences in this metric overall.
Discriminator measure.
To check whether the generated data cannot be easily told apart from the original data, we train a Random Forest discriminator (with hyperparameter tuning) on a mix of the generated train set (with label 0 ) and the original train set (with label 1 ).
We then report the test accuracy on a test data set, which contains equal shares of samples from the generated test set and
[Table]
the real test set.
Scores are shown in Table 2 and demonstrate the superior performance of the GReaT approach, which on average (across all used data sets) decreases the discriminator performance by 16.2 % over other competitive baselines for tabular data generation.
Bivariate joint distribution plots.
We qualitatively compare the generated feature distributions by the baselines and the GReaT approach to that of the original data.
As an illustrative example, we present joint density plots for the Longitude and Latitude features in the California Housing data set in Fig.
While CTGAN and CopulaGAN fail to model the strong dependency between these variables (indicated by out-of-distribution samples having both high latitude and longitude or low values for both), Distill-GReaT and GReaT yield densities that align well with the ground truth boundaries.
TVAE shows mediocre performance but still places density mass outside the bounds.
Effects of pretraining and permutations.
Having obtained impressive results with the complete setup, we investigate the role of the individual components towards its success.
We compare the full model (with permutation and pretraining step) to a model without permutation and a model without pretraining.
Tab.
3 presents ML efficiency and discriminator results for the modified models.
On all but the very small “Travel” data set, we observe pretraining to help boost the performance.
This might be due to the context knowledge made available to GReaT through the extensive adaptation to large text corpora.
The results regarding the feature order permutation step are mixed – the MLE performance decreases with permutations, but the results for the discriminator metric are inconclusive.
With permutations, the learning problem is undoubtedly more demanding.
It provides models with the ability to perform generation based on arbitrary conditioning.
However, we conjecture that it might also increase performance in some cases because it does not introduce any possibly unnatural, fixed feature ordering into the tabular data.
Reproducibility details.
We utilize pretrained generative language models from the established
HuggingFace framework (Wolf et al., 2020).
Its routines are also used for the fine-tuning and sampling steps.
We plan to open-source all code online.
Further details about all experiments, such as hyperparameters for each baseline and experiment setups are provided in Appendix C.
C ONCLUSION
In our recent work, we investigate how state-of-the-art generative language models can be leveraged to synthesize highly realistic tabular data samples.
Instead of following the usual path of encoding heterogeneous tabular data in a numerical format, we devise a textual encoding strategy and represent it with sentences that capture each record’s semantics.
The resulting transformer-decoder network fine-tuned on this data exhibits unprecedented generative performance and outstanding flexibility at the same time.
We term our method G eneration of Rea listic T abular data (GReaT).
GReaT unites several remarkable characteristics that address key problems in tabular data modeling: First, minimal preprocessing is required, which is efficient and results in the least possible information loss, thereby tackling the issue of possibly lossy preprocessing .
Second, leveraging random feature order permutations, we exploit the capability of arbitrary conditioning and are thus equipped with full control over the probabilistic sampling procedure for tabular data.
Finally, through pretraining, we can incorporate contextual and semantic knowledge extracted from terabytes of textual data for a more authentic tabular data synthesis.
In conclusion, we see our work as a door opener leading to yet-undiscovered possibilities in the domain of heterogeneous data generation.
E THICS S TATEMENT
Various critical applications rely on heterogeneous tabular data (e g., healthcare, finance, and so forth) and would profit from free data sharing.
However, this always requires ethics and privacy considerations.
The present work proposes a novel synthetic tabular data generation approach based on pretrained large language models.
Before sharing any data (including data generated by the proposed method), the authors thus strongly encourage possible owners of proprietary data sets to verify that reverse identification is impossible or is prevented by regulatory means.
Other than that, we see no further ethical questions related to this work
A D ISCUSSION
Processing of numerical values with LLMs in the GReaT approach.
Since we convert heterogeneous tabular data into text (Sec. 3), continuous and discrete numerical values are represented as character sequences.
While this might seem counter-intuitive at first, multiple independent studies have shown that transformer-based models are capable of understanding and processing numerical data encoded in such a way (Wallace et al., 2019; ? ).
This is even true in the one-shot setting for inference tasks (Dinh et al., 2022).
The impressive performance of GReaT that encompasses the numerical features aligns with these previous observations.
Having said that, smarter encodings for numerical values can also be considered a possible path to further improvement.
Unification of multiple modalities through transformer models.
Along with the usual numerical values, tabular data frequently contains textual metadata, e.g., feature names, named categories (“male”, “female”), and open text features (e.g., remarks ).
The recent evolution of transformer neural networks now permits to holistically unite this information, which was processed separately in the past, and learn context-specific, robust, and meaningful representations.
In the case of tabular data, information comes from the textual and numerical data modalities.
The GReaT approach and the results presented in this work provide an initial clue of the range of possibilities and opportunities that may lie in this line of research.
B A DDITIONAL E XPERIMENTAL R ESULTS B. F URTHER MLE MEASURES
Next to the accuracy measures (Tab. 1) we also report the ROCAUC score and the F1 score for the
ML efficiency experiment (Sec. 4).
B. A VERAGE N EGATIVE L OG -L IKELIHOOD M ETRIC FOR SYNTHETIC DATA
The generated data should be likely under the training data’s distribution.
Therefore, we generate three common synthetic data sets, and compute both the likelihood of the sampled data under the training data’s density ( L syn ).
This can however be prone to overfitting, which is why we additionally deploy the likelihood fitness metric proposed by Xu et al.
( L test ).
To compute this metric, a parametric density model (BNs and GMMs respectively) is fitted to the generated data.
Then the likelihood of the original test samples is computed.
The results in Table 5 indicate that
LLMs are comparable with state-of-the-art deep neural networks when modeling high-dimensional mixture distributions.
[Table]
[Figure]
B. D ISTANCE TO C LOSEST R ECORD R ESULTS
We compare the distribution of the minimal distances of the generated samples to the training data set.
Figure 4 shows the distribution for the California Housing data set and Figure 5 for the HELOC data set.
The results indicate that the generated samples are close to the original ones without coping them exactly.
B. A DDITIONAL Q UALITATIVE A NALYSIS R ESULTS
In Figure 6 we additionally compared the joint feature distribution of two exemplary features from the Adult Income data set, Age and EducationNum .
The joint plots were computed using a kernel density estimator.
[Table]
[Figure]
B. R UNTIME C OMPARISON
We also analyse the training/fine-tuning and sampling time between baseline models and two versions of the GReaT method using only two GPUs (Sec. C).
Table 6 summarizes our timebenchmarking results.
To make a fair comparison, we used the latest available versions of the corresponding implementations.
Also, we utilize the same DL framework - PyTorch (Paszke et al., 2019), and the same number of epochs 200 as well as the sampling size - 1000 , for each model.
While the proposed model requires significantly more time for the fine-tuning step compared to other baselines, Distill-GReaT and GReaT models demonstrate top performance for the vast majority of benchmarks across various datasets (Tab.1, Tab.2, Tab. 4).
Also, there are critical applications such as healthcare (Hernandez et al., 2022) or finance (Assefa et al., 2020), where high quality synthetic tabular data is required, regardless of the possible computation cost, therefore in our experiments we introduced two data sets from medical machine learning field (Tab. 8).
Lastly, with more computation resources, the proposed model’s sampling and fine-tuning times can be drastically decreased.
[Table]
[Table]
C R EPRODUCIBILITY D ETAILS
We fine-tune the Distill-GReaT model for each data set for 200 epochs, except for the California housing and Diabetes (Strack et al., 2014) data sets, for them, we fine-tune them for 100 epochs.
The
GReaT baseline is fine-tuned for 110 , 310 , 400 , 255 , 150 , 85 , epochs for California Housing, Adult
Income, Travel, Home Equity Line of Credit (HELOC), Sick (Dua & Graff, 2017), and Diabetes data sets, respectively.
Depending on the GPU memory limitations, we vary the batch size fromto 124 .
For the sampling step, we set the temperature parameter T to 0 .
7 for all experiments and data sets.
We sample new synthetic data using the name-value pair preconditioning (Sec. 3), starting with the target feature for all data sets (see an example in the supplementary materials).
[Figure]
We utilize the AdamW optimizer (Loshchilov & Hutter, 2017) for the proposed generative models, with the learning rate 5 × 10 − 5 .
We plan to share trained weights for the GReaT models.
The baseline models are trained for 200 epochs for each data set.
Our hardware setup consisted of two NVIDIA 2080RTX GPUs with 12 GB RAM each,
GB system RAM, and AMD Ryzen 3960X with 24 cores, we use the Ubuntu 20.04 operation system.
For the ML efficiency and discriminator experiments (Sec, 4) we additionally use linear/logistic regression, decision tree, and random forest models from the Scikit-Learn package (Buitinck et al., 2013), we report the exact hyperparameters for each model in Table 7.
For the discriminator measure experiment (Table 2), we tune hyperparameters for each data set using the 5 -fold cross-validation.
[Table]
[Table]
[Table]
Besides providing important information for the experiment reproducibility
[Table]
we also share the synthetic data sets generated by our GReaT method
[Table]
For this experiment, we select only three variables Education , Income , and Age from the
Adult Income data set (Dua & Graff, 2017).
However, the proposed method can be scaled to the arbitrary number of conditions.
The results obtaining by changing the input textual sequence, e.g., Education is HS-school, Income is >50K, Age is , after we obtain the conditional discriminate distribution p ( Age | income = > 50 K, education = HS − school ) .
Interestingly, the GReaT method successfully learned that there is only two options for the Income variable.
The arbitrary sampling is supported by our Python programming framework.