HYTREL: Hypergraph-enhanced Tabular Data Representation Learning
This work addresses the need for better tabular data models in machine learning by incorporating inductive biases, though it is incremental as it builds on existing language model approaches.
The paper tackled the problem of tabular data representation learning by proposing HYTREL, a model that captures structural properties like permutation invariances using hypergraphs, resulting in consistent outperformance of baselines on four downstream tasks with minimal pretraining.
Language models pretrained on large collections of tabular data have demonstrated their effectiveness in several downstream tasks. However, many of these models do not take into account the row/column permutation invariances, hierarchical structure, etc. that exist in tabular data. To alleviate these limitations, we propose HYTREL, a tabular language model, that captures the permutation invariances and three more structural properties of tabular data by using hypergraphs - where the table cells make up the nodes and the cells occurring jointly together in each row, column, and the entire table are used to form three different types of hyperedges. We show that HYTREL is maximally invariant under certain conditions for tabular data, i.e., two tables obtain the same representations via HYTREL iff the two tables are identical up to permutations. Our empirical results demonstrate that HYTREL consistently outperforms other competitive baselines on four downstream tasks with minimal pretraining, illustrating the advantages of incorporating the inductive biases associated with tabular data into the representations. Finally, our qualitative analyses showcase that HYTREL can assimilate the table structures to generate robust representations for the cells, rows, columns, and the entire table.