This work proposes FedSurvBoost, a federated learning pipeline for survival analysis based on the AdaBoost.F algorithm, which iteratively aggregates the best local weak hypotheses. Our method extends AdaBoost.F by removing the dependence on the number of classes coefficient from the computation of the weights of the best model. This makes it suitable for regression tasks, such as survival analysis. We show the effectiveness of our approach by comparing it with state-of-the-art methods, specifically developed for survival analysis problems, on two common survival datasets. Our code is available at https://github.com/oussamaHarrak/FedSurvBoost .

Federated AdaBoost for Survival Analysis

Bruno Casella
Co-first
;
Samuele Fonio
Co-first
;
Piero Fariselli;Gianluca Mittone;Cesare Rollo;Tiziana Sanavia;Marco Aldinucci
Last
In corso di stampa

Abstract

This work proposes FedSurvBoost, a federated learning pipeline for survival analysis based on the AdaBoost.F algorithm, which iteratively aggregates the best local weak hypotheses. Our method extends AdaBoost.F by removing the dependence on the number of classes coefficient from the computation of the weights of the best model. This makes it suitable for regression tasks, such as survival analysis. We show the effectiveness of our approach by comparing it with state-of-the-art methods, specifically developed for survival analysis problems, on two common survival datasets. Our code is available at https://github.com/oussamaHarrak/FedSurvBoost .
In corso di stampa
2nd Workshop on Advancements in Federated Learning
Vilnius
13/09/2024
Proceedings of the ECML-PKDD Workshops
Springer
1
9
Oussama Harrak; Bruno Casella; Samuele Fonio; Piero Fariselli; Gianluca Mittone; Cesare Rollo; Tiziana Sanavia; Marco Aldinucci
File in questo prodotto:
File Dimensione Formato  
WAFL___ECMLPKDD___FedSurvBoost.pdf

Accesso aperto

Tipo di file: PREPRINT (PRIMA BOZZA)
Dimensione 476.75 kB
Formato Adobe PDF
476.75 kB Adobe PDF Visualizza/Apri

I documenti in IRIS sono protetti da copyright e tutti i diritti sono riservati, salvo diversa indicazione.

Utilizza questo identificativo per citare o creare un link a questo documento: https://hdl.handle.net/2318/2030498
Citazioni
  • ???jsp.display-item.citation.pmc??? ND
  • Scopus ND
  • ???jsp.display-item.citation.isi??? ND
social impact