← Search

Yiwei Bai

4 accepted papers

2024

Scaling Up Pareto Optimization for Tree Structures with Affine Transformations: Evaluating Hybrid Floating Solar-Hydropower Systems in the Amazon

AAAI 2024technical

Sustainability challenges inherently involve the consideration of multiple competing objectives. The Pareto frontier – the set of all optimal solutions that cannot be improved with respect to one objective without negatively affecting another – is a crucial decision-making tool for navigating sustai…

Cited by 3Source ↗PDF ↗Scholar ↗
2023

Unsupervised Learning for Solving the Travelling Salesman Problem

NeurIPS 2023poster

We propose UTSP, an Unsupervised Learning (UL) framework for solving the Travelling Salesman Problem (TSP). We train a Graph Neural Network (GNN) using a surrogate loss. The GNN outputs a heat map representing the probability for each edge to be part of the optimal path. We then apply local search t…

2020

Deep Reasoning Networks for Unsupervised Pattern De-mixing with Constraint Reasoning

ICML 2020poster

We introduce Deep Reasoning Networks (DRNets), an end-to-end framework that combines deep learning with constraint reasoning for solving pattern de-mixing problems, typically in an unsupervised or very-weakly-supervised setting. DRNets exploit problem structure and prior knowledge by tightly combini…

Cited by 31Source ↗PDF ↗Scholar ↗