用决策树生成表格数据,解决回归任务中目标值稀疏问题。
CARTGen-IR: Synthetic Tabular Data Generation for Imbalanced Regression
- 基于决策树构建采样策略,不设阈值直接处理连续目标。
- 在极端值预测任务上性能媲美顶尖方法,速度更快且可解释。
- 适合需要透明、高效数据增强的工业级回归场景。
回归任务中的目标分布不均衡问题长期存在,关键目标值的稀缺会显著影响模型性能。现有数据层面解决方案多沿用分类方法,对连续目标强行设定阈值,导致问题定义失真。深度生成模型虽灵活但计算成本高且难以解释。本文提出一种面向表格数据不平衡回归的基于决策树的合成采样方法(CARTGen-IR),通过融合相关性与密度引导的采样机制,在无需阈值的情况下填补稀疏目标区域,并利用特征驱动的树结构生成包含异质特征与非线性交互的真实样本。在基准极端值预测数据集上的实验表明,该方法在性能上可媲美当前最优重采样与生成方法,同时具备更快速的执行效率和更高的可解释性,展现出作为可扩展、可解释的数据级策略在不平衡回归领域的潜力。
原文摘要 · Abstract (English)
Handling imbalanced target distributions in regression poses a persistent challenge, as the underrepresentation of relevant target values can significantly hinder model performance. Existing data-level solutions often adapt classification-oriented techniques, introducing arbitrary thresholds over the continuous target and leading to artificial and potentially misleading problem formulations. Deep generative models offer flexible sample synthesis but are computationally intensive and difficult to interpret. We propose a CART-based synthetic sampling method specifically designed for imbalanced regression on tabular data. The method integrates relevance- and density-guided sampling to address sparse target regions without thresholding, and employs a feature-driven tree structure to generate realistic tabular samples across heterogeneous features and non-linear interactions. Experiments on benchmark datasets for extreme-value prediction show that the proposed approach is competitive with state-of-the-art resampling and generative methods while offering faster execution and greater transparency. These results highlight its potential as a scalable and interpretable data-level strategy for improving regression models in imbalanced domains.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。