Daily Arxiv

This is a page that curates AI-related papers published worldwide.
All content here is summarized using Google Gemini and operated on a non-profit basis.
Copyright for each paper belongs to the authors and their institutions; please make sure to credit the source when sharing.

Survival Analysis with Adversarial Regularization

Created by
  • Haebom

Author

Michael Potter, Stefano Maxenti, Michael Everett

Outline

This paper proposes a method to introduce adversarial robustness to improve the performance of survival analysis (SA) models using neural networks (NNs). While neural networks are utilized to overcome the limitations of conventional generalized linear models, which often fail to capture complex data patterns, we propose an adversarial regularization-based loss function to address the performance degradation caused by data uncertainty. We utilize the CROWN-IBP technique to reduce the computational cost of the min-max optimization problem. Experimental results using 10 SurvSet datasets demonstrate that the proposed method (SAWAR) outperforms existing adversarial learning methods and state-of-the-art deep SA models in terms of NegLL, IBS, and CI metrics, achieving up to 150% performance improvement over baseline models. This demonstrates that the proposed method mitigates data uncertainty and improves generalization across diverse datasets.

Takeaways, Limitations

Takeaways:
Suggesting the possibility of improving the performance and reliability of neural network-based survival analysis models through adversarial robustness.
Proof of SAWAR's superiority, demonstrating consistent performance improvements across diverse datasets.
A novel approach to ensure model robustness to data uncertainty is presented.
Expanding the potential applications of neural networks in survival analysis.
Limitations:
The dataset used is limited to SurvSet, so verification of generalization performance on other types of datasets is necessary.
The computational cost of the CROWN-IBP technique may still be high. Research into more efficient optimization techniques is needed.
It's possible that we've focused only on certain types of data uncertainty. Further research is needed on other types of uncertainty.
👍