A Deep Attention-Multiple Instance Learning Framework to Predict Survival of Soft-Tissue Sarcoma from Whole Slide Images

Van Linh Le, Audrey Michot, Amandine Crombé, Carine Ngo, Charles Honoré, Jean Michel Coindre, Olivier Saut, Francois Le-Loarer

    Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

    Abstract

    Soft-tissue sarcomas are heterogeneous cancers of the mesenchymal lineage that can develop anywhere in the body. A precise prediction of sarcomas patients’ prognosis is critical for clinicians to define an adequate treatment plan. In this paper, we proposed an end-to-end Deep learning framework via Multiple Instance Learning (MIL), Deep Attention-MIL framework, for the survival predictions: Overall survival (OS), Metastasis-free survival (MFS), and Local-recurrence free survival (LRFS) of sarcomas patients, by studying the features from Whole Slide Images (WSIs) of their tumors. The Deep Attention-MIL framework consists of three steps: tiles selection from the WSIs to choose the relevant tiles for the study; tiles feature extraction by using a pre-trained deep learning model; and a Deep Attention-MIL model to predict the risk score for each patient via MIL approach. The risk scores outputted from the Deep Attention-MIL model are used to divide the patients into low/high-risk groups and predict survival time. The framework was trained and validated on a local dataset including 220 patients, then it was used to predict the survival for 48 patients in an external validation dataset. The experiments showed the proposed framework yielded satisfactory and promising results and contributed to accurate cancer survival predictions on both the validation and external testing datasets: By using the WSIs feature only, we obtained an average C-index (of 5-fold cross-validation) of 0.6901, 0.7179, and 0.6211 for OS, MFS, and LRFS tasks on the validation dataset, respectively. On the external testing dataset, these scores are 0.6294, 0.682, and 0.76 for the three tasks (OS, MFS, LRFS), respectively. By adding the clinical features, these scores have been improved both on validation and external testing datasets. We obtained an average C-index of 0.7835/0.6378, 0.7389/0.6885, and 0.6883/0.7272 for the three tasks (OS, MFS, LRFS) on validation/external testing datasets.

    Original languageEnglish
    Title of host publicationCancer Prevention Through Early Detection - 2nd International Workshop, CaPTion 2023, Held in Conjunction with MICCAI 2023, Proceedings
    EditorsSharib Ali, Fons van der Sommen, Maureen van Eijnatten, Iris Kolenbrander, Bartłomiej W. Papież, Yueming Jin
    PublisherSpringer Science and Business Media Deutschland GmbH
    Pages3-16
    Number of pages14
    ISBN (Print)9783031453496
    DOIs
    Publication statusPublished - 1 Jan 2023
    Event2nd International Workshop on Cancer Prevention through early detecTion, CaPTion 2023 - Vancover, Canada
    Duration: 12 Oct 202312 Oct 2023

    Publication series

    NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
    Volume14295 LNCS
    ISSN (Print)0302-9743
    ISSN (Electronic)1611-3349

    Conference

    Conference2nd International Workshop on Cancer Prevention through early detecTion, CaPTion 2023
    Country/TerritoryCanada
    CityVancover
    Period12/10/2312/10/23

    Keywords

    • Deep Attention model
    • Multiple Instance Learning
    • Soft-tissue sarcoma
    • Survival prediction
    • Whole Slide Image

    Cite this