Structured Probabilistic Coding
Dou Hu, Lingwei Wei, Yaxin Liu, Wei Zhou, Songlin Hu
Abstract
Survival analysis is critical in many real-world do-mains where predicting time-to-event outcomes under censoring is essential. While deep learning has advanced survival modeling, existing methods struggle to jointly optimize discrimination, cali-bration, and uncertainty quantification. Probabilistic approaches like variational autoencoders offer uncertainty estimates but suffer from information loss due to encoder-decoder architectures and lack principled treatment of censored data. We propose Survival Structured Probabilistic Coding (Survival-SPC), a novel framework with two key mechanisms. First, an encoder-only architecture directly encodes features to probabilistic hazard representations, reducing information loss while preserving uncer-tainty. Second, censoring-aware structured regularization lever-ages partial information from censored observations to encourage diversity in latent representations. Unlike previous approaches, Survival-SPC enables efficient end-to-end training of calibrated survival distributions. Comprehensive experiments on six real-world datasets demonstrate superior performance, improving concordance index by up to 4.2% over strongest baselines while providing well-calibrated estimates. The method shows particular advantages under limited data and high censoring, establishing a robust solution for clinical applications. Code is available at Suvival-SPC Repository.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext edf364a0-32ce-4b7d-b78d-aa7b1afcffbeCited by top-tier papers4
- Structural Entropy Guided Probabilistic CodingXiang Huang, Hao Peng, Li Sun, Hui Lin et al.AAAI 2025 · 4 citations
- An Information-theoretic Multi-task Representation Learning Framework for Natural Language UnderstandingDou Hu, Lingwei Wei, Wei Zhou, Songlin HuAAAI 2025 · 3 citations
- Representation Learning with Conditional Information Flow MaximizationDou Hu, Lingwei Wei, Wei Zhou, Songlin HuACL 2024
- Impartial Multi-task Representation Learning via Variance-invariant Probabilistic DecodingDou Hu, Lingwei Wei, Wei Zhou, Songlin HuACL 2025
Builds on10
- Supervised Contrastive LearningPrannay Khosla, Piotr Teterwak, Chen Wang, Aaron Sarna et al.NeurIPS 2020 · 7,049 citations
- Supervised Contrastive Learning for Pre-trained Language Model Fine-tuningBeliz Gunel, Jingfei Du, Alexis Conneau, Veselin StoyanovICLR 2021 · 595 citations
- Probabilistic Face EmbeddingsYichun Shi, Anil K. JainICCV 2019 · 362 citations
- InfoBERT: Improving Robustness of Language Models from An Information Theoretic PerspectiveBoxin Wang, Shuohang Wang, Yu Cheng, Zhe Gan et al.ICLR 2021 · 132 citations
- Variational Information Bottleneck for Effective Low-Resource Fine-TuningRabeeh Karimi Mahabadi, Yonatan Belinkov, James HendersonICLR 2021 · 88 citations
Related papers
- Estimating Calibrated Individualized Survival Curves with Deep LearningFahad Kamran, Jenna WiensAAAI 2021 · 21 citations
- NeuralSurv: Deep Survival Analysis with Bayesian Uncertainty QuantificationMélodie Monod, Alessandro Micheli, Samir BhattNeurIPS 2025 · 5 citations
- Censor Dependent Variational InferenceChuanhui Liu, Xiao WangICML 2025
- Toward a Well-Calibrated Discrimination via Survival Outcome-Aware Contrastive LearningDongjoon Lee, Hyeryn Park, Changhee LeeNeurIPS 2024 · 6 citations
- Explicitly Modeling Censoring Produces Superior Survival PredictorsShi-ang Qi, Yakun Yu, Russell GreinerICML 2026
