针对表格回归中不规则目标函数难题,提出新框架提升模型性能。
APAR: Modeling Irregular Target Functions in Tabular Regression via Arithmetic-Aware Pre-Training and Adaptive-Regularized Fine-Tuning
- 通过算术感知预训练捕捉连续标签的样本关系
- 自适应正则化微调使模型在10个数据集上RMSE提升9.43%~20.37%
- 适合处理金融、医疗等含复杂规律的表格数据任务
表格数据在金融、基因组学和医疗等领域广泛应用。本文聚焦表格回归任务,针对深度学习模型因表格式数据固有的不规则目标函数而表现不稳定的问题,提出一种新的算术感知预训练与自适应正则化微调框架(APAR)。该框架在预训练阶段引入算术感知代理任务,从连续标签角度捕捉样本间的复杂关系;在微调阶段采用基于一致性的自适应正则化技术,自动学习合适的数据增强策略。在10个数据集上的大量实验表明,APAR在均方根误差(RMSE)上相比现有基于GBDT、监督神经网络及预训练-微调神经网络的方法提升9.43%至20.37%,并实证验证了预训练任务的有效性,包括对算术运算的分析。代码与数据已公开于https://github.com/johnnyhwu/APAR。
原文摘要 · Abstract (English)
Tabular data are fundamental in common machine learning applications, ranging from finance to genomics and healthcare. This paper focuses on tabular regression tasks, a field where deep learning (DL) methods are not consistently superior to machine learning (ML) models due to the challenges posed by irregular target functions inherent in tabular data, causing sensitive label changes with minor variations from features. To address these issues, we propose a novel Arithmetic-Aware Pre-training and Adaptive-Regularized Fine-tuning framework (APAR), which enables the model to fit irregular target function in tabular data while reducing the negative impact of overfitting. In the pre-training phase, APAR introduces an arithmetic-aware pretext objective to capture intricate sample-wise relationships from the perspective of continuous labels. In the fine-tuning phase, a consistency-based adaptive regularization technique is proposed to self-learn appropriate data augmentation. Extensive experiments across 10 datasets demonstrated that APAR outperforms existing GBDT-, supervised NN-, and pretrain-finetune NN-based methods in RMSE (+9.43% $\sim$ 20.37%), and empirically validated the effects of pre-training tasks, including the study of arithmetic operations. Our code and data are publicly available at https://github.com/johnnyhwu/APAR.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。