arXiv:2512.13003stat.MLcs.LG2025-12

提出一种融合模型与子空间感知的新型分布外检测方法。

General OOD Detection via Model-aware and Subspace-aware Variable Priority

  • 基于模型预测构建局部邻域,聚焦关键特征方向
  • 在真实癌症生存数据中准确识别手术方式引起的分布偏移
  • 适用于回归与生存分析,无需全局距离或密度估计

分布外(OOD)检测对于判断监督模型是否遇到与训练分布有显著差异的输入至关重要。尽管分类任务中的OOD检测已广泛研究,但回归和生存分析因缺乏离散标签且难以量化预测不确定性,相关研究仍有限。本文提出一种同时具备模型感知与子空间感知能力的检测框架,并将可变优先级直接嵌入检测步骤。该方法利用拟合的预测器,在每个测试样本周围构建局部邻域,突出模型学习关系所依赖的特征方向,弱化对预测不重要的方向。该框架不依赖全局距离度量或全特征密度估计,即可生成OOD得分。其适用范围涵盖不同输出类型;在实现中采用随机森林,因其规则结构能产生透明的邻域与有效的评分。在设计用于隔离功能偏移的合成与真实数据基准上,性能持续优于现有方法。进一步在食管癌生存分析中验证,与淋巴结清扫相关的分布偏移揭示了符合外科指南的临床模式。

原文摘要 · Abstract (English)

Out-of-distribution (OOD) detection is essential for determining when a supervised model encounters inputs that differ meaningfully from its training distribution. While widely studied in classification, OOD detection for regression and survival analysis remains limited due to the absence of discrete labels and the challenge of quantifying predictive uncertainty. We introduce a framework for OOD detection that is simultaneously model aware and subspace aware, and that embeds variable prioritization directly into the detection step. The method uses the fitted predictor to construct localized neighborhoods around each test case that emphasize the features driving the model's learned relationship and downweight directions that are less relevant to prediction. It produces OOD scores without relying on global distance metrics or estimating the full feature density. The framework is applicable across outcome types, and in our implementation we use random forests, where the rule structure yields transparent neighborhoods and effective scoring. Experiments on synthetic and real data benchmarks designed to isolate functional shifts show consistent improvements over existing methods. We further demonstrate the approach in an esophageal cancer survival study, where distribution shifts related to lymphadenectomy identify patterns relevant to surgical guidelines.

OOD检测生存分析随机森林分布外

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