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 (WAFL) held at the European Conference on Machine Learning and Principles and Practice of Knowledge Discovery in Database (PKDD and ECML combined from 2008)
Vilnius, Lituania
9-13 settembre 2024
Machine Learning and Principles and Practice of Knowledge Discovery in Databases: International Workshops of ECML PKDD 2024
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