Beyond Point Predictions: Distribution-Informed Prediction-Powered Inference
Abstract
Prediction-powered inference (PPI) typically relies on point predictions on unlabeled data. When predictive distributions are available as in a wide range of applications, including predictions from LLMs, we introduce distribution-informed prediction-powered inference (DiPPI), a general framework for further improving statistical efficiency by using predictive distributions as auxiliary information in the spirit of PPI. We characterize the optimal use of this information through score calibration, derive the oracle efficiency for a finite-dimensional representation of the predictive distribution, and provide theoretical guarantees for positive learning with the cross-fitted DiPPI estimator. Through simulations and three real-data applications, we show that DiPPI achieves better efficiency than PPI methods based on point predictions. These gains arise when the predictive distribution contains score-relevant information that is partially lost in point predictions.
Related
BibTeX
@misc{zhang2026beyond,
title = {{Beyond Point Predictions: Distribution-Informed Prediction-Powered Inference}},
author = {Maoyu Zhang and Jingfei Zhang and Xuming He},
howpublished = {arXiv},
year = {2026},
month = sep,
doi = {10.48550/arxiv.2609.32006},
eprint = {2609.32006},
archivePrefix = {arXiv},
url = {https://arxiv.org/abs/2609.32006},
}
RIS
TY - UNPB
TI - Beyond Point Predictions: Distribution-Informed Prediction-Powered Inference
AU - Maoyu Zhang
AU - Jingfei Zhang
AU - Xuming He
PY - 2026
DA - 2026/09/25
PB - arXiv
DO - 10.48550/arxiv.2609.32006
UR - https://arxiv.org/abs/2609.32006
AB - Prediction-powered inference (PPI) typically relies on point predictions on unlabeled data. When predictive distributions are available as in a wide range of applications, including predictions from LLMs, we introduce distribution-informed prediction-powered inference (DiPPI), a general framework for further improving statistical efficiency by using predictive distributions as auxiliary information in the spirit of PPI. We characterize the optimal use of this information through score calibration, derive the oracle efficiency for a finite-dimensional representation of the predictive distribution, and provide theoretical guarantees for positive learning with the cross-fitted DiPPI estimator. Through simulations and three real-data applications, we show that DiPPI achieves better efficiency than PPI methods based on point predictions. These gains arise when the predictive distribution contains score-relevant information that is partially lost in point predictions.
ER -