Maxway CRT: improving the robustness of the model-X inference
Journal of the Royal Statistical Society: Series B (Statistical Methodology) · 2023 · Volume 85 · Issue 5
Abstract
The model-X conditional randomisation test (CRT) is a flexible and powerful testing procedure for testing the hypothesis ⫫ X ⫫ Y ∣ Z . However, it requires perfect knowledge of X ∣ Z and may lose its validity when there is an error in modelling X ∣ Z . This problem is even more severe when Z is of high dimensionality. In response to this, we propose the Maxway CRT, which learns the distribution of Y ∣ Z and uses it to calibrate the resampling distribution of X to gain robustness to the error in modelling X. We prove that the type-I error inflation of the Maxway CRT can be controlled by the learning error for a low-dimensional adjusting model plus the product of learning errors for X ∣ Z and Y ∣ Z , interpreted as an ‘almost doubly robust’ property. Based on this, we develop implementing algorithms of the Maxway CRT in practical scenarios including (surrogate-assisted) semi-supervised learning (SA-SSL) and transfer learning (TL). Through simulations, we demonstrate that the Maxway CRT achieves significantly better type-I error control than existing model-X inference approaches while preserving similar powers. Finally, we apply our methodology to two real examples of SA-SSL and TL.
Keywords
conditional randomisation test, double robustness, machine learning, semi-supervised learning, surrogate, transfer learning
Enable JavaScript for citation tools and additional paper interactions.