Search papers, labs, and topics across Lattice.
The paper introduces `genriesz`, a Python package for automating debiased machine learning using generalized Riesz regression, which unifies various causal inference and density estimation techniques. The package implements automatic regressor balancing (ARB) to ensure moment-matching optimality. `genriesz` offers a modular interface for specifying target functionals, representer models, and Bregman generators, and it returns various estimators with statistical inference tools, demonstrating its utility for ATE, ATT, and average marginal effect estimation.
Forget hand-tuning causal inference pipelines: `genriesz` automates debiased machine learning by intelligently balancing regressors, offering a unified framework for estimating causal effects with statistical guarantees.
Efficient estimation of causal and structural parameters can be automated using the Riesz representation theorem and debiased machine learning (DML). We present genriesz, an open-source Python package that implements automatic DML and generalized Riesz regression, a unified framework for estimating Riesz representers by minimizing empirical Bregman divergences. This framework includes covariate balancing, nearest-neighbor matching, calibrated estimation, and density ratio estimation as special cases. A key design principle of the package is automatic regressor balancing (ARB): given a Bregman generator $g$ and a representer model class, genriesz} automatically constructs a compatible link function so that the generalized Riesz regression estimator satisfies balancing (moment-matching) optimality conditions in a user-chosen basis. The package provides a modulr interface for specifying (i) the target linear functional via a black-box evaluation oracle, (ii) the representer model via basis functions (polynomial, RKHS approximations, random forest leaf encodings, neural embeddings, and a nearest-neighbor catchment basis), and (iii) the Bregman generator, with optional user-supplied derivatives. It returns regression adjustment (RA), Riesz weighting (RW), augmented Riesz weighting (ARW), and TMLE-style estimators with cross-fitting, confidence intervals, and $p$-values. We highlight representative workflows for estimation problems such as the average treatment effect (ATE), ATE on treated (ATT), and average marginal effect estimation. The Python package is available at https://github.com/MasaKat0/genriesz and on PyPI.
