用混合自编码器提升小样本表格数据的建模能力
Hybrid Autoencoders for Tabular Data: Leveraging Model-Based Augmentation in Low-Label Settings
- 神经编码器+无感知软决策树编码器,各自通过门控网络做特征选择
- 在多个低标签表格数据集上分类与回归性能均优于基线模型
- 适合小样本场景下的表格数据建模,尤其对高频率结构敏感的任务
深度神经网络在表格数据上表现不佳,主要因其对无关特征敏感且偏好平滑、低频函数。在标签稀少的情况下,这种缺陷更显著,难以捕捉定义表格结构的尖锐高频信号。尽管自监督学习(SSL)在此类场景中具有潜力,但表格领域缺乏有效的数据增强方法。本文提出一种混合自编码器:结合神经编码器与无感知软决策树(OSDT)编码器,每种编码器由独立的随机门控网络引导进行样本特定特征选择。两者通过共享解码器和交叉重建损失联合训练,分别学习反映其归纳偏置的独特但对齐的表示。其中,对噪声鲁棒且擅长建模局部高频结构的OSDT编码器,可引导神经编码器学习更符合表格数据特性的表示。推理阶段仅使用神经编码器,保持灵活性并兼容自监督学习。谱分析验证了两种编码器的差异性归纳偏置。实验表明,该方法在多种表格数据集上实现了稳定的低标签分类与回归性能提升,优于深度模型与树模型的监督基线。
原文摘要 · Abstract (English)
Deep neural networks often under-perform on tabular data due to their sensitivity to irrelevant features and a spectral bias toward smooth, low-frequency functions. These limitations hinder their ability to capture the sharp, high-frequency signals that often define tabular structure, especially under limited labeled samples. While self-supervised learning (SSL) offers promise in such settings, it remains challenging in tabular domains due to the lack of effective data augmentations. We propose a hybrid autoencoder that combines a neural encoder with an oblivious soft decision tree (OSDT) encoder, each guided by its own stochastic gating network that performs sample-specific feature selection. Together, these structurally different encoders and model-specific gating networks implement model-based augmentation, producing complementary input views tailored to each architecture. The two encoders, trained with a shared decoder and cross-reconstruction loss, learn distinct yet aligned representations that reflect their respective inductive biases. During training, the OSDT encoder (robust to noise and effective at modeling localized, high-frequency structure) guides the neural encoder toward representations more aligned with tabular data. At inference, only the neural encoder is used, preserving flexibility and SSL compatibility. Spectral analysis highlights the distinct inductive biases of each encoder. Our method achieves consistent gains in low-label classification and regression across diverse tabular datasets, outperforming deep and tree-based supervised baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。