帕累托最优光子张量核的零样本自动电路拓扑搜索.pdf

1、ADEPT-Z:Zero-Shot Automated Circuit Topology Search for Pareto-Optimal Photonic Tensor Cores1Ziyang Jiang1,Pingchuan Ma1,Meng Zhang2,Rena Huang2,Jiaqi Gu11Arizona State University,2Rensselaer Polytechnic Institute1School of Electrical,Computer and Energy Engineeringzjian124asu.edujiaqiguasu.edu|scop

2、ex-asu.github.ioCopyright 2022 Arizona Board of RegentsPhotonic ML Accelerators2Evolve from electronics to integrated photonicsSource:Mitchell A.Nahmias,Bhavin J.Shastri,Alexander N.Tait,Thomas Ferreira de Lima and Paul R.Prucnal,“Neuromorphic photonics,”Optics&Photonics News,Jan 2018.Optical-electr

3、onic hybrid chips(10 TOPS/W)Fully-optical chipsNeuromorphic photonics(1E6 TOPS/W)Energy Efficiency(TOPS/W)Compute Density(TOPS/mm2)Speed-of-light,Massive parallelism,low-powerpotential105Copyright 2022 Arizona Board of RegentsPTC:Photonic Tensor Core3Basic Optical ComponentsSVD-based ONNs with MZIs

4、Nature17,Shen+WButterfly-style PTC ASP-DAC20,Gu+PTC with MMI Devices Sci.Rep.24,Zelaya+Modular Coherent PTC DesignSRepeated block structures-Phase shifter arrays(s)-Coupler arrays()-Crossing arrays()DCCRPScouplercrossingphase shifterMatrix:=222111Coherent PTCs:Complex-valued Matrix MultiplicationCop

5、yright 2022 Arizona Board of RegentsManual PTC Design Auto-Searched Design?MMI ArrayDA for PhotonicsManual Design Automated PTC DesignCan we automatically&efficiently find Pareto-optimal photonic circuits designs?Copyright 2022 Arizona Board of RegentsPrior Work on Auto PTC DesignRelax as continuous

6、 optimizationGradient descent to search(similar to DARTS in one-shot neural architecture search)Challenges in Differentiable ADEPT Limited design space:fixed slots for 2x2 coupler,cannot handle multi-port couplers Optimization difficulty:hard to handle multiple non-differentiable objectives/constrai