Learning U-Statistics with Active Inference

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Wang, Xiaoning, Huo, Yuyang, Peng, Liuhua, Zou, Changliang
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910211405512704
author Wang, Xiaoning
Huo, Yuyang
Peng, Liuhua
Zou, Changliang
author_facet Wang, Xiaoning
Huo, Yuyang
Peng, Liuhua
Zou, Changliang
contents $U$-statistics play a central role in statistical inference. In many modern applications, however, acquiring the labels required for $U$-statistics is costly. Motivated by recent advances in active inference, we develop an active inference framework for $U$-statistics that selectively queries informative labels to improve estimation efficiency under a fixed labeling budget, while preserving valid statistical inference. Our approach is built on the augmented inverse probability weighting $U$-statistic, which is designed to incorporate the sampling rule and machine learning predictions. We characterize the optimal sampling rule that minimizes its variance and design practical sampling strategies. We further extend the framework to $U$-statistic-based empirical risk minimization. Experiments on real datasets demonstrate substantial gains in estimation efficiency over baseline methods, while maintaining target coverage.
format Preprint
id arxiv_https___arxiv_org_abs_2605_11638
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Learning U-Statistics with Active Inference
Wang, Xiaoning
Huo, Yuyang
Peng, Liuhua
Zou, Changliang
Machine Learning
$U$-statistics play a central role in statistical inference. In many modern applications, however, acquiring the labels required for $U$-statistics is costly. Motivated by recent advances in active inference, we develop an active inference framework for $U$-statistics that selectively queries informative labels to improve estimation efficiency under a fixed labeling budget, while preserving valid statistical inference. Our approach is built on the augmented inverse probability weighting $U$-statistic, which is designed to incorporate the sampling rule and machine learning predictions. We characterize the optimal sampling rule that minimizes its variance and design practical sampling strategies. We further extend the framework to $U$-statistic-based empirical risk minimization. Experiments on real datasets demonstrate substantial gains in estimation efficiency over baseline methods, while maintaining target coverage.
title Learning U-Statistics with Active Inference
topic Machine Learning
url https://arxiv.org/abs/2605.11638