English
Related papers

Related papers: Ensemble-Based Survival Models with the Self-Atten…

200 papers

Machine learning models that aim to predict dementia onset usually follow the classification methodology ignoring the time until an event happens. This study presents an alternative, using survival analysis within the context of machine…

Machine Learning · Computer Science 2023-06-21 Daniel Stamate , Henry Musto , Olesya Ajnakina , Daniel Stahl

This tutorial shows how various Bayesian survival models can be fitted using the integrated nested Laplace approximation in a clear, legible, and comprehensible manner using the INLA and INLAjoint R-packages. Such models include accelerated…

Models for predicting the time of a future event are crucial for risk assessment, across a diverse range of applications. Existing time-to-event (survival) models have focused primarily on preserving pairwise ordering of estimated event…

Machine Learning · Statistics 2021-01-14 Paidamoyo Chapfuwa , Chenyang Tao , Lawrence Carin , Ricardo Henao

The use of cumulative incidence functions for characterizing the risk of one type of event in the presence of others has become increasingly popular over the past decade. The problems of modeling, estimation and inference have been treated…

Methodology · Statistics 2021-06-25 Youngjoo Cho , Annette M. Molinaro , Chen Hu , Robert L. Strawderman

Ensemble learning is a mainstay in modern data science practice. Conventional ensemble algorithms assign to base models a set of deterministic, constant model weights that (1) do not fully account for individual models' varying accuracy…

Methodology · Statistics 2019-04-02 Jeremiah Zhe Liu , John Paisley , Marianthi-Anna Kioumourtzoglou , Brent A. Coull

New models of random forests jointly using the attention and self-attention mechanisms are proposed for solving the regression problem. The models can be regarded as extensions of the attention-based random forest whose idea stems from…

Machine Learning · Computer Science 2022-07-12 Lev V. Utkin , Andrei V. Konstantinov

Survival analysis is essential for clinical decision-making, as it allows practitioners to estimate time-to-event outcomes, stratify patient risk profiles, and guide treatment planning. Deep learning has revolutionized this field with…

Machine Learning · Computer Science 2026-02-03 Pinar Erbil , Alberto Archetti , Eugenio Lomurno , Matteo Matteucci

The modeling of time-to-event data, also known as survival analysis, requires specialized methods that can deal with censoring and truncation, time-varying features and effects, and that extend to settings with multiple competing events.…

Machine Learning · Statistics 2021-04-20 Andreas Bender , David Rügamer , Fabian Scheipl , Bernd Bischl

Introduction. We investigate the generalization ability of models built on datasets containing a small number of subjects, recorded in single study protocols. Next, we propose and evaluate methods combining these datasets into a single,…

Machine Learning · Computer Science 2023-12-05 Gideon Vos , Kelly Trinh , Zoltan Sarnyai , Mostafa Rahimi Azghadi

Sequential recommendation models the dynamics of a user's previous behaviors in order to forecast the next item, and has drawn a lot of attention. Transformer-based approaches, which embed items as vectors and use dot-product self-attention…

Information Retrieval · Computer Science 2022-03-08 Ziwei Fan , Zhiwei Liu , Alice Wang , Zahra Nazari , Lei Zheng , Hao Peng , Philip S. Yu

We develop a Bayesian tree ensemble model to estimate heterogeneous treatment effects in censored survival data with high-dimensional covariates. Instead of imposing sparsity through the tree structure, we place a horseshoe prior directly…

Methodology · Statistics 2026-05-08 Tijn Jacobs , Wessel N. van Wieringen , Stéphanie L. van der Pas

Time-to-event analysis is a branch of statistics that has increased in popularity during the last decades due to its many application fields, such as predictive maintenance, customer churn prediction and population lifetime estimation. In…

Machine Learning · Computer Science 2024-03-13 Camila Fernandez , Chung Shue Chen , Chen Pierre Gaillard , Alonso Silva

Survival analysis concerns the study of timeline data where the event of interest may remain unobserved (i.e., censored). Studies commonly record more than one type of event, but conventional survival techniques focus on a single event…

Machine Learning · Computer Science 2023-03-01 Hylke C. Donker , Harry J. M. Groen

Unsupervised outlier detection is attractive because it eliminates the need for labeled data. Moreover, forming multi-model ensembles can improve detection robustness. However, composing an ensemble without labeled data is challenging.…

Machine Learning · Computer Science 2026-05-19 Hong-Phuc Phan , Tuan-Anh Vu , Tung Kieu , Son Ha Xuan , Bin Yang , Christian S. Jensen

Applications of machine learning in healthcare often require working with time-to-event prediction tasks including prognostication of an adverse event, re-hospitalization or death. Such outcomes are typically subject to censoring due to…

Machine Learning · Computer Science 2022-08-04 Chirag Nagpal , Willa Potosnak , Artur Dubrawski

Observational studies are often conducted to estimate causal effects of treatments or exposures on event-time outcomes. Since treatments are not randomized in observational studies, techniques from causal inference are required to adjust…

Methodology · Statistics 2023-10-23 Han Ji , Arman Oganisian

Machine learning is becoming ubiquitous. From finance to medicine, machine learning models are boosting decision-making processes and even outperforming humans in some tasks. This huge progress in terms of prediction quality does not…

Machine Learning · Computer Science 2025-04-11 Marco Anisetti , Claudio A. Ardagna , Alessandro Balestrucci , Nicola Bena , Ernesto Damiani , Chan Yeob Yeun

Estimating heterogeneous treatment effects (HTEs) is crucial for personalized decision-making. However, this task is challenging in survival analysis, which includes time-to-event data with censored outcomes (e.g., due to study dropout). In…

Machine Learning · Computer Science 2025-05-20 Dennis Frauen , Maresa Schröder , Konstantin Hess , Stefan Feuerriegel

Survival analysis, which estimates the probability of event occurrence over time from censored data, is fundamental in numerous real-world applications, particularly in high-stakes domains such as healthcare and risk assessment. Despite…

Machine Learning · Computer Science 2025-05-26 Yu Liu , Weiyao Tao , Tong Xia , Simon Knight , Tingting Zhu

We study the problem of inferring heterogeneous treatment effects from time-to-event data. While both the related problems of (i) estimating treatment effects for binary or continuous outcomes and (ii) predicting survival outcomes have been…

Machine Learning · Computer Science 2022-01-25 Alicia Curth , Changhee Lee , Mihaela van der Schaar