arXiv:2512.06652cs.LGcs.AI2025-12

针对重症患者呼吸机需求预测,提出自适应测试时训练方法提升模型跨中心泛化能力。

Adaptive Test-Time Training for Predicting Need for Invasive Mechanical Ventilation in Multi-Center Cohorts

  • 通过自监督学习增强主任务与辅助任务的特征对齐
  • 在多中心数据上实现优于基准的分类准确率
  • 适合需部署于不同医院的ICU临床预测系统

准确预测重症监护室(ICU)患者是否需要侵入性机械通气(IMV)对及时干预和资源分配至关重要。然而,不同机构间患者群体、临床实践及电子健康记录(EHR)系统的差异导致领域偏移,降低预测模型的泛化性能。测试时训练(TTT)通过在推理阶段动态调整模型,无需目标域标签即可缓解此类偏移。本文提出自适应测试时训练(AdaTTT),专用于基于EHR的ICU IMV预测。我们推导出测试时预测误差的信息论边界,表明其受限于主任务与辅助任务间的不确定性。为增强任务对齐,引入自监督学习框架,包含重构和掩码特征建模两种预训练任务,并采用动态掩码策略突出主任务关键特征。此外,为提升对领域偏移的鲁棒性,结合原型学习并使用部分最优传输(POT)实现灵活、部分的特征对齐,同时保持临床可解释的患者表征。在多个中心的ICU队列上实验表明,该方法在不同测试时适应基准下表现优异。

原文摘要 · Abstract (English)

Accurate prediction of the need for invasive mechanical ventilation (IMV) in intensive care units (ICUs) patients is crucial for timely interventions and resource allocation. However, variability in patient populations, clinical practices, and electronic health record (EHR) systems across institutions introduces domain shifts that degrade the generalization performance of predictive models during deployment. Test-Time Training (TTT) has emerged as a promising approach to mitigate such shifts by adapting models dynamically during inference without requiring labeled target-domain data. In this work, we introduce Adaptive Test-Time Training (AdaTTT), an enhanced TTT framework tailored for EHR-based IMV prediction in ICU settings. We begin by deriving information-theoretic bounds on the test-time prediction error and demonstrate that it is constrained by the uncertainty between the main and auxiliary tasks. To enhance their alignment, we introduce a self-supervised learning framework with pretext tasks: reconstruction and masked feature modeling optimized through a dynamic masking strategy that emphasizes features critical to the main task. Additionally, to improve robustness against domain shifts, we incorporate prototype learning and employ Partial Optimal Transport (POT) for flexible, partial feature alignment while maintaining clinically meaningful patient representations. Experiments across multi-center ICU cohorts demonstrate competitive classification performance on different test-time adaptation benchmarks.

重症监护机器学习自适应训练医疗预测

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