Now Solving: Global Optimal AI

We focus on Scalable Optimization, Learning, and Intelligent Decision-making (SOLID).


SOLID Lab is based in the Department of Chemical and Biological Engineering at The University of British Columbia .


Explore Our Research Meet Our People

Latest Highlight

Jiayang received the Wall Research Award.  | June 2025

Prof. Yankai Cao gave a talk at the MIT PSE Seminar.  | May 2025

Congratulations to Jingyi on her PhD graduation!  | April 2025

Jason Zhao received the Four Year Doctoral Fellowship.  | March 2025

Yixiu Wang has published a paper in Nature communication as a co-first author. The paper has been cited for more than 500 times.  | April 2022

→ More lab news

Recruiting

We are seeking future postdoctoral researchers, graduate students, visiting students, and undergraduate students. → Join Us


Research

SOLID Lab focuses on the design and implementation of large-scale local and global optimization algorithms to solve problems that arise in diverse decision-making paradigms such as machine learning, data analysis, stochastic optimization,and optimal control. Our algorithms combine mathematical techniques and emerging high-performance computing hardware (e.g., multi-core CPUs, GPUs, and computing clusters) to achieve computational scalability. The goal is to make these developments accessible to academic and industrial users by implementing algorithms on easy-to-use and extensible software libraries.

Furthermore, We have applied the algorithms and tools to help collaborators address engineering and scientific questions that arise in diverse application domainsincluding conflict resolution in energy system design, robust control of crystallization systems, predictive control of wind turbines, power management in large networks, estimation of microbial growth models, and image classification for contaminant detection.

Selected Publications → See Full Publications

Differentiable Decision Tree via ``ReLU+Argmin'' Reformulation. Advances in Neural Information Processing Systems (NeurIPS), Spotlight Paper, 2025.
A scalable deterministic global optimization algorithm for training optimal decision tree. Advances in Neural Information Processing Systems (NeurIPS) 35, 8347-8359, 2022.
Global optimal k-medoids clustering of one million samples. Advances in Neural Information Processing Systems (NeurIPS) 35, 982-994, 2022.
Global optimization of k-center clustering. International Conference on Machine Learning (ICML), Spotlight Paper, 19956-19966, 2022.
A Scalable deterministic global optimization algorithm for clustering problems. International Conference on Machine Learning (ICML),Spotlight Paper, 4391-4401, 2021.

SOLID Lab

We focus on Scalable Optimization, Learning, and Intelligent Decision-making (SOLID). SOLID Lab is based in the Department of Chemical and Biological Engineering at The University of British Columbia.

Contact US

Yankai Cao
Associate Professor
Tel: 1 604 822 1346
Email: yankai.cao@ubc.ca
Office: CHBE 237
2360 East Mall
Vancouver, BC, Canada