paper-with-me

Papers

Prediction-Powered Inference with Imputed Covariates and Nonuniform Sampling

2025-01-30 · Dan M. Kluger, Kerri Lu, Tijana Zrnic, Sherrie Wang, Stephen Bates

Machine learning models are increasingly used to produce predictions that serve as input data in subsequent statistical analyses. For example, computer vision predictions of economic and environmental indicators based on satellite imagery are used in downstream regressions; similarly, language models are widely used to approximate human ratings and opinions in social science research. However, failure to properly account for errors in the machine learning predictions renders standard statistical procedures invalid. Prior work uses what we call the Predict-Then-Debias estimator to give valid confidence intervals when machine learning algorithms impute missing variables, assuming a small complete sample from the population of interest. We expand the scope by introducing bootstrap confidence intervals that apply when the complete data is a nonuniform (i.e., weighted, stratified, or clustered) sample and to settings where an arbitrary subset of features is imputed. Importantly, the method can be applied to many settings without requiring additional calculations. We prove that these confidence intervals are valid under no assumptions on the quality of the machine learning model and are no wider than the intervals obtained by methods that do not use machine learning predictions.

📄 PDF Abstract BibTeX arXiv:2501.18577

Code (3)

DanKluger/Predict-Then-Debias_Bootstrap 공식 구현
dankluger/ptdboot 공식 구현
earth-intelligence-lab/ppi_regression 공식 구현

Tasks

valid

Similar Papers 제목 키워드 기반

Semi-Supervised Treatment Effect Estimation with Unlabeled Covariates for Prediction-Powered Causal Inference

2025-11-11 · Masahiro Kato arxiv

This study investigates treatment effect estimation in the semi-supervised setting, also can be interpreted as prediction-powered inference. In our setting, we can use not only the standard triple of covariates, treatmen…

Causal Inference

Predictions as Surrogates: Revisiting Surrogate Outcomes in the Age of AI

2025-01-16 · Wenlong Ji, Lihua Lei, Tijana Zrnic

We establish a formal connection between the decades-old surrogate outcome model in biostatistics and economics and the emerging field of prediction-powered inference (PPI). The connection treats predictions from pre-tra…

Prediction-Powered Conditional Inference

2026-03-05 · Yang Sui, Jin Zhou, Hua Zhou, Xiaowu Dai arxiv

We study prediction-powered conditional inference in the setting where labeled data are scarce, unlabeled covariates are abundant, and a black-box machine-learning predictor is available. The goal is to perform statistic…

Intermittent Inference with Nonuniformly Compressed Multi-Exit Neural Network for Energy Harvesting Powered Devices

2020-04-23 · Yawen Wu, Zhepeng Wang, Zhenge Jia, Yiyu Shi 외

This work aims to enable persistent, event-driven sensing and decision capabilities for energy-harvesting (EH)-powered devices by deploying lightweight DNNs onto EH-powered devices. However, harvested energy is usually w…

Pre-Trained AI Model Assisted Online Decision-Making under Missing Covariates: A Theoretical Perspective

2025-07-10 · Haichen Hu, David Simchi-Levi arxiv

We study a sequential contextual decision-making problem in which certain covariates are missing but can be imputed using a pre-trained AI model. From a theoretical perspective, we analyze how the presence of such a mode…