Structured Probabilistic Coding
Dou Hu, Lingwei Wei, Yaxin Liu, Wei Zhou, Songlin Hu
摘要
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.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
引用它的顶会 Paper4
- Structural Entropy Guided Probabilistic CodingXiang Huang, Hao Peng, Li Sun, Hui Lin 等AAAI 2025 · 被引用 4 次
- An Information-theoretic Multi-task Representation Learning Framework for Natural Language UnderstandingDou Hu, Lingwei Wei, Wei Zhou, Songlin HuAAAI 2025 · 被引用 3 次
- 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
它引用的顶会 Paper10
- Supervised Contrastive LearningPrannay Khosla, Piotr Teterwak, Chen Wang, Aaron Sarna 等NeurIPS 2020 · 被引用 7,049 次
- Supervised Contrastive Learning for Pre-trained Language Model Fine-tuningBeliz Gunel, Jingfei Du, Alexis Conneau, Veselin StoyanovICLR 2021 · 被引用 595 次
- Probabilistic Face EmbeddingsYichun Shi, Anil K. JainICCV 2019 · 被引用 362 次
- InfoBERT: Improving Robustness of Language Models from An Information Theoretic PerspectiveBoxin Wang, Shuohang Wang, Yu Cheng, Zhe Gan 等ICLR 2021 · 被引用 132 次
- Variational Information Bottleneck for Effective Low-Resource Fine-TuningRabeeh Karimi Mahabadi, Yonatan Belinkov, James HendersonICLR 2021 · 被引用 88 次
相关 Paper
- Estimating Calibrated Individualized Survival Curves with Deep LearningFahad Kamran, Jenna WiensAAAI 2021 · 被引用 21 次
- NeuralSurv: Deep Survival Analysis with Bayesian Uncertainty QuantificationMélodie Monod, Alessandro Micheli, Samir BhattNeurIPS 2025 · 被引用 5 次
- 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 次
- Explicitly Modeling Censoring Produces Superior Survival PredictorsShi-ang Qi, Yakun Yu, Russell GreinerICML 2026
