arXiv:2412.05390cs.LGstat.ML2024-12

用张量收缩层与Transformer生成表格数据,提升分布估计精度

Tabular data generation with tensor contraction layers and transformers

  • 将表格特征嵌入连续矩阵,结合张量收缩层与Transformer建模
  • 在OpenML CC18数据集上,密度估计指标显著提升
  • 适合关注表格数据生成与分布建模的研究者

表格数据生成在深度学习领域受到广泛关注,目标是估计数据的潜在分布。然而,表格数据包含多种类型特征,模型难以学习其内部关系。一种方法是通过分词将每个特征嵌入连续矩阵,另一种是利用Transformer架构捕捉变量间关系。本文实验研究了在变分自编码器中使用嵌入表示的潜力,结合张量收缩层与Transformer建模表格数据的潜在分布。我们比较了四种结构:基线VAE模型、仅用张量收缩层的变体、仅用Transformer的变体,以及两者融合的混合模型。实验基于OpenML CC18套件中的多个数据集,评估密度估计和机器学习效率指标。结果表明,借助张量收缩层的嵌入表示能显著改善密度估计性能,同时保持机器学习效率的竞争力。

原文摘要 · Abstract (English)

Generative modeling for tabular data has recently gained significant attention in the Deep Learning domain. Its objective is to estimate the underlying distribution of the data. However, estimating the underlying distribution of tabular data has its unique challenges. Specifically, this data modality is composed of mixed types of features, making it a non-trivial task for a model to learn intra-relationships between them. One approach to address mixture is to embed each feature into a continuous matrix via tokenization, while a solution to capture intra-relationships between variables is via the transformer architecture. In this work, we empirically investigate the potential of using embedding representations on tabular data generation, utilizing tensor contraction layers and transformers to model the underlying distribution of tabular data within Variational Autoencoders. Specifically, we compare four architectural approaches: a baseline VAE model, two variants that focus on tensor contraction layers and transformers respectively, and a hybrid model that integrates both techniques. Our empirical study, conducted across multiple datasets from the OpenML CC18 suite, compares models over density estimation and Machine Learning efficiency metrics. The main takeaway from our results is that leveraging embedding representations with the help of tensor contraction layers improves density estimation metrics, albeit maintaining competitive performance in terms of machine learning efficiency.

表格生成变分自编码器张量收缩Transformer

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。