arXiv Science⌕ Search
Home/Archive/arXiv

arXiv · 2610.37227

Doubly Robust Deep Kernel Tests for Structured Outcomes

Robin Leman·Bariscan Bozkurt·Sofiane Tanji·Arthur Gretton·Houssam Zenati
View original source ↗Read source document ↗Export metadata ↓

Abstract

We propose a test of whether two interventions induce the same outcome distribution when outcomes are structured and high-dimensional, such as images, text and graphs. We represent the interventional distributions as mean embeddings on learned deep features, such that the treatment effect is a doubly robust representation of the difference in mean embeddings. Our test, DR-KTE-Deep, optimises the learned features so as to maximise an estimated power criterion on this treatment effect representation. To motivate our learning objective, we show that the power criterion characterises the asymptotic signal-to-noise ratio of the test under fixed alternatives, and we derive a uniform finite-sample bound for its regularised estimate. We show that, with the learned kernel fixed, a held-out test is asymptotically valid under the null given suitable nuisance-estimation rates. Empirically, DR-KTE-Deep achieves substantially higher power than fixed and bandwidth-optimised kernel baselines across image, text and graph outcomes, while maintaining well-calibrated Type-I error.

Explore related subjects

stat.MLcs.LG

Keep this discovery

Explore connections, maps & timelines

BibTeXRIS

Robin Leman, Bariscan Bozkurt, Sofiane Tanji, Arthur Gretton, Houssam Zenati. 2026-10-27. Doubly Robust Deep Kernel Tests for Structured Outcomes. https://arxiv.org/abs/2610.37227

Cite the original work for its findings. Save a collection to share your selection of sources.

KEEP EXPLORING

Related papers

arXiv2026

Non-asymptotic Estimation of Sparse Partial Correlation via Joint Partial Regression

Partial correlation analysis is widely used in modern multivariate analysis. However, prior methods lack non-asymptotic guarantees and have slow asymptotic estimation rates. We present a method for simultaneously estimating sparse high-dimensional partial correlation matrices and inverse covariance with non-asymptotic rates. The method is a two-stage estimation which exploits the connection between the inverse covariance matrix and linear regression estimation method while enforcing positive semi-definiteness simultaneously. An efficient proximal splitting algorithm for numerically computing the estimate is also derived, and we provide a fast implementation of the algorithm. The effectiveness of the proposed method is demonstrated on both synthetic and real-world data.

stat.ML↗
arXiv2026

Foreclassing: From Forecasts to Decisions

Time series forecasts are widely used to inform decisions under uncertainty. In many applications, these decisions involve classifying a time series based on outcomes that have yet to be observed. In this paper, we introduce Foreclassing, a new machine learning problem in which the class label of a time series is defined as a function of its future observations. We provide a mathematical formulation of Foreclassing and establish several theoretical results characterizing the problem. These results describe how additional future information affects Bayes classification risk, provide bounds on the classification error of learned Foreclassing models, and demonstrate how future observations used during training can yield finite-sample improvements over conventional classifiers. To provide a benchmark for future research, we propose ForeClassNet, a family of neural network models that jointly perform forecasting and classification while accounting for uncertainty in future outcomes. We investigate a variety of model architectures and evaluate their performance against conventional time series classifiers and forecasting-then-classify approaches. Experiments on datasets spanning weather, energy, and finance demonstrate the potential of Foreclassing to improve classification performance, probabilistic prediction, and downstream decision-making. Together, these theoretical and empirical contributions establish a foundation for future research on Foreclassing.

stat.ML↗
arXiv2026

Expert-Guided Class-Conditional Goodness-of-Fit Scores for Interpretable Classification with Informative Missingness

We study a classification problem with three key challenges: pervasive informative missingness, the integration of partial prior expert knowledge into the learning process, and the need for interpretable decision rules. We propose a framework that encodes prior knowledge through an expert-guided class-conditional model for one or more classes, and use this model to construct a small set of interpretable goodness-of-fit features. The features quantify how well the observed data agree with the expert model, isolating the contributions of different aspects of the data, including both observed and missing components. These features are combined with a few transparent auxiliary summaries in a simple discriminative classifier, resulting in a decision rule that is easy to inspect and justify. We develop and apply the framework in the context of seismic monitoring used to assess compliance with the Comprehensive Nuclear-Test-Ban Treaty. We show that the method has strong potential as a transparent screening tool, reducing workload for expert analysts. A simulation designed to isolate the contribution of the proposed framework shows that this interpretable expert-guided method can even outperform strong standard machine-learning classifiers, particularly when training samples are small.

stat.ML↗