Skip to content

abdoush/SurVED

Repository files navigation

The Concordance Index Decomposition

C-index (CI) is a weighted harmonic average of the C-indices defined for the subsets ee (events vs. events) and ec (events vs. censored cases).

$\frac{1}{CI} = \alpha \frac{1}{CI_{ee}} + (1 - \alpha) \frac{1}{CI_{ec}}$

To use the C-index decompostion, download the file Utils/metrics.py and use the function c_index_decomposition. The function will return the following terms:

  • Cee: The C-index of the ee pairs.
  • Cec: The C-index of the ec pairs.
  • alpha: The weight alpha.
  • alpha_deviation: The deviation from the optimal alpha.
  • C: The total C-index.

For more details, see the full paper The Concordance Index decomposition: A measure for a deeper understanding of survival prediction models

BibTeX Citation

@article{ALABDALLAH2024102781,
   title = {The Concordance Index decomposition: A measure for a deeper understanding of survival prediction models},
   journal = {Artificial Intelligence in Medicine},
   volume = {148},
   pages = {102781},
   year = {2024},
   issn = {0933-3657},
   doi = {https://doi.org/10.1016/j.artmed.2024.102781},
   url = {https://www.sciencedirect.com/science/article/pii/S093336572400023X},
   author = {Abdallah Alabdallah and Mattias Ohlsson and Sepideh Pashami and Thorsteinn Rögnvaldsson},
   keywords = {Survival analysis, Evaluation metric, Concordance Index, Variational encoder–decoder}
}

SurVED

Survival Analysis with Variational Encoder Decoder.

Reproducing the results

SurVED

  • Run the file surved_final_test.py to reproduce the 100-fold test results of the four datasets.
  • Run the file surved_change_censoring.py to reproduce the SurVED model results on the SUPPORT dataset with changing censoring levels.

DeepHit:

  • Run the file OtherModels\DeepHit\deephit_final_test.py to reproduce the 100-fold test results of the four datasets.
  • Run the file OtherModels\DeepHit\deephit_change_censoring.py to reproduce the DeepHit model results on the SUPPORT dataset with changing censoring levels.

DeepSurv:

  • Run the file OtherModels\DeepSurv\deepsurv_final_test.py to reproduce the 100-fold test results of the four datasets.
  • Run the file OtherModels\DeepSurv\deepsurv_change_censoring.py to reproduce the DeepSurv model results on the SUPPORT dataset with changing censoring levels.

RSF (Random Survival Forest)

  • Run the file OtherModels\RSF\rsf_final_test.py to reproduce the 100-fold test results of the four datasets.
  • Run the file OtherModels\RSF\rsf_change_censoring.py to reproduce the RSF model results on the SUPPORT dataset with changing censoring levels.

CPH (Cox Proportional Hazard)

  • Run the file OtherModels\CPH\cph_final_test.py to reproduce the 100-fold test results of the four datasets.
  • Run the file OtherModels\CPH\cph_change_censoring.py to reproduce the CPH model results on the SUPPORT dataset with changing censoring levels.

DATE:

  • Run the file OtherModels\DATE\date_final_test.py to reproduce the 100-fold test results of the four datasets.

  • Run the file OtherModels\DATE\date_change_censoring.py to reproduce the DATE model results on the SUPPORT dataset with changing censoring levels.

    Note: DATE repository should be downloaded from DATE_Repo and placed in the same folder

VSI:

  • Run the file OtherModels\VSI\vsi_final_test.py to reproduce the 100-fold test results of the four datasets.

  • Run the file OtherModels\VSI\vsi_change_censoring.py to reproduce the VSI model results on the SUPPORT dataset with changing censoring levels.

    Note: VSI repository should be downloaded from VSI_Repo and placed in the same folder

Copies of the four datasets are provided in the Data folder for convenience.

Releases

No releases published

Packages

No packages published

Languages