Classification and Treatment Learning with Constraints via Composite Heaviside Optimization: a Progressive MIP Method
Abstract: This paper proposes a Heaviside composite optimization approach and presents a progressive (mixed) integer programming (PIP) method for solving multi-class classification and multi-action treatment problems with constraints. A Heaviside composite function is a composite of a Heaviside function (i.e., the indicator function of either the open $( \, 0,\infty )$ or closed $[ \, 0,\infty \, )$ interval) with a possibly nondifferentiable function. Modeling-wise, we show how Heaviside composite optimization provides a unified formulation for learning the optimal multi-class classification and multi-action treatment rules, subject to rule-dependent constraints stipulating a variety of domain restrictions. A Heaviside composite function has an equivalent discrete formulation, and the resulting optimization problem can in principle be solved by integer programming (IP) methods. Nevertheless, for constrained learning problems with large data sets, a straightforward application of off-the-shelf IP solvers is usually ineffective in achieving global optimality. To alleviate such a computational burden, our major contribution is the proposal of the PIP method by leveraging the effectiveness of state-of-the-art IP solvers for problems of modest sizes. We provide the theoretical advantage of the PIP method with the connection to continuous optimization and show that the computed solution is locally optimal for a broad class of Heaviside composite optimization problems. The numerical performance of the PIP method is demonstrated by extensive computational experimentation.
- Athey S, Wager S (2021) Policy learning with observational data. Econometrica 89(1):133–161.
- Bertsimas D, Dunn J (2017) Optimal classification trees. Machine Learning 106:1039–1082.
- De Maio FG (2007) Income inequality measures. Journal of epidemiology and community health (10):849–852.
- Gini C (1997) Concentration and dependency ratios. Rivista di Politica Economica 769–789.
- Gurobi (2023) Gurobi Optimizer Reference Manual (https://www.gurobi.com/documentation/10.0/refman/index.html).
- Kitagawa T, Tetenov A (2018) Who should be treated? empirical welfare maximization methods for treatment choice. Econometrica 86(2):591–616.
- Kitagawa T, Tetenov A (2021) Equality-minded treatment choice. Journal of Business & Economic Statistics 39(2):561–574.
- Mangasarian OL (1965) Linear and nonlinear separation of patterns by linear programming. Operations Research (3):444–452.
- Scholtes S (2002) Introduction to piecewise differentiable equations. Springer Briefs in Optimization .
- Schutz RR (1951) On the measure of income inequality. American Economic Review 107–122.
Paper Prompts
Sign up for free to create and run prompts on this paper using GPT-5.
Top Community Prompts
Collections
Sign up for free to add this paper to one or more collections.