MPI-INF Logo
Campus Event Calendar

Event Entry

What and Who

Structured Synthesis for Neural Operators

Zhengyuan Su
Tsinghua University
PhD Application Talk
AG 1, AG 2, AG 3, INET, AG 4, AG 5, D6, SWS, RG1, MMCI  
AG Audience
English

Date, Time and Location

Monday, 27 January 2025
14:30
30 Minutes
Virtual talk
zoom

Abstract

To optimize neural networks for better prediction accuracy and higher execution performance, researchers often rely on Neural Architecture Search (NAS) and tensor compilers. However, these methods are limited to optimizing existing, manually designed operators. In this presentation, I will introduce Syno, an end-to-end framework I developed to automate the discovery of novel neural operators with better accuracy and/or speed. Syno constructs novel neural operators based on a novel set of fine-grained primitives, guides the synthesis with heavy canonicalization and pruning techniques, and leverages Monte Carlo tree search algorithms to explore the design space. This work, which has been submitted to ASPLOS 25, discovers better operators with an average of 2.06x speedup and less than 1% accuracy loss, even on NAS-optimized models, demonstrating its potential to advance neural network optimization.

Contact

Ina Geisler
+49 681 9325 1802
--email hidden

Virtual Meeting Details

Zoom
passcode not visible
logged in users only

Ina Geisler, 01/24/2025 10:57 -- Created document.