精读笔记

Problem Setting

这篇论文处理的是多地域 AI 数据中心的联合 workload-energy scheduling：训练任务可延迟、可跨数据中心调度；推理请求必须低延迟服务，但可以跨地域路由并选择 serving configuration；每个数据中心还有 renewable generation、battery、grid purchase 和 carbon intensity。真正难点不在单个模块，而在这些灵活性同时被 forecast uncertainty 约束：renewable、training arrival、inference arrival 都有误差，而且误差不是同分布静态噪声，而是随天气、时间、历史负载等 covariates 改变。

以前方法的卡点很明确：传统 RO 给一个全局 uncertainty set，可靠但保守；SO/CCP 需要分布假设，遇到 JCC 和高维 uncertainty 后很难解；DRCC 有理论吸引力，但大规模 scheduling 中通常计算代价高，还常要 Bonferroni/CVaR 等近似。任务的关键矛盾是：为了 carbon-aware scheduling，系统必须大胆利用时空灵活性；为了服务可靠性，又必须对 forecast error 保守。本文试图通过 context-dependent uncertainty set 把“该保守的地方保守，不该保守的地方放松”。

Motivation

作者的核心观察是 forecast error 的尺度和形状明显依赖上下文：晴天中午 PV forecast error、夜间 workload arrival error、不同 LLM request class 的到达波动，不应共享同一个 residual envelope。静态 uncertainty set 把所有 operating regimes 混在一起，会导致低风险场景下过度保守，高风险场景下又未必覆盖。

另一个动机是 AI data center 与传统 general-purpose data center 的 workload abstraction 不同。training 的 flexibility 主要是 deadline 和可用数据中心集合；inference 的 flexibility 主要是 routing 和 serving configuration，并受 TTFT/TBT/response SLO 约束。把它们压成 flexible / inflexible 会丢掉最能产生 carbon-aware benefit 的结构。

因此论文缺口不是“再提出一个鲁棒调度模型”，而是：如何把 covariate information 直接用于 JCC 的 uncertainty representation，同时避免求解完整 conditional distribution 或大规模 DRCC。

Core Idea

核心思想可以概括为：不学习点预测再用 residual 构造不确定性，而是直接学习一个由上下文参数化的 uncertainty set，并用校准集把该 set 调整到目标覆盖率。随后把 contextual chance constraint 转化成 robust constraint：只要真实 uncertainty vector 落在集合中的概率至少为 1-ε，那么对集合内所有 realization 可行的解，就满足原始 chance constraint。

这个思路改变了信息流。传统 predict-then-optimize 是 X → point forecast → residual set → optimization；本文是 X → uncertainty set → robust optimization。新的 inductive bias 是：优化器真正需要的不是最准确的均值预测，而是 constraint-relevant coverage shape。IQL 学上下界，MVL 学均值-方差椭球，本质是在学习条件误差结构，而 calibration 则把学习模型的不确定性转化为 distribution-free 覆盖保证。

和 prior 的本质区别在于，它没有显式估计 conditional distribution，也没有把 covariates 塞进 ambiguity set 后再做复杂 DRCC，而是用 conformal calibration + RO 的组合绕开了难解的概率约束。这也是其 scalability 的主要来源。

Method

方法中真正关键的机制只有几个。

1）AI workload 的结构化聚合建模。training 不做 job-level scheduling，而是按 class 聚合成 GPU-equivalent demand，并通过 deadline window 和可选数据中心集合描述灵活性。这解决的是生产级规模下 job-level MIP 不可解的问题。代价是牺牲细粒度调度真实性，但换来可优化结构。

2）inference 建模显式纳入 serving configuration。推理请求按任务和 token length class 分类，配置 p 表示 TP degree / GPU allocation 等，影响 latency、service rate 和功耗。这个机制的作用是让 carbon-aware routing 不只是“把请求送到绿色数据中心”，还要考虑为了满足 SLO 需要开多少实例、用多大 TP、消耗多少 GPU power。

3）contextual uncertainty learning。box set 用 interval quantile loss 预测上下界，ellipsoid 用 mean-variance loss 预测中心和尺度。它解决的是 residual distribution 随上下文变化的问题。核心变化是 uncertainty set 从全局静态对象变成 sample-conditional object。

4）calibration for JCC。对 box 使用 joint interval violation score，对 ellipsoid 使用 normalized residual score，然后取校准分位数扩张集合。因为 score 是对整个 uncertainty vector 计算的，所以得到的是 joint coverage，而不是逐维 marginal coverage。这一点对 JCC 很重要。

5）RO reformulation。box 变成 MILP，ellipsoid 变成 MISOCP。这里没有新的求解理论，主要是把难 CCP/DRCC 转成标准 solver 能处理的形式。

Key Insight / Why It Works

最核心的有效性来自两个因素：contextual heteroscedasticity 被利用，以及 conformal calibration 把 ML 模型变成可用于约束可行性的对象。换句话说，性能提升并不是因为优化器更聪明，而是因为 uncertainty set 更贴近当前 operating regime，从而减少不必要的 worst-case padding。

这篇论文最有价值的技术点是 coverage-to-feasibility 的组织方式：先对 uncertain vector 做 joint coverage，再用 robust feasibility 推出 JCC feasibility。这个逻辑简单但实用，特别适合 energy scheduling 这类约束线性、uncertainty 进入方式清楚的问题。它避免了直接求 conditional chance constraint，也避免了 DRCC 中 ambiguity set 选择和近似带来的复杂度。

我认为 IQL/MVL 本身不是特别新的贡献，更像是把已有 quantile regression、heteroscedastic regression、conformal calibration 和 robust optimization 拼成一个工程上干净的 pipeline。真正新增的信息是：把这个 pipeline 对齐到 AI data center scheduling 的 JCC 结构，并展示它在大规模调度中比 ReDRO / CVaR-DJCC 更可扩展。

哪些可能只是辅助：AI workload 功耗和 latency 模型虽然让问题更贴近 AI 场景，但多数是从已有系统论文和 profiling 假设移植过来；ESS、电价、碳价部分是标准 energy scheduling。实验中 cost/emission gain 可能部分来自 baseline 过保守或 ReDRO/CVaR formulation 计算负担，而不是 uncertainty learning 本身的绝对优势。

这不是 scaling、retrieval、memory reuse 或 test-time compute 类型的贡献；本质是 better inductive bias + data coverage。它假设上下文能解释误差结构，并用校准控制剩余风险。若数据分布稳定，这个假设很强；若出现 regime shift，优势可能迅速退化。

Relation To Prior Work

它最接近三条路线：carbon-aware geo-distributed data center scheduling、data-driven robust optimization with covariates、conformal prediction / calibration for uncertainty sets。

相对于 Google-style carbon-aware load shifting 或 general data center scheduling，本文的实质差异是 workload representation：training 和 inference 被分开建模，并与 GPU capacity、SLO、serving config、power model 联动。这比 flexible/inflexible abstraction 更贴近 AI workloads，但仍是聚合级模型。

相对于 classical RO/ARO，区别是 uncertainty set 从 covariate-independent 变成 covariate-dependent。这个差异是实质性的，因为它直接降低保守性，也是主要实验增益来源。

相对于 DRCC/CVaR-DJCC，本文不追求刻画 worst-case distribution，而是用 calibrated set coverage 替代概率分布建模。因此理论保证更弱但更可计算：保证 feasibility coverage，不保证对所有分布 ambiguity 下的最优性。

相对于 ReDRO 和 conditional robust optimization，本文看似新的是“loss-based uncertainty learning”，但思想上是已有 uncertainty set learning + conformal calibration 的重组。实质创新在于把 joint nonconformity score 和 JCC 可行性保证明确连起来，并在 AI data center 这个复杂调度问题上保持 MILP/MISOCP 级可解。

Dataset / Evaluation

评估覆盖了几个必要场景：Alibaba training trace 提供训练任务 GPU 需求，Azure LLM traces 提供推理请求时序与类别，renewable generation 由历史天气模拟，carbon intensity 来自 CAISO/ENTSO-E。这些数据足以验证论文核心 claim 中的“上下文误差 + 多地域 carbon-aware scheduling + AI workload 异质性”。

实验支持了三个结论：第一，contextual uncertainty set 比静态 set 更省成本和碳排；第二，ULRO 在 empirical violation rate 上能接近目标可靠性；第三，相比 ReDRO/CVaR-DJCC，ULRO-B 的求解时间随样本数增长更小，scalability 更好。

但 evaluation 仍主要是 offline simulation，不是真实 production deployment。renewable generation 是模拟而非数据中心实测；LLM serving 的 latency/power 参数来自文献和 profiling 假设，不等于真实多租户集群；training scheduling 是 aggregate VCC 风格，不验证 job-level side effects。benchmark 是否充分验证“statistical guarantees under operational nonstationarity”这一点并不充分。

还有一个归因问题：文中没有强 ablation 来区分 IQL/MVL 学到的 set shape、conformal calibration、AI-specific workload model、solver tractability 各自贡献。增益来源不清，尤其是与强 residual conformal baseline 的比较缺失。

Limitation

第一，finite-sample guarantee 的前提是 calibration samples 与 test sample exchangeable。真实数据中心中 workload、model serving pattern、weather forecast model、grid carbon intensity 都可能 nonstationary；一旦分布漂移，coverage guarantee 变成经验假设。

第二，保证对象是 uncertainty set coverage 和由此推出的 chance feasibility，不是 closed-loop reliability。真实系统中有 receding horizon、online correction、queue spillover、job cancellation、network congestion、GPU fragmentation，这些都可能破坏静态调度模型的可行性。

第三，AI workload 模型仍是聚合近似。training 用 GPU-equivalent workload 和 class-level deadline，无法表达 placement constraints、distributed training communication、checkpoint/preemption cost。inference queue 模型也较理想化，把 active instance 近似为独立队列，未充分处理 batching、KV cache、prefill/decode disaggregation、multi-model interference。

第四，scalability 上限取决于 MILP/MISOCP 的整数变量规模。ULRO-B 目前看起来快，是因为 box set 和聚合模型结构简单；当 class、DC、serving configuration、time horizon 继续扩展时，未必保持线性可扩展。

第五，所谓 contextual learning 是否真正 generalizable 文中未充分说明。周期性 workload trace 和天气特征可能让模型主要利用 data coverage，而不是学到可迁移的不确定性结构。跨 region、跨模型族、跨业务突发流量的泛化没有验证。

第六，risk allocation 处理较粗。多个 ICC/JCC 各自的 ε 如何选择、全系统可靠性如何组合、不同约束 violation 的业务代价是否等价，文中没有深入讨论。

Takeaway

• 1）对调度优化而言，学习点预测不是最优接口；更合理的接口可能是直接学习 constraint-relevant uncertainty object，再用 calibration 转成可行性保证。
• 2）contextual uncertainty set 是降低 RO 保守性的有效路径，尤其适合误差明显 heteroscedastic、但完整 conditional distribution 难估计的场景。
• 3）conformal calibration + robust optimization 是一个值得迁移的范式：只要约束对 uncertainty 结构清晰、robust counterpart 可解，就可以用 coverage guarantee 替代复杂概率建模。
• 4）未来真正值得做的是动态 / online calibrated scheduling：处理 regime shift、rolling horizon、risk reallocation，以及真实 serving stack 下的 feedback，而不是继续堆更复杂的 ambiguity set。

一句话总结

这篇论文把 AI 数据中心碳感知调度从静态鲁棒 / 难解 DRCC 推向“上下文不确定性学习 + 校准覆盖 + 标准鲁棒优化”的实用谱系，真正贡献在于用可计算的 coverage-to-feasibility 机制降低 JCC 调度的保守性。