Borrowing From the Future: Enhancing Early Risk Assessment through Contrastive Learning

2508.11210v1 cs.LG, stat.ML 2025-08-19
Авторы:

Minghui Sun, Matthew M. Engelhard, Benjamin A. Goldstein

Резюме на русском

#### Контекст Риск-ассессменты для детского населения часто проводятся на различных стадиях развития: преждевременно, после рождения и во время регулярных приемов у врача. Хотя оценки на поздних стадиях демонстрируют высокую точность, желательно делать надежные прогнозы в начале периода мониторинга. Основная задача исследования — улучшить прогностическую эффективность в первых стадиях, чтобы обеспечить ранний интервенционный подход к потенциальным проблемам. Предлагаемый подход, **"Borrowing From the Future" (BFF)**, представляет собой контрастивный подход с многомодальной архитектурой. Он использует все доступные данные по всему периоду мониторинга для обучения модели. Затем модель проводит оценку рисков, опираясь только на текущие данные. Этот подход позволяет "питать" модель сигналами из будущих стадий (например, приемов у врача в детстве), чтобы улучшить ее прогностическую точность в первоначальных стадиях мониторинга. #### Метод **BFF** состоит из нескольких ключевых компонентов: 1. **Многомодальная архитектура** — каждая стадия мониторинга (преждевременная, после рождения, Well-Child) играет роль отдельной модальности. 2. **Контрастивное обучение** — модель обучается с использованием всей информации по всему периоду, но производит прогнозы только на основе текущих данных. 3. **Имплицитное научение** — отложенные сигналы (например, от Well-Child приёмов) используются для "научения" модели в первые стадии мониторинга. Модель обучается с использованием метода градиентного спуска и использует архитектуру, основанную на предсказании риска. Изучаемые данные включают многочисленные реальные клинические наблюдения. #### Результаты **BFF** протестирована на двух реальных клинических задачах: 1. Прогноз ранних клинических результатов у детей. 2. Оценка риска заболеваний на разных стадиях мониторинга. Использовались многочисленные реальные данные, полученные из клинических баз. Результаты показали, что **BFF** показывает постоянные улучшения в ранних оценках риска по сравнению с контрольными моделями. Эти улучшения связаны с возможностью "питания" модели данными из будущих стадий. #### Значимость **BFF** может быть применена в различных клинических сценариях, где ранняя оценка риска ключевая. Особенно эффективна в задачах, где доступны данные из будущих стадий мониторинга. Её преимущества: - Улучшение точности в ранних стадиях. - Эффективное использование информации из будущих приёмов. Потенциальное влияние заключается в повышении качества раннего вмешательства, снижении рисков и повышении эффективности медицински

Abstract

Risk assessments for a pediatric population are often conducted across multiple stages. For example, clinicians may evaluate risks prenatally, at birth, and during Well-Child visits. Although predictions made at later stages typically achieve higher precision, it is clinically desirable to make reliable risk assessments as early as possible. Therefore, this study focuses on improving prediction performance in early-stage risk assessments. Our solution, \textbf{Borrowing From the Future (BFF)}, is a contrastive multi-modal framework that treats each time window as a distinct modality. In BFF, a model is trained on all available data throughout the time while performing a risk assessment using up-to-date information. This contrastive framework allows the model to ``borrow'' informative signals from later stages (e.g., Well-Child visits) to implicitly supervise the learning at earlier stages (e.g., prenatal/birth stages). We validate BFF on two real-world pediatric outcome prediction tasks, demonstrating consistent improvements in early risk assessments. The code is available at https://github.com/scotsun/bff.

Ссылки и действия