A Prescriptive Machine Learning Approach to Mixed-Integer Convex Optimization

被引:6
作者
Bertsimas, Dimitris [1 ]
Kim, Cheol Woo [2 ]
机构
[1] MIT, Sloan Sch Management, Cambridge, MA 02139 USA
[2] MIT, Operat Res Ctr, Cambridge, MA 02139 USA
关键词
mixed-integer optimization; prescriptive analytics; artificial intelligence; decision trees; computational methods;
D O I
10.1287/ijoc.2022.0188
中图分类号
TP39 [计算机的应用];
学科分类号
081203 ; 0835 ;
摘要
We introduce a prescriptive machine learning approach to speed up the process of solving mixed-integer convex optimization (MICO) problems. We solve multiple optimization instances and train a machine learning model in advance, which we use to solve new instances. Previous works have shown that the predictions of classification algorithms enable us to solve optimization problems much faster than commercial solvers. What distinguishes this paper from the previous work is that we use a prescriptive algorithm, Optimal Policy Trees (OPT), instead of classification algorithms. Whereas classification algorithms aim to predict the correct label and consider all other labels equally undesirable, a prescriptive approach takes into account all the available decision options and their counterfactuals. We first introduce an algorithm that is purely based on OPT and also its extension. We compare their performance with Optimal Classification Trees (OCT) on various tion, facility location, and hybrid vehicle control. We also experiment on real-world instances taken from the Mixed Integer Programming Library. OPT-based methods have a significant edge on finding feasible solutions, whereas OCT-based methods have a slight edge on the degree of suboptimality. The proposed extension of the pure OPT algorithm improves on the suboptimality of the solutions the algorithm produces.
引用
收藏
页码:1225 / 1241
页数:18
相关论文
共 30 条
[1]   A Machine Learning-Based Approximation of Strong Branching [J].
Alvarez, Alejandro Marcos ;
Louveaux, Quentin ;
Wehenkel, Louis .
INFORMS JOURNAL ON COMPUTING, 2017, 29 (01) :185-195
[2]   Optimal policy trees [J].
Amram, Maxime ;
Dunn, Jack ;
Zhuo, Ying Daisy .
MACHINE LEARNING, 2022, 111 (07) :2741-2768
[3]  
Balcan M. F., 2020, PR MACH LEARN RES
[4]   How Much Data Is Sufficient to Learn High-Performing Algorithms? Generalization Guarantees for Data-Driven Algorithm Design [J].
Balcan, Maria-Florina ;
DeBlasio, Dan ;
Dick, Travis ;
Kingsford, Carl ;
Sandholm, Tuomas ;
Vitercik, Ellen .
STOC '21: PROCEEDINGS OF THE 53RD ANNUAL ACM SIGACT SYMPOSIUM ON THEORY OF COMPUTING, 2021, :919-932
[5]  
Balcan MF, 2022, PREPRINT
[6]  
Balcan MF, 2020, P 34 AAAI C ART INT
[7]   Machine learning for combinatorial optimization: A methodological tour d'horizon [J].
Bengio, Yoshua ;
Lodi, Andrea ;
Prouvost, Antoine .
EUROPEAN JOURNAL OF OPERATIONAL RESEARCH, 2021, 290 (02) :405-421
[8]  
Bertsimas D, 2019, Machine learning under a modern optimization lens
[9]   Online Mixed-Integer Optimization in Milliseconds [J].
Bertsimas, Dimitris ;
Stellato, Bartolomeo .
INFORMS JOURNAL ON COMPUTING, 2022, 34 (04) :2229-2248
[10]   The voice of optimization [J].
Bertsimas, Dimitris ;
Stellato, Bartolomeo .
MACHINE LEARNING, 2021, 110 (02) :249-277