精读笔记
Problem Setting
这篇论文把水填充问题推进到一个竞争式版本：发射端在总功率和空间约束下最大化信息率，干扰端在总干扰预算下选择最坏干扰分布。它瞄准的不是经典 fixed-noise water-filling，而是多运营商 LEO/NTN 频谱共享中“对方干扰既不固定也不可信”的场景。
困难点有两层。第一，高斯输入下问题还能被组织成相对干净的 convex-concave saddle game，水位变量有明确闭式结构；但离散星座下，mutual information 通过 I-MMSE 引入 modulation-dependent curvature，mercury/water-filling 的 proximal step 不再闭式，干扰侧甚至不保证全局凸性。第二，现实部署中通道数量、约束矩阵、调制格式持续变化，传统 solver 每个实例从头迭代，无法 amortize 已经学到的水位搜索规律。
关键矛盾是：AWF 的解看似是高维 allocation，但真正控制结构是低维 dual water levels 和稀疏约束耦合；已有方法要么精确但慢，要么学习快但固定维度/固定任务。论文试图用结构化 learned optimizer 把二者折中起来。
Motivation
已有路线缺的是跨实例复用能力。经典水填充/PDHG/Mirror-Prox 能利用问题结构，但每个新 channel realization、constraint graph、modulation table 都要重新跑；学习式无线资源分配能加速，但常常是固定拓扑、固定维度、固定信道分布上的 supervised regression。
作者的核心观察是：AWF 并不是一般黑盒非凸优化。即使在 mercury/water-filling 下，KKT 仍然显示出很强的结构：活跃通道上 marginal utility 被 global water level 和 constraint-induced shift 对齐；线性约束通过 A^T theta 只以稀疏图形式耦合通道；通道本身是 unordered set。因此，缺口不是“再设计一个 solver”，而是“学习一个尊重这些不变性和低维 dual 结构的 solver dynamics”。
这也是为什么论文会走 foundation model 叙事：它不是想训练一个单一实例的神经解算器，而是想把 AWF 族问题的共享搜索动力学参数化出来。
Core Idea
论文真正的核心思想是：把 adversarial water-filling 的可迁移结构从解空间转移到动力学空间。模型不直接学习从 channel/constraint 到最优 p,n 的静态映射，而是学习一个带投影的 primal-dual extragradient 更新过程。这样做的好处是，约束满足、预算守恒、dual feedback、水位调整都可以在迭代结构中显式体现，网络只需要学习初始化、表示和步长/修正，而不是学习完整优化问题。
本质区别在于 inductive bias 的强度。普通 GNN/MLP resource allocation 方法把信道图输入后输出功率，泛化主要靠数据；这里的信息流被 KKT 结构重新组织：set encoder 处理 permutation invariance，constraint graph 近似 A^T theta，global latent 表示水位状态，projected extragradient 保留 saddle dynamics。这使模型更像 amortized Mirror-Prox / learned PDHG，而不是普通预测器。
理论上它可能有效，是因为 AWF 的高维 allocation 被低维 dual state 强约束；如果训练分布覆盖了足够多的水位 regime，模型学到的是“如何快速找水位”，而不是“记住某个维度下的功率模板”。这也是其跨 m、跨约束、跨调制能成立的主要直觉。
Method
1. AWF 建模：论文先把频域和空间约束下的发射功率 p 与 adversarial interference n 写成 max-min。高斯情形给出强凸凹/水位闭式关系，作用是建立一个可解释的结构模板；mercury 情形用 I-MMSE 替换 log utility，保留 KKT 形式但放弃全局凸凹保证。
2. dual water-level 结构：关键不是公式本身，而是说明最优解由 global simplex dual、constraint dual theta 和 interference dual 共同控制。空间约束把单一水位变成 channel-wise effective water level：1 / (nu + A^T theta)_i。这为后面的 graph module 提供了明确目标：学习约束诱导的 level shift。
3. 结构化实例表示：通道集合用 permutation-invariant encoder，解决变长和通道重排问题；约束矩阵 A 被看作 channel-constraint bipartite graph，解决线性约束导致的局部/稀疏耦合问题。这里的核心变化是模型输入不再是固定维向量，而是优化问题的结构化表示。
4. learned projected extragradient：离散星座下每步 proximal 需要解大量一维非线性方程，论文用可学习步长的 extragradient 代替显式 implicit solve。投影到 simplex 和非负 dual cone 保证预算与 dual 可行性，训练目标再用 gap、ineq violation、KKT residual 引导有限步 stationarity。
5. distribution token：用 mutual information 和 derivative 表在 SNR grid 上编码调制格式。它解决的是 modulation-dependent curvature 的条件化问题，使同一个 learned dynamics 可以根据星座曲线调整更新。
Key Insight / Why It Works
最核心的有效性来源是 representation alignment：网络结构和 AWF 的 KKT/dual geometry 对齐了。活跃通道的最优性不是任意模式，而是 marginal utility 对齐；约束只通过 A^T theta 改变局部水位；预算约束通过 simplex projection 自动实现全局归一。这些都大幅降低了需要学习的函数复杂度。
第二个有效来源是 amortized optimization，而不是真正意义上的通用推理。模型通过大量随机实例学习了典型 water-level search 的步长、初始化和修正策略，因此在测试时用 64 步左右的 learned dynamics 替代数千步稳定 Mirror-Prox。这里的 speedup 很可能主要来自 fixed finite-step inference + GPU-friendly architecture + training distribution amortization，而不是发现了新的优化复杂度理论。
第三个来源是 test-time compute 仍然存在。模型不是 one-shot predictor，而是 unrolled extragradient；因此它保留了一定自校正能力。这个设计比直接输出 p,n 更稳，也解释了为什么 KKT residual 不如 Mirror-Prox 但 objective 接近：有限步 learned solver 找到了足够好的水位区间，但没有完全收敛。
哪些是核心贡献？我认为是“AWF minimax + water-level/KKT 结构化 learned extragradient”这个组合。Perceiver、GNN、distribution token 分别是合理工程化承载，但单独看都不是新思想。所谓 foundation model 的说法偏强，更准确地说是 domain-specific amortized primal-dual optimizer。
增益归因并不完全清楚。实验中的 Mirror-Prox 是稳定固定步长 baseline，若加入 warm start、adaptive step、Anderson acceleration、GPU batching、problem-specific stopping，runtime gap 可能缩小。论文没有充分拆分：加速来自 learned initialization、learned step sizes、architecture inductive bias，还是只是固定 T 与 baseline 迭代次数差异。文中未充分说明这一点。
Relation To Prior Work
最接近的技术谱系有三条：经典 water-filling / mercury-water-filling，convex-concave saddle / PDHG / Mirror-Prox，以及 learning to optimize / GNN-based wireless resource allocation。
相对 classical water-filling，论文新增的是 adversarial interference 侧和空间线性约束下的 minimax 组织方式；但高斯部分的水位推导本质上是已知 KKT/dual water-filling 思路的扩展，创新性主要在把它作为 learned model 的结构先验。
相对 PDHG/Mirror-Prox，论文没有提出新的优化算法，而是把 extragradient unroll 成可学习动态。理论也不是证明一个新算法全局收敛，而是说明在 fixed point/contractivity 等条件下 learned dynamics 与 KKT/local convergence 一致。这里的实质差异是 amortization：把求解成本从测试时转移到训练时。
相对 GNN resource allocation，论文更强的地方在于没有只学习 graph-to-power，而是显式保留 saddle gradient、simplex projection、dual variable 和 KKT residual。它属于 model-based deep learning / learned optimizer，而不是纯数据驱动无线资源管理。
看似新的“wireless foundation model”更多是叙事包装。真正新增的信息不是 foundation scale，而是把 AWF 族问题的物理/优化不变性编码成可变规模神经求解器。
Dataset / Evaluation
评估覆盖了三个维度：channel size 从训练内规模扩展到未见的 1024，调制从训练的 16QAM/64QAM扩展到 mixed 和 held-out 256QAM，约束从随机 sparse 扩展到 group/prefix/dense。这些实验确实验证了模型在合成分布族内具备一定 zero-shot transfer，尤其是跨 m 的稳定性支持 permutation-invariant + graph 表示的必要性。
但 evaluation 仍然是模拟/合成实例，没有真实 LEO 轨道、beam pattern、regulatory mask、traffic dynamics 或多运营商策略数据。所谓多场景泛化主要是程序生成分布上的结构变化，不等同于真实部署泛化。
结果支持“objective 接近 Mirror-Prox 且更快”，但不完全支持“stationary solution quality 同等”。表中模型的 transmit-side KKT residual 往往明显高于 Mirror-Prox，只是 objective 差距小。这说明它更像 fast approximate solver，而不是高精度求解器。
benchmark 还有一个潜在偏置：baseline 是固定步长稳定 Mirror-Prox，且运行时间很长；如果对 baseline 做同等 GPU 工程优化、adaptive stopping 或 warm-start，速度优势的实际幅度可能不同。文中未充分说明 baseline 是否是最强工程实现。
Limitation
1. 理论保证很弱且条件化。KKT consistency 要求 extragradient fixed point 还满足 underlying projected first-order fixed point；local convergence 直接假设 Jacobian spectral radius < 1。这些结果解释了“如果模型已经学到收缩映射会怎样”，但没有证明训练会得到这样的映射。
2. 非凸 mercury 情形没有全局保证。离散星座下干扰侧非凸，模型输出的只是局部 stationary 或近似 saddle residual 低的点。不同初始化可能对应不同局部点，论文没有系统讨论局部解质量和 worst-case robustness。
3. 泛化可能主要来自数据覆盖。训练实例在线生成，覆盖了多种 m、A、预算、调制，因此测试的所谓 zero-shot 很可能仍在相近生成机制内。对真实 OOD constraint geometry、极端 SNR、非平稳干扰、多峰调制混合的泛化未被证明。
4. foundation model 定位偏夸张。模型规模、数据规模、任务开放性都更像专用 amortized solver，不像跨无线任务的 foundation model。它没有展示跨优化目标、跨协议层或跨真实网络环境迁移。
5. 速度收益归因不清。learned finite-step solver 与长迭代 Mirror-Prox 比较天然有优势，但代价是 KKT residual 较高。论文没有给出同等精度-时间 Pareto 曲线，也没有清楚回答“若 baseline 只跑到相同 objective gap，会慢多少”。
6. 真实部署鸿沟大。LEO 共存问题涉及时变 channel、预测 horizon、beam scheduling、regulatory constraints、multi-agent strategic behavior。本文 AWF 是静态单阶段 minimax，planner 并没有长期状态建模；把它直接称为解决 LEO coexistence 仍有明显距离。
Takeaway
- 1. 最值得迁移的 insight 是：无线资源分配中的高维 allocation 往往受低维 dual variables 控制；学习器应该对齐 dual/KKT 结构，而不是盲学 allocation map。
- 2. 对非闭式 proximal / mercury-water-filling 这类问题，learned extragradient 是一个合理折中：保留投影和梯度物理结构，用学习来 amortize 难解的局部搜索。
- 3. 跨规模泛化的关键不是更大的网络，而是正确的不变性：permutation invariance、constraint graph sparsity、budget projection、modulation curve conditioning。
- 这些比普通 GNN 堆叠更重要。
一句话总结
这篇论文把 adversarial/mercury water-filling 组织成带 KKT 水位结构的 minimax 问题，并用结构化 learned extragradient 做 amortized 求解，是从 model-based water-filling solver 向 domain-specific neural primal-dual optimizer 演化的一步。