← Search

Lei Mao

3 accepted papers

2022

Structural Pruning via Latency-Saliency Knapsack

NeurIPS 2022accept

Structural pruning can simplify network architecture and improve inference speed. We propose Hardware-Aware Latency Pruning (HALP) that formulates structural pruning as a global resource allocation optimization problem, aiming at maximizing the accuracy while constraining latency under a predefined…

2021

Tesseract: Tensorised Actors for Multi-Agent Reinforcement Learning

ICML 2021spotlight

Reinforcement Learning in large action spaces is a challenging problem. This is especially true for cooperative multi-agent reinforcement learning (MARL), which often requires tractable learning while respecting various constraints like communication budget and information about other agents. In thi…

Cited by 45SourcePDFScholar
2020

Bongard-LOGO: A New Benchmark for Human-Level Concept Learning and Reasoning

NeurIPS 2020spotlight

Humans have an inherent ability to learn novel concepts from only a few samples and generalize these concepts to different situations. Even though today's machine learning models excel with a plethora of training data on standard recognition tasks, a considerable gap exists between machine-level pat…