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

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Pereira, Shovon Niverd, Khadka, Krishna, Lei, Yu
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_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