← 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 5SourcePDFScholar
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 31SourcePDFScholar