ATEST 框架:按 GPU 负载动态省 48 倍推理能耗

Mobina Kashaniyan · hf · 2026-10-22

爱荷华州立大学论文提出 ATEST,一种资源感知的测试时扩展(TTS)框架。核心观察:固定采样预算既忽略题目难度,也忽略 GPU 争用下推理能耗的变化——同样的 4 样本查询,GPU 负载升高时能耗最多增加 4.1 倍。

两种模式

实验结果:在 A100 上用 Phi-3-mini 于四个推理/QA 数据集评测,并迁移到 L40S 和 Qwen2.5-3B。高负载下的 GSM8K 上,ATEST-Online 达 60.0% 准确率且零延期,而 Calibrated 仅 44.0% 且 42.4% 查询被延期。计入离线剖析,Online 模式在评估负载下比 Calibrated 少用 42 倍 GPU 时间、48 倍能耗。消融显示在线成本学习是负载适应的关键;Online 的代价是短任务尾延迟更高、吞吐更低。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →