针对表格数据中的系统性误差,提出自监督鲁棒学习框架。
Distributionally robust self-supervised learning for tabular data
- 用掩码语言建模训练编码器-解码器,学习鲁棒表征
- 结合JTT与DFR方法,提升各特征子群体表现
- 适合关注表格数据泛化能力的从业者
基于经验风险最小化(ERM)训练的机器学习模型在表格数据的特定子群体上常出现系统性误差,称为误差切片。在自监督预训练阶段,由于高基数特征和误差集构建复杂,学习鲁棒表征尤为困难。现有鲁棒学习方法主要聚焦于计算机视觉中的监督场景,缺乏针对表格数据的针对性方案。本文提出一种在自监督预训练阶段学习鲁棒表征的框架,采用掩码语言建模(MLM)损失训练编码器-解码器模型。引入Just Train Twice(JTT)和Deep Feature Reweighting(DFR)方法,在预训练中对易错样本加权或为特定分类特征构建平衡数据集,生成每特征专用模型,并通过集成方式提升下游分类性能。实验在多个数据集上验证了该方法的有效性,显著增强跨切片鲁棒性与整体泛化能力。代码已公开:https://github.com/amazon-science/distributionally-robust-self-supervised-learning-for-tabular-data。
原文摘要 · Abstract (English)
Machine learning (ML) models trained using Empirical Risk Minimization (ERM) often exhibit systematic errors on specific subpopulations of tabular data, known as error slices. Learning robust representation in presence of error slices is challenging, especially in self-supervised settings during the feature reconstruction phase, due to high cardinality features and the complexity of constructing error sets. Traditional robust representation learning methods are largely focused on improving worst group performance in supervised setting in computer vision, leaving a gap in approaches tailored for tabular data. We address this gap by developing a framework to learn robust representation in tabular data during self-supervised pre-training. Our approach utilizes an encoder-decoder model trained with Masked Language Modeling (MLM) loss to learn robust latent representations. This paper applies the Just Train Twice (JTT) and Deep Feature Reweighting (DFR) methods during the pre-training phase for tabular data. These methods fine-tune the ERM pre-trained model by up-weighting error-prone samples or creating balanced datasets for specific categorical features. This results in specialized models for each feature, which are then used in an ensemble approach to enhance downstream classification performance. This methodology improves robustness across slices, thus enhancing overall generalization performance. Extensive experiments across various datasets demonstrate the efficacy of our approach. The code is available: https://github.com/amazon-science/distributionally-robust-self-supervised-learning-for-tabular-data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。