Geodesic Flow Kernels for Semi-Supervised Learning on Mixed-Variable Tabular Dataset

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Hwang, Yoontae, Lee, Yongjae
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916703316738048
author Hwang, Yoontae
Lee, Yongjae
author_facet Hwang, Yoontae
Lee, Yongjae
contents Tabular data poses unique challenges due to its heterogeneous nature, combining both continuous and categorical variables. Existing approaches often struggle to effectively capture the underlying structure and relationships within such data. We propose GFTab (Geodesic Flow Kernels for Semi- Supervised Learning on Mixed-Variable Tabular Dataset), a semi-supervised framework specifically designed for tabular datasets. GFTab incorporates three key innovations: 1) Variable-specific corruption methods tailored to the distinct properties of continuous and categorical variables, 2) A Geodesic flow kernel based similarity measure to capture geometric changes between corrupted inputs, and 3) Tree-based embedding to leverage hierarchical relationships from available labeled data. To rigorously evaluate GFTab, we curate a comprehensive set of 21 tabular datasets spanning various domains, sizes, and variable compositions. Our experimental results show that GFTab outperforms existing ML/DL models across many of these datasets, particularly in settings with limited labeled data.
format Preprint
id arxiv_https___arxiv_org_abs_2412_12864
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Geodesic Flow Kernels for Semi-Supervised Learning on Mixed-Variable Tabular Dataset
Hwang, Yoontae
Lee, Yongjae
Machine Learning
Tabular data poses unique challenges due to its heterogeneous nature, combining both continuous and categorical variables. Existing approaches often struggle to effectively capture the underlying structure and relationships within such data. We propose GFTab (Geodesic Flow Kernels for Semi- Supervised Learning on Mixed-Variable Tabular Dataset), a semi-supervised framework specifically designed for tabular datasets. GFTab incorporates three key innovations: 1) Variable-specific corruption methods tailored to the distinct properties of continuous and categorical variables, 2) A Geodesic flow kernel based similarity measure to capture geometric changes between corrupted inputs, and 3) Tree-based embedding to leverage hierarchical relationships from available labeled data. To rigorously evaluate GFTab, we curate a comprehensive set of 21 tabular datasets spanning various domains, sizes, and variable compositions. Our experimental results show that GFTab outperforms existing ML/DL models across many of these datasets, particularly in settings with limited labeled data.
title Geodesic Flow Kernels for Semi-Supervised Learning on Mixed-Variable Tabular Dataset
topic Machine Learning
url https://arxiv.org/abs/2412.12864