TabKD: Tabular Knowledge Distillation through Interaction Diversity of Learned Feature Bins

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Pereira, Shovon Niverd, Khadka, Krishna, Lei, Yu
Format: Preprint
Publié: 2026
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866910054828998656
author Pereira, Shovon Niverd
Khadka, Krishna
Lei, Yu
author_facet Pereira, Shovon Niverd
Khadka, Krishna
Lei, Yu
contents Data-free knowledge distillation enables model compression without original training data, critical for privacy-sensitive tabular domains. However, existing methods does not perform well on tabular data because they do not explicitly address feature interactions, the fundamental way tabular models encode predictive knowledge. We identify interaction diversity, systematic coverage of feature combinations, as an essential requirement for effective tabular distillation. To operationalize this insight, we propose TabKD, which learns adaptive feature bins aligned with teacher decision boundaries, then generates synthetic queries that maximize pairwise interaction coverage. Across 4 benchmark datasets and 4 teacher architectures, TabKD achieves highest student-teacher agreement in 14 out of 16 configurations, outperforming 5 state-of-the-art baselines. We further show that interaction coverage strongly correlates with distillation quality, validating our core hypothesis. Our work establishes interaction-focused exploration as a principled framework for tabular model extraction.
format Preprint
id arxiv_https___arxiv_org_abs_2603_15481
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle TabKD: Tabular Knowledge Distillation through Interaction Diversity of Learned Feature Bins
Pereira, Shovon Niverd
Khadka, Krishna
Lei, Yu
Machine Learning
Artificial Intelligence
Data-free knowledge distillation enables model compression without original training data, critical for privacy-sensitive tabular domains. However, existing methods does not perform well on tabular data because they do not explicitly address feature interactions, the fundamental way tabular models encode predictive knowledge. We identify interaction diversity, systematic coverage of feature combinations, as an essential requirement for effective tabular distillation. To operationalize this insight, we propose TabKD, which learns adaptive feature bins aligned with teacher decision boundaries, then generates synthetic queries that maximize pairwise interaction coverage. Across 4 benchmark datasets and 4 teacher architectures, TabKD achieves highest student-teacher agreement in 14 out of 16 configurations, outperforming 5 state-of-the-art baselines. We further show that interaction coverage strongly correlates with distillation quality, validating our core hypothesis. Our work establishes interaction-focused exploration as a principled framework for tabular model extraction.
title TabKD: Tabular Knowledge Distillation through Interaction Diversity of Learned Feature Bins
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2603.15481