Photonic KAN: a Kolmogorov-Arnold network inspired efficient photonic neuromorphic architecture

Yiwei Peng,Sean Hooten,Xinling Yu,Thomas Van Vaerenbergh,Yuan Yuan,Xian Xiao,Bassem Tossoun,Stanley Cheung,Marco Fiorentino,Raymond Beausoleil
2024-08-16
Abstract:Kolmogorov-Arnold Networks (KAN) models were recently proposed and claimed to provide improved parameter scaling and interpretability compared to conventional multilayer perceptron (MLP) models. Inspired by the KAN architecture, we propose the Photonic KAN -- an integrated all-optical neuromorphic platform leveraging highly parametric optical nonlinear transfer functions along KAN edges. In this work, we implement such nonlinearities in the form of cascaded ring-assisted Mach-Zehnder Interferometer (MZI) devices. This innovative design has the potential to address key limitations of current photonic neural networks. In our test cases, the Photonic KAN showcases enhanced parameter scaling and interpretability compared to existing photonic neural networks. The photonic KAN achieves approximately 65$\times$ reduction in energy consumption and area, alongside a 50$\times$ reduction in latency compared to previous MZI-based photonic accelerators with similar performance for function fitting task. This breakthrough presents a promising new avenue for expanding the scalability and efficiency of neuromorphic hardware platforms.
Optics,Emerging Technologies
What problem does this paper attempt to address?