MDTransformer: A Hardware-Software Co-Design of Mode-Division Photonic Transformer Accelerator with Inverse-Designed Coherent Crossbar

LLM MQIA AI Systems and Hardware TPU/NPU
2026年07月28日
近期,光子Transformer加速器(PTA)已在加速Transformer推理任务方面,显著超越电子加速器,实现了更高的运行速度与能效。然而,当前最先进的PTA方案依赖昂贵的多波长光源生成技术及体积庞大的点积运算单元,原因在于其采用了有源相位调制器元件,导致整体方案效率低下且缺乏实用性。为解决这一问题,我们提出了MDTransformer——一种基于模分复用(mode-division)光数据流与光运算范式的新型光电协同设计PTA架构。具体而言,MDTransformer利用空间模式干涉执行复杂的矩阵运算,将逆向设计的多模耦合器、多模交叉波导以及马赫–曾德尔型IQ调制器高度集成于一个紧凑的模分光子张量核(MPTC)中,从而在光域内直接完成矩阵乘法运算。该架构中每根波导所支持的各阶导模(即TE₀–TE₃模)均作为独立的计算通道,无需光谱滤波或自由光谱范围(FSR)限制,即可实现单波导四路并行计算。此外,其相干探测与IQ调制协同工作,可同时编码光场的幅度与相位信息,从而支持复数值运算,满足Transformer中全部算子的全范围计算需求。MDTransformer提供模拟域乘法运算,等效精度达4比特以下,模间串扰低于–30 dB;其逆向设计方法亦确保了良好的可扩展性,并完全兼容单激光器连续波(CW)工作模式(中心波长1550 nm)。实验结果表明,在不同典型负载(包括DeiT-Tiny/Small/Base与BERT-Base/Large)下,相较于当前最先进的PTA方案,MDTransformer可减少40.4%的芯片面积、降低63.6%的功耗、节省40.6%的能量消耗,同时保持相当的端到端延迟。上述成果表明,MDTransformer为构建高性能、高能效的Transformer基系统提供了一种切实可行的技术路径。
Recently, photonic transformer accelerators (PTAs) have successfully achieved significant speedup and energy efficiency improvements over electronic accelerators for expediting Transformer inference. However, state-of-the-art rely on expensive multi-wavelength light generation and large dot-product units due to active phase-shifter components, thus making their approach inefficient and impractical. To address this, we propose MDTransformer, a novel hardware-software co-design of PTA based on mode-division optical dataflow and operations. Specifically, MDTransformer performs complex matrix operations using spatial-mode interference, that leverages the inverse-designed multi-mode couplers, crossings, and Mach-Zehnder IQ modulators into a compact mode-division photonic tensor core (MPTC), capable of executing matrix multiplications in the optical domain. Its each guided mode (i.e., TE0-TE3) acts as an independent computational lane, enabling four-fold parallelism-per-waveguide without spectral filtering or free-spectral-range limitations. Moreover, its coherent detection and IQ modulation jointly encode amplitude and phase, realizing complex-valued arithmetic for full-range operations in transformers. MDTransformer offers analog multiplication with sub-4-bit effective precision and inter-modal crosstalk below -30 dB. Its inverse-designed approach also offers scalable and full compatibility with single-laser continuous-wave operation at 1550 nm. Experimental results show that MDTransformer achieves 40.4% area reduction, 63.6% power saving, 40.6% energy saving, and comparable latency over the state-of-the-art PTA across different workloads (i.e., DeiT-Tiny/Small/Base and BERT-Base/Large). These results show that MDTransformer offers a practical solution for high-performance and energy-efficient transformer-based systems.
许愿