arXiv:2606.01566cs.LG2026-06

解决科学数据小样本下特征选择不稳定与性能高估问题

RobustModelMaker: Coupling Bootstrap Stability Selection with Leakage-Safe Nested Cross-Validation for Scientific Machine Learning

论文配图:RobustModelMaker: Coupling Bootstrap Stability Selection with Leakage-Safe Nested Cross-Validation for Scientific Machine Learning
图 1 · 摘自论文原文
  • 用自助法稳定性选择+严格嵌套交叉验证,确保每轮训练独立
  • 在三个真实科学数据集上,性能优于或持平主流选择器
  • 兼顾预测性能与特征选择稳定性,适合追求可复现的研究

小到中等规模的科学数据集使机器学习流程面临双重挑战:单次运行的特征选择对训练数据微小扰动敏感,导致结果不稳;而使用相同数据进行选择、调参和评估会带来性能估计的乐观偏差。这两种失效模式常被分开处理,但在科学数据实际场景中它们相互作用——选择不稳定会放大本已偏高的评分方差,且针对一种问题的常规方法往往无法解决另一问题。RobustModelMaker 是一个 Python 框架,将自助法稳定性选择与严格的嵌套交叉验证结合,所有预处理与特征选择均在每折内部完成,输出经稳定性验证的特征子集及防泄漏的性能估计。支持二分类、多分类与回归任务中的九种算法。通过三组真实科学数据集上的确定性测试(涵盖单元、性能与可复现性检查),对比 ANOVA F 检验、基于交叉验证的递归特征消除与 Boruta 三种选择器,结果表明 RobustModelMaker 在各项预测得分上与最优替代方法相当,并在联合得分-稳定性前沿上表现独特,未被任何其他方法覆盖。两个应用案例——来自 PLCO 试验的卵巢癌生物标志物发现与 UCI 超导数据集上的临界温度回归——展示了其实际使用方式,揭示当稳定性被作为首要交付目标时所显现的关键权衡。

原文摘要 · Abstract (English)

Small-to-medium scientific datasets place machine learning pipelines under two compounding pressures. Single-run feature selection produces feature sets that change substantially under small perturbations of the training data, and any procedure that uses the same data for selection, tuning, and evaluation produces optimistically biased performance estimates. The two failure modes are routinely treated as separable, but in the regimes where scientific data live, they interact: an unstable selection inflates the variance of an already-optimistic score, and standard remedies for one rarely address the other. RobustModelMaker is a Python framework that couples bootstrap stability selection with strict nested cross-validation, performs all preprocessing and selection inside each fold, and produces a stability-tested feature subset together with a leakage-safe performance estimate. The framework supports nine algorithms across binary classification, multiclass classification, and regression. Behaviour is verified by a deterministic test suite spanning unit, performance, and reproducibility checks on three real scientific datasets comparing to three alternative selectors (ANOVA F-test, recursive feature elimination with cross-validation, and Boruta) on both predictive score and a Jaccard measure of selection stability. RobustModelMaker is competitive in score with the best alternative selector on each dataset, and occupies a position on the joint score-stability frontier that none of the alternatives match across all three task types. Two example applications, ovarian cancer biomarker discovery from the PLCO Trial and critical-temperature regression on the UCI Superconductivity Data, illustrate how the framework is used in practice and what trade-offs become visible when stability is treated as a first-class deliverable rather than an emergent property.

特征选择稳健建模交叉验证科学计算

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。