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

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Sun, Minghui, Engelhard, Matthew M., Goldstein, Benjamin A.
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866915447467671552
author Sun, Minghui
Engelhard, Matthew M.
Goldstein, Benjamin A.
author_facet Sun, Minghui
Engelhard, Matthew M.
Goldstein, Benjamin A.
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2508_11210
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Borrowing From the Future: Enhancing Early Risk Assessment through Contrastive Learning
Sun, Minghui
Engelhard, Matthew M.
Goldstein, Benjamin A.
Machine Learning
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.
title Borrowing From the Future: Enhancing Early Risk Assessment through Contrastive Learning
topic Machine Learning
url https://arxiv.org/abs/2508.11210