Published January 30, 2025 | Version v1
Journal article Open

Addressing discretization-induced bias in demographic prediction

  • 1. Cornell University
  • 2. University of Chicago
  • 3. University of Michigan
  • 4. Cornell Tech

Description

Racial and other demographic imputation is necessary for many applications, especially in auditing disparities and outreach targeting in political campaigns. The canonical approach is to construct continuous predictions—e.g. based on name and geography—and then to often discretize the predictions by selecting the most likely class (argmax), potentially with a minimum threshold (thresholding). We study how this practice produces discretization bias. For example, we show that argmax labeling, as used by a prominent commercial voter file vendor to impute race/ethnicity, results in a substantial under-count of Black voters, e.g. by 28.2% points in North Carolina. This bias can have substantial implications in downstream tasks that use such labels. We then introduce a joint optimization approach—and a tractable data-driven threshold heuristic—that can eliminate this bias, with negligible individual-level accuracy loss. Finally, we theoretically analyze discretization bias, show that calibrated continuous models are insufficient to eliminate it, and that an approach such as ours is necessary. Broadly, we warn researchers and practitioners against discretizing continuous demographic predictions without considering downstream consequences.

Data availability

The replication dataset in SI Appendix C.1 is public at Barber and Argyle (57) and the result of work by Argyle and Barber (21). The replication dataset in SI Appendix C.4 is public at Greengard and Gelman (58) and the result of work by Greengard and Gelman (18). The specific code and jupyter notebook used for these analyses are available at https://github.com/evan-dong/demographic-prediction-argmax-bias. A more general repository of code with a jupyter notebook for other researchers and practitioners to discretize and analyze their own model outputs is at https://github.com/evan-dong/demographic-discretization. The commercial dataset used in our analysis is privately owned by TargetSmart, a political data and analytics company, a copy of which we accessed with a research license from PredictWise, a campaign analytics firm. We are unable to provide public access to this proprietary dataset. Researchers can apply for access to TargetSmart data by contacting TargetSmart at: https://targetsmart.com/contact-us/.

Files

pgaf027.pdf

Files (20.2 MB)

Name Size Download all
Article
md5:9e771b8a7740dbe0f3c7168860a9a409
7.1 MB Preview Download
Supplementary material
md5:033c24611b40ec4eb5778a096ef701d9
13.1 MB Preview Download

Additional details

Identifiers

DOI
10.1093/pnasnexus/pgaf027
Other
oai:uchicago.tind.io:14756

UChicago Information

Division(s)
Physical Sciences Division
Department(s)
Statistics
Center(s) or Institute(s)
Data Science Institute