TabINR: An Implicit Neural Representation Framework for Tabular Data Imputation

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Ochs, Vincent, Bieder, Florentin, Hadramy, Sidaty el, Friedrich, Paul, Taha-Mehlitz, Stephanie, Taha, Anas, Cattin, Philippe C.
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866909818846969856
author Ochs, Vincent
Bieder, Florentin
Hadramy, Sidaty el
Friedrich, Paul
Taha-Mehlitz, Stephanie
Taha, Anas
Cattin, Philippe C.
author_facet Ochs, Vincent
Bieder, Florentin
Hadramy, Sidaty el
Friedrich, Paul
Taha-Mehlitz, Stephanie
Taha, Anas
Cattin, Philippe C.
contents Tabular data builds the basis for a wide range of applications, yet real-world datasets are frequently incomplete due to collection errors, privacy restrictions, or sensor failures. As missing values degrade the performance or hinder the applicability of downstream models, and while simple imputing strategies tend to introduce bias or distort the underlying data distribution, we require imputers that provide high-quality imputations, are robust across dataset sizes and yield fast inference. We therefore introduce TabINR, an auto-decoder based Implicit Neural Representation (INR) framework that models tables as neural functions. Building on recent advances in generalizable INRs, we introduce learnable row and feature embeddings that effectively deal with the discrete structure of tabular data and can be inferred from partial observations, enabling instance adaptive imputations without modifying the trained model. We evaluate our framework across a diverse range of twelve real-world datasets and multiple missingness mechanisms, demonstrating consistently strong imputation accuracy, mostly matching or outperforming classical (KNN, MICE, MissForest) and deep learning based models (GAIN, ReMasker), with the clearest gains on high-dimensional datasets.
format Preprint
id arxiv_https___arxiv_org_abs_2510_01136
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle TabINR: An Implicit Neural Representation Framework for Tabular Data Imputation
Ochs, Vincent
Bieder, Florentin
Hadramy, Sidaty el
Friedrich, Paul
Taha-Mehlitz, Stephanie
Taha, Anas
Cattin, Philippe C.
Machine Learning
Artificial Intelligence
Tabular data builds the basis for a wide range of applications, yet real-world datasets are frequently incomplete due to collection errors, privacy restrictions, or sensor failures. As missing values degrade the performance or hinder the applicability of downstream models, and while simple imputing strategies tend to introduce bias or distort the underlying data distribution, we require imputers that provide high-quality imputations, are robust across dataset sizes and yield fast inference. We therefore introduce TabINR, an auto-decoder based Implicit Neural Representation (INR) framework that models tables as neural functions. Building on recent advances in generalizable INRs, we introduce learnable row and feature embeddings that effectively deal with the discrete structure of tabular data and can be inferred from partial observations, enabling instance adaptive imputations without modifying the trained model. We evaluate our framework across a diverse range of twelve real-world datasets and multiple missingness mechanisms, demonstrating consistently strong imputation accuracy, mostly matching or outperforming classical (KNN, MICE, MissForest) and deep learning based models (GAIN, ReMasker), with the clearest gains on high-dimensional datasets.
title TabINR: An Implicit Neural Representation Framework for Tabular Data Imputation
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2510.01136