Optimizing Black-box Metrics with Iterative Example Weighting
Gaurush Hiranandani, Jatin Mathur, Harikrishna Narasimhan, Mahdi Milani Fard, Sanmi Koyejo
Abstract
We consider learning to optimize a classification metric defined by a black-box function of the confusion matrix. Such black-box learning settings are ubiquitous, for example, when the learner only has query access to the metric of interest, or in noisy-label and domain adaptation applications where the learner must evaluate the metric via performance evaluation using a small validation sample. Our approach is to adaptively learn example weights on the training dataset such that the resulting weighted objective best approximates the metric on the validation sample. We show how to model and estimate the example weights and use them to iteratively post-shift a pre-trained class probability estimator to construct a classifier. We also analyze the resulting procedure's statistical properties. Experiments on various label noise, domain shift, and fair classification setups confirm that our proposal compares favorably to the state-of-the-art baselines for each application.
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.
Cited by top-tier papers3
- Zeroth-Order Optimization with Trajectory-Informed Derivative EstimationYao Shu, Zhongxiang Dai, Weicong Sng, Arun Verma et al.ICLR 2023
- Logits are All We Need to Adapt Closed ModelsGaurush Hiranandani, Haolun Wu, Subhojyoti Mukherjee, Sanmi KoyejoICML 2025
- Refining Adaptive Zeroth-Order Optimization at EaseYao Shu, Qixin Zhang, Kun He, Zhongxiang DaiICML 2025
Builds on4
- Rethinking Importance Weighting for Deep Learning under Distribution ShiftTongtong Fang, Nan Lu, Gang Niu, Masashi SugiyamaNeurIPS 2020 · 179 citations
- Robust Optimization for Fairness with Noisy Protected GroupsSerena Lutong Wang, Wenshuo Guo, Harikrishna Narasimhan, Andrew Cotter et al.NeurIPS 2020 · 134 citations
- Optimizing Black-box Metrics with Adaptive SurrogatesQijia Jiang, Olaoluwa Adigun, Harikrishna Narasimhan, Mahdi Milani Fard et al.ICML 2020 · 19 citations
- Optimization and Analysis of the pAp@k Metric for Recommender SystemsGaurush Hiranandani, Warut Vijitbenjaronk, Sanmi Koyejo, Prateek JainICML 2020 · 8 citations
Related papers
- Gradient-Based Sample Selection for Black-Box Universal Domain AdaptationQiuyan He, Minghua DengAAAI 2025
- Graph-Smoothed Bayesian Black-Box Shift Estimator and Its Information GeometryMasanari KimuraNeurIPS 2025 · 1 citation
- Match: A Maximum-Likelihood Approach for Classification under Label ShiftZahra Donyavi, Feiyu Li, Yunrui Zhang, Diego Furtado Silva et al.KDD 2025 · 1 citation
- Maximum Likelihood with Bias-Corrected Calibration is Hard-To-Beat at Label Shift AdaptationAmr Alexandari, Anshul Kundaje, Avanti ShrikumarICML 2020 · 123 citations
- Domain Adaptation under Open Set Label ShiftSaurabh Garg, Sivaraman Balakrishnan, Zachary C. LiptonNeurIPS 2022 · 57 citations
