Nature:模拟光学计算机,让AI推理能效提升百倍?

2025-09-04 14:00:57   阅览:296

【导语】人工智能(AI)与组合优化正引领科学研究与工业应用的飞速发展,但其能耗问题日益凸显,对数字计算的可持续性构成挑战。当前新型计算系统往往局限于单一领域,且频繁的数字转换导致效率低下。微软英国剑桥研究院团队提出了一种创新方案——模拟光学计算机(AOC),它无需数字转换,能同时高效处理AI推理与组合优化任务。相关研究已发表于权威期刊Nature,展现了AOC在能效与扩展潜力上的显著优势,为更高速、更可持续的计算提供了新路径。未来,AOC有望成为推动硬件与算法创新的关键力量,助力实现可持续计算的目标。

人工智能(AI)和组合优化正在推动科学研究和工业应用的快速发展,但它们日益增长的能耗也对数字计算的可持续性提出了严峻挑战。

同时,目前大多数新型计算系统要么只擅长 AI,要么只擅长优化,而且还需频繁进行高能耗的数字转换,效率受限。另外,在实际应用中,这些系统也常难与硬件高效配合,无论是处理内存受限的神经网络、解决复杂优化问题,还是应对模拟计算噪声,都表现不佳。

如果换一种思路,不再依赖“0”与“1”的切换,而是用光与模拟信号去计算呢?这种不需要频繁转换、不依赖数字逻辑的计算方式,能否打(dǎ)破现有的限制?

基于此,微软英国剑桥研究院团队及其合作者提出了“模拟光学计算机”(Analog Optical Computer,AOC),其在无需数字转换的情况下,能够同时高效完成 AI 推理与组合优化任务,并具备显著的扩展潜力与能效优势。

相关研究论文已发表在权威科学期刊 Nature 上。上海交大校友、微软英国剑桥研究院首席研究员褚嘉琦(Jiaqi Chu)为该论文的作者之一。

据介绍,AOC 实现了将模拟电子(zi)技(jì)术(shù)与(yǔ)三(sān)维(wéi)光(guāng)学(xué)技术相结合,使同一平台能够同时加速 AI 推理和组合优化任务。这种“双领域能力”得益于快速的定点搜索方法,无需数字转换,同时也增强抗噪声能力。基于这种定点抽象,AOC 可运行具备递归推理能力的新型计算密集型神经模型,并采用先进的梯度下降方法解决高表现力优化问题。

图|AOC 架构与应用示意图

研究团队表示,AOC 架构基于可扩展的消费级技术构建,为实现更高速、更可持续的计算提供了有前景的路径。其原生支持迭代式、计算密集型模型,为 AI 与优化领域的未来创新提供了可扩展的模拟计算平台

模拟光学计算机:如何加速AI和优化任务?

AOC 在应用层面主要面向两类任务:机器学习推理任务和组合优化任务。研究团队通过四个典型案例展示了 AOC 在这两类任务中的能力。这项研究也体现了硬件与抽象层协同设计的优势,呼应了数字加速器与深度学习模型共同演进的趋势。

AOC 硬件结合了 3D 光学技术与模拟(nǐ)电(diàn)子(zi)技(jì)术(shù),并(bìng)基(jī)于(yú)平(píng)衡(héng)模(mó)型(xíng)实(shí)现(xiàn)了(le)两(liǎng)个(gè)机(jī)器(qì)学(xué)习(xí)推(tuī)理(lǐ)任(rèn)务(wu):图(tú)像(xiàng)分(fēn)类(lèi)与(yǔ)非(fēi)线(xiàn)性(xìng)回(huí)归(guī)。在(zài)这(zhè)两(liǎng)项(xiàng)任(rèn)务(wu)中(zhōng),模(mó)型(xíng)均(jūn)通(tōng)过(guò) AOC-DT 进(jìn)行(xíng)数(shù)字(zì)训(xun)练(liàn),并(bìng)直(zhí)接(jiē)部(bù)署(shǔ)到(dào)硬(yìng)件(jiàn)上,无需进一步校准。这对硬件精度提出了较高要求,同时也要求 AOC-DT 具备较高的保真度。

在图像分类实验中,AOC 的结果证明了采用数字训练并将权重转移至光电模拟推理硬件的可行性。将 AOC 的结果与线性分类器进行比较时,运行在 AOC 上的平衡模型所作出的贡献更加明显。研究人员还训练了一个简单的前馈模型,线性分类器和前馈模型都具有与 AOC 硬件相同数量的参数。尽管 AOC 实现了略高的准确率,但 MNIST 和 Fashion-MNIST 数据集本身较为简单,难以充分展现自递归模型的全部潜力。

**研究表明,AOC 硬件可以运行非线性回归模型。**他们选择两个非线性函数进行回归:高斯曲线和正弦曲线。**硬件准确地重现了这两个函数。**相比高斯曲线,正弦曲线由于存在多个极小值和极大值,对拟合精度提出了更高要求,因此需要更高精度的可微分数字孪生模型(AOC-DT)。此外,AOC 为运行在硬件上的平衡模型提供了支持。

图|AOC 在机器学习推理中的应用

QUMO 代表了一类广泛的组合优化问题,旨在最小化目标函数,QUMO 问题的求解过程即为找到一组使目标函数最小化的变量赋值。研究团队在 AOC 硬件上展示了两个典型的 QUMO 应用场景:医学图像重建与金融交易结算

他们在 AOC 硬件上实现了压缩感知,这是一种可以用更少的测量实现准确信号重建的技术。最终的图像重建结果与原始线条高度一致。所有 QUMO 实例均以完全模拟方式求解,未使用任何数字后处理。为了(le)验证压缩感知在大规模下的 QUMO 表达形式,研究人员使用 AOC-DT 从 FastMRI 数据集中重建了一幅脑部扫描图像。这一问题包含超过 200000 个变量。在典型的 4 倍和 8 倍欠采样率下,重建的均方误差(MSE)均低于 0.07。

在金融领域的优化任务中,他们使用 AOC 硬件解决了一个交易结算问题。每一笔证券交易都是以支付换取证券的交换,清算所会处理这类交易的批次,在每个交易批次中,交易结算的目标是最大化已结算交易的总数或总价值。鉴于交易数量庞大,同时受到法律约束和其他附加要求的限制,这成为一项复杂的优化问题。在该交易结算场景中,AOC 硬件在 7 个块坐标下(xià)降(jiàng)法(fǎ)(BCD)步(bù)骤(zhòu)内(nèi)找(zhǎo)到(dào)了(le)全局(jú)最(zuì)优(yōu)解(jiě)。相(xiāng)比(bǐ)之(zhī)下(xià),量(liàng)子(zi)硬(yìng)件(jiàn)在(zài)相(xiāng)同(tóng)问(wèn)题(tí)上(shàng)的(de)成(chéng)功(gōng)率(lǜ)仅(jǐn)为(wèi) 40–60%

图(tú)|AOC 在(zài)优(yōu)化(huà)中(zhōng)的应用

利用 AOC-DT,研究人员还在 QPLIB 基准测试中最难的具有线性不等式约束的二次二元问题上验证了算法性能,这些问题被表述为 QUMO 实例。AOC 方法与商用求解器 Gurobi 进行了对比,后者在这些问题上通常需要超过一分钟才能达到当前已知的最优解。

这种全模拟操作最大限度地减少了模数转换的开销。

未来潜力:实现100倍能效提升

值得注意的是,现实应用对硬件的扩展能力提出(chū)了(le)更高要求。使用 AOC 处理实际任务,需要其硬件能够支持从数亿到数十亿个权重的可扩展性。

研究团队表示,AOC 通过模块化架构具备满足这一需求的潜力,该架构可将核心的光学矩阵–向量乘法运算分解为较小的子向量与子矩阵乘法,从而实现可扩展的内存计算。

研究团队预计,AOC 可支持参数规模在 1 亿至 20 亿之间的模型,对应需要 50 至 1000 个光学模块。如果单个光学模块能够同时处理正负权重,则所需模块数量可减少一半。AOC 所采用的所有组件,包括 microLED、光电探测器、SLM 及模拟电子器件,均已具备持续扩展的制造生态系统,能够支持晶圆级的生产。

AOC 的运行速度和功耗决定了其能效。其速度受光电组件带宽限制,通常为 2 GHz 或更高。对于一个 1 亿权重的矩阵,使用 25 个 AOC 模块时,功耗估计为 800 W,可实现 400 Peta-OPS 的计算速度,在 8 位权重精度下的能效为每瓦 500 TOPS。相比之下,最新的 GPU 在相(xiāng)同(tóng)精(jīng)度(dù)下(xià)处(chù)理(lǐ)稠(chóu)密(mì)矩(ju)阵(zhèn)时(shí),其(qí)系(xì)统(tǒng)能(néng)效(xiào)最(zuì)高(gāo)仅(jǐn)为(wèi)每(měi)瓦(wǎ) 4.5 TOPS。

总(zǒng)之(zhī),AOC 架(jià)构(gòu)在(zài)扩(kuò)展(zhǎn)到(dào)实(shí)际(jì)的(de)机(jī)器(qì)学(xué)习(xí)和(hé)优(yōu)化(huà)任(rèn)务(wu)方(fāng)面(miàn)展(zhǎn)现(xiàn)出(chū)良(liáng)好(hǎo)前(qián)景(jǐng),有(yǒu)望(wàng)在(zài)能(néng)效(xiào)方(fāng)面(miàn)实(shí)现(xiàn)约(yuē) 100 倍(bèi)的(de)提(tí)升(shēng)。

展(zhǎn)望(wàng)未(wèi)来(lái),AOC 的(de)协(xié)同(tóng)设(shè)计(jì)方(fāng)法(fǎ)——将(jiāng)硬(yìng)件(jiàn)与(yǔ)机(jī)器(qì)学(xué)习和优化算法紧密对齐——有望持续推动硬件与算法的创新飞轮,这对实现可持续计算至关重要。

注意:封面图为版权图库图片,转载使用可能引发版权纠纷。