LIGHT / ARCTN LIGHT
VS Cotengra Auto
- 优化时间快 1.62×
- FLOPs低 61.32%
- 总写入元素数低 64.16%
- 最大中间张量低 59.39%
- 峰值存活元素低 61.30%
MODULE / 01 · TENSOR NETWORK
面向大规模量子线路模拟,以多链退火搜索更低开销的张量网络收缩路径,并通过 Rust 工程实现提升搜索与评估效率。
METHOD / 01
不同温度的搜索链承担探索与收敛任务,并周期性交换候选解。高温链帮助跳出局部最优,低温链保留更有价值的张量网络收缩序;Rust 实现用于承载高频候选评估和并行调度。
从量子线路得到张量及其指标连接关系。
多条温度链独立扰动和评估候选张量网络收缩序。
按接受概率交换状态,扩大有效搜索范围。
综合计算量与内存指标选择可执行方案。
RESULTS / 02 · PERFORMANCE OVERVIEW
轻量优化场景比较 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 次 |
| 项目 | 参数 |
|---|---|
| CPU | Intel 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 寻路 |
| 软件 | 实验使用版本 |
|---|---|
| Python | 3.12.3 |
| ArcTN | 0.1.0 |
| Cotengra | 0.8.2 |
| cotengrust | 0.2.1 |
| OMECO | 0.2.6 |
| Google TNCO | 0.4.0 |
| NumPy | 2.2.6 |
| opt_einsum | 3.4.0 |
| Optuna | 4.9.0 |
| CMA-ES(cmaes) | 0.13.0 |
| KaHyPar | 1.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
实验从运算量、数据写入和中间态内存三个维度比较不同方法。图中横轴为相对本图最低值的倍数,数值越低越好。
四项实验指标图表。
衡量完成收缩所需的浮点运算量。
反映收缩过程中累计产生的数据写入规模。
观察单个中间结果可能带来的内存压力。
衡量执行过程中同时驻留数据的峰值规模。
LIGHT / ARCTN LIGHT
HEAVY / ARCTN HEAVY
HEAVY / ARCTN HEAVY
DETAILS / 04
深度优化场景逐条呈现不同规模与结构的测试电路,对比 ArcTN Heavy 与基线方法在收缩计算量和路径搜索时间上的表现,观察整体优势在具体电路上的稳定性。
轻量优化场景逐条呈现不同规模与结构的测试电路,对比 ArcTN Light 与 Cotengra Auto/CMA-ES、AutoHQ/Optuna、Hyper128/Optuna 在收缩计算量和路径搜索时间上的表现,检验有限搜索预算下的优化效率。
HEAVY · 结果质量 在不同线路规模和结构下比较各方法得到的收缩计算量。
HEAVY · 搜索成本 对比生成张量网络收缩序所需时间,补充衡量方法的实际可用性。
LIGHT · 结果质量 在不同线路规模和结构下比较各方法得到的收缩计算量。
LIGHT · 搜索成本 对比生成张量网络收缩序所需时间,补充衡量方法的实际可用性。
REPRODUCIBILITY / 05