arXivDaily arXiv每日学术速递 周一至周五更新

大厂专区

Hewlett Packard Enterprise(慧与)

至 收录 19
2607.15123 2026-07-17 cs.AR cs.AI 新提交

NIFA: Nonlinear IMC enhanced FPGA for efficient ML inference

NIFA:用于高效机器学习推理的非线性IMC增强型FPGA

Jiajun Hu, Ruthwik Reddy Sunketa, Lei Zhao, Archit Gajjar, Luca Buonanno, Aman Arora

机构 * Arizona State University(亚利桑那州立大学) Hewlett Packard Enterprise Labs(惠普企业实验室)

AI总结 研究针对传统FPGA中IMC设计局限,提出含无ADC的IMC块的新颖FPGA架构,通过设计空间探索和高效映射,在CNN和Transformer基准测试中显著提升能源和面积效率,推进FPGA设计。

详情
AI中文摘要

近期的FPGA通过专用张量块和片上RAM计算提高了深度学习推理效率。基于电阻式随机存取存储器(ReRAM)的模拟内存计算(IMC)进一步提升了效率,在ReRAM交叉开关内直接执行向量矩阵乘法(VMM),相比传统数字逻辑,计算密度和能源效率提高了一个数量级。传统IMC设计仅支持静态权重VMM,将非线性操作和动态矩阵乘法(DIMM)留给FPGA结构。因此,IMC的优势主要局限于静态权重模型,而基于Transformer的模型受益有限。此外,每个IMC块内的模数转换器(ADC)消耗了超过70%的面积和功耗,进一步限制了系统效率和可扩展性。为解决这些限制,我们提出了一种新颖的FPGA架构,集成了无ADC的IMC块,用模拟内容可寻址存储器(ACAM)取代传统ADC,ACAM可在块内原生执行非线性操作。为充分利用此块,我们进行了FPGA感知的设计空间探索,确定最佳交叉开关尺寸,同时平衡FPGA面积、灵活性和DL性能,并开发了一种高效映射,利用ACAM执行DIMM操作,将IMC的适用性扩展到注意力计算。在基于CNN和Transformer的基准测试中,所提出的架构分别实现了高达40倍和1.9倍的更高能源效率,以及4.1倍和2.5倍的更高面积效率。总体而言,它显著提高了FPGA DL推理效率,并在基于Transformer的长输入序列工作负载上保持了强劲的增益增长。

英文摘要

Recent FPGAs have improved deep learning (DL) inference efficiency through dedicated tensor blocks and in-BRAM computation. ReRAM-based analog in-memory computing (IMC) pushes efficiency further, offering an order-of-magnitude improvement in compute density and energy efficiency over conventional digital logic by performing vector-matrix multiplication (VMM) directly within the ReRAM crossbar; prior work has integrated such IMC blocks into FPGAs for DL inference. However, conventional IMC designs support only static-weight VMM, leaving nonlinear operations and dynamic matrix-matrix multiplication (DIMM) to the FPGA fabric. As a result, the benefits of IMC are largely confined to static-weight models, whereas Transformer-based models, which rely on frequent nonlinear and DIMM operations, gain only limited improvement. Moreover, the ADCs within each IMC block consume more than 70% of its area and power, further limiting system efficiency and scalability. To address these limitations, we propose a novel FPGA architecture that integrates an ADC-free IMC block, replacing the conventional ADC with analog content-addressable memories (ACAMs) that natively perform nonlinear operations inside the block. To fully exploit this block, we conduct an FPGA-aware design-space exploration that determines optimal crossbar dimensions while balancing FPGA area, flexibility, and DL performance, and we develop an efficient mapping that leverages ACAMs to carry out DIMM operations, extending the applicability of IMC to attention computation. On CNN and Transformer-based benchmarks, the proposed architecture achieves up to 40x and 1.9x higher energy efficiency and 4.1x and 2.5x higher area efficiency, respectively. Overall, it significantly improves FPGA DL inference efficiency and sustains robust gains on Transformer-based workloads across long input sequences, advancing domain-specialized FPGA design.

URL PDF HTML 收藏
2607.03932 2026-07-07 cs.LG cs.AI 新提交

MPSelectTune: Prompt-type Selection for Fine-tuning improves Concept Unlearning in LLMs

MPSelectTune:微调中的提示类型选择可提高大语言模型中的概念遗忘能力

Shubhadip Nag, Srinjoy Das, Agniva Saha, Anushree Ghosh, Soumi Das, Tarun Kumar, Suparna Bhattacharya, Sourangshu Bhattacharya

机构 * IIT Kharagpur(印度理工学院卡拉格布尔分校) MPI-SWS(马克斯·普朗克软件系统研究所) HPE Labs(慧与实验室)

AI总结 研究大语言模型中去除有害概念难题,提出对抗方法用联合提示预测主任务和概念任务,MPSelectTune两阶段方法可提升平均遗忘性能,实验显示能提高主任务精度并降低概念精度。

Comments Accepted at NeurIPS 2025 - Reliable ML Workshop (https://openreview.net/forum?id=Jk8sOL97Tg#discussion)

详情
AI中文摘要

大语言模型可通过适当提示适应多样任务,但预训练模型中的偏见或有害概念会导致不良响应。现有方法难跨不同提示类型去除这些概念。本文探索新对抗方法,用联合提示预测主任务和概念任务。实验表明所提MPSelectTune方法可提升性能。

英文摘要

LLMs can be conveniently adapted to a diverse set of tasks, e.g, prediction, question-answering tasks, etc, using appropriate prompts with few-shot examples. Biased or harmful concepts, e.g. gender or bio-weapons, present in pre-trained LLMs can lead to unsafe or unethical responses for many such prompts. Removing such undesirable concepts robustly across different prompt types remains a challenging problem, since existing unlearning methods typically ignore the impact of prompt variation. In this paper, we explore a novel adversarial approach to use a joint prompt for the main task and concept task prediction. We show that fine-tuning using the ``worst prompt type'' for concept prediction (with the highest concept accuracy) improves the average unlearning performance over a fine-tuning method that uses a combination of all prompt types. Our proposed method, MPSelectTune, is a two-stage approach that minimizes the concept accuracy of the highest accuracy-prompt type, after fine-tuning using a novel multi-task loss using multiple prompt types. Experimental results on four benchmarks show $2 - 15\%$ main task accuracy improvements over recent baselines and while reducing the worst-case concept accuracy by up to $17\%$ compared to recent baselines.

URL PDF HTML 收藏
2606.30389 2026-06-30 cs.LG

Predict, Reuse, and Repair: Accelerating Dynamic Sparse Attention for Long-Context LLM Decoding

预测、重用与修复:加速长上下文LLM解码的动态稀疏注意力

Tianyu Wang, Gourav Rattihalli, Aditya Dhakal, Junbo Li, Zhiwei Ren, Dejan Milojicic, Longfei Shangguan

机构 * University of Pittsburgh(匹兹堡大学) HPE Labs(HPE实验室)

AI总结 提出PRR运行时系统,利用动态稀疏注意力选择的时间局部性,通过预测、推测和增量修复来减少解码延迟,最高可降低40%的每令牌延迟,同时保持下游任务精度。

Comments 9 pages body plus 3 pages appendix, 13 pages total

详情
AI中文摘要

动态稀疏注意力(DSA)通过仅关注与每个查询最相关的top-K KV块来加速长上下文LLM解码,但它引入了一个串行的选择到注意力依赖,这成为了新的延迟瓶颈。我们提出了PRR,一个推测-重用-修复运行时,它利用DSA选择中的时间局部性来预测可能的块,在选择进行时推测性地对它们进行注意力计算,并在真实选定集合已知后增量地修复遗漏的块。PRR使用轻量级的基于EMA的预测器、一个基于性能分析的推测预算(使推测工作不处于关键路径上),以及一个基于FlashAttention的修复内核,该内核使用在线softmax统计将遗漏的块折叠到部分注意力状态中。在长上下文基准测试和代表性DSA方法上,PRR将每令牌解码延迟降低了高达40%,同时保持了下游任务精度。Github: https://github.com/Tianyu9748/Incremental_FlashAttention

英文摘要

Dynamic sparse attention (DSA) accelerates long-context LLM decoding by attending to only the top-K KV blocks relevant to each query, but it introduces a serialized selection-to-attention dependency that emerges as a new latency bottleneck. We present PRR, a speculate-reuse-repair runtime that exploits temporal locality in DSA selections to predict likely blocks, speculate the attention over them while selection is in flight, and incrementally repair missed blocks once the true selected set is known. PRR uses a lightweight EMA-based predictor, a profiling-guided speculation budget that keeps speculative work off the critical path, and a FlashAttention-based repair kernel that folds missed blocks into the partial attention state using online-softmax statistics. Across long-context benchmarks and representative DSA methods, PRR reduces per-token decoding latency by up to 40% while preserving downstream task accuracy. Github: https://github.com/Tianyu9748/Incremental_FlashAttention

URL PDF HTML 收藏
2606.11387 2026-06-11 cs.CL cs.AI cs.LG 新提交

Small Experiments, Cheaper Decisions: A Case Study in Staged Promotion for Micro-Pretraining

小实验,更经济的决策:微预训练中分阶段提升的案例研究

Felipe Chavarro Polania

机构 * Hewlett Packard Enterprise(慧与科技公司)

AI总结 研究微预训练中分阶段提升协议,通过固定预算筛选配置,在Windows A100和Linux L40S上验证,发现早期排名不稳定,但最终协议以144 GPU小时找到最优配置,成本低于全量筛选。

Comments 14 pages, 5 figures; 12-hour dual-host micro-pretraining promotion study; source package includes curated ancillary artifacts

详情
AI中文摘要

短预训练运行可以降低实验成本,但它们也可能过度推广那些仅在小预算下表现良好的配置。我们针对固定微预训练运行器在两个异构主机块(Windows A100和Linux L40S)上研究了一种可审计的分阶段提升协议。从12个预先筛选的配置开始,我们使用2分钟、5分钟、10分钟、60分钟和12小时的分阶段预算,并在昂贵的延续之前设置固定的提升规则。早期筛选被有意视为不稳定:5分钟和10分钟的排名对主机敏感,而最终的12小时排名最优条件并非复制10分钟门控下的平均最佳条件。由于不同阶段的种子范围不同,这些变化是操作性的提升证据,而非种子内曲线。复制60分钟门控将分阶段因子筛选桥接参考保留在提升集中,它在所有四个60分钟主机-种子单元中排名第一。在最终的12小时确认包中,桥接条件在两个种子的所有四个主机-种子单元中排名第一;贪婪比较器未满足固定的0.010 val_bpb近似等价规则;更便宜的d8/ar48(深度8,宽高比48)哨兵未满足固定的0.020平均差距规则。执行的12小时分支花费144 GPU小时,完整的分阶段协议记录169.2训练GPU小时(包括筛选阶段)。继续所有四个60分钟候选将花费192 GPU小时,而继续所有九个复制10分钟候选将花费432 GPU小时。后者是未运行延续的会计反事实,并非表明跳过的候选不可能超越参考。结果是一个有界成本分配发现,而非全局最优性、容量归一化优越性或优于自适应超参数优化方法的声明。

英文摘要

Short pretraining runs can reduce experimental cost, but they can also over-promote configurations that only look strong at tiny budgets. We study an auditable staged-promotion protocol for a fixed micro-pretraining runner on two heterogeneous host blocks: Windows A100 and Linux L40S. Starting from twelve prior-screened configurations, we use staged budgets of 2 minutes, 5 minutes, 10 minutes, 60 minutes, and 12 hours, with frozen promotion rules before expensive continuations. The early screens are intentionally treated as unstable: the 5- and 10-minute rankings are host-sensitive, and the eventual 12-hour top-ranked condition is not the mean-best condition at the replicated 10-minute gate. Because seed ranges differ across stages, these changes are operational promotion evidence, not within-seed curves. A replicated 60-minute gate keeps the Staged Factorial Screening bridge reference in the promoted set, where it ranks first in all four 60-minute host-seed cells. In the final 12-hour confirmation package, the bridge condition ranks first in all four host-seed cells across two seeds; the greedy comparator does not meet the frozen 0.010 val_bpb near-equivalence rule; and the cheaper d8/ar48 (depth-8, aspect-48) sentinel does not meet the frozen 0.020 mean-gap rule. The executed 12-hour branch spends 144 GPU-hours, and the full staged protocol records 169.2 training GPU-hours including screening stages. Continuing all four 60-minute candidates would spend 192 GPU-hours, while continuing all nine replicated 10-minute candidates would spend 432 GPU-hours. The latter numbers are accounting counterfactuals for unrun continuations, not evidence that skipped candidates could not have overtaken the reference. The result is a bounded cost-allocation finding, not a claim of global optimality, capacity-normalized superiority, or superiority over adaptive hyperparameter optimization methods.

URL PDF HTML 收藏
2606.05186 2026-06-05 cs.LG cs.CL

Staged Factorial Screening for Budget-Constrained Micro-Pretraining

预算受限的微预训练中的分阶段因子筛选

Felipe Chavarro Polania

机构 * Hewlett Packard Enterprise(惠普企业)

AI总结 针对预算受限的微预训练,提出分阶段分数因子设计方法,通过短时筛选识别高惩罚方向并确认有效锚点,在共享加速器上实现高效配方筛选。

Comments 23 pages, 4 figures

详情
AI中文摘要

预算受限的微预训练通常需要在共享加速器上对许多候选配方进行分诊,然后才能花费更大的搜索预算。我们研究了分阶段分数因子工作流是否能在这种设置中恢复稳定的早期效应结构。在固定的自动研究衍生的单GPU训练循环上,我们运行了613个实验,包括在2、5和10分钟时的试点和后续筛选;5和10分钟时的完整16条件种子重运行;有针对性的种子锚点检查;同主机贪婪和匹配成本随机基线;一个60分钟的桥接包;以及通过24小时的有界Windows A100和Linux L40S锚点延续。总批次、深度和宽度的主要惩罚在短预算时最大,并随预算增加而放松。在预先声明的种子全屏系列中,D、A、B和C在预算内Benjamini-Hochberg校正后,在5和10分钟时保留非零估计,而E则没有。随机搜索可以在这个32条件空间中达到强当前最优,但反复在相同的低惩罚区域,且没有因子归因。60分钟桥接锚点具有最低均值,尽管该包没有将工作流改进与更大桥接模型的能力优势分开。在两个主机上的有界12小时和24小时三锚点延续中,桥接具有最低样本均值,而非桥接顺序保持主机敏感。因此,我们提出了一个有界方法结果:使用短设计筛选来识别高惩罚方向,在重复运行下确认有希望的锚点,并在缩减空间内局部细化。证据支持在24小时内两个主机上的以桥接为中心的推荐,而不是硬件不变的排名或通用超参数优化的优越性。

英文摘要

Budget-constrained micro-pretraining often requires triaging many candidate recipes on a shared accelerator before larger search budgets are spent. We study whether a staged fractional-factorial workflow can recover stable early effect structure in this setting. On a fixed autoresearch-derived single-GPU training loop, we run 613 experiments across pilot and follow-up screens at 2, 5, and 10 minutes; full 16-condition seeded reruns at 5 and 10 minutes; targeted seeded anchor checks; same-host greedy and matched-cost random baselines; a 60-minute bridge package; and bounded Windows A100 and Linux L40S anchor continuations through 24 hours. Main penalties from total batch, depth, and width are largest at short budgets and relax as budget increases. Within the predeclared seeded full-screen families, D, A, B, and C retain non-zero estimates at 5 and 10 minutes after within-budget Benjamini-Hochberg correction, while E does not. Random search can reach strong incumbents in this 32-condition space, but repeatedly in the same low-penalty region and without factor attribution. The 60-minute bridge anchor has the lowest mean, although that package does not separate workflow refinement from the larger bridge model's capacity advantage. In bounded 12-hour and 24-hour three-anchor continuations on both hosts, the bridge has the lowest sample mean while the non-bridge ordering stays host-sensitive. We therefore present a bounded methods result: use short designed screens to identify high-penalty directions, confirm promising anchors under repeated runs, and refine locally inside the reduced space. The evidence supports a bridge-centered recommendation through 24 hours on two hosts, not hardware-invariant ranking or general hyperparameter-optimization superiority.

URL PDF HTML 收藏
2605.30018 2026-06-01 cs.CL cs.LG

Latent Performance Profiling of Large Language Models

大型语言模型的潜在性能剖析

Tanmoy Chakraborty, Ayan Sengupta, Suparna Bhattacharya, Partha Pratim Chakrabarti, Amlan Chakrabarti, Supratik Chakraborty, Partha Pratim Das, Lipika Dey, Richa Singh, Mayank Vatsa

机构 * Department of Electrical Engineering, Indian Institute of Technology Delhi(印度理工学院德里分校电子工程系) Yardi School of Artificial Intelligence, Indian Institute of Technology Delhi(印度理工学院德里分校人工智能学院) Hewlett Packard Enterprise, India(印度惠普企业公司) Department of Computer Science & Engineering, Indian Institute of Technology Kharagpur(印度理工学院Khargapur分校计算机科学与工程系) A.K.Choudhury School of Information Technology, University of Calcutta, India(印度加尔各答大学信息科技学院) Department of Computer Science & Engineering, Indian Institute of Technology Bombay(印度理工学院孟买分校计算机科学与工程系) Department of Computer Science, Ashoka University, India(阿什oka大学计算机科学系) Department of Computer Science & Engineering, Indian Institute of Technology Jodhpur(印度理工学院朱罗普分校计算机科学与工程系)

AI总结 提出潜在性能剖析(LPP)框架,通过隐藏激活和输出分布提取任务无关的诊断指标,揭示模型内在特性,补充传统基准评估。

详情
AI中文摘要

大型语言模型(LLMs)在标准化基准测试中经常取得令人印象深刻的分数,但仅凭准确性对能力的了解有限。通过排行榜评估开源LLMs面临持续的问题,如数据污染、任务范围狭窄以及与真实世界可靠性的弱对齐。基于基准的评估(如MMLU PRO、BBH或IFEval)主要捕捉模型在固定测试集上的输出,而非其如何处理信息、校准不确定性或构建内部知识。在本文中,我们主张从以基准为中心的评估转向对LLMs进行互补的、以状态为中心的内在评估。为此,我们引入了潜在性能剖析(LPP)——一个从隐藏激活和输出分布中提取任务无关诊断的框架。LPP在模型的潜在表示和动态上定义了一组标量指标,揭示了与规模无关的特征,从而实现可解释的比较并揭示隐藏的脆弱性。与静态准确性分数不同,LPP在相似规模的模型间提供稳定、对架构敏感的签名。通过对八个LLMs(规模范围0.5B-14B)的广泛实证分析,我们证明了具有相似基准分数的模型可能表现出对比的潜在特征,例如熵或适应性的差异。在这些见解的指导下,我们设计了用于不确定性和符号推理的合成探针,这些探针与内在指标一致,同时与排行榜偏差解耦。我们建议将LPP与基准一起报告,以提供对模型行为更深入、可解释的理解,从而实现更可靠的模型选择、安全评估以及超越表面准确性的评估。

英文摘要

Large language models (LLMs) frequently achieve impressive scores on standardized benchmarks, yet accuracy alone offers a limited view of their capabilities. Evaluating open-source LLMs through leaderboards faces persistent issues like data contamination, narrow task scope, and weak alignment with real-world reliability. Benchmark-based evaluations such as MMLU PRO, BBH, or IFEval primarily capture what a model outputs on fixed test sets, not how it processes information, calibrates uncertainty, or structures internal knowledge. In this article, we advocate for a shift from benchmark-centric evaluation toward a complementary, state-centered intrinsic assessment of LLMs. To this end, we introduce Latent Performance Profiling (LPP) -- a framework that derives task-agnostic diagnostics from hidden activations and output distributions. LPP defines a set of scalar metrics on a model's latent representations and dynamics, revealing scale-independent traits that enable interpretable comparisons and uncover hidden vulnerabilities. Unlike static accuracy scores, LPP provides stable, architecture-sensitive signatures across models of similar size. With extensive empirical analyses across eight LLMs, spanning a size range of 0.5B-14B, we demonstrate that models with similar benchmark scores can exhibit contrasting latent profiles, such as differences in entropy or adaptability. Guided by these insights, we design synthetic probes for uncertainty and symbolic reasoning that align with intrinsic metrics while decoupling from leaderboard bias. We recommend that reporting LPP alongside benchmarks provides a deeper, interpretable understanding of model behavior, enabling more reliable model selection, safety assessment, and evaluation beyond surface-level accuracy.

URL PDF HTML 收藏
2605.22850 2026-05-25 cs.DC cs.AI

ObjectCache: Layerwise Object-Storage Retrieval for KV Cache Reuse

ObjectCache: 用于KV缓存重用的分层对象存储检索

Yu Zhu, Aditya Dhakal, Yunming Xiao, Dejan Milojicic, Gustavo Alonso

机构 * ETH Zurich(苏黎世联邦理工学院) HPE Labs(惠普实验室) The Chinese University of Hong Kong, Shenzhen(香港中文大学(深圳))

AI总结 提出ObjectCache系统,利用S3兼容对象存储存储KV缓存,通过协同设计存储协议和传输调度,在GPU消费顺序下交付数据,重叠数据传输与计算,从而在不增加集群成本的前提下最小化TTFT延迟。

详情
AI中文摘要

前缀KV缓存已成为LLM服务中的关键机制:它通过避免共享前缀(即系统提示)的请求之间的冗余计算来减少首令牌时间(TTFT)。然而,累积的KV缓存通常超过GPU内存和本地DRAM的容量。为了保持低延迟,当前系统将KV缓存保存在远程DRAM池中,从而增加了服务集群的规模和成本。在本文中,我们探索了一种不同的方法:将KV缓存存储在S3兼容的对象存储中,使容量不再成为约束,同时最小化对TTFT的影响。我们提出了ObjectCache,它协同设计存储协议和传输调度,使存储服务器按照GPU消费的顺序交付KV缓存数据,并在并发请求之间重叠数据传输与计算。我们在一个100 Gbps的RoCE集群上使用NIXL(一个抽象存储和内存的推理库)、Ceph RGW(一个用于集群的对象网关)和DAOS(一个开源存储系统)对ObjectCache进行了原型实现。对于当今系统中常见的64K上下文,ObjectCache相比本地DRAM仅增加5.6%的延迟;对于4K上下文,由于可用于掩盖传输的计算较少,ObjectCache相比最优的本地逐层基线增加了56-75毫秒。在共享带宽限制下,我们的调度器相比等带宽共享将增加的TTFT减少了1.2-1.8倍。

英文摘要

Prefix KV caching has become a key mechanism in LLM serving: it reduces time to first token (TTFT) by avoiding redundant computation across requests that share a prefix (i.e., the system prompt). However, the accumulated KV cache is often larger than what GPU memory and local DRAM can hold. To preserve latency, current systems keep the KV cache in remote DRAM pools, increasing serving-cluster size and cost. In this paper, we explore a different approach: storing the KV cache in S3-compatible object storage so that capacity is no longer the constraint, while minimizing the impact on TTFT. We propose ObjectCache, which co-designs the storage protocol and transfer schedule so that the storage server delivers KV cache data in the order the GPU consumes it, overlapping data transfer with compute across concurrent requests. We prototype ObjectCache on a 100 Gbps RoCE cluster with NIXL (an inference library that abstracts storage and memory), Ceph RGW (an Object Gateway for clusters), and DAOS (an open source storage system). For 64K contexts, common in today's systems, ObjectCache adds only 5.6\% latency over local DRAM; for 4K contexts, where less compute is available to mask transfer, ObjectCache adds 56--75\,ms over the optimal local layerwise baseline. Under shared bandwidth caps, our scheduler reduces added TTFT by 1.2--1.8x compared with equal bandwidth sharing.

URL PDF HTML 收藏
2503.16589 2026-05-18 cs.LG cs.ET math.ST stat.TH

A Statistical Analysis for Per-Instance Evaluation of Stochastic Optimizers: Avoiding Unreliable Conclusions

对随机优化器的实例评估的统计分析:避免不可靠的结论

Moslem Noori, Elisabetta Valiante, Thomas Van Vaerenbergh, Masoud Mohseni, Ignacio Rozada

机构 * QB Information Technologies (1QBit)(1QB信息科技(1QBit)) Hewlett Packard Labs, Hewlett Packard Enterprise(惠普实验室,惠普企业)

AI总结 本文研究了随机优化器性能评估中重复次数对统计精度的影响,提出统计分析方法和实验设计指南,以提高评估的可靠性和准确性。

Journal ref Physical Review Applied 25, no. 3 (2026): 034081

详情
AI中文摘要

随机优化器的一个关键特性是,解决相同问题时多次运行可能产生不同结果。因此,其性能需通过多次重复运行来评估。然而,估计性能指标的准确性依赖于运行次数,并应使用统计工具进行研究。本文对常用指标进行了统计分析,开发了实验设计指南,以在高置信度和准确性下测量优化器性能。首先讨论了指标的置信区间及其与实验重复次数的关系。然后推导了保证达到给定精度的重复次数下限。利用此下限,我们提出了一种自适应调整所需重复次数的算法。仿真结果展示了分析的实用性,表明该方法能够进行可靠的基准测试、超参数调优,并防止对随机优化器性能得出过早结论。

英文摘要

A key trait of stochastic optimizers is that multiple runs of the same optimizer in attempting to solve the same problem can produce different results. As a result, their performance is evaluated over several repeats, or runs, on the problem. However, the accuracy of the estimated performance metrics depends on the number of runs and should be studied using statistical tools. We present a statistical analysis of the common metrics, and develop guidelines for experiment design to measure the optimizer's performance using these metrics to a high level of confidence and accuracy. To this end, we first discuss the confidence interval of the metrics and how they are related to the number of runs of an experiment. We then derive a lower bound on the number of repeats in order to guarantee achieving a given accuracy in the metrics. Using this bound, we propose an algorithm to adaptively adjust the number of repeats needed to ensure the accuracy of the evaluated metric. Our simulation results demonstrate the utility of our analysis and how it allows us to conduct reliable benchmarking as well as hyperparameter tuning and prevent us from drawing premature conclusions regarding the performance of stochastic optimizers.

URL PDF HTML 收藏
2605.13496 2026-05-14 cs.DC cs.LG

MARLIN: Multi-Agent Game-Theoretic Reinforcement Learning for Sustainable LLM Inference in Cloud Datacenters

MARLIN:多智能体博弈强化学习用于云数据中心可持续LLM推理

H. Moore, S. Qi, D. Milojicic, C. Bash, S. Pasricha

机构 * Colorado State University(科罗拉多州立大学) Hewlett Packard Labs(惠普实验室)

AI总结 本文提出MARLIN框架,通过博弈强化学习优化LLM推理的TTFT、碳排放、用水量和能耗,实现至少18%的TTFT、33%的碳排放、43%的用水量和11%的能耗降低。

详情
AI中文摘要

大型语言模型(LLMs)在云平台中日益普及,推动了AI驱动的消费和企业服务。LLM推理请求尤其占总LLM生命周期能耗的90%,远高于训练能耗。随着LLM推理请求量的增加,环境足迹,特别是碳排放和用水量也显著增加。为提高云数据中心LLM推理服务的可持续性,我们提出了一种新颖的多智能体博弈强化学习框架MARLIN,用于协同优化TTFT、碳排放、用水量和与LLM推理相关的能耗。MARLIN相比最先进的LLM推理管理框架,在TTFT、碳排放、用水量和能耗方面分别减少了至少18%、33%、43%和11%。

英文摘要

Large Language Models (LLMs) have become increasingly prevalent in cloud-based platforms, propelled by the introduction of AI-based consumer and enterprise services. LLM inference requests in particular account for up to 90% of total LLM lifecycle energy use, dwarfing training energy costs. The rising volume of LLM inference requests is increasing environmental footprints, particularly carbon emissions and water consumption. To improve sustainability for LLM inference serving in cloud datacenter environments, we propose a novel multi-agent game-theoretic reinforcement learning framework called MARLIN to co-optimize time-to-first token (TTFT), carbon emissions, water usage, and energy costs associated with LLM inference. MARLIN demonstrates a reduction of at least 18% in TTFT, 33% in carbon emissions, 43% in water usage, and 11% in energy costs compared to state-of-the-art LLM inference management frameworks.

URL PDF HTML 收藏
2605.11847 2026-05-13 cs.ET cs.LG

A Fast and Energy-Efficient Latch-Based Memristive Analog Content-Addressable Memory

一种快速且节能的基于锁存的忆阻型模拟内容寻址存储器

Paul-Philipp Manea, Aishwarya Natarajan, Jim Ignowski, John Paul Strachan, Luca Buonanno

机构 * PGI-14, Forschungszentrum Jülich(Jülich 研究中心 PGI-14) RWTH Aachen University(亚琛工业大学) Hewlett Packard Enterprise(惠普企业)

AI总结 本文提出一种基于锁存的忆阻型模拟内容寻址存储器(SALM),通过动态电流竞赛比较器替代静态电压分压,实现高再生增益和近零静态搜索功率,提升能效与可扩展性。

Comments This work has been submitted to the IEEE for possible publication

详情
AI中文摘要

基于忆阻器的模拟内容寻址存储器(aCAMs)为边缘AI和嵌入式智能应用提供了高效能大规模关联计算的可行路径。它们已被成功应用于决策树推理,并扩展了计算内存(CIM)架构的能力。然而,传统设计如6T2M架构存在静态搜索功率、有限电压增益和显著的匹配线串扰,限制了模拟精度和可扩展性。我们引入了一种强臂锁存忆阻器(SALM) aCAM单元,通过动态电流竞赛比较器替代静态电压分压,实现高再生增益、固有结果锁存和近零静态搜索功率。与6T2M相比,SALM在相同延迟下将读取能量减少33%,同时消除限制6T2M扩展到大规模阵列的增益和串扰限制。SALM进一步实现了可扩展的顺序和并行锁存共享,并通过数据集感知优化框架暴露了显式的能量-延迟权衡,实现了在代表性工作负载上高达50%的能量减少,延迟增加3倍。为支持架构探索,我们开发了一个基于SPICE查找表的电路准确行为模型,采用22nm FD-SOI技术,捕捉匹配线动态和串扰。将此框架整合到X-TIME决策树编译器中,证明SALM在高维数据集上保持接近软件精度,而基线设计由于增益有限和累积串扰而退化。

英文摘要

Analog content-addressable memories (aCAMs) based on memristors provide a promising pathway toward energy-efficient large-scale associative computing for Edge AI and embedded intelligence applications. They have been successfully applied to decision-tree inference and extend the capabilities of compute-in-memory (CIM) architectures beyond conventional vector-matrix multiplication. However, conventional designs such as the 6T2M architecture suffer from static search power, limited voltage gain, and pronounced match-line crosstalk, constraining analog precision and scalability. We introduce a strong-arm latched memristor (SALM) aCAM cell that replaces static voltage division with a dynamic current-race comparator, enabling high regenerative gain, intrinsic result latching, and near-zero static search power. Compared to 6T2M, SALM reduces read energy by 33% at identical latency while eliminating the gain and crosstalk limitations that prevent 6T2M from scaling to large arrays. SALM further enables scalable sequential and parallel latch sharing, and a dataset-aware optimization framework exposes an explicit energy-latency tradeoff, achieving up to 50% energy reduction at 3x latency across representative workloads. To enable architectural exploration, we develop a circuit-accurate behavioral model derived from SPICE lookup tables in 22 nm FD-SOI technology, capturing match-line dynamics and crosstalk. Integrated into the X-TIME decision-tree compiler, this framework demonstrates that SALM maintains near-software accuracy for high-dimensional datasets, whereas baseline designs degrade due to limited gain and cumulative crosstalk.

URL PDF HTML 收藏
2605.00068 2026-05-04 cs.LG cs.AI physics.plasm-ph

Human-in-the-Loop Meta Bayesian Optimization for Fusion Energy and Scientific Applications

人机协同元贝叶斯优化用于聚变能与科学应用

Ricardo Luna Gutierrez, Sahand Ghorbanpour, Ejaz Rahman, Varchas Gopalaswamy, Riccardo Betti, Vineet Gundecha, Aarne Lees, Soumyendu Sarkar

机构 * Hewlett Packard Enterprise(惠普企业公司) University of Rochester(罗切斯特大学)

AI总结 本文提出HL-MBO框架,结合专家知识与少样本不确定性感知机器学习,加速数据稀缺的高风险科学领域发现,优于现有BO方法在聚变能优化和超导材料优化中的表现。

Comments Accepted at IJCAI 2026 (35th International Joint Conference on Artificial Intelligence)

详情
AI中文摘要

人机协同元贝叶斯优化用于聚变能与科学应用通过整合专家知识与少样本不确定性感知机器学习,加速数据稀缺的高风险科学领域发现,在聚变能优化和超导材料优化中优于现有BO方法。

英文摘要

Inertial Confinement Fusion (ICF) holds transformative promise for sustainable, near-limitless clean energy, yet remains constrained by prohibitively high costs and limited experimental opportunities. This paper presents Human-in-the-Loop Meta Bayesian Optimization (HL-MBO), a framework that integrates expert knowledge with few-shot, uncertainty-aware machine learning to accelerate discovery in data-scarce, high-stakes scientific domains. HL-MBO introduces a meta-learned surrogate model with an expert-informed acquisition function to recommend candidate experiments. To foster trust and enable informed decisions, HL-MBO also provides interpretable explanations of its suggestions. We show HL-MBO outperforms current BO methods on ICF energy yield optimization, as well as benchmarks in molecular optimization and critical temperature maximization for superconducting materials.

URL PDF HTML 收藏
2604.22990 2026-04-29 cs.CV cs.AI

Hard to See, Hard to Label: Generative and Symbolic Acquisition for Subtle Visual Phenomena

难以察觉,难以标注:生成与符号获取用于微妙的视觉现象

Renjith Prasad, Rishabh Sharma, Andrew E. Shao, Annmary Justine Koomthanam, Shreyas Kulkarni, Suparna Bhattacharya, Martin Foltin, Amit Sheth, David Orozco, Matthew Quinn, Brian Sammuli

机构 * University of South Carolina(南卡罗来纳大学) AI Research Lab, HPE Labs, Hewlett Packard Enterprise(HPE实验室人工智能研究部) Indian AI Research Organization(印度人工智能研究组织) General Atomics(通用原子公司)

AI总结 本文提出GSAL框架,结合生成模型与语义覆盖先验,解决细微视觉异常标注难题,提升稀有类别检索效率。

Comments Accepted at CVPR 2026 SVC Workshop

详情
AI中文摘要

细微视觉异常如微裂纹、亚毫米空洞和低对比包含物在结构上不典型但视觉模糊,使标注困难且易被忽略。标准获取启发式方法基于判别不确定性或特征多样性常过度选择主导模式而忽视稀疏重要区域。在工业缺陷检测中尤为严重,因为异常可能低发且难以区分周围结构。为此,本文提出GSAL,结合扩散模型生成难度信号与层次语义覆盖先验,通过平衡视觉难度与语义覆盖提升稀有目标检索。实验显示在薄膜缺陷、Pascal VOC和MS COCO数据集上,GSAL在标注效率和稀有类别检索方面优于基于不确定性、多样性及混合基线。

英文摘要

Subtle visual anomalies such as hairline cracks, sub-millimeter voids, and low-contrast inclusions are structurally atypical yet visually ambiguous, making them both difficult to annotate and easy to overlook during active learning. Standard acquisition heuristics based on discriminative uncertainty or feature diversity often overselect dominant patterns while underexploring sparse yet important regions of the data space. This failure mode is especially severe in industrial defect inspection, where anomalies may be both low-prevalence and difficult to distinguish from surrounding structure. To resolve this, we propose GSAL, an active learning framework for object detection that combines a diffusion-based difficulty signal with a hierarchical semantic coverage prior. The diffusion component scores images and proposals using reconstruction discrepancy and denoising variability, prioritizing visually atypical or ambiguous examples. However, diffusion alone does not prevent acquisition from repeatedly favoring hard samples within dominant semantic modes. The semantic component therefore organizes candidate samples in a three-level concept graph and promotes coverage of underrepresented semantic regions while providing interpretable acquisition rationales. By balancing visual difficulty with semantic coverage, GSAL improves retrieval of subtle and rare targets that are often missed by uncertainty-only selection. Experiments on a proprietary thin-film defect, Pascal VOC and MS COCO dataset show consistent gains in label efficiency and rare-class retrieval over uncertainty-, diversity-, and hybrid-based baselines

URL PDF HTML 收藏
2604.12005 2026-04-15 cs.LG cs.AI

BayMOTH: Bayesian optiMizatiOn with meTa-lookahead -- a simple approacH

BayMOTH:基于元前瞻的贝叶斯优化——一种简单方法

Rahman Ejaz, Varchas Gopalaswamy, Ricardo Luna, Aarne Lees, Vineet Gundecha, Christopher Kanan, Soumyendu Sarkar, Riccardo Betti

机构 * Laboratory for Laser Energetics(激光能实验室) Department of Mechanical Engineering(机械工程系) Hewlett Packard Labs(惠普实验室) Department of Computer Science(计算机科学系) Department of Physics(物理系)

AI总结 本文提出一种简单高效的元贝叶斯优化算法,通过在统一框架中利用相关任务信息或回溯优化,提升样本效率并保持低相关性任务下的性能。

详情
AI中文摘要

贝叶斯优化(BO)在序列优化昂贵的黑盒函数方面已显示出在许多现实场景中的实用性和有效性。元贝叶斯优化(meta-BO)旨在通过利用相关任务的信息来提高BO的样本效率。尽管在任务结构转移时meta-BO是样本高效的,但meta训练与测试任务之间的不匹配可能导致在线优化过程中建议次优查询。为此,我们提出了一种简单的meta-BO算法,在确定相关任务信息有用时利用它,否则回溯,所有都在统一的框架内。我们展示了该方法在函数优化任务中与现有方法的竞争力,同时在测试任务与meta训练集结构有限时仍保持强大的性能。

英文摘要

Bayesian optimization (BO) has for sequential optimization of expensive black-box functions demonstrated practicality and effectiveness in many real-world settings. Meta-Bayesian optimization (meta-BO) focuses on improving the sample efficiency of BO by making use of information from related tasks. Although meta-BO is sample-efficient when task structure transfers, poor alignment between meta-training and test tasks can cause suboptimal queries to be suggested during online optimization. To this end, we propose a simple meta-BO algorithm that utilizes related-task information when determined useful, falling back to lookahead otherwise, within a unified framework. We demonstrate competitiveness of our method with existing approaches on function optimization tasks, while retaining strong performance in low task-relatedness regimes where test tasks share limited structure with the meta-training set.

URL PDF HTML 收藏
2510.21884 2026-04-03 cs.CL cs.AI

Support-Contra Asymmetry in LLM Explanations

支持-矛盾不对称性在大语言模型解释中的体现

Avinash Patil

机构 * Hewlett Packard Enterprise(慧与科技公司)

AI总结 研究探讨了大语言模型生成的解释与外部模型提取的预测词汇证据之间的关系,发现正确预测的解释更倾向于引用支持性词汇,而错误预测的解释则更多引用矛盾性词汇,揭示了解释与证据之间的不对称性。

Comments 17 Pages, 12 Figures, 4 tables

详情
AI中文摘要

大型语言模型(LLMs)越来越多地在预测的同时生成自然语言解释,但尚不清楚这些解释是否引用了输入文本中的预测性词汇。本文通过实证研究,探讨了LLM生成的解释在文本分类任务中如何与外部模型提取的可解释特征重要性信号对齐。为了分析这种关系,我们比较了解释内容与来自透明线性分类器的可解释特征重要性信号。这些参考模型使我们能够将预测性词汇线索划分为支持性和矛盾性证据,相对于预测标签。在三个基准数据集(WIKIONTOLOGY、AG NEWS和IMDB)上,我们观察到一种一致的实证模式,我们称之为支持-矛盾不对称性。伴随正确预测的解释倾向于引用更多支持性词汇线索和更少矛盾性线索,而与错误预测相关的解释则引用大量矛盾性证据。这种模式在数据集、参考模型家族(逻辑回归和线性SVM)以及多个特征检索深度上都是一致的。这些结果表明,LLM解释往往反映了预测任务中具有预测性的词汇信号,当预测正确时,而错误预测更常与引用输入中误导性线索的解释相关。我们的发现提供了一个简单的实证视角,展示了解释-证据对齐,并说明了如何利用外部预测性证据来分析LLM生成解释的行为。

英文摘要

Large Language Models (LLMs) increasingly produce natural language explanations alongside their predictions, yet it remains unclear whether these explanations reference predictive cues present in the input text. In this work, we present an empirical study of how LLM-generated explanations align with predictive lexical evidence from an external model in text classification tasks. To analyze this relationship, we compare explanation content against interpretable feature importance signals extracted from transparent linear classifiers. These reference models allow us to partition predictive lexical cues into supporting and contradicting evidence relative to the predicted label. Across three benchmark datasets-WIKIONTOLOGY, AG NEWS, and IMDB-we observe a consistent empirical pattern that we term support-contra asymmetry. Explanations accompanying correct predictions tend to reference more supporting lexical cues and fewer contradicting cues, whereas explanations associated with incorrect predictions reference substantially more contradicting evidence. This pattern appears consistently across datasets, across reference model families (logistic regression and linear SVM), and across multiple feature retrieval depths. These results suggest that LLM explanations often reflect lexical signals that are predictive for the task when predictions are correct, while incorrect predictions are more frequently associated with explanations that reference misleading cues present in the input. Our findings provide a simple empirical perspective on explanation-evidence alignment and illustrate how external sources of predictive evidence can be used to analyze the behavior of LLM-generated explanations.

URL PDF HTML 收藏
2604.01215 2026-04-02 cs.LG cs.AI physics.ao-ph

The Recipe Matters More Than the Kitchen:Mathematical Foundations of the AI Weather Prediction Pipeline

配方比厨房更重要:AI天气预测流水线的数学基础

Piyush Garg, Diana R. Gergel, Andrew E. Shao, Galen J. Yacalis

机构 * Commercial AI Lab, RWE Trading Americas Inc.(RWE贸易美洲公司商业AI实验室) AI Research Lab, Hewlett Packard Enterprise(慧与AI研究实验室)

AI总结 本文研究了AI天气预测中流水线的重要性,通过理论和实证分析,揭示了训练方法、损失函数设计和数据多样性对预测精度的影响,提出了一种综合评估模型。

详情
AI中文摘要

AI天气预测已迅速发展,但尚无统一的数学框架解释影响预测能力的因素。现有理论仅关注特定架构选择,而操作证据表明训练方法、损失函数设计和数据多样性至少与架构选择同样重要。本文理论上构建了一个基于球面逼近理论、动力系统理论、信息论和统计学习理论的框架,涵盖完整学习流水线(架构、损失函数、训练策略、数据分布)。我们建立了学习流水线误差分解,证明在当前规模下,估计误差(依赖损失和数据)主导逼近误差(依赖架构)。我们发展了损失函数谱理论,正式化了MSE诱导的球面谐波坐标中的谱模糊,并推导了超出分布外推界限,证明数据驱动模型系统性低估破纪录极端事件,偏倚随纪录突破率线性增长。实证上,我们通过推理验证了十个架构各异的AI天气模型,使用NVIDIA Earth2Studio和ERA5初始条件,评估了六个指标在30个初始化日期上所有季节的结果。结果确认了MSE训练模型在高波数处普遍的谱能量损失,上升的误差共识比率表明大多数预测误差在不同架构间共享,以及极端事件中的线性负偏倚。一个综合模型评估分数提供了统一的多维评估,而一个指导性框架使在训练前可以数学评估所提出的流水线。

英文摘要

AI weather prediction has advanced rapidly, yet no unified mathematical framework explains what determines forecast skill. Existing theory addresses specific architectural choices rather than the learning pipeline as a whole, while operational evidence from 2023-2026 demonstrates that training methodology, loss function design, and data diversity matter at least as much as architecture selection. This paper makes two interleaved contributions. Theoretically, we construct a framework rooted in approximation theory on the sphere, dynamical systems theory, information theory, and statistical learning theory that treats the complete learning pipeline (architecture, loss function, training strategy, data distribution) rather than architecture alone. We establish a Learning Pipeline Error Decomposition showing that estimation error (loss- and data-dependent) dominates approximation error (architecture-dependent) at current scales. We develop a Loss Function Spectral Theory formalizing MSE-induced spectral blurring in spherical harmonic coordinates, and derive Out-of-Distribution Extrapolation Bounds proving that data-driven models systematically underestimate record-breaking extremes with bias growing linearly in record exceedance. Empirically, we validate these predictions via inference across ten architecturally diverse AI weather models using NVIDIA Earth2Studio with ERA5 initial conditions, evaluating six metrics across 30 initialization dates spanning all seasons. Results confirm universal spectral energy loss at high wavenumbers for MSE-trained models, rising Error Consensus Ratios showing that the majority of forecast error is shared across architectures, and linear negative bias during extreme events. A Holistic Model Assessment Score provides unified multi-dimensional evaluation, and a prescriptive framework enables mathematical evaluation of proposed pipelines before training.

URL PDF HTML 收藏
2601.19132 2026-01-28 cs.NI cs.AI cs.AR cs.PF cs.SY eess.SY

In-Network Collective Operations: Game Changer or Challenge for AI Workloads?

网络内集体操作:人工智能工作负载中的变革者还是挑战?

Torsten Hoefler, Mikhail Khalilov, Josiah Clark, Surendra Anubolu, Mohan Kalkunte, Karen Schramm, Eric Spada, Duncan Roweth, Keith Underwood, Adrian Caulfield, Abdul Kabbani, Amirreza Rastegari

机构 * ETH Zürich(苏黎世联邦理工学院) Microsoft(微软) AMD(超威半导体) Broadcom Inc.(博通公司) Hewlett Packard Enterprise(惠普企业)

AI总结 本文探讨了网络内集体操作在人工智能工作负载中的潜力与挑战,分析了边缘和核心INC的性能优势及阻碍其应用的关键障碍,并预测其未来发展方向。

Journal ref IEEE Computer Jan. 2026

详情
AI中文摘要

本文总结了网络内集体操作(INC)在加速人工智能工作负载中集体操作方面的机遇。我们提供了足够的细节,使该重要领域对人工智能或网络领域的非专家变得易于理解,促进这两个社区之间的联系。考虑两种类型的INC:边缘INC,其中系统在节点层面实现,以及核心INC,其中系统嵌入在网络交换机中。我们概述了潜在的性能优势以及在边缘INC和核心INC背景下可能阻碍其采用的六个关键障碍。最后,我们提出了一组关于INC未来发展和应用的预测。

英文摘要

This paper summarizes the opportunities of in-network collective operations (INC) for accelerated collective operations in AI workloads. We provide sufficient detail to make this important field accessible to non-experts in AI or networking, fostering a connection between these communities. Consider two types of INC: Edge-INC, where the system is implemented at the node level, and Core-INC, where the system is embedded within network switches. We outline the potential performance benefits as well as six key obstacles in the context of both Edge-INC and Core-INC that may hinder their adoption. Finally, we present a set of predictions for the future development and application of INC.

URL PDF HTML 收藏
2512.14137 2025-12-17 cs.CV

Erasing CLIP Memories: Non-Destructive, Data-Free Zero-Shot class Unlearning in CLIP Models

Ashish Mishra, Tarun Kumar, Gyanaranjan Nayak, Arpit Shah, Suparna Bhattacharya, Martin Foltin

机构 * Hewlett Packard Labs(惠普实验室)

详情
英文摘要

We introduce a novel, closed-form approach for selective unlearning in multimodal models, specifically targeting pretrained models such as CLIP. Our method leverages nullspace projection to erase the target class information embedded in the final projection layer, without requiring any retraining or the use of images from the forget set. By computing an orthonormal basis for the subspace spanned by target text embeddings and projecting these directions, we dramatically reduce the alignment between image features and undesired classes. Unlike traditional unlearning techniques that rely on iterative fine-tuning and extensive data curation, our approach is both computationally efficient and surgically precise. This leads to a pronounced drop in zero-shot performance for the target classes while preserving the overall multimodal knowledge of the model. Our experiments demonstrate that even a partial projection can balance between complete unlearning and retaining useful information, addressing key challenges in model decontamination and privacy preservation.

URL PDF HTML 收藏
2512.14113 2025-12-17 cs.CV

Selective, Controlled and Domain-Agnostic Unlearning in Pretrained CLIP: A Training- and Data-Free Approach

Ashish Mishra, Gyanaranjan Nayak, Tarun Kumar, Arpit Shah, Suparna Bhattacharya, Martin Foltin

机构 * Hewlett Packard Labs(惠普实验室)

详情
英文摘要

Pretrained models like CLIP have demonstrated impressive zero-shot classification capabilities across diverse visual domains, spanning natural images, artistic renderings, and abstract representations. However, real-world applications often demand the removal (or "unlearning") of specific object classes without requiring additional data or retraining, or affecting the model's performance on unrelated tasks. In this paper, we propose a novel training- and data-free unlearning framework that enables three distinct forgetting paradigms: (1) global unlearning of selected objects across all domains, (2) domain-specific knowledge removal (e.g., eliminating sketch representations while preserving photo recognition), and (3) complete unlearning in selective domains. By leveraging a multimodal nullspace through synergistic integration of text prompts and synthesized visual prototypes derived from CLIP's joint embedding space, our method efficiently removes undesired class information while preserving the remaining knowledge. This approach overcomes the limitations of existing retraining-based methods and offers a flexible and computationally efficient solution for controlled model forgetting.

URL PDF HTML 收藏
2508.10520 2025-08-15 cs.LG cond-mat.dis-nn

Nonlocal Monte Carlo via Reinforcement Learning

Dmitrii Dobrynin, Masoud Mohseni, John Paul Strachan

机构 * Peter Grünberg Institut (PGI-14), Forschungszentrum Jülich GmbH, Jülich, Germany(julich研究所(PGI-14)) RWTH Aachen University, Aachen, Germany(亚琛工业大学) Emergent Machine Intelligence, Hewlett Packard Labs, CA, USA(新兴机器智能,惠普实验室)

详情
英文摘要

Optimizing or sampling complex cost functions of combinatorial optimization problems is a longstanding challenge across disciplines and applications. When employing family of conventional algorithms based on Markov Chain Monte Carlo (MCMC) such as simulated annealing or parallel tempering, one assumes homogeneous (equilibrium) temperature profiles across input. This instance independent approach was shown to be ineffective for the hardest benchmarks near a computational phase transition when the so-called overlap-gap-property holds. In these regimes conventional MCMC struggles to unfreeze rigid variables, escape suboptimal basins of attraction, and sample high-quality and diverse solutions. In order to mitigate these challenges, Nonequilibrium Nonlocal Monte Carlo (NMC) algorithms were proposed that leverage inhomogeneous temperature profiles thereby accelerating exploration of the configuration space without compromising its exploitation. Here, we employ deep reinforcement learning (RL) to train the nonlocal transition policies of NMC which were previously designed phenomenologically. We demonstrate that the resulting solver can be trained solely by observing energy changes of the configuration space exploration as RL rewards and the local minimum energy landscape geometry as RL states. We further show that the trained policies improve upon the standard MCMC-based and nonlocal simulated annealing on hard uniform random and scale-free random 4-SAT benchmarks in terms of residual energy, time-to-solution, and diversity of solutions metrics.

URL PDF HTML 收藏