全部动态

本页 40 条
10月12日 周一
arXiv cs.LG77

超越分布内性能保持:通过面向脆弱性的调优恢复量化 VLA 的泛化能力

arXiv:2610.05745v2 公告类型:替换 摘要:已有研究表明,训练后量化能够在标准评估条件下保持 VLA 的性能,但其能否保留全精度模型的鲁棒性和泛化能力,仍缺乏充分探索。本研究系统考察了训练后量化的 VLA 策略在环境扰动下的鲁棒性和泛化能力。实证结果表明,量化策略尽管保持了相近的分布内性能,却可能对细微的环境变化变得脆弱。我们进一步观察到,动作差异集中在策略执行轨迹中的一小部分状态上,而教师指导的效果随差异程度而呈现相反趋势:它能提升高差异状态下的泛化能力,却可能削弱低差异状态下的泛化能力。这些发现揭示,有效的量化后恢复需要对脆弱状态进行选择性干预,而不是对学生模型进行全局蒸馏。因此,我们提出 Policy-Induced Vulnerability-Oriented Tuning(PIVOT-Q),这是一个具有脆弱性感知能力的同策略蒸馏(OPD)框架,以冻结的全精度策略作为教师,选择性地纠正量化学生模型在执行轨迹中遇到的脆弱状态。PIVOT-Q 利用短时域内的折扣累积差异识别脆弱状态,实施阶段均衡的稀疏监督,并使用 Behavioral Anchor 防止不必要的改变。在 LIBERO-Plus 的七种环境变化下开展的实验表明,该方法在多种 VLA 骨干模型和量化方法上均能实现稳定的性能恢复。值得注意的是,PIVOT-Q 在所有设置下都持续优于全状态蒸馏,而所用的状态级蒸馏预算仅为后者的 7.4%。我们的代码已发布于 https://github.com/ruanruan-andy/PIVOT-Q。

arXiv cs.LG73

Green-Routed Neural Operators:物理决定网络在何处读取

arXiv:2610.05337v2 公告类型:替换 摘要:我们发现,输运场在许多偏微分方程(PDE)中的物理作用与其在神经算子中的通常作用之间存在不匹配:PDE 利用输运场选择读取坐标,而神经算子通常仅将其视为输入值。为解决这一不匹配,我们提出 Green-Routed Neural Operator(GRNO),利用控制方程确定潜在特征的采样位置。一个无参数的方程适配器计算诊断关系,并构建以读取坐标为取值的出发点映射。多尺度编码器—解码器结合潜在特征的中心读取与路由读取,学习完整的有限时间更新。在五个二维和三维 PDE 系统的 40 步自回归评估中,GRNO 在其中四个系统上取得了最低的最终相对 $L^2$ 误差均值,并在 Keller-Segel 系统上保持了竞争力。固定权重下的路由干预表明,四个系统对方向和空间对齐具有较强依赖,而 Keller-Segel 系统的依赖较弱。在独立训练的消融实验中,相较于仅将输运场作为输入特征、用学习得到的位移替代方程指定的路由,或以空间错位方式应用路由的各个变体,GRNO 在全部五个系统上均取得了更低的平均误差。它还显著优于直接对物理状态进行平流输运并学习剩余更新的方式,表明方程指定的读取坐标为长时域 PDE 预测提供了有效的结构先验。

arXiv cs.LG70

Graph World Models for Constrained Epidemic Policy Planning

arXiv:2609.35545v2 Announce Type: replace Abstract: Epidemic policy planning often requires coordination between geographical regions, taking into account mobility-driven spillovers and how to make use of limited resources. Existing methods either lack action-conditioned models of coupled dynamics or cannot guarantee per-period feasibility. We present EpiMind, a graph world model framework for constrained epidemic policy planning across regions. A graph-factored recurrent state-space model generates joint policy-conditioned rollouts from regional latent beliefs, while graph-temporal ADMM optimizes regional interventions, enforces shared-resource feasibility through projection, and evaluates temporal specifications under the learned model. EpiMind reduces admission RMSE by 29% relative to graph-free dynamics modeling, plans within 1-5% of the best feasible constant policy with guaranteed shared-budget feasibility, and outperforms all deployable baselines across three resource budgets in real-context evaluation. These results demonstrate that graph-structured policy imagination with explicit constrained coordination supports effective epidemic interventions from learned dynamics.

arXiv cs.LG70

LLM4Trust: Exploring the Capabilities of Large Language Models for Trust Evaluation

arXiv:2609.33521v2 Announce Type: replace Abstract: Trust evaluation plays a critical role in cybersecurity by supporting risk mitigation and decision-making. A variety of trust evaluation methods have been proposed, with learning-based approaches offering high accuracy and automation. However, they often require substantial ground truth, suffer from low training efficiency, lack support for basic trust properties, and provide limited explainability. Large Language Models (LLMs) offer a compelling alternative due to their strong zero-/few-shot reasoning abilities and broad knowledge. To this end, we propose LLM4Trust, the first benchmark framework that systematically explores the capabilities of LLMs for trust evaluation. We first construct diverse trust graphs to model five basic trust properties and design corresponding property understanding tasks. We then assess the ability of eight representative LLMs to understand these properties under nine prompt methods. Based on this exploration, we identify the most effective LLM-prompt combinations and apply them to five real-world datasets for validating LLMs' trust evaluation capability. During this process, we propose two strategies to extract key information from large-scale trust graphs, addressing the context window limitations of LLMs. Extensive experiments show that LLMs can effectively understand basic trust properties and have great potential for real-world trust evaluation, particularly under limited supervision. However, they remain vulnerable to attacks targeting trust graphs and demonstration examples used in few-shot prompting, and incur high inference costs. Accordingly, we propose a defense mechanism and batch inference to improve the robustness and efficiency of LLM-based trust evaluation. The source code of LLM4Trust is available at https://github.com/Jieerbobo/LLM4Trust

arXiv cs.LG74

表格基础模型注意力机制的基准测试

arXiv:2609.31306v3 公告类型:替换 摘要:TabPFN、Mitra 或 ConTextTab 等表格上下文学习器依赖于在潜在嵌入的二维序列上交替进行行注意力和列注意力计算。这些注意力模式与语言模型中的一维情形显著不同:行注意力涉及较长的序列,而列注意力处理的序列短得多,且表格数据的跨步内存布局使得生成连续张量的成本很高。此外,当前模型使用的隐藏维度相较于近期的语言模型较小。然而,高效注意力的研究主要集中于一维序列,二维表格场景尚未得到探索。为此,我们构建了一套可复现的基准测试环境,并在多个后端上研究表格注意力的独有特征——包括 Torch SDPA(efficient 和 cuDNN)、FlashAttention-2/3/4,以及仅用于推理的后端 vLLM 和 SageAttention——在三代 GPU(A100、H100、B200)上,针对符合实际的表格张量形状测量前向和反向计算的吞吐量。我们发现,列注意力与行注意力的最优后端选择不同,并且会随硬件和模型具体配置而变化:尽管针对各代 GPU 定制的 FlashAttention 实现总体表现最佳,但在序列较长的列注意力场景中,CuDNN 有时表现更优,性能交叉点取决于注意力头维度。在仅用于推理的后端中,SageAttention 在行注意力以及超过 16,000 行的大规模序列上表现良好。我们可复现的基准测试为未来改进表格原生注意力机制奠定了基础。自包含的基准测试与评估代码已公开发布于:https://github.com/SAP-samples/tabular-attention-benchmark

arXiv cs.LG71

RLVR$^{2}$:基于可验证评分准则排名的强化学习

arXiv:2609.23457v2 公告类型:替换 摘要:基于可验证奖励的强化学习(RLVR)正从数学和代码等具有明确定义的正确性信号的任务,拓展至由多维评分准则规定的多方面质量要求。由于策略优化对每次 rollout 仅使用一个标量,基于评分准则的流程必须将多个标准的分数映射为一个标量奖励。这种聚合通常被视为分数缩放,但它实际上隐式决定了训练过程中各质量维度之间的权衡方式。目前普遍采用的做法是对每项标准进行归一化,再取线性组合,这假定不同标准的基数分数差异具有可比性,且某项标准上的提升可以弥补另一项标准上的不足;当各标准在语义上存在异质性时,这两个假设都不可靠。我们提出基于可验证评分准则排名的强化学习(RLVR$^2$),这是一种面向基于评分准则的 RLVR 的可验证排名范式。对于每项标准,RLVR$^2$ 将评分准则分数转换为该标准特定的组内序数结果,从所得的比较矩阵中恢复潜在效用,并将这些效用合并为一个训练信号。通过仅保留组内排序并舍弃原始分数的大小,RLVR$^2$ 避免了对异质评分准则量表进行校准。它还支持保持目标不变的属性调整:与观测排名相关、但并非训练目标的辅助属性可以纳入估计,而无须扩展评分准则或直接对这些属性给予奖励。在三种模型规模和 16 项基准测试中,RLVR$^2$ 均持续优于具有代表性的基于评分准则的基线方法,并在每种规模下的大多数基准测试中取得最佳综合表现。分析表明,它在保持质量目标不变的同时,能够控制与推理效率和回复格式相关的系统性影响。

arXiv cs.LG74

SUN:在强化学习中追寻新颖性

arXiv:2609.08642v2 公告类型:替换 摘要:强化学习(RL)中的探索仍是一项根本性挑战。近期的目标条件化 RL 策略(通过选择目标来促进更广泛的状态覆盖)已展现出良好效果,但尚无策略能同时依据新颖性和可达性对目标进行评分:这两种信号要么被人为权衡,要么被依次应用,要么其中一种被完全忽略。本文提出一种可达性感知的目标选择框架,显式整合这两个方面,并可无缝融入任何离策略 RL 算法。为此,我们提出 SUccessor-to-Novelty(SUN),这是一种从后继价值函数推导出的指标,用于识别既新颖又可达的目标。我们证明,SUN 是一种按计数缩放的目标达成奖励所对应的价值,可为短时域内的命中概率提供界限,并能排除不可达目标。我们进一步提出一种利用这些性质的自适应目标选择策略,以及一种准确且轻量级的伪计数方法,以避免经典方法的开销。我们通过全面的基准测试验证了所有论断:在包含不可达或难以到达的状态、不可逆转移、障碍物、迷宫及无界空间的标准环境和新环境中,SUN 均持续优于当前最先进的方法。代码见 https://github.com/sparisi/rl-anvil

arXiv cs.LG74

评估方案对跨 Trust-Hub 家族的基于特征的硬件木马检测的影响

arXiv:2609.07199v2 公告类型:替换 摘要:Trust-Hub 会重复使用宿主电路:若干文件的主要差异在于植入的木马。当同源变体中的逻辑门同时进入训练折和测试折时,检测器可能会受益于其已经见过的宿主逻辑。我们测量了这一效应,而非提出另一种分类器。该语料库包含来自 16 个网表的 49,124 个逻辑门,分为五个宿主家族。我们保持解析器、36 项逻辑门特征、类别权重、模型设置、阈值及家族级聚合方式不变,仅改变一个选项:测试集的划分边界。三种设置分别从混合语料库中抽取测试逻辑门、留出一个完整网表,以及留出某个宿主的所有变体。这一选择会产生显著影响。随机森林在混合逻辑门设置下的 F1/AP 为 0.914/0.978,在留出一个网表时为 0.636/0.851,在留出一个宿主家族时为 0.460/0.577。在相同的比较中,XGBoost 从 0.946/0.976 降至 0.464/0.544。逻辑回归的 AP 也有所下降,但其固定阈值下的 F1 并非单调变化。每个家族从混合划分到家族留出划分都呈现相同的变化方向。特征移除、使用多个模型与模拟器随机种子重复实验、分数归一化、与解析器相关的样本排除,以及缩小样本规模,都会改变差距的大小,但不会逆转这一趋势。聚合方式同样重要:按逻辑门数量加权的平均值由较大的 ISCAS 文件主导,因此主要报告值对每个宿主家族赋予相同权重。自助法和刀切法的汇总结果显示差距仍为正,但其各折重复使用了训练家族。我们将五个家族对应的结果行视为描述性证据,而非独立试验。五个宿主家族不足以支持对总体作出断言,而且该实验无法说明向新标准单元库或工业设计迁移的情况。它支持一个更有限的结论:同源基准变体可能夸大表观迁移能力。对于包含同一宿主电路多个变体的基准,应在报告混合划分得分的同时,报告考虑家族结构的留出评估结果以及全部五个家族的结果。

arXiv cs.LG77

DASC:面向混合线性注意力推理服务的衰减感知状态压缩

arXiv:2608.30386v2 公告类型:替换 摘要:混合线性注意力架构近期已扩展至大型开放权重模型,在显著降低键/值(KV)缓存增长的同时,提供了可与全注意力媲美的质量。然而,其原地递归状态更新使缓存管理变得复杂:前缀复用需要在全注意力 KV 缓存之外保存状态检查点,而完整存储状态检查点会增加内存压力,导致更多缓存驱逐和重复预填充。通过分析 Gated DeltaNet(GDN)和 Kimi Delta Attention(KDA)的衰减结构,我们发现,不同注意力头和通道保留前缀信息的时间尺度存在显著差异,我们将其称为\emph{保留时域}。这种差异表明,持久保存的状态检查点具有很大的压缩潜力。基于这一观察,我们提出了\emph{衰减感知状态压缩}(DASC):从模型权重推导保留时域,选择长时域状态单元,并将其打包为不规则的状态检查点布局。为与张量并行推理引擎高效集成,DASC 进一步在各个 TP 秩之间均衡分配压缩后的状态检查点。复用时,DASC 可以将被省略的单元填零,也可以付出额外计算开销,利用长度受限的后缀刷新这些单元。在 Kimi-Linear 上的检索和端到端推理基准测试中,保守的 DASC 配置在将 KDA 递归状态检查点压缩 $2.63\times$ 的同时,保持了接近完整缓存的表现。在固定的状态检查点内存预算下,由此获得的容量提升使平均首个词元生成时间(TTFT)降低了 42.6\%,输入吞吐量提高了 68.4\%。在更高压缩比下,后缀刷新能够恢复因更激进省略而损失的大部分准确率,但代价是额外的重放计算。采用 GDN 的 Qwen 也呈现出类似的质量—效率趋势,表明 DASC 可以从按通道处理的 KDA 扩展到按注意力头处理的 GDN。

arXiv cs.LG70

The Fourth Quadrant: A Stylized View of Benign Misfitting

arXiv:2608.01032v2 Announce Type: replace Abstract: Training error is what we can observe on a training set; test error is the quantity we care about. We study squared-error linear regression in a deterministic $(d+1)$-dimensional single-spike model. Each training vector has spike coordinate $\sqrt{\gamma}$, $\gamma>1$, and equal-norm, mutually orthogonal nuisance components. Training labels are $1$. Fresh test points are centered Gaussian with covariance $\operatorname{diag}(\gamma,1,\ldots,1)$ and noise-free label $x[1]/\sqrt{\gamma}$. We compute the exact train--test tradeoff for linear predictors in the training span. When the spike direction is unknown, allowing all linear predictors does not improve this tradeoff, on average or in the worst case over directions consistent with the data. There is a range of training-set sizes where every span predictor that generalizes well must fit the training data worse than the zero predictor. We call this benign misfitting, or the fourth quadrant. At its sharpest, test error tends to zero while training error diverges, with both zero-predictor errors normalized to one. The best span predictor begins to generalize when $n\gg d/\gamma^2$, while interpolation needs $n\gg d/\gamma$. In the intervening window $d/\gamma^2\ll n\ll d/\gamma$, useful span prediction lies beyond interpolation: training predictions overshoot the labels. One-pass stochastic gradient descent with a large constant learning rate attains small test error throughout this window, matching the best span predictor up to a logarithmic factor, while leaving large empirical training error. Finally, the nuisance component responsible for the training misfit also controls adversarial sensitivity.

arXiv cs.LG77

一阶上下文 Shapley:面向 LLM 对齐与评估的可扩展数据审计

arXiv:2607.22766v2 公告类型:替换 摘要:大语言模型的后训练对齐日益受到数据质量的制约:偏好与指令微调语料中不断积累标签噪声、安全性偏好倒置和标注者间的矛盾,这些问题无法通过去重发现,而使用 LLM 对整个语料库进行评判的成本又过高。尽管 Data Shapley 提供了具有理论依据的价值评估框架,但精确计算需要进行难以承受的重新训练,而 KNN-Shapley 等基于嵌入的替代方法无法区分高效用与低效用示例(与生成效用的相关性为 $\rho = 0.0003$)。我们提出了一种仅需推理的价值评估流程,在语义 $k$-NN 邻域上构建上下文合作博弈,无需更新参数,仅通过 $O(k)$ 次前向传播即可计算其一阶项;该方法在全部目标上均与多示例蒙特卡洛上下文 Shapley 呈正相关,而计算量仅为后者的 1/19。通过将成对的对数似然变化投影到有向影响图上,我们用三个信号刻画每条记录:出向示范优势、入向受示范优势,以及用于保护稀有提示的拓扑独特性得分。预筛选将 LLM 双模型仲裁的范围限制在 1%–5% 的记录内,同时在 HelpSteer2 和 Anthropic HH-RLHF 上分别实现了 4.5 倍和 2.2 倍的已确认错误富集率,并捕获了零样本似然和间隔筛选遗漏的矛盾。在 DPO 训练期间,已确认存在矛盾的记录比干净记录更频繁地与更新方向相悖(在 LoRA 梯度中占比分别为 56.7% 和 37.0%),且其奖励准确率始终低于随机水平。最后,对 HH-RLHF 基准的审计揭示了标签错误:对于这些错误,对齐模型与人工标签不一致的比例达到 73%–83%;在修复或过滤被标记的记录后重新进行 DPO 训练,即使不使用 LLM 仲裁,也能使整体奖励准确率提高 0.5–0.6 个百分点,并使已修复的偏好倒置样本上的奖励准确率提高 3.1–3.5 个百分点。

arXiv cs.LG70

Learning from All-Failed Groups via First-Visit Coverage in Long-Horizon Agentic Tasks

arXiv:2607.22724v2 Announce Type: replace Abstract: Group-based policy optimization has been increasingly used to train large language model (LLM) agents from sparse outcome rewards by comparing trajectories or steps within a group. However, on difficult long-horizon agentic tasks, we found this comparison suffers from a sampling imbalance: repeated or low-effect actions dominate the high-probability region of the policy, leaving useful state-changing actions under-sampled. This imbalance produces many all-failed rollout groups, where the relative advantages within such a group provide no direction for updating the policy. Consequently, the sampling imbalance and the all-failed groups collapse into a self-perpetuating credit trap: failure-dominated sampling yields no learning signal, allowing repeated low-effect actions to persist. To break this trap, we propose Coverage-conditioned Group Policy Optimization (CovGPO), which uses first-visit observation coverage only when all samples in a group receive zero outcome reward. Specifically, within such groups, CovGPO assigns higher relative advantages to state-exploring actions without corrupting the original reward signal, as reaching new observations is a prerequisite for task success. Experiments on two challenging agentic benchmarks, ALFWorld and WebShop, show that CovGPO consistently improves over group-based baselines, with particularly large gains on hard tasks.

arXiv cs.LG73

使用 Transformers 进行规划:计算链与结构化上下文窗口

arXiv:2607.17710v2 公告类型:替换 摘要:大语言模型(LLMs)已对机器学习的诸多领域产生了显著影响。然而,近期研究表明,它们难以可靠地解决规划问题,尤其是与经典规划器相比。即使面向推理的模型近期取得了进展,LLMs 究竟能在多大程度上真正执行规划,仍是一个悬而未决的问题。与此同时,理论结果表明,作为现代 LLMs 核心架构的 transformers 具有图灵完备性。在本研究中,我们探究了 LLMs 的理论计算能力与实际规划表现之间这一明显差距。我们提出了 Chain of Computation(COC),这是一种将基于 transformer 的语言模型(LM)置于迭代循环中,并利用其作为模式匹配系统的优势的计算架构。COC 使用 Structured Context Window(SCW)作为外部记忆,使模型在每个规划步骤中仅访问累积上下文中所需的部分,从而将输入规模维持在近似恒定的水平。在这一架构中,LM 学习规划策略和世界模型,并执行规划过程中所需的算术运算。我们表明,从零开始训练的较小规模 LM 能够从少量训练实例中学会规划程序,并泛化到未见过的问题,在 BlocksWorld 和煎饼排序谜题上的成功率超过 99%。我们进一步表明,COC 可以被形式化为下推自动机(PDA)。这一形式化方案能够求解最多包含 20 个圆盘的汉诺塔(TOH)实例,并使规划器能够执行深度优先搜索(DFS)算法。在理论层面,我们刻画了在标准全上下文设置下,无关上下文的累积如何增加规划失败的概率,以及 SCW 如何缓解这一影响。

arXiv cs.LG73

通过对抗蒸馏提升可认证鲁棒性

arXiv:2606.31653v2 公告类型:替换 摘要:可认证训练旨在生成其预测能够被形式化验证为可抵御对抗扰动的模型,通常通过优化允许扰动集合上最坏情况损失的上界来实现。对于神经网络,完全基于紧松弛界的可认证训练方法能够生成易于认证的网络,但会牺牲标准准确率。相反,对抗训练往往能带来更强的经验鲁棒性和更高的标准准确率,但所得模型通常难以用神经网络验证器进行认证。近期文献表明,将对抗训练目标与基于区间界传播(IBP)的宽松过近似相结合,可以在标准准确率与可认证准确率之间取得更好的权衡,其实质是在最坏情况损失的下界与上界之间进行插值。在此基础上,我们提出 AD-CERT,这是一种将对抗蒸馏与 IBP 上界相结合的可认证训练目标。我们表明,在 logit 空间中从具有经验鲁棒性的教师模型蒸馏对抗信息,可以为可认证训练提供有效的下界替代目标;AD-CERT 在多个面向 ReLU 网络的鲁棒性基准上取得了最先进的可认证性能。此外,在统一的实验设置下,研究表明,与鲁棒特征空间蒸馏目标相比,在 logit 层面蒸馏对抗信息可将可认证准确率提高最多 5.40 个百分点。我们的代码已发布于 https://github.com/matteo-melis/ad-cert。

arXiv cs.LG76

FlowAWR:通过优势加权校正实现在线自适应流强化学习

arXiv:2606.30376v2 公告类型:替换 摘要:通过在线强化学习对齐连续空间中的生成式流模型,受到轨迹似然难以计算的限制。现有基于密度近似的策略梯度方法依赖随机 SDE 采样器来构建可计算的转移核,这会引入训练与推理之间的不一致,并且需要无分类器引导(CFG)。尽管 DiffusionNFT 等隐式框架直接优化前向过程的速度场,但其启发式固定幅度校正使优化强度无法随组内相对质量而调整。我们提出 \textit{Flow Advantage-Weighted Rectification}(\textbf{FlowAWR}),这一范式将连续生成策略优化重新表述为朝向理论最优速度场的监督回归。从 KL 约束下奖励最大化的最优策略出发,FlowAWR 推导出具有幅度感知、优势加权校正形式的最优速度场,实现了无需 SDE 的优化和无需 CFG 的生成。在 SD3.5-Medium 上的对比评估中,FlowAWR 在提升对齐性能的同时,相较于 DiffusionNFT 实现了 2$\times$ 至 5$\times$ 的收敛加速(例如,FlowAWR 在 1.2k 步时达到 24.12 的 PickScore,而 DiffusionNFT 在 2.0k 步时达到 23.82,FlowGRPO 在 $>$4k 步时达到 23.50)。在多奖励约束下,FlowAWR 能够维持生成质量,在满足结构规则的同时保持稳定的域外性能。代码见 \href{https://github.com/Fuzm7/FlowAWR}{FlowAWR}。

arXiv cs.LG79

已被抹除,却未消失:输出遗忘并非真正遗忘

arXiv:2606.25001v2 公告类型:替换 摘要:机器遗忘(MU)通常通过输出遗忘来评判,例如遗忘集上的低准确率或成员推断能力的降低。然而,如果输出层面的成功可以与表征空间中不符合重新训练结果的残留同时存在,那么当前评估究竟在验证哪一种遗忘?我们通过与重新训练一致的表征遗忘来研究这一问题,将一组重新训练的模型(即不使用待遗忘数据、从头训练的模型)作为理想遗忘的可操作参考分布。针对多种遗忘方法、数据集和模型,我们的理论分析与实证结果表明,标准的输出层面评估可能系统性地高估机器遗忘的成功程度。在这一更严格的视角下,当前方法往往在输出层看似已完成遗忘,但相对于重新训练仍呈现结构性失配。它们在待遗忘样本上与重新训练部分对齐,在保留样本上则表现出更强的不一致性,并且往往留下沿重新训练相关方向局部分布的残余差异,而非弥散于整个表征空间。这种结构性失配的特征包括遗忘与保留之间的不对称性、方向性失配,以及沿重新训练相关方向集中的残留。在许多情况下,这些差异超出了独立重新训练的变异范围,并带来在重新学习攻击中恢复信息的风险。总体而言,结果表明,当前对 MU 的评估往往针对的是表面遗忘,而非与重新训练一致的遗忘。更广泛地说,重新训练揭示了输出遗忘所掩盖的内容。

arXiv cs.LG75

VLA-FAIL:面向微调视觉-语言-动作模型的高效任务失败检测

arXiv:2606.21386v2 公告类型:替换 摘要:视觉-语言-动作模型(VLA)在许多机器人操作任务上达到了最先进的性能,但在分布外场景中仍可能表现出不可预测的行为。因此,运行时失败检测对于 VLA 在现实世界中的安全部署至关重要。然而,现有的任务失败检测器要么需要计算成本高昂的动作采样,要么基于限制其对 VLA 适用性的架构假设,要么需要获取失败轨迹。我们提出 VLA-FAIL,这是一种轻量且广泛适用的 VLA 失败检测框架,以极低的额外开销结合了两种新颖的失败检测器,且无需失败数据。第一种是末层马氏距离(LLMD),通过逐词元测量末层特征相对于训练数据的偏离程度来检测分布外状态。第二种是动作块一致性(ACC),利用滚动时域控制产生的时间重叠,在连续动作块变得不一致时检测失败。为刻画检测准确性与检测延迟之间的权衡,我们引入了 \textsc{AUCPDT},这是一种不依赖阈值的指标,可联合评估精确率、召回率和检测时间。通过大量现实世界和仿真实验,我们证明 LLMD 和 ACC 能够捕捉互补的失败模式,两者结合可在多种任务中实现可靠且及时的失败检测,并且经常优于计算成本高得多的基线方法。项目页面:https://vla-fail.github.io/vla-fail-2026

arXiv cs.LG

Urban Power Grid Topology and Hierarchy Identification from Open Data

arXiv:2606.21352v2 Announce Type: replace Abstract: Understanding the complex topology and hierarchy of urban power grid is crucial for energy prognosis, power flow management, and system resilience analysis. However, detailed grid information remains largely proprietary. This information, nevertheless, is crucial especially for analyzing the last-mile distribution networks connecting individual buildings. This paper addresses this challenge by developing an open-data-driven framework to identify urban power grid topology, from high-voltage transmission down to individual building connections. We integrate public infrastructure data (power-lines, substations, transformers, poles) to map the high and medium-voltage grid skeleton. We then leverage geospatial machine learning on OpenStreetMap building data to cluster loads of building energy demand. Upon the clustering, we develop the physical topology of the final distribution lines linking the clustered buildings and low-voltage distribution grids. Our derived grid topology incorporates physical and geographical constraints like voltage level and road networks, thus leading to reasonable topology and hierarchy of grid networks. We apply the developed framework to the districts of Alna in Oslo, Norway, and Waiheke Island in New Zealand, and we reconstruct the complete grid topology that connects all the buildings, power lines, and electricity infrastructure assets. With the developed framework, we provide a critical tool that facilitates power system analysis, e.g., power flow optimization, cascading failure simulation, and grid resilience against the penetration of distributed renewable generation. This work also opens the opportunity to build energy digital twins that fuse power assets and grid topology, geographic constraints, and fine-grained power demand profiles.

arXiv cs.LG74

用于解释大语言模型的可扩展回路学习

arXiv:2606.16939v2 公告类型:替换 摘要:机制可解释性领域的一个重要研究方向是学习大语言模型组件之间的稀疏回路,以揭示这些组件如何共同产生模型行为。然而,原始神经元具有多义性,导致学到的回路难以解释。稀疏自编码器(SAE)特征缓解了这一问题,但其高维性使现有基于干预的回路学习方法面临难以承受的计算成本。我们提出 CircuitLasso,这是一种基于稀疏线性回归的可扩展回路学习方法。在基准数据上,CircuitLasso 恢复出的回路在结构准确性上可媲美最先进的基于干预的方法,而计算成本仅为其一小部分。在可解释性方面,CircuitLasso 能高效揭示 SAE 特征之间的关系,展示人类可理解的语义特征如何在模型中传播并影响其预测。最后,我们利用所学回路提供的洞见,在一项领域泛化任务上以显著更低的成本实现了相当的性能,从而验证了这些回路的实用价值。

arXiv cs.LG66

Learning Transfers: Kan Extensions of Topological Invariants of Learned Representations

arXiv:2606.07627v4 Announce Type: replace Abstract: When a learned representation is reused for a new task, which of its structure should survive? A task is a small category, a change of task is a functor $J$, and the invariants of a trained representation form a functor $F$. The left Kan extension $\operatorname{Lan}_JF$ is the invariant that the source and the change of task force on the target: by its universal property, every comparison of the source with an observed target invariant factors through it. The transfer discrepancy is the largest distance, over the target objects, between this extension and the observed invariant. For persistent homology the extension is a cokernel over comma categories, and a functorial version of the discrepancy, which also compares the maps between target objects, is $1$-Lipschitz under interleavings. On finite samples the discrepancy is a bottleneck distance, computed without approximation. For data glued from two pieces whose exclusive parts lie farther apart than the filtration scale, a Mayer-Vietoris sequence identifies what extending homology loses, while extending chains loses nothing. For a merge of labelled point clouds, the degree-zero discrepancy vanishes if and only if some $1$-Lipschitz classifier has a margin above the filtration scale at every sample point. In experiments on sampled manifolds, trained encoders and pretrained image models, every exact prediction holds on every sample. The discrepancy of truncated invariants recovers a hidden merge of separated parts far above chance and tells the intended target of a merge from a permuted control, but across pretrained models it does not track the accuracy of a linear probe.

arXiv cs.LG75

教师何时应更新?自同策略蒸馏中的时间耦合与稳定性

arXiv:2606.03532v2 公告类型:替换 摘要:自同策略蒸馏使用由学生策略自身参数历史构建的教师来训练学生策略,但教师的更新调度——决定教师与学生之间的\emph{时间耦合}——尚未作为稳定性变量得到系统研究。通过在 Qwen3-8B 上开展受控的调度扫描实验,我们证实,\emph{隔离期}(即两次更新之间完全冻结教师的时段)才是实现稳定学习的关键结构性属性,而非教师的年龄。为刻画这些底层训练动态,我们提出了一个涵盖时间 KL 结构、刷新冲击和长度尾部风险的诊断框架。该框架进一步揭示了\emph{状态无感知崩溃}:在短期训练中最优的固定调度,在长期训练中会发生灾难性失效,因为时钟驱动的刷新可能通过一次不可逆的操作,将暂时发生漂移的学生复制到教师中。这种失效模式在短期评估中无法察觉,且在机制上不同于 EMA 的持续性污染。为解决这一问题,我们提出 \emph{Consolidation-Gated Teacher Refresh}(CGTR),该方法保留隔离期,同时根据奖励提升与长度尾部安全性的联合证据,对每次刷新进行门控,确保教师的每次更新都响应学生真正的学习巩固,而非时钟信号。CGTR 仅使用一套共享参数,无需针对各数据集重新调参,就在全部四项任务(Chemistry、Biology、Physics、ToolUse)上实现了\textbf{零崩溃}和最佳最终得分,并能根据各任务的学习动态自主调节刷新频率。

arXiv cs.LG72

动态医疗治疗强化学习的统一基准

arXiv:2606.01028v4 公告类型:替换 摘要:医疗治疗推荐给强化学习(RL)带来了若干挑战:患者的生理状态在连续时间中演变,测量和干预以不规则的时间间隔进行,且治疗效果在个体间存在显著差异。然而,现有的 RL 建模方法和仿真环境均基于具有固定决策间隔的离散时间马尔可夫决策过程(MDP)。因此,仍难以评估 RL 方法能否处理依赖于时间间隔的疾病进展、个性化治疗反应,以及连续两次测量之间的安全性。为填补这一空白,我们推出了 MedGym,一个用于动态治疗推荐的基准环境。MedGym 在连续时间框架下对患者状态的纵向演变进行建模,并利用物理信息神经网络,根据临床数据构建可配置的医疗 RL 基准。由此构建的基准能够在不规则治疗时机和患者特异性动态变化条件下,直接比较离散时间方法与连续时间方法。此外,MedGym 支持从个性化和轨迹层面的安全性等临床重要角度进行评估。通过为连续时间动态治疗提供标准化且可配置的基准,MedGym 能够对医疗 RL 方法进行更贴近现实、更具参考价值的评估。

arXiv cs.LG75

深度集成真的能捕捉图神经网络中的不确定性吗?

arXiv:2605.22593v2 公告类型:替换 摘要:尽管深度集成被广泛视为深度学习中不确定性量化的默认方法,但人们往往仅凭其在计算机视觉等领域的成功,就假定它对图结构数据同样有效。我们专门研究了标准深度集成在消息传递图神经网络中的表现。通过在涵盖不同任务和复杂度的八个数据集上进行基准测试,我们发现,集成相较于单个模型仅带来微小改进。这些微小收益主要源于对点预测中的优化噪声进行平均,而非不确定性估计得到实质性改善。通过对偶然不确定性和认知不确定性进行分解,我们发现了认知不确定性坍缩现象:独立训练的网络总是收敛到过于相似的预测结果。由于预测分歧是集成捕捉认知不确定性的根本机制,这种多样性的缺失抵消了其关键优势。为解释这一现象,我们推导了一个由图拓扑和 Dirichlet 能量预算决定的认知方差界,将认知不确定性坍缩与消息传递的平滑偏置联系起来。我们的结果表明,深度集成的成功并不能直接迁移到图机器学习中。

arXiv cs.LG66

Stochastic MeanFlow Policies: One-Step Generative Control with Entropic Mirror Descent

arXiv:2605.21282v3 Announce Type: replace Abstract: Combining entropy regularisation with mirror descent in online off-policy reinforcement learning requires expressive policies with efficient sampling and explicit entropy control. Gaussian policies struggle to represent multimodal targets, whereas generative policies often require iterative sampling or lack tractable entropy estimates. We introduce Stochastic MeanFlow Policies (SMFP), a one-step generative policy class that combines MeanFlow-based noise-to-action mappings with Gaussian reparameterisation. The latent mapping captures diverse action modes, while a learnable conditional Gaussian scale provides an analytic entropy lower bound. We use this bound to regulate stochasticity alongside direct value guidance and advantage-weighted MeanFlow regression, enabling entropy-regularised mirror-descent learning with one-step policies. Across seven MuJoCo benchmarks, SMFP achieves strong performance against Gaussian and generative baselines with single-step inference. Additional experiments demonstrate multimodal action modelling and effective exploration in sparse-reward navigation.

arXiv cs.LG73

SURF:调控标量化权重以均匀遍历 Pareto 前沿

arXiv:2605.20619v2 公告类型:替换 摘要:标量化因其简单性和可扩展性而被广泛用于多目标优化。在许多应用中,目标是生成能够代表多样化用户偏好的解,理想情况下均匀覆盖 Pareto 前沿(PF)。然而,均匀采样标量化权重通常会导致对 PF 的非均匀覆盖。我们通过标量化路径的几何性质解释这一不匹配现象。随着权重变化,相应的解以通常并不均匀的速度遍历 PF。对该速度积分可得到累积弧长。用总长度归一化后,它便定义了权重上的累积分布函数(CDF),其反函数可选取弧长间隔相等的点。基于这一洞见,我们提出 SURF(沿 Pareto 前沿均匀采样)。对于包括双目标多臂老虎机在内的结构化问题,我们推导了该 CDF 映射及由此得到的 PF 感知权重采样规则的解析表达式。对于一般问题,SURF 交替执行 CDF 重建与权重采样。在理论上,我们证明,在双目标 LS 的可证明条件下,SURF 的 CDF 重建误差线性收敛至 $O(N^{-2})$ 的离散化误差下限。在实证方面,多臂老虎机、multi-objective-gymnasium 和多目标 LLM 对齐上的实验表明,SURF 能够高效地实现比基线方法更均匀的 PF 覆盖。

arXiv cs.LG68

Quadratic Characterizations for Reachability Analysis of Neural Networks

arXiv:2605.20482v2 Announce Type: replace Abstract: Quadratic constraints (QCs) are widely used to characterize nonlinearities and uncertainties, but generic analytical characterizations can be conservative on bounded domains. This paper develops a framework for constructing verified quadratic characterizations of scalar relations in the two-dimensional real plane. Candidate quadratic inequalities are locally generated by solving convex quadratic programs using samples from the relation and exterior sample points. They are then verified globally using sum-of-squares certificates over an exact semialgebraic description or, in the case of nonpolynomial relations, over relaxed polynomial descriptions. The resulting verified constraints define a sound overapproximation of the scalar relations over the considered domains. These constraints are directly compatible with existing analysis frameworks based on QCs and pointwise integral quadratic constraints (IQCs) for static nonlinearities and uncertainties, and they can also be embedded in QC-based semidefinite programs for reachability and safety analysis of feedforward neural networks. For smooth activations such as $\tanh$, the method yields domain-dependent quadratic characterizations that constitute an alternative to generic sector- or slope-based descriptions. For ReLU networks, we give methods to reduce conservatism in QC-based reachability analysis of feedforward networks by exploiting dependencies between neurons and tighter local bounds. Numerical examples demonstrate improved reachability results for smooth activations, reduced conservatism for ReLU networks, and applicability beyond neural networks through an example involving saturation.

arXiv cs.LG73

基于采样的安全强化学习

arXiv:2605.19469v2 公告类型:替换 摘要:安全探索仍然是强化学习(RL)中的一项根本性挑战,限制了 RL 智能体在现实世界中的部署。我们提出 Sampling-Based Safe Reinforcement Learning(SBSRL),这是一种基于模型的 RL 算法,通过在有限的动力学样本集上联合施加约束,在整个学习过程中维持安全性。该形式化方法近似求解了针对不确定动力学的难以处理的最坏情况优化问题,并能在连续域中提供实用的安全保证。我们进一步引入了一种基于约束认知不确定性的探索策略,无需显式的探索奖励。在正则性条件下,我们推导了整个学习过程中的高概率安全保证,以及用于获得近似最优策略的有限时间样本复杂度界。实验表明,SBSRL 在仿真和真实机器人硬件上均能实现安全、高效的探索,并且易于扩展为实用的深度集成实现,以适应高维连续控制问题。

arXiv cs.LG70

Bayesian Nonparametric Mixed-Effect ODEs with Gaussian Processes

arXiv:2605.13088v2 Announce Type: replace Abstract: Dynamical modelling in pharmacometrics, systems biology, and epidemiology often involves subjects with related but distinct dynamics, observed through temporally sparse and irregular measurements. Classical nonlinear mixed-effects ODE models represent this heterogeneity but rely on parametric vector fields, making them vulnerable to structural misspecification. Existing nonparametric ODE methods typically assume shared dynamics rather than an explicit mixed-effect hierarchy. We propose MEGPODE, a Bayesian nonparametric mixed-effect ODE model that decomposes each subject's vector field into a shared population component and a subject-specific deviation, both endowed with Gaussian process priors. To avoid repeated ODE solves during training, we represent latent trajectories with state-space GP priors and penalize discrepancies between their derivatives and the vector fields on a finite time grid. A single variational objective enables approximate joint inference over trajectories and vector fields, accounting for state uncertainty and dependence between population and subject-specific dynamics. Across heterogeneous oscillatory and biomedical ODE benchmarks, MEGPODE improves population-field recovery and subject-level trajectory prediction relative to strong baselines.

arXiv cs.LG62

On the Approximation Complexity of Matrix Product Operator (MPO) Born Machines

arXiv:2605.11471v2 Announce Type: replace Abstract: Matrix product operator Born machines (MPO-BMs) are tractable tensor-network models for probabilistic modeling, yet their approximation capabilities remain poorly understood in theory. In this paper, we characterize these capabilities from both negative and positive perspectives. As a negative result, we prove that constructing a polynomial-bond MPO-BM approximation to an arbitrary continuous target with bounded KL error is NP-hard. On the positive side, for score-based variational inference, we prove that targets satisfying a log-Sobolev inequality admit polynomial-bond MPO-BM approximations with provable KL guarantees when the loss-induced Hamiltonian is local and has a constant spectral gap. Furthermore, under the same conditions, we construct local importance-sampling estimators and prove that polynomially many local score queries suffice to preserve the KL guarantees.

arXiv cs.LG64

Model Capacity Determines Grokking through Competing Memorisation and Generalisation Speeds

arXiv:2605.09724v2 Announce Type: replace Abstract: Existing accounts of grokking explain the phenomena in terms of mechanistic frameworks such as circuit efficiency or lazy-to-rich transitions. However, despite a known dependence between grokking and model size, how model capacity shapes grokking remains an open question. We give an information-theoretic account of this relationship on the task of modular arithmetic, showing that grokking does not immediately occur when a model becomes large enough to memorise the training set, but rather emerges as the outcome of a competition between two measurable timescales: a memorisation speed $T_{\text{mem}}(P)$ and a generalisation speed $T_{\text{gen}}(P)$, both of which are functions of model parameter count $P$. Adapting the information capacity framework of Morris et al. (2025), we estimate $T_{\text{mem}}(P)$ on random-label data of equivalent complexity and $T_{\text{gen}}(P)$ on the modular task itself, and show that grokking emerges close to the parameter scale where these timescales intersect. The framework also suggests an empirical model for predicting memorisation speed given model capacity and dataset complexity, recovering the previously reported empirical observation that larger models memorise faster. Overall, we motivate the formalisation of different learning timescales as important abstractions to study when explaining how model capacity shapes grokking on algorithmic tasks.

arXiv cs.LG79

FactoryNet:面向工业时间序列基础模型的大规模数据集

arXiv:2605.09081v5 公告类型:替换 摘要:工业机器是受驱动系统,但大多数开放时间序列语料库仅记录传感器测量结果;少数记录了指令信号的语料库也采用因机器而异的厂商特定格式,因此无法跨实体形态比较指令意图与测量响应。我们提出 FactoryNet,一个用于工业时间序列预训练的语料库:涵盖七种实体形态、四类任务的 5.7 万次端到端任务执行(1.5 万次真实执行、4.2 万次合成执行),共计 1.13 亿个数据点,并提供 27 种带标注的异常类型、正常基线及反事实配对。所有来源中的每个信号均映射至统一的控制理论模式 Setpoint、Effort、Feedback、Context(S-E-F-C),该模式以通用表示框架描述任意受驱动系统,并将指令与实际响应之间的动态差异呈现为显式预测残差。利用该语料库,我们发现,仅使用仿真训练、不使用任何真实阶段标签的任务阶段模型,在两种机器人上均能达到真实数据性能上限的约三分之二;迁移在同一机器人系列内有效,但跨厂商时鲜有成效。故障特征则表现不同:未经校准的仿真器在两种机器人上均无法提供可检测的故障信号,而对齐特征分布也无法弥补这一缺陷。有限的区分能力来自 Effort:在实测工具运动的故障判别表现接近随机水平时,电机电流仍能区分故障;而通过校准仿真器以复现这一个通道,即可在真实机器人上识别出一种负载故障。我们据此认为,仿真能够捕捉任务的稳定结构,而故障行为仍具有多样性,且取决于具体故障类型。我们发布 FactoryNet,旨在为工业基础模型提供持续扩展、涵盖多种实体形态的数据基础。

arXiv cs.LG69

Reinforcement learning for inverse silhouette design and rapid laser cutting of compact parallelogram quad kirigami prototypes

arXiv:2605.08098v2 Announce Type: replace Abstract: Kirigami is an increasingly useful fabrication method to produce shape-programmable metamaterial structures. However, inverse design remains difficult because deployment is nonlinear, and feasible cut layouts must satisfy discrete compatibility rules, avoid overlap, and map one target shape to valid designs. We present RL-Kirigami, an inverse design framework that combines optimal-transport conditional flow matching (OT-CFM) with reinforcement learning to generate compatible ratio fields for compact reconfigurable parallelogram quad kirigami. A marching decoder enforces global geometric compatibility, and Group Relative Policy Optimization (GRPO) aligns the generator with nondifferentiable rewards for silhouette matching, feasibility, and ratio-field regularity. Across procedurally generated target shape instances, a single sample from the pretrained OT-CFM prior reached $94.2\%~\mathrm{sIoU}$ and outperformed solver baselines while reducing forward simulator evaluations from hundreds to $1$. GRPO improved accuracy to $94.91\%~\mathrm{sIoU}$ and, with regularity included, reduced $\mathrm{TV}(\mathbf{x})$ from $0.95$ to $0.81$ while maintaining $94.83\%~\mathrm{sIoU}$. Generated layouts were exported to DXF and laser cut in $50~\mu\mathrm{m}$ polymeric sheets to produce deployable prototypes in $8.0\pm 1.0$ minutes per part. These results support a manufacturing-aware inverse design workflow for deployable kirigami metamaterials under geometric feasibility constraints. Code is available at https://github.com/milad1378yz/RL-Kirigami.

arXiv cs.LG71

超越成对比较:你的语言模型正在暗中优化偏好图

arXiv:2605.08037v2 公告类型:替换 摘要:直接偏好优化(DPO)利用成对偏好比较来对齐语言模型,为基于人类反馈的强化学习(RL)提供了一种简单有效的替代方案。然而,在许多实际场景中,训练数据包含每个提示对应的多个采样输出,由此形成了丰富的偏好结构,而成对 DPO 无法利用这些结构。将此类数据简化为相互独立的样本对会丢弃传递性,引入冗余或冲突的监督,并可能导致优化不稳定。我们提出 Graph Direct Preference Optimization(GraphDPO),这是 DPO 的一种结构化推广,在由采样输出排名形成的有向无环偏好图上进行优化。GraphDPO 将支配关系编码为边,并优化一个受 Plackett--Luce 启发的图结构目标函数,该目标函数在被支配邻域上聚合监督信息,传播输入偏好的传递结构,同时将标准 DPO 涵盖为一种特例。GraphDPO 通过等价类构造自然地处理偏好并列和不可比较的回答:具有相同偏好信号的回答构成图层,不可比较的回答共享同一层且彼此之间不连边,不存在的边对损失的贡献为零,从而避免严格全序目标函数产生的伪梯度。在这种分层构造下,GraphDPO 通过高效的 log-sum-exp 聚合,将每个提示的计算复杂度保持在线性水平。我们进一步引入可选的真实答案锚定机制,将经过验证的解作为支配节点插入,并采用退火调度,在稳定早期训练的同时逐步放松预言机监督。在推理和程序合成任务上的实验表明,该方法具有更优的性能,说明图结构偏好建模是一种可扩展且稳健的方案,可替代成对和列表式对齐目标。

arXiv cs.LG71

用于离散选择模型的 Constrained Triple-Head Price Optimization (C3PO) 网络

arXiv:2605.06941v2 公告类型:替换 摘要:我们提出了 Constrained Triple-Head Price Optimization (C3PO) 网络,用于离散选择环境中的实时双层决策。在这些场景中,卖方选择提供的产品或服务以实现业务目标,而异质用户根据自身偏好做出个性化的接受或拒绝决策,最终决定卖方的收益结果。现有方法利用合成数据微调预训练的表格基础模型,但受限于模仿学习范式,且无法强制执行常见的定价规则,包括跨产品约束和全局约束。C3PO 通过一种新颖的基于 transformer 的架构解决了这些局限,该架构联合整合价格模仿学习、多任务收入预测和上下文弹性学习,无需重新训练即可生成跨领域的稳健定价建议。一个轻量级可微约束层使生成的价格在优化收入目标的同时满足业务约束。我们在仿真、合成和真实世界数据集上展示了该模型出色的分布外性能。在经过校准的 MNL 模型相对于真实选择模型存在设定偏误的仿真基准中,C3PO 相较于经过校准的 MNL 基线将预期收入提高了 1.4 至 7.9 个百分点。约束层将平均绝对约束违反量降低了 50% 以上。我们进一步将该模型部署到真实世界应用中,包括医疗招标定价和航空附加服务定价。在这些应用中,以涨价和降价召回率衡量,C3PO 将定价方向有效性提高了 18 至 34 个百分点,具体增幅取决于所比较的基线,包括经过校准的 MNL、经典优化方法和现成的表格基础模型。

arXiv cs.LG72

逆向玩转网络:一种博弈论归因框架

arXiv:2605.06212v2 公告类型:替换 摘要:归因方法用于解释哪些输入特征驱动了模型的预测,因此在模型调试和机制可解释性研究中具有核心地位。然而,包括梯度、LRP 和 transformer 专用规则在内的反向归因方法,缺乏一个可用于比较其底层反向计算的统一框架。我们基于 Gaubert 和 Vlassopoulos 的 ReLU Net Game,将反向归因重新表述为扩展网络图上的双人博弈,从而提出了这样一个框架。在特定均衡下,梯度和整个 alpha-beta-LRP 方法族均可表示为对博弈轨迹的积分,因此归因图成为轨迹分布的投影,而不再是首要研究对象。定位聚焦性、对输入噪声的鲁棒性或稳定的注意力路由等期望的解释属性,可以通过策略正则化、风险厌恶和扩展动作集等博弈论概念来刻画,并直接转化为对现有经典反向规则的新型改进。在 ViT-B/16 上,我们选取的一种此类 alpha-beta-LRP 改进方法,在所有考察的定位指标上均优于已有的 transformer 专用反向方法。

arXiv cs.LG76

通过最优传输训练不可微网络

arXiv:2605.01928v4 公告类型:替换 摘要:硬阈值、量化和离散路由可能导致训练损失出现平坦区域和跳跃,使普通梯度消失或无定义。我们提出 PolyStep,这是一种仅需前向计算的优化器,通过评估旋转多胞体探测点,并沿探测方向的加权平均方向移动参数块。我们从单侧熵正则化传输中推导权重,并在主要实验中使用其非耦合 softmax 解。我们的分析解释了探测点代价之间的差异何时会产生移动,以及这种移动何时会降低损失。在正单纯形上,非常量的代价总会给出非零方向。对于单调脊函数损失,在任意正温度下,softmax 更新都不会增加损失;扰动界则给出了在弯曲跳跃面附近实现下降的充分条件。对于有界可测损失,我们将探测半径随机化,并确定了一种精确的平滑化方式,其梯度在相差一个比例因子的意义下等于按代价线性加权的方向的期望。这个恒等式为一种理想化的固定温度变体给出了平稳性界:在比我们实际训练配置所满足的条件更强的正则性和采样假设下,该界包含一个 $O(T^{-1/2})$ 项和一个持续存在的偏差下限。我们在包含硬操作的网络、离散优化和策略搜索任务上评估了这一实用方法。在采用硬阈值脉冲神经元的 MNIST 任务上,PolyStep 达到 $93.0 \pm 0.2\%$,而在相同评估次数下,经过最佳调优的无梯度基线为 $79.6 \pm 5.2\%$。这些提升伴随着每个新步骤中与搜索维度成正比的查询成本,因此限制了固定预算下可执行的更新次数。

arXiv cs.LG71

Approximating Uniform Random Rotations by Two-Stage Structured Hadamard Rotations in High Dimensions

arXiv:2604.23418v2 Announce Type: replace Abstract: Uniform random rotations are useful in fast Johnson--Lindenstrauss embeddings, kernel approximation, communication-efficient learning, and recent AI compression pipelines, but are computationally expensive to generate and apply in high dimensions. A practical alternative repeats Walsh--Hadamard transforms with random sign diagonals. Applying this structured rotation twice has shown empirical benefits, but supporting theory remains limited. We study the approximation quality of the two-stage structured Hadamard rotation. Here $d$ is the dimension of the rotated vector. On the positive side, we prove explicit, nonasymptotic upper bounds of order $\sqrt{\log d/d}$ in both Kolmogorov and Wasserstein-1 distance for corresponding coordinates scaled by $\sqrt d$, uniformly over unit inputs and coordinates. For the input $u=e_1$, both distances also have lower bounds of order $d^{-1/2}$ at every coordinate. On the negative side, we prove an explicit lower bound on the Wasserstein-1 distance between the full vector distributions, showing a substantial discrepancy that does not vanish as $d$ grows for the input $u=e_1$. For the same input, we also prove a matching asymptotic upper bound, showing that the lower-bound scale is sharp. Taken together, the results identify a separation between one-dimensional marginal behavior, where approximation improves with dimension, and the full-vector law, where a nonvanishing Wasserstein-1 discrepancy remains. The marginal bounds control coordinate threshold probabilities and expected scalar Lipschitz losses, including expected averages over coordinates. They do not establish joint-coordinate approximation, and the full-vector separation does not imply failure of a particular algorithm. Connecting these distributional results to performance guarantees for specific algorithms remains open.

arXiv cs.LG62

Koopman-Based Nonlinear Identification and Model Predictive Control of a Turbofan Engine

arXiv:2604.01730v3 Announce Type: replace Abstract: This paper investigates Koopman operator-based approaches for multivariable control of a two-spool turbofan engine. A physics-based component-level model is developed to generate training data and validate the controllers. A meta-heuristic extended dynamic mode decomposition is adapted, with a cost function designed to accurately capture both spool-speed dynamics and the engine pressure ratio (EPR), enabling the construction of a single Koopman model that can be reused across multiple control strategies. Using the identified time-varying Koopman model, an adaptive Koopman-based model predictive controller (AKMPC) with a disturbance observer is developed and compared with a Koopman-based feedback linearization controller (K-FBLC) and its integrator-augmented version (K-FBLC-I). The Koopman representation further enables nonlinear GTE output limiters, such as rotor-acceleration and turbine-inlet-temperature limits, to be expressed as linear constraints in the AKMPC. The controllers are evaluated for two control configurations of spool speeds and EPR, under both sea-level and varying flight conditions. The results demonstrate that the proposed identification approach enables accurate predictions of both spool speeds and EPR, allowing the Koopman model to be reused flexibly across different control formulations. While all strategies achieve comparable performance in sea-level conditions, the AKMPC demonstrates improved performance under varying flight conditions due to its ability to capture nonlinear dynamics, handle constraints, and compensate for model mismatch. Moreover, the EPR control strategy improves the thrust response. The study highlights the applicability of Koopman-based control and the advantages of the AKMPC framework for robust turbofan engine control.

arXiv cs.LG73

用于在线生成模型选择的 Mixture-Greedy:多样性感知多臂老虎机是否需要 UCB?

arXiv:2603.21716v3 公告类型:替换 摘要:在现代生成式 AI 中,从多个生成模型中进行高效选择变得越来越重要,因为从次优模型中采样的成本很高。该问题可视为一项多臂老虎机(MAB)任务。在多样性感知评估分数下,生成器的非退化混合可以优于任何单个模型,这使得此类 MAB 设定不同于经典的最优臂识别问题。以往方法在混合目标中加入置信上界(UCB)探索奖励。然而,在多个数据集和评估指标上,我们观察到 UCB 项始终会减缓收敛并降低样本效率。相比之下,不采用显式 UCB 式乐观机制的简单 Mixture-Greedy 策略收敛更快,甚至能取得更好的性能,尤其是在 FID 和 Vendi 等难以构造紧致置信界的常用指标上。我们从理论上解释了这一现象:在特定结构条件下,多样性感知目标通过偏好内部混合解来诱导隐式探索,从而促使所有臂都得到采样,并为基于多样性的目标提供次线性遗憾保证。这些结果表明,在多样性感知多臂老虎机问题中,例如生成模型选择问题,探索可以内生于目标函数的几何结构。

arXiv cs.LG77

权重聚类揭示相对排序在大语言模型中的重要性

arXiv:2603.17917v2 公告类型:替换 摘要:大语言模型(LLMs)包含数十亿个参数,但许多权重的精确数值并非必不可少。我们对预训练模型应用标量 $K$ 均值聚类,将每个权重矩阵中的权重替换为仅 16--64 个共享值。对于 Llama~3.1-8B-Instruct 和 SmolLM2-135M,这种方法无需重新训练即可保持较高的准确率,为压缩磁盘上存储的 LLMs 提供了一种简单方法。仅微调聚类中心,就能进一步挽回剩余准确率损失的 30--40\%。随后,我们将聚类作为分析工具,固定聚类分配,仅扰动共享值,以研究这些值的哪些属性最为重要。我们发现,保留它们的\emph{相对排序}(哪些连接比其他连接更强或更弱)比保留其精确大小更重要。破坏这种排序会大幅增加困惑度,而在许多中间层和后期层中,改变数值但保持排序只会造成很小的损失。然而,仅保留排序还不够。模型性能还取决于是否维持了重构权重的方差,而早期层仍然更为敏感;对聚类分配本身进行扰动表明,其空间几何结构也起着关键的支撑作用。我们在来自两个架构家族的四个 LLMs 中观察到了相同的行为。这些结果表明,权重的排序及其整体方差在 LLM 压缩和鲁棒性方面发挥着核心作用。