Component-Wise Composite Likelihood Distillation for Censored Time-to-Event Data
Feiyang Deng, Lingfeng Luo, Jiayu Zhou, Kevin He
Abstract
Accurate survival modeling in biomedical studies is often hindered by rare events, limited effective sample sizes, and settings with limited or partially observed information (e.g., covariates of interest that are difficult or expensive to collect, highly structured sampling designs, or nuisance parameters omitted by conditioning). Knowledge distillation can leverage external predictive information without sharing individual-level data, but existing approaches are largely built for fully specified likelihoods or probability-based survival models and do not extend to settings where outcome distributions are only partially specified. To address this challenge, we propose a knowledge distillation framework based on a composite-likelihood Kullback--Leibler divergence that aligns teacher and student models within components. Our key insight is that, although composite likelihoods do not define a global outcome distribution, each likelihood component induces a well-defined probability model on its restricted outcome space, enabling a principled KL divergence. Simulation studies and biomedical case studies show improved discrimination and predictive accuracy in rare-event, heterogeneous settings without requiring access to external individual-level data.
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.
Builds on2
Related papers
- Discrete Survival Knowledge Distillation for Competing Risks AnalysisFeiyang Deng, Lingfeng Luo, Di Wang, Qinmengge Li et al.ICML 2026
- Knowledge Distillation as Semiparametric InferenceTri Dao, Govinda M. Kamath, Vasilis Syrgkanis, Lester MackeyICLR 2021 · 4 citations
- Generative Diffusion Prior Distillation for Long-Context Knowledge TransferNilushika Udayangani Hewa Dehigahawattage, Nandakishor Desai, Marimuthu PalaniswamiICLR 2026
- Knowledge Distillation with Auxiliary VariableBo Peng, Zhen Fang, Guangquan Zhang, Jie LuICML 2024 · 7 citations
- Distilled Prompt Learning for Incomplete Multimodal Survival PredictionYingxue Xu, Fengtao Zhou, Chenyu Zhao, Yihui Wang et al.CVPR 2025
