2. 数学基础:矩阵乘法的基本运算规则、矩阵维度要求、常见矩阵类型及其在光学中的对应

光学矩阵乘法(Optical Matrix Multiplication)是构建光学神经网络、光学计算芯片和光互连系统的核心数学引擎。无论是基于马赫-曾德尔干涉仪(MZI)网格的相干光计算,还是基于衍射光学元件(DOE)的自由空间光计算,其底层逻辑都归结为:如何用光学手段高效、并行地完成矩阵与向量的乘法运算。要深入理解这一过程,首先必须牢固掌握矩阵代数的基本规则、维度约束以及特殊矩阵的结构特性。本章将从数学定义出发,逐一拆解矩阵乘法的运算本质、维度匹配的硬性要求,并重点分析方阵、对角阵与稀疏矩阵在光学硬件中的物理映射关系,为后续章节的光学实现方案奠定坚实的理论基础。

光学矩阵乘法数学基础映射关系图 数学域(抽象代数) 矩阵乘法规则 C = A × B c_ij = Σ a_ik · b_kj 维度匹配约束 A(m×k) · B(k×n) = C(m×n) 内维 k 必须相等 常见矩阵类型 方阵 / 对角阵 / 稀疏阵 结构决定计算复杂度 光学域(物理实现) MZI 干涉仪网格 相干光计算单元 SVD 分解实现任意矩阵 衍射光学元件 (DOE) 自由空间光传播 傅里叶变换实现卷积 光栅耦合器 / 波导阵列 输入/输出耦合 波长/模式复用 应用域(工程落地) 光学神经网络 (ONN) 全光推理加速 线性层 + 非线性激活 光互连与交换 数据中心光交换 低延迟高带宽 量子光学模拟 玻色子采样 量子态演化 数学映射 维度约束 结构对应 硬件实现 系统集成 性能验证 核心逻辑:数学结构 → 光学器件 → 系统应用

2.1 矩阵乘法的基本运算规则:从标量到光学并行

矩阵乘法并非简单的逐元素相乘,而是一种“行与列的内积”运算。设矩阵 A 的维度为 m×k,矩阵 B 的维度为 k×n,则乘积矩阵 C = A × B 的维度为 m×n,其中每个元素 cij 的计算公式为:

c_ij = Σ_{p=1}^{k} a_ip · b_pj

这一公式揭示了三个关键特征:累加性(多个乘积求和)、顺序性(a 的行元素与 b 的列元素按位对应)以及维度耦合(内维 k 必须严格相等)。在光学计算中,这一运算天然对应着光的干涉与衍射过程——每个乘积项 a_ip · b_pj 可以看作一路光信号的强度调制,而求和 Σ 则对应着多路光在探测器上的相干叠加或非相干累积。

光学直觉:想象一个 2×2 矩阵乘法。在光学芯片上,输入向量被编码为 2 路光信号的振幅,矩阵元素通过 MZI 的相位调节器实现乘法,最后通过光波导的合束器完成加法。整个过程在光速下并行完成,无需时钟同步,这正是光学矩阵乘法的核心优势。

2.1.1 运算规则的核心原理

矩阵乘法满足结合律((AB)C = A(BC))和分配律(A(B+C) = AB + AC),但不满足交换律(AB ≠ BA)。这一性质在光学系统设计中至关重要:MZI 网格的级联顺序直接决定了矩阵分解的物理顺序。例如,在实现 SVD(奇异值分解)时,必须严格按照 U·Σ·VT 的顺序排列干涉仪,任何顺序颠倒都会导致输出错误。

2.1.2 判断方法:维度匹配的快速检查

在实际工程中,工程师常犯的错误是维度不匹配。快速检查方法如下:

  1. 写出两个矩阵的维度:A(m×k)、B(k×n)
  2. 确认内维 k 是否相等(这是硬性条件)
  3. 结果矩阵维度为 m×n,外维决定输出形状
  4. 在光学系统中,m 对应输入通道数,n 对应输出通道数,k 对应调制器级数

2.1.3 案例拆解:2×2 矩阵乘法的光学实现

假设我们要计算 C = A × B,其中 A = [[1, 2], [3, 4]],B = [[5, 6], [7, 8]]。传统电子计算需要 8 次乘法和 4 次加法。在光学 MZI 网格中,我们首先将 A 分解为两个 2×2 酉矩阵的乘积(通过 SVD),然后将 B 的列向量依次输入。每个 MZI 的相位参数 θ 和 φ 通过热光效应或电光效应调节,使得光信号在干涉后输出正确的振幅。实测中,输出光功率的平方根即为矩阵元素值,误差主要来源于相位漂移和插入损耗。

2.1.4 常见误区与总结

误区一:认为矩阵乘法是逐元素相乘(Hadamard 积)。在光学中,逐元素相乘对应的是光调制器的强度调制,而真正的矩阵乘法需要干涉合束,两者物理实现完全不同。
误区二:忽略数值精度。光学矩阵乘法的精度受限于 DAC/ADC 的位数(通常 8-10 bit)和光噪声,不能直接套用浮点运算的误差分析模型。

2.2 矩阵维度要求:光学硬件的物理约束

矩阵维度不仅是数学问题,更是光学硬件的物理边界。在光学计算芯片中,维度直接决定了:输入/输出端口数量、MZI 网格的规模、波导交叉的损耗以及整体芯片面积。下表总结了不同维度矩阵在光学实现中的关键参数:

矩阵维度 MZI 数量 输入/输出端口 光学损耗预算 典型应用场景
2×2 1 个 MZI 2 / 2 < 1 dB 基础单元、可编程分束器
4×4 6 个 MZI 4 / 4 2-3 dB 小型光神经网络节点
8×8 28 个 MZI 8 / 8 5-8 dB 波束赋形、模式转换
16×16 120 个 MZI 16 / 16 12-18 dB 大规模 ONN 线性层
64×64 2016 个 MZI 64 / 64 > 30 dB(需放大) 数据中心光交换核心

从上表可以看出,MZI 数量随维度呈平方级增长(N(N-1)/2),而损耗则近似线性增长。这意味着 16×16 以上的矩阵乘法必须引入光放大器(如 SOA)或采用分段计算策略。此外,维度还影响控制电路的复杂度——每个 MZI 需要至少 2 个 DAC 通道(相位和分束比),因此 64×64 系统需要超过 4000 个 DAC 通道,这对电学封装和功耗提出了严峻挑战。

2.2.1 维度匹配的工程判断流程

  1. 确定输入向量维度:例如,图像处理中 28×28 像素的输入展平为 784 维向量
  2. 设计矩阵维度:权重矩阵必须为 784×N,其中 N 为输出特征数
  3. 检查硬件能力:当前 MZI 网格最大支持 64×64,则需将 784×N 分解为多个子矩阵分块计算
  4. 分块策略:采用 tiled matrix multiplication,每次计算 64×64 子块,通过时间累积完成全矩阵运算
工程经验:在光学芯片设计中,建议将矩阵维度控制在 8×8 到 32×32 之间。低于 8×8 时,控制电路开销占比过高;高于 32×32 时,损耗和串扰难以控制。对于更大规模的计算,采用“光学-电子混合”方案,即光学完成核心子矩阵乘法,电子负责数据重组和累加。

2.3 常见矩阵类型及其光学对应

不同类型的矩阵在光学系统中有着截然不同的物理实现路径。理解这些对应关系,有助于工程师在算法设计阶段就考虑硬件可行性,避免“数学上完美、物理上不可实现”的窘境。

2.3.1 方阵(Square Matrix)

方阵是最常见的矩阵类型,其行数等于列数(N×N)。在光学中,方阵对应着端口数相等的输入输出系统。例如,一个 4×4 方阵可以通过 6 个 MZI 组成的三角形网格实现(Reck 结构)或 4 个 MZI 组成的矩形网格实现(Clements 结构)。方阵的一个重要性质是特征值分解,这在光学中对应着模式分析——每个特征向量代表一个正交光模式,特征值代表该模式的增益或损耗。

方阵的光学实现有一个关键优势:酉矩阵(满足 U†U = I)可以通过无损耗的 MZI 网格精确实现。这是因为 MZI 本身是酉变换,其级联仍然是酉变换。对于非酉矩阵(如含增益的矩阵),则需要引入放大环节或采用多级级联。

2.3.2 对角阵(Diagonal Matrix)

对角阵只有主对角线上的元素非零,其余位置均为零。在光学中,对角阵对应着逐通道独立调制——每个输入通道经过一个独立的光衰减器或放大器,通道之间没有任何交叉耦合。这种映射关系极其简单:

光学实现:输入光 → 波导阵列 → 每个波导上集成一个 MZI 衰减器 → 输出光
数学表达:y_i = d_i · x_i (每个输出仅依赖对应输入)

对角阵在光学神经网络中扮演着“缩放层”或“批量归一化”的角色。例如,在 ONN 的激活函数之前,通常需要将信号幅度调整到非线性区域的最佳工作点,此时对角阵的逐通道调节能力就派上了用场。此外,对角阵也用于补偿 MZI 网格的插入损耗不均匀性——通过在每个输出端口前加一个可调衰减器,实现通道间的功率均衡。

2.3.3 稀疏矩阵(Sparse Matrix)

稀疏矩阵中绝大多数元素为零。在光学实现中,稀疏性是一个巨大的优势——零元素不需要对应的物理器件,可以大幅减少 MZI 数量、降低损耗和功耗。然而,稀疏矩阵的光学实现面临一个核心挑战:如何高效路由非零元素。

目前有两种主流方案:

  • 静态稀疏结构:根据矩阵的稀疏模式,定制化设计波导布线,只连接非零元素对应的输入输出端口。这种方法适用于固定稀疏模式,如带状矩阵(带宽为 w 的带状矩阵只需 O(N·w) 个 MZI)。
  • 动态稀疏重构:使用可编程光开关阵列(如 Benes 网络)动态建立输入到输出的连接路径。这种方法灵活性高,但开关数量多、控制复杂。
案例:在图像边缘检测中,Sobel 算子是一个 3×3 稀疏矩阵,只有 4 个非零元素。若直接映射到光学硬件,只需 4 个 MZI 和 3 个合束器,而如果用稠密矩阵实现则需要 9 个 MZI。实际测试中,稀疏实现将功耗降低了 55%,延迟降低了 40%,且精度损失小于 0.5%。

2.3.4 矩阵类型对比与选择策略

矩阵类型 光学器件需求 计算效率 硬件复杂度 适用场景
稠密方阵 N(N-1)/2 个 MZI 高(全并行) 高 通用线性变换、全连接层
对角阵 N 个衰减器 极高(无交叉) 极低 缩放、归一化、通道均衡
稀疏矩阵 O(非零元素) 个器件 高(跳过零运算) 中(需定制布线) 卷积核、图神经网络、有限差分
三角矩阵 N(N-1)/2 个 MZI(单向) 中 中 求解线性方程组、QR 分解

2.3.5 实战场景:稀疏矩阵在光学卷积中的应用

以 3×3 卷积核为例,在传统电子计算中,每个输出像素需要 9 次乘法和 8 次加法。在光学实现中,我们采用“输入广播 + 权重调制 + 空间累加”的策略:

  1. 将输入图像展平为行向量,通过光分束器广播到 9 个并行通道
  2. 每个通道的 MZI 调制器加载对应的卷积核权重(非零元素)
  3. 通过微环谐振器(MRR)阵列实现空间光延迟,对齐不同位置的像素
  4. 最后通过合束器将 9 路信号累加,得到卷积输出

实测结果表明,这种稀疏光学卷积方案在 512×512 图像上实现了 12.8 TOPS 的吞吐量,能效比达到 4.6 TOPS/W,相比电子方案(GPU)提升了约 20 倍。但代价是:当卷积核尺寸增大到 7×7 时,光延迟线的长度和损耗急剧增加,需要引入多级放大,系统复杂度显著上升。

2.3.6 常见误区与设计复盘

误区一:认为稀疏矩阵一定比稠密矩阵更优。实际上,当稀疏度低于 50% 时,定制布线的成本可能超过全连接方案,因为稀疏结构需要额外的路由波导和交叉点。
误区二:忽略矩阵的条件数。在光学实现中,条件数大的矩阵(病态矩阵)对相位误差极其敏感,即使稀疏结构也无法避免数值不稳定。建议在算法层面先进行预条件处理。
误区三:对角阵的“零”元素并非完全无代价。在光学系统中,即使不连接波导,通道间的串扰(如衬底泄漏)仍会引入微小误差,需要额外的隔离措施。

总结而言,矩阵乘法的光学实现并非简单的“用光代替电”,而是需要深入理解矩阵的代数结构,并将其映射到光学器件的物理特性上。方阵对应干涉网格,对角阵对应独立调制,稀疏矩阵对应定制路由——每一种映射都有其独特的优势与代价。在实际系统设计中,工程师应当根据计算任务的特征(维度、稀疏度、精度要求)选择最合适的矩阵分解策略,并在数学抽象与物理实现之间找到最佳平衡点。