Estimation Beyond Data Reweighting: Kernel Method of Moments
Heiner Kremer, Yassine Nemmour, Bernhard Schölkopf, Jia-Jie Zhu
Abstract
Moment restrictions and their conditional counterparts emerge in many areas of machine learning and statistics ranging from causal inference to reinforcement learning. Estimators for these tasks, generally called methods of moments, include the prominent generalized method of moments (GMM) which has recently gained attention in causal inference. GMM is a special case of the broader family of empirical likelihood estimators which are based on approximating a population distribution by means of minimizing a -divergence to an empirical distribution. However, the use of -divergences effectively limits the candidate distributions to reweightings of the data samples. We lift this long-standing limitation and provide a method of moments that goes beyond data reweighting. This is achieved by defining an empirical likelihood estimator based on maximum mean discrepancy which we term the kernel method of moments (KMM). We provide a variant of our estimator for conditional moment restrictions and show that it is asymptotically first-order optimal for such problems. Finally, we show that our method achieves competitive performance on several conditional moment restriction tasks.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 7383c3f2-253a-4cbc-802b-4b8a8c75e8f4Cited by top-tier papers3
- Posterior Sampling Based on Gradient Flows of the MMD with Negative Distance KernelPaul Hagemann, Johannes Hertrich, Fabian Altekrüger, Robert Beinert et al.ICLR 2024 · 32 citations
- Functional Generalized Empirical Likelihood Estimation for Conditional Moment RestrictionsHeiner Kremer, Jia-Jie Zhu, Krikamol Muandet, Bernhard SchölkopfICML 2022 · 9 citations
- Geometry-Aware Instrumental Variable RegressionHeiner Kremer, Bernhard SchölkopfICML 2024
Builds on7
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- Minimax Estimation of Conditional Moment ModelsNishanth Dikkala, Greg Lewis, Lester Mackey, Vasilis SyrgkanisNeurIPS 2020 · 125 citations
- Learning Deep Features in Instrumental Variable RegressionLiyuan Xu, Yutian Chen, Siddarth Srinivasan, Nando de Freitas et al.ICLR 2021 · 85 citations
- Non-parametric Models for Non-negative FunctionsUlysse Marteau-Ferey, Francis R. Bach, Alessandro RudiNeurIPS 2020 · 65 citations
- Efficient Policy Learning from Surrogate-Loss Classification ReductionsAndrew Bennett, Nathan KallusICML 2020 · 20 citations
Related papers
- Learning Causal Models from Conditional Moment Restrictions by Importance WeightingMasahiro Kato, Masaaki Imaizumi, Kenichiro McAlinn, Shota Yasui et al.ICLR 2022 · 7 citations
- Distributional Reinforcement Learning via Moment MatchingThanh Nguyen-Tang, Sunil Gupta, Svetha VenkateshAAAI 2021 · 44 citations
- Understanding Deep Generative Models with Generalized Empirical LikelihoodsSuman V. Ravuri, Mélanie Rey, Shakir Mohamed, Marc Peter DeisenrothCVPR 2023
- Learning from Biased Data: A Semi-Parametric ApproachPatrice Bertail, Stéphan Clémençon, Yannick Guyonvarch, Nathan NoiryICML 2021 · 6 citations
- Optimal Rates for Regularized Conditional Mean Embedding LearningZhu Li, Dimitri Meunier, Mattes Mollenhauer, Arthur GrettonNeurIPS 2022 · 69 citations
