arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2002.05120cs.LGcs.AIstat.ML
收藏

Parameterizing Branch-and-Bound Search Trees to Learn Branching Policies

Mila(米拉人工智能研究所)

Giulia Zarpellon, Jason Jo, Andrea Lodi, Yoshua Bengio

英文摘要:

Branch and Bound (B&B) is the exact tree search method typically used to solve Mixed-Integer Linear Programming problems (MILPs). Learning branching policies for MILP has become an active research area, with most works proposing to imitate the strong branching rule and specialize it to distinct classes of problems. We aim instead at learning a policy that generalizes across heterogeneous MILPs: our main hypothesis is that parameterizing the state of the B&B search tree can aid this type of generalization. We propose a novel imitation learning framework, and introduce new input features and architectures to represent branching. Experiments on MILP benchmark instances clearly show the advantages of incorporating an explicit parameterization of the state of the search tree to modulate the branching decisions, in terms of both higher accuracy and smaller B&B trees. The resulting policies significantly outperform the current state-of-the-art method for "learning to branch" by effectively allowing generalization to generic unseen instances.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑