Semi-supervised approaches to efficient evaluation of model prediction performance

被引:26
作者
Gronsbell, Jessica L. [1 ]
Cai, Tianxi [1 ]
机构
[1] Harvard Univ, Boston, MA 02115 USA
基金
美国国家卫生研究院;
关键词
Model evaluation; Perturbation resampling; Receiver operating characteristic curve; Risk prediction; Semi-supervised learning; ELECTRONIC MEDICAL-RECORDS; NONCONCAVE PENALIZED LIKELIHOOD; DISCOVERY RESEARCH; TESTS; INFERENCE; DISEASE; LASSO;
D O I
10.1111/rssb.12264
中图分类号
O21 [概率论与数理统计]; C8 [统计学];
学科分类号
020208 ; 070103 ; 0714 ;
摘要
In many modern machine learning applications, the outcome is expensive or time consuming to collect whereas the predictor information is easy to obtain. Semi-supervised (SS) learning aims at utilizing large amounts of unlabelled' data along with small amounts of labelled' data to improve the efficiency of a classical supervised approach. Though numerous SS learning classification and prediction procedures have been proposed in recent years, no methods currently exist to evaluate the prediction performance of a working regression model. In the context of developing phenotyping algorithms derived from electronic medical records, we present an efficient two-step estimation procedure for evaluating a binary classifier based on various prediction performance measures in the SS setting. In step I, the labelled data are used to obtain a non-parametrically calibrated estimate of the conditional risk function. In step II, SS estimates of the prediction accuracy parameters are constructed based on the estimated conditional risk function and the unlabelled data. We demonstrate that, under mild regularity conditions, the estimators proposed are consistent and asymptotically normal. Importantly, the asymptotic variance of the SS estimators is always smaller than that of the supervised counterparts under correct model specification. We also correct for potential overfitting bias in the SS estimators in finite samples with cross-validation and we develop a perturbation resampling procedure to approximate their distributions. Our proposals are evaluated through extensive simulation studies and illustrated with two real electronic medical record studies aiming to develop phenotyping algorithms for rheumatoid arthritis and multiple sclerosis.
引用
收藏
页码:579 / 594
页数:16
相关论文
共 46 条
  • [1] Improving Case Definition of Crohn's Disease and Ulcerative Colitis in Electronic Medical Records Using Natural Language Processing: A Novel Informatics Approach
    Ananthakrishnan, Ashwin N.
    Cai, Tianxi
    Savova, Guergana
    Cheng, Su-Chun
    Chen, Pei
    Perez, Raul Guzman
    Gainer, Vivian S.
    Murphy, Shawn N.
    Szolovits, Peter
    Xia, Zongqi
    Shaw, Stanley
    Churchill, Susanne
    Karlson, Elizabeth W.
    Kohane, Isaac
    Plenge, Robert M.
    Liao, Katherine P.
    [J]. INFLAMMATORY BOWEL DISEASES, 2013, 19 (07) : 1411 - 1420
  • [2] [Anonymous], 2006, Semi-Supervised Learning, DOI DOI 10.7551/MITPRESS/9780262033589.003.0003
  • [3] [Anonymous], 2004, Advances in neural information processing systems
  • [4] [Anonymous], 2008, Advances in Neural Information Processing Systems
  • [5] [Anonymous], 2009, Advances in neural information processing systems
  • [6] [Anonymous], 2000, Proceedings of the Seventeenth International Conference on Machine Learning
  • [7] The central role of receiver operating characteristic (ROC) curves in evaluating tests for the early detection of cancer
    Baker, SG
    [J]. JOURNAL OF THE NATIONAL CANCER INSTITUTE, 2003, 95 (07) : 511 - 515
  • [8] Baluja Shumeet., 1998, NIPS, P854
  • [9] Semi-supervised learning on Riemannian manifolds
    Belkin, M
    Niyogi, P
    [J]. MACHINE LEARNING, 2004, 56 (1-3) : 209 - 239
  • [10] Belkin M, 2006, J MACH LEARN RES, V7, P2399