Prediction Intervals and Group Variable Importance for Classification Models in University Enrollment Yield

Read full text
Authors: Xiaohui Yin, Yingfa Xie, Jun Yan, Siyan Wang, Pang-Yu Liu, Jeffrey Gagnon, Elton Thompson, Lawrence Walsh, Nathan Fuerst, Ming-Hui Chen
Cited by: 1
References: 25

Abstract

Accurate forecasting of student yield is critical for academic institutions because enrollment projections directly affect resource allocation, course offerings, housing capacity, and budget decisions. Existing enrollment models primarily focus on predicting individual matriculation probabilities, whereas institutional planning depends on accurate prediction of the aggregate enrollment count together with its uncertainty quantification. Here we develop statistical methods for aggregate enrollment prediction and interval estimation under logistic, regularized regression, tree-based, and ensemble classification models. For unpenalized logistic regression, we derive an asymptotic prediction interval and a parametric-simulation interval that propagates coefficient uncertainty. For regularized and tree-based learners, we develop a model-agnostic bootstrap interval that captures selection, estimation, and predictive uncertainty in a unified framework. To interpret predictive structure, we introduce partial and marginal AUC measures to quantify the unique and standalone contributions of feature groups. Simulation studies show that the proposed intervals attain close-to-nominal coverage under well-calibrated and bootstrap-stable learners. Applied to University of Connecticut in-state freshman cohorts from 2020–2025, the proposed intervals contain the observed 2025 enrollment count while identifying the feature groups that contribute most strongly to the enrollment prediction.

Fields of Study & Subjects

Online Learning and Analytics Advanced Causal Inference Techniques Higher Education Research Studies

Publication details

Journal / source
Journal of Data Science
Publisher
School of Statistics, Renmin University of China
Pages
523–543
Publication date
1 January 2026
Language
EN