MODULE / 01 · TENSOR NETWORK

张量网络收缩序
优化

面向大规模量子线路模拟,以多链退火搜索更低开销的张量网络收缩路径,并通过 Rust 工程实现提升搜索与评估效率。

ArcTN多链协同搜索
三条不同温度的搜索链分别改进候选收缩树,链间交换状态,经代价评估后选出低开销收缩路径。
探索、交换,寻找更低开销的路径每个节点代表候选收缩树,绿色箭头表示链间状态交换。
核心方法
Parallel Tempering
计算对象
Tensor Network
工程实现
Rust Native

METHOD / 01

让多条搜索链
协同寻找更优路径

不同温度的搜索链承担探索与收敛任务,并周期性交换候选解。高温链帮助跳出局部最优,低温链保留更有价值的张量网络收缩序;Rust 实现用于承载高频候选评估和并行调度。

  1. 01构建张量网络

    从量子线路得到张量及其指标连接关系。

  2. 02并行路径搜索

    多条温度链独立扰动和评估候选张量网络收缩序。

  3. 03链间交换

    按接受概率交换状态,扩大有效搜索范围。

  4. 04输出低开销序列

    综合计算量与内存指标选择可执行方案。

RESULTS / 02 · PERFORMANCE OVERVIEW

ArcTN 优化性能总览

轻量优化场景比较 ArcTN Light 与 Cotengra Auto;深度优化场景比较 ArcTN Heavy 与 Cotengra AutoHQ、OMECO。图中优化时间与 log10 FLOPs 均越低越好。

32 个不同电路 · 每个电路重复 5 次 · 每次使用 8 个物理核 · 单次限时 3,600 秒

测试环境与配置 适用于本页全部性能图表

测试仅包含收缩路径搜索,不包含实际张量数值收缩。优化时间为路径搜索耗时,FLOPs 表示所得收缩路径的计算成本。

寻路任务配置
配置项实验设置
测试项目收缩路径搜索(仅规划,planning-only)
输入网络EinsumBenchmark32,包含 32 个不同量子电路
重复次数每个电路重复 5 次,各方法一致
优化目标纯 FLOPs
CPU 分配每次运行使用 8 个物理核
时间上限单次运行外部墙钟时间上限为 3,600 秒
测试规模32 个不同电路 × 5 次重复,共 160 次
运行设备
项目参数
CPUIntel Xeon Platinum 8253 @ 2.20 GHz
整机物理核心8 插槽 × 16 核,共 128 核
CPU 拓扑256 个逻辑 CPU,8 个 NUMA 节点
系统内存约 2.95 TiB
操作系统Ubuntu 24.04.4 LTS
Linux 内核6.8.0-117-generic
运行平台x86_64;glibc 2.39;CPU 寻路
主要框架与依赖版本
软件实验使用版本
Python3.12.3
ArcTN0.1.0
Cotengra0.8.2
cotengrust0.2.1
OMECO0.2.6
Google TNCO0.4.0
NumPy2.2.6
opt_einsum3.4.0
Optuna4.9.0
CMA-ES(cmaes)0.13.0
KaHyPar1.3.5

Google TNCO SA 采用轻量级路径搜索,但所得收缩路径质量相对较低。 在完整的 32 个测试电路上,其 FLOPs 几何平均值分别为 ArcTN Light 的 11.00× 和 ArcTN Heavy 的 15.05×; 相较于 Google TNCO SA,ArcTN Light 与 ArcTN Heavy 的端到端加速比分别为 1.06× 和 1.25×。 逐电路展开时不将其纳入对比

RESULTS / 03 · METRIC BREAKDOWN

ArcTN 张量网络收缩序优化
能力细览

实验从运算量、数据写入和中间态内存三个维度比较不同方法。图中横轴为相对本图最低值的倍数,数值越低越好。

LIGHT / ARCTN LIGHT

VS Cotengra Auto

  • 优化时间快 1.62×
  • FLOPs低 61.32%
  • 总写入元素数低 64.16%
  • 最大中间张量低 59.39%
  • 峰值存活元素低 61.30%

HEAVY / ARCTN HEAVY

VS Cotengra AutoHQ

  • 优化时间快 7.21×
  • FLOPs低 43.83%
  • 总写入元素数低 34.96%
  • 最大中间张量低 39.85%
  • 峰值存活元素低 38.81%

HEAVY / ARCTN HEAVY

VS OMECO

  • 优化时间快 1.33×
  • FLOPs低 11.85%
  • 总写入元素数低 6.89%
  • 最大中间张量低 16.88%
  • 峰值存活元素低 18.72%

DETAILS / 04

ArcTN 具体电路
性能表现(Heavy

深度优化场景逐条呈现不同规模与结构的测试电路,对比 ArcTN Heavy 与基线方法在收缩计算量和路径搜索时间上的表现,观察整体优势在具体电路上的稳定性。

HEAVY · 结果质量 在不同线路规模和结构下比较各方法得到的收缩计算量。

REPRODUCIBILITY / 05

继续阅读与复现