mlsanityData to 7 Oct 2026

A new perspective on classification: optimally allocating limited resources to uncertain tasks

arXiv 2202.04369 ↗First posted 9 Feb 2022cs.LG stat.MLcited by 5

Not on the map. Why?

Abstract (arXiv)

A central problem in business concerns the optimal allocation of limited resources to a set of available tasks, where the payoff of these tasks is inherently uncertain. In credit card fraud detection, for instance, a bank can only assign a small subset of transactions to their fraud investigations team. Typically, such problems are solved using a classification framework, where the focus is on predicting task outcomes given a set of characteristics. Resources are then allocated to the tasks that are predicted to be the most likely to succeed. However, we argue that using classification to address task uncertainty is inherently suboptimal as it does not take into account the available capacity. Therefore, we first frame the problem as a type of assignment problem. Then, we present a novel solution using learning to rank by directly optimizing the assignment's expected profit given limited, stochastic capacity. This is achieved by optimizing a specific instance of the net discounted cumulative gain, a commonly used class of metrics in learning to rank. Empirically, we demonstrate that our new method achieves higher expected profit and expected precision compared to a classification approach for a wide variety of application areas and data sets. This illustrates the benefit of an integrated approach and of explicitly considering the available resources when learning a predictive model.

Key facts. Cited by 5 papers in the map (data through 7 Oct 2026). Builds on 8 of its 72 references that are in the map. First posted on arXiv 9 Feb 2022.

Builds on

8 of 72 references.

References · 8

  1. 2018Machine Learning for Combinatorial Optimization: a Methodological Tour d'Horizoncited by 612
  2. 2017Smart "Predict, then Optimize"cited by 332
  3. 2018Melding the Data-Decisions Pipeline: Decision-Focused Learning for Combinatorial Optimizationcited by 220
  4. 2021End-to-End Constrained Optimization Learning: A Surveycited by 154
  5. 2019Smart Predict-and-Optimize for Hard Combinatorial Optimization Problemscited by 88
  6. 2016Matching while Learningcited by 19
  7. 2020Learning to rank for uplift modelingcited by 18
  8. 2017Profit Driven Decision Trees for Churn Predictioncited by 4

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2024: 1; 2025: 3; 2026: 1.

5 papers.