# Robust Treatment-Effect Learning Beyond Binary Treatments: Stabilized Weighting with Deep Representation

> **NSF 01002627DB NSF RESEARCH & RELATED ACTIVIT** · University of California-Riverside (CA) · $199,065

## Abstract

Modern scientific, medical, and policy decisions increasingly rely on observational data to evaluate the effects of interventions such as medical treatments, public programs, and behavioral exposures. Many of these interventions are multi-level or continuous, such as medication dosage or program participation intensity, rather than simple binary choices. However, existing statistical methods, including widely used propensity score approaches, often become unstable or unreliable in these settings, particularly when applied to high-dimensional data with complex relationships. This project addresses these challenges by developing new methods for reliable and interpretable causal analysis. The research will enhance the ability to draw credible conclusions from large-scale data, with applications in public health, healthcare delivery, social policy programs, and digital health. By improving the stability and accuracy of treatment-effect estimation, the project supports evidence-based decision making that can improve health outcomes, inform policy design, and promote societal well-being. In addition, the project will contribute to workforce development through student training, interdisciplinary collaboration, and outreach activities that broaden participation in data science and statistics.

This research advances the theory and methodology of causal inference for complex data with multi-level and continuous treatments, where classical inverse propensity weighting methods can fail due to unstable weights and ill-conditioned estimation. It develops a unified and scalable framework for treatment-effect estimation beyond binary interventions by leveraging stabilized weighting, empirical likelihood, and deep learning representations. The project introduces new methodologies for counterfactual distribution estimation, dose–response analysis, and general loss-based causal inference. The proposed approach replaces unstable inverse weighting with stabilized weighting strategie

## Key facts

- **NSF award ID:** 2610432
- **Awardee organization:** University of California-Riverside (CA)
- **SAM.gov UEI:** MR5QC5FCAVH5
- **PI:** Shujie Ma
- **Primary program:** 01002627DB NSF RESEARCH & RELATED ACTIVIT
- **All programs:** Artificial Intelligence (AI), Machine Learning Theory, Biotechnology
- **Estimated total:** $199,065
- **Funds obligated:** $199,065
- **Transaction type:** Standard Grant
- **Period:** 07/01/2026 → 06/30/2029

## Primary source

NSF Award Search: https://www.nsf.gov/awardsearch/showAward?AWD_ID=2610432

## Citation

> US National Science Foundation, Award 2610432, Robust Treatment-Effect Learning Beyond Binary Treatments: Stabilized Weighting with Deep Representation. Retrieved via AI Analytics 2026-07-20 from https://api.ai-analytics.org/grant/nsf/2610432. Licensed CC0.

---

*[NSF Awards dataset](/datasets/nsf-awards) · CC0 1.0*