全部动态

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

State-OPD:面向多轮智能体的基于状态的同策略蒸馏

arXiv:2610.13238v1 公告类型:新论文 摘要:同策略蒸馏(OPD)已成为将前沿模型的推理能力迁移至较小模型的强大范式,但将其成功推广到多轮智能体环境仍面临挑战。我们从交互状态的角度研究智能体 OPD,将客观任务和已执行的动作—观测记录与学生模型推理的内部语义轨迹区分开来。通过这一视角,我们发现学生模型推理在多轮 OPD 中具有两种作用:先前的推理会扭曲教师模型的监督信号,而当前的推理则会影响在同一状态下采样得到的候选动作。具体而言,在匹配的交互状态下,移除学生模型先前的推理可以提高教师模型的后续执行成功率,并改变教师模型的词元级目标,且在较后轮次中,后续执行成功率的提升更大。同时,与在固定推理轨迹下重新采样动作相比,重新采样完整的推理—动作响应能在同一状态下产生更多样的动作。我们提出基于状态的同策略蒸馏框架,将教师模型的监督建立在交互状态之上,并重新分配采样预算,在响应熵较高的状态下生成多个完整的推理—动作响应,同时保留学生模型基于完整历史的生成方式。在 ALFWorld、ScienceWorld 和 WebShop 上的全面评估表明,无论采用 Qwen3-1.7B 还是 Qwen3-4B 作为学生模型,State-OPD 在所评估的方法中均取得了最高的宏平均成功率。值得注意的是,对于 Qwen3-4B,State-OPD 将宏平均成功率从最强基线的 26.07\% 提高至 31.74\%,并在 ALFWorld Seen 上超越了其教师模型 Qwen3-32B。进一步分析表明,State-OPD 使用更少的受监督轮次响应即可取得更高的任务成功率,并显著减少对失败动作的持续重复;定性轨迹展示了基于交互反馈进行的决策修正。

arXiv cs.LG60

Learning Schwarzschild Geodesic Motion with Lagrangian Neural Networks

arXiv:2610.13234v1 Announce Type: new Abstract: Lagrangian Neural Networks (LNNs) learn dynamics from trajectory data by parameterizing a scalar Lagrangian and deriving accelerations through the Euler--Lagrange equation, providing a variational inductive bias that improves physical consistency over purely data-driven models. LNNs have recently been extended to general relativistic settings, leaving Schwarzschild geodesics as an open challenge. Training on complex relativistic systems is hindered by the diversity of the dynamics, as well as numerical instabilities and the difficulty of adequately sampling physically relevant, diverse orbital regimes. We introduce two coordinated improvements: a decoupled position--velocity normalization scheme that independently rescales each coordinate and its time derivatives, reweighting acceleration targets to balance contributions across all channels; and an energy-constrained log-uniformly sampled dataset generation strategy informed by the innermost stable circular orbit (ISCO) structure. With these improvements, our model accurately reproduces perihelion precession and relativistic time dilation, conserves energy and angular momentum to within a few percent even after completing multiple orbits around the central mass and predicts orbital trajectories across eight orbital scenarios in close, medium, and far-field regimes. These results establish that LNNs, augmented with physically motivated normalization and sampling, can reliably learn geodesic dynamics in curved spacetime. Our work also represents the first successful training of LNNs on Schwarzschild geodesics and offers a promising direction for data-driven modeling of complex relativistic systems.

arXiv cs.LG75

ReSFlow:校正统计流形上的类别流以实现单步生成

arXiv:2610.13225v1 公告类型:新稿 摘要:通过在统计流形上学习连续向量场,将简单先验输运至目标离散分布,离散数据生成建模取得了显著进展。然而,这些连续时间流并未被约束为沿统计流形上的测地线运动,因此需要通过多步数值积分进行模拟,导致推理阶段的计算成本高昂。在本研究中,受 Rectified Flow 在连续欧几里得空间中“直且快”特性的启发,我们提出了 Rectified Statistical Flow(ReSFlow)模型。该模型通过显式学习遵循内在黎曼几何的测地线轨迹,实现统计流形上的单步类别生成。通过校正统计流形上的流,我们从理论上保证了先验分布与目标分布之间的轨迹逼近测地线。这种校正使学习到的轨迹直接沿统计流形上的测地线对齐,从而显著降低输运成本,使模型能够在单步内将纯噪声映射为目标离散数据。通过在涵盖图像、文本和生物领域的多种真实世界生成任务上开展广泛实验,我们展示了 ReSFlow 卓越的单步生成质量,真正实现了测地线意义下最优且快速的类别生成。我们的代码已公开于 https://github.com/xiaohuanjia/ReSFlow。

arXiv cs.LG66

Tensor-based Fusion of Multi-modal Data for Predicting Cancer Risk Levels

arXiv:2610.13221v1 Announce Type: new Abstract: Cancer diagnosis and treatment increasingly rely on multi-modal diagnostic technologies that generate high-dimensional datasets. Although these datasets are rich in pathological information, they pose considerable challenges in interpretation, dimensionality reduction, and clinical applicability. Conventional machine learning approaches often fail to extract meaningful cross-modal features, thereby limiting predictive accuracy in patient risk stratification. To address these limitations, this paper proposes an unsupervised machine learning framework that integrates feature extraction, tensor decomposition, and clustering to improve cancer risk prediction. The framework employs tensor decomposition to fuse heterogeneous genomic and clinical data, enabling the extraction of latent, biologically meaningful features that outperform unimodal feature sets. Multiple feature extraction strategies are investigated, and the resulting features are subsequently clustered using a range of clustering algorithms. Comparative evaluations based on internal and external metrics are then used to identify the most effective configurations. The framework is validated on LinkedOmics data comprising 11,158 patients across 32 cancer types. For patient risk stratification, clinically relevant attributes such as tumor purity, pathological stage, and survival outcomes serve as ground-truth indicators. Experimental results demonstrate the superiority of tSVD+GMM, which consistently outperforms other configurations across clinical benchmarks.

arXiv cs.LG75

Monarch-MoE:用于多模态持续指令微调的结构化专家

arXiv:2610.13215v1 公告类型:新投稿 摘要:在多模态持续指令微调中,学习新任务可能导致对先前所获知识的灾难性遗忘。尽管多专家 LoRA 将学习分散到多个专家中,但对稠密低秩专家的更新仍可能影响许多输出特征。我们提出 Monarch-MoE,将 LoRA 专家中的稠密因子替换为通过固定置换连接的块对角因子。这种结构提供了分块梯度隔离:对单个块对的更新仅影响一个输出块,同时保持所有输入块与输出块之间的连接。为了使这些紧凑的专家能够在显存受限的硬件上切实使用,我们结合了批量计算、融合前向算子和优化的反向传播过程。我们使用 LLaVA-1.5-7B,在 MLLM-DCL 和 UCIT 上评估了三种持续学习方法;Monarch-MoE 提高了首个任务的知识保留能力,同时将适配器参数量分别减少约 75% 和 50%。在主要对比中,MLLM-DCL 上首个任务的遗忘幅度低至 0.52 个百分点,UCIT 上则低至 0.10 个百分点,而最终性能上的权衡因方法而异。我们的实现支持在单张 24 GB RTX 4090 上进行微调和推理,在所评估的设置下,实测显存占用更低,运行时间也具有竞争力。这些算法和系统层面的改进共同使结构化专家成为在显存受限硬件上进行持续适应的实用选择。

arXiv cs.LG71

每次预测需要多少计算量?两种时间序列基础模型中的结构化块跳过

arXiv:2610.13208v1 公告类型:新论文 摘要:时间序列基础模型对每个输入都使用相同的网络深度,尽管预测窗口在结构和难度上各不相同。我们在两种预训练模型 Chronos-Bolt 和 TimesFM 2.5 中检验能否在推理时移除整个 Transformer 块。我们的旁路机制避开了原始块的计算,并在启用所有块时精确保留稠密模型的行为。掩码仅在校准窗口上选择,并在来自 ETTh1、Traffic、Weather 和 Exchange 的 1,088 个留出窗口上评估。在适度稀疏的情况下,保留 7/11 个候选块的 Chronos-Bolt 和保留 16/20 个块的 TimesFM 在各数据集上的平均相对预测质量变化很小,同时将 A100 上批大小为 1 的推理延迟中位数分别降低了 11.2% 和 13.4%。更激进的 TimesFM 块跳过可进一步加速推理,但会显著损害其在 Traffic 和 Weather 上的表现。在这两种模型中,不同的等预算掩码在不同窗口上各有优势,而事后使用的理想选择器在每个数据集上都显著优于最佳静态掩码。然而,数据集特定掩码的迁移表现并不一致,Chronos 表征也无法支持可靠的未见域路由。这些结果证实了跨架构的结构化冗余以及随输入而异的优化潜力,但尚未形成可部署的动态推理方法。

arXiv cs.LG74

重新思考同策略蒸馏中的相对策略差异迁移

arXiv:2610.13205v1 公告类型:新论文 摘要:同策略蒸馏(OPD)利用更强教师提供的词元级监督,在目标策略自身生成的轨迹上训练目标策略,但通常依赖教师的能力显著强于目标策略。近期研究通过迁移专家策略与参考策略之间的相对策略差异,而非专家策略的绝对分布,放宽了这一要求。然而,哪些差异能够提供有效监督,以及哪些目标策略能从中受益,仍不清楚。我们提出了一种统一视角:代理专家与代理参考策略之间的对数概率比,为 KL 正则化策略优化定义了隐式奖励。这一表述导出了目标策略的同策略更新,将传统 OPD 纳入其中作为特例,并支持在规模、训练阶段或代际不同的模型之间进行任意到任意的迁移。我们的系统性实验得出了三项发现。首先,更强的专家未必是更好的监督来源:由特定任务训练(包括强化学习和监督微调)产生的差异能够稳定迁移,而仅由模型规模产生的差异则迁移效果微弱,甚至完全无法迁移。其次,专家与参考策略发挥互补作用:专家提供待迁移的能力,而参考策略过滤掉共有行为,分离出相关的残余差异。第三,即使是有用的差异,也无法迁移至所有目标策略:成功迁移要求所诱导的更新使目标策略向专家靠近,并能泛化到采样轨迹之外。原始策略距离无法预测可迁移性,而恰当的参考策略能够挽救原本失败的迁移。这些结果刻画了相对策略差异何时能够为 OPD 提供有效监督。

arXiv cs.LG75

S$^3$R:通过自稀疏推测解码实现快速且忠实的强化学习轨迹生成

arXiv:2610.13202v1 公告类型:新投稿 摘要:在具有可验证奖励的长上下文强化学习(RLVR)中,快速生成轨迹需要在不改变用于策略优化的响应组的前提下,减少 KV 缓存访问流量。在我们进行性能剖析的 RLVR 工作负载中,注意力计算占自回归解码内核执行时间的 72--76\%。稀疏注意力可以减少这些读取操作,但直接从稀疏策略中采样会改变用于学习的响应组。对于组相对优化,我们证明,即使对每个响应进行精确的重新加权,通常也无法恢复稠密策略的期望更新,因为每个响应的优势取决于组内其他成员的奖励。我们提出了 Self-Sparse Speculative Rollouts(S$^3$R)。该方法使用稀疏注意力提出候选词元,再通过全注意力验证和精确推测采样生成响应。草拟与验证共享同一套策略参数,因此无需训练或更新单独的草拟模型。由此生成的响应组遵循稠密策略的分布。我们刻画了 S$^3$R 实现加速的条件,以及草拟和验证的成本。我们发现,稠密验证仍然需要完整的历史 KV 缓存。因此,具备 KV 感知能力的调度器会在保持轨迹生成并行度的同时,限制 KV 缓存的反复卸载与恢复。% 或抖动 在采用激进稀疏草拟预算的匹配消融实验中,S$^3$R 保持了稳定的训练,而采用训练端校正的直接稀疏采样则变得不稳定。在长上下文数学推理任务中,S$^3$R 在 Qwen3 模型上实现了自回归轨迹生成 $1.39\times~1.35\times$ 的端到端训练吞吐量,同时保持了相当的任务性能。

arXiv cs.LG72

在部分可观测强化学习中,训练循环网络重要吗?

arXiv:2610.13197v1 公告类型:新论文 摘要:部分可观测条件下的强化学习(RL)通常依赖循环网络来编码时序信息,并将循环模块与策略或价值函数联合训练。这引出了一个自然的问题:训练循环网络真的有必要吗?本文研究冻结的循环网络能否提供足够的记忆,以求解部分可观测马尔可夫决策过程(POMDP)。受储备池计算(RC)和广义储备池计算(GRC)理论基础的启发,我们研究了一种简单架构,将固定的循环网络与可训练的非线性输出模块相结合。我们在记忆需求、观测模态和动作空间各异的任务上评估了这种方法。令人惊讶的是,冻结的循环网络能够回忆数百个时间步之前的信息,并在多种 POMDP 中取得具有竞争力的控制性能,而所需的可训练参数数量不到对应的全可训练网络的一半。这些结果表明,求解 POMDP 并不总是需要训练循环记忆模块。因此,冻结的循环网络为部分可观测强化学习提供了一种简单、参数高效且表现出乎意料地强劲的基线。我们的源代码已发布于 https://github.com/Aphlios/FRN-POMDP

arXiv cs.LG70

WaSA: Wavelet Scattering Autoencoders for Unsupervised Offline Detection of Malicious Clients in Federated Learning

arXiv:2610.13186v1 Announce Type: new Abstract: Federated Learning (FL) enables collaborative model training without exposing raw data, yet remains vulnerable to malicious clients injecting poisoned datasets. Existing defenses are typically reactive (operating during training) or supervised (requiring prior knowledge of attack types) - assumptions that rarely hold in open-ended IoT deployments. We propose WaSA, a fully unsupervised, offline framework for malicious client detection. Before training, each client transmits a single, privacy-preserving spectral summary of its data to the server. A denoising autoencoder, trained solely on clean auxiliary data, scores clients based on their deviation from the learned benign manifold, identifying corruptions regardless of the attack type. WaSA requires no attack labels, generalizes to unseen anomalies, and introduces a Boltzmann sampling strategy for client selection that smoothly interpolates between hard exclusion and standard aggregation. Experiments on MNIST, CIFAR-10, and CIFAR-100 under diverse attack scenarios (up to 90 % malicious clients) demonstrate consistent improvements over supervised baselines and robust aggregation methods, with negligible communication overhead. Code is available at https://github.com/alelicciardi99/wavelet_autoencoders_malicious.

arXiv cs.LG74

通过合成自蒸馏对小模型进行持续后训练

arXiv:2610.13113v1 公告类型:新提交 摘要:持续后训练面临能力习得与知识保留的双重挑战:模型必须在学习新能力的同时保留已有知识。尽管自蒸馏微调(SDFT)利用上下文示例进行自监督来缓解这一问题,但它依赖于较强的上下文学习和指令遵循能力。因此,SDFT 在对低延迟、端侧应用至关重要的小模型上失效。为使较小规模的模型能够持续学习,我们首先确定了失效的根本原因。我们表明,能力习得不佳和知识遗忘与后训练模型的分布偏移呈线性相关,揭示了分布内学习的必要性。因此,我们提出合成自蒸馏(SynthSelf),将改写任务交由规模仅略大、同系列的对齐模型完成,以模仿 SDFT 的分布内改写。结合离策略监督和拒绝采样,这种方法能够在不破坏小模型原始分布的前提下,提供其所需的高质量监督。在 4 个能力习得数据集(包括一个混合数据集)和 14 个知识保留基准上的评估表明,SynthSelf 在 3 种小模型规模上均持续优于 6 种方法(包括离策略蒸馏和同策略蒸馏)。大量进一步分析证实,SynthSelf 为有效实现小模型持续学习铺平了道路。

arXiv cs.LG75

通过迭代细化扩展 RNA 基础模型

arXiv:2610.13106v1 公告类型:新论文 摘要:RNA 的生物学功能编码于从局部序列基序到长程结构依赖的多个尺度之中。传统 RNA 基础模型在固定架构内将参数容量与推理深度耦合,为每次推理规定固定的计算图。本文提出 Eco-RNA,这是一个拥有 8000 万参数的 RNA 基础模型,通过学习到的迭代细化机制实现推理深度的可调节性。其共享的、以输入为条件的 Transformer 核心通过反复应用来细化序列表示。迭代次数在训练后选定,可在模型参数固定的情况下控制有效深度。我们在不同循环深度下开展训练,采用掩码重建、状态正则化,以及将更深层重建锚定于浅层参考的单侧损失。在包含 70 项测定的 RNAGym 适应度基准上,将细化次数从一次增加到四次,可使汇总的 Spearman 相关系数绝对值从 0.242 提升至 0.286,且这一增益在迭代次数增加至八次时仍得以保持。在四次迭代下,尽管参数量仅为当前最先进模型 Evo 2 40B 的 1/500,Eco-RNA 在全部三项汇总适应度指标上均超越了该模型。其不同迭代配置推进了受评模型的计算量—性能帕累托前沿。它还在非编码 RNA 家族分类、RNA 二级结构预测及化学探测谱预测任务中优于受评基线模型。这些结果确立了内部推理计算作为 RNA 基础模型有效扩展维度的价值。

arXiv cs.LG71

Can Failed Trajectories Reveal Where Agents Go Wrong? Agentic Failure Attribution Without Step-Level Labels

arXiv:2610.13105v1 Announce Type: new Abstract: Failure attribution for LLM agents aims to identify decisive steps responsible for task failure, which is critical for debugging and improving agentic systems. Existing methods rely on costly LLM prompting, step-level error annotations, or verified successful trajectories that lack failure signals for assistance. In this paper, we study failure attribution from step-unlabeled failure trajectories, where each trajectory is known to have failed but its decisive step is unknown. We propose SOAP (Spectral scOring with Attention-guided Propagation), an effective framework requiring neither step-level training labels nor a separate success-only reference set. SOAP first constructs a spectral reference space from LLM representations and assigns each step a base error score according to its projection behavior. Importantly, later steps may exhibit stronger failure signals because they inherit an earlier mistake, causing independent scorers to favor downstream consequences over the decisive error. To address this downstream contamination, SOAP derives soft step dependencies from attention patterns and propagates downstream error evidence toward its likely upstream sources. Extensive experiments show that SOAP consistently improves attribution performance over competitive baselines.

arXiv cs.LG64

Human and LLM Facilitation in Collaborative Design Meetings: A Computational Behavioral Comparison

arXiv:2610.13082v1 Announce Type: new Abstract: Large language models (LLMs) are increasingly being explored as facilitators of group discussion and collaborative work, but the behavioral differences between LLM-generated and human facilitation remain insufficiently characterized. We compare human Project Manager facilitation turns from scenario-based collaborative design meetings with Llama 3.1 8B Instruct responses across 199 paired contexts from the AMI Meeting Corpus. The analysis covers five confirmatory dimensions: contextual semantic distance, a lexical elaboration proxy, empathyand-hedge lexical density, topic divergence, and phase-vocabulary alignment. An exploratory directiveness measure failed domain validation and is not interpreted. The sampled human Project Manager turns were higher on contextual semantic distance (r = 0.644) and on empathyand-hedge lexical density (r = 0.203). The second contrast is carried mainly by hedge terms, so we read it as lexical hedging rather than psychological empathy. Llama-generated facilitation was higher on topic divergence (r = 0.703), the lexical elaboration proxy (r = 0.288; lengthconfounded), and instruction-conditioned phase-vocabulary alignment (r = 0.382). A secondary no-phase check left the Llama-higher phase contrast in place. The system prompt still requested a facilitation move, so that check is not evidence of spontaneous phase awareness. These differences should not be interpreted as a ranking of facilitation quality. The resulting behavioral profile provides a reference point for later comparisons of LLM facilitators, human facilitators, prompts, models, and collaborative settings.

arXiv cs.LG75

Double-Take:提升双向记忆在长上下文中的容量、绑定能力与选择性

arXiv:2610.13073v1 公告类型:新论文 摘要:能够处理长序列,并不意味着远距离信息仍然可用。随着上下文增长,固定大小的记忆必须保留大量关联、连接不同位置的信息,并保留相关结构,同时避免被所观察到的全部信息淹没。这些是彼此不同的要求:记忆可能存储了更多信息,却没有表征检索所需的关系;也可能传递了这些关系,却未能保留其中重要的关系。因此,挑战不仅在于扩展上下文长度,更在于构建一种在上下文增长时仍然可用的记忆。我们在大分块测试时训练(TTT)中研究这一问题,并发现了两种基本失效模式:单步写入会累积干扰,从而限制容量;自关联写入则无法形成跨位置绑定。我们提出 Double-Take(DT),通过多步非锚定更新和跨位置混合修复这些问题,并引入对变化敏感的双向写入机制,以实现选择性保留。仅提升容量时,端到端召回率仍停留在 $0.10$ 的随机猜测基线;加入绑定机制后,召回率提升至 $0.92$。DT 仅在 $32\,\mathrm{kb}$ 的基因组上下文上训练,便在 $131\,\mathrm{kb}$ 的上下文上达到 $0.927$ 的 AUROC,在接近 $1\,\mathrm{Mb}$ 的上下文上达到 $0.856$,同时实现了从人类到小鼠的零样本迁移。这些结果共同表明,容量、绑定能力和选择性是构建可用长程记忆的三项必要条件。

arXiv cs.LG71

CB-Orchestrator: Adaptive Workflow Optimization for LLM Agents via Contextual Bandits

arXiv:2610.13064v1 Announce Type: new Abstract: Large Language Model (LLM) agents have demonstrated remarkable capabilities in tackling complex tasks through agentic workflows. However, manually designing these workflows is labor-intensive and lacks flexibility to adapt to diverse queries. While automated workflow optimization is a promising direction, existing methods often incur expensive API costs during training-phase feedback collection. Furthermore, they fail to adaptively reuse high-quality workflows, which creates a bottleneck for both performance and cost-effectiveness. To address these limitations, we propose CB-Orchestrator, an adaptive framework that decouples workflow generation from selection. In the first stage, we construct a diverse pool of candidate workflows via evolutionary search. In the second stage, we formulate workflow selection as a Contextual Bandit problem, which enables sample-efficient learning by balancing exploration and exploitation, thereby significantly reducing the feedback required for training. During inference, the model adaptively selects the optimal workflow tailored to each query. Evaluations across five diverse benchmarks demonstrate that CB-Orchestrator consistently outperforms all baselines. Notably, on the MATH dataset, CB-Orchestrator achieves better performance than strong recent baselines while reducing training-phase API token overhead by no less than 40.55% and total end-to-end training (including workflow pool generation) token overhead by no less than 29.73%.

arXiv cs.LG79

迈向更易调参的训练算法:AlgoPerf 的新结果

arXiv:2610.13036v1 公告类型:新论文 摘要:评估深度学习优化器具有挑战性,因为超参数调优、优化器与模型的协同设计等混杂因素往往会掩盖算法的真实性能。我们展示了更新后的 AlgoPerf V1 基准测试结果,在固定硬件上,以达到目标所需的实际运行时间为指标,对一组新的训练算法在九种工作负载上的表现进行了评估。本报告重点介绍该基准测试中自调优规则集的结果,该规则集要求参赛方案在所有工作负载中使用同一套配置。在使用 PyTorch 和 JAX 的 13 个参赛方案中,Schedule-Free AdamW 排名第一。除实际运行时间得分外,我们还分析了迭代步效率与实际运行时间之间的权衡,发现采用矩阵预条件的方法和双循环方法具有良好的样本效率,但需要更大力度的计算优化,才能在运行时间上具备竞争力。最后,我们对超参数显著性和保留情况的分析表明,学习率和动量在不同领域和算法中均表现出高度敏感性,这凸显了开发对超参数调优更具鲁棒性、且能良好泛化到新任务的方法的必要性。

arXiv cs.LG73

用于轨迹外推的目标约束池化

arXiv:2610.13034v1 公告类型:新提交 摘要:根据部分目标轨迹进行轨迹外推是欠定问题,但已完成的相关轨迹可为尚未确定的后续轨迹提供信息。我们提出目标约束池化(TCP),将有限时域的终点与后续轨迹形状分离,并根据两种互补的源信号估计该终点。均匀池化捕捉最终响应水平,而匹配增量池化则在对齐源状态与目标状态后捕捉剩余变化。TCP 选择具有代表性的源轨迹,将两种终点估计投影到与目标一致的区域,将其转换为对无约束预测的修正,并通过强度校准和退化感知加权控制其影响。在来自五个领域、涵盖两个预测方向的 25,245 个目标上,TCP 将全部十种仅基于目标的预测器的平均 RMSE 降低了 35.8% 至 42.1%,并在预测器、数据集和方向的全部 100 组比较中取得改进。两种终点信号的联合使用始终优于单独使用任一种信号;同时,代表性选择仅使用完整供体池 1/12.7 的供体输入和 1/138.8 的匹配距离计算量,与完整供体池的性能差距仍保持在 0.64% 至 1.02% 之间。

arXiv cs.LG74

用于可解释算子学习的低秩 Kolmogorov-Arnold 生成元流

arXiv:2610.13028v1 公告类型:新论文 摘要:从数据中学习物理定律的神经算子应当能够泛化到训练分布之外,并以闭式形式呈现所学定律。Kolmogorov-Arnold 神经算子通过学习覆盖所有位置和频率的稠密伪微分符号,在位置依赖动力学中实现了这两个目标,但其计算成本随分辨率的平方增长,且闭式表达式必须手动提取。这种稠密符号并无必要:有限阶变系数微分算子的符号,其分离秩以系数函数的数量为上界,与分辨率无关;而该网络在其自身基准测试中学到的符号的秩仅为二。我们提出 Kolmogorov-Arnold flow(KAFlow),将物理算子的生成元学习为由一元 Kolmogorov-Arnold 函数构成的低秩符号,通过固定这些函数的规范,使每一项都有唯一的归属,使用精确指数积分器对动力学进行积分,并自动提炼其闭式表达式。在 Kolmogorov-Arnold 神经算子的合成基准测试中,KAFlow 将分布外损失降至原来的 1/3.4 至 1/11,无需符号模板即可将每个系数的恢复误差控制在 10^-4 以内,达到数据的精度极限。在长时域量子动力学中,它将量子态的不保真度最多降低 6 个数量级;即使仅使用位置和动量测量结果进行训练,也能将其降至原来的 1/30。KAFlow 学习变阶分数阶算子的精度还比现有算子网络高 15 倍,并能发现稀疏回归无法识别的非局域、分数阶和非多项式色散关系。

arXiv cs.LG73

轨迹感知策略优化:弥合长时域 LLM 强化学习中的状态分布差距

arXiv:2610.13020v1 公告类型:新论文 摘要:强化学习(RL)已成为增强 LLM 推理能力的主流范式。然而,长时域自回归轨迹采样带来了一项根本性局限:局部策略失配会沿轨迹累积,造成严重的状态分布偏移。现有 LLM RL 方法(如 GRPO 和 GSPO)在词元或序列层面控制策略失配,却忽视了这种由轨迹引起的偏移,导致训练不稳定,并随着时域延长而最终崩溃。为弥合状态分布差距,我们提出 Trajectory-Aware Policy Optimization(TAPO),这一框架具有坚实的理论基础,从离策略策略梯度出发,在理论上推导出状态分布校正项,并将其纳入 RL 目标函数。TAPO 进一步采用几何平均估计器来降低方差,同时结合长度感知的动态裁剪,防止整条轨迹上的更新偏离过大。大量实验表明,TAPO 在八项竞赛级基准测试中显著优于现有 LLM RL 算法,相比 GRPO,在数学推理和算法编程任务上分别取得最高 48% 和 12% 的提升,同时大幅增强了训练稳定性。

arXiv cs.LG75

广义近端策略优化:从散度视角统一 LLM 强化学习算法

arXiv:2610.13015v1 公告类型:新提交 摘要:强化学习已成为增强 LLM 推理能力的基础范式。然而,主流方法近端策略优化(PPO)依赖于一种几何退化的裁剪代理目标,无法自适应地调整对策略偏移的阻力,导致过早裁剪和探索能力下降。后续变体通过零散的启发式修改缓解了这些问题。我们提出广义近端策略优化(GPPO),这是一个原理清晰的统一框架,用于从散度视角理解和设计 LLM 强化学习算法。我们建立了具有性能单调改进保证的广义 $f$-散度信赖域理论,并表明不同的 $f$-散度会诱导不同的几何结构和策略更新动力学。我们进一步引入了新颖的估计器拓扑结构,相较于传统的单样本蒙特卡洛估计,能够提供更忠实的散度估计。在这一框架下,PPO 及 DAPO、DPPO 和 RIPO 等近期变体均可视为特例,由此建立了具有理论依据的 {散度 $\times$ 估计器} 二维设计空间。在八个竞赛级数学和编程基准上的广泛实验表明,这两个维度都至关重要。表现最佳的 GPPO 实例在数学和编程任务上分别比标准 GRPO 提升最高 50% 和 14%,达到当前最先进的性能水平。

arXiv cs.LG58

Evidential Multimodal Transformers with Dempster-Shafer Fusion for Emotion Recognition

arXiv:2610.13013v1 Announce Type: new Abstract: Softmax-based multimodal emotion recognition methods often assign high confidence even when sensor data are corrupted or contradictory, for example when facial expressions suggest positive valence while vocal prosody indicates negative arousal. We present an Evidential Multimodal Transformer with Dempster-Shafer Fusion (EMT-DS) that addresses this problem through formal uncertainty quantification. EMT-DS models predictions as Subjective Opinions (Dirichlet distributions), separating epistemic uncertainty (lack of evidence) from aleatoric uncertainty (sensor noise). Modality-specific transformer encoders produce belief mass distributions that are fused with Dempster's Rule of Combination, which normalizes the joint belief mass by a conflict coefficient 1/(1-c), so that contradictory evidence is explicitly quantified as conflict and the fused prediction becomes diffuse rather than being forced into an unjustified consensus. Under subject-independent and speaker-independent evaluation on four benchmarks, EMT-DS achieves 84.3% accuracy on IEMOCAP versus 81.7% for the best baseline (p < 0.01), 82.4% on DEAP under leave-one-subject-out evaluation versus 78.1%, and 96.8% on the highly imbalanced KEMDy20, and it improves accuracy on CMU-MOSEI by 2.4 points over the strongest evidential baseline. In controlled ablation, it reduces Expected Calibration Error from 0.153 (softmax transformer) to 0.042. It maintains 71.2% accuracy under severe Gaussian noise on audio features (sigma = 1.0), where attention baselines drop to 45.3%, and degrades gracefully when entire modalities are absent. The fusion step has O(MK^2) complexity, compared with O(M^2 d^2) for cross-modal attention. The learned conflict metric also flags likely errors: reviewing 20% of samples captures 80% of misclassifications, which supports selective prediction in clinical and safety-critical systems.

arXiv cs.LG67

Gaussian-Width Bounds for Offline-to-Online Linear Bandits

arXiv:2610.12990v1 Announce Type: new Abstract: We study how offline data can help online learning in stochastic linear bandits. We focus on partially covered offline data, where only $k$ of the $d$ feature directions are poorly covered. Existing analyses of algorithms based on phased elimination and optimism give regret bounds of order $\widetilde O(\sqrt{dkT})$ in this regime. We show that this dependence arises from loose control of the reward prediction error, and obtain tighter bounds using concentration of Lipschitz functions of Gaussian vectors. Based on this analysis, we develop a phased elimination algorithm with regret $\widetilde O(k\sqrt T)$. For the unit Euclidean ball, we prove a matching regret lower bound under partial coverage, as well as matching upper and lower bounds under uniform offline coverage. We also study pure exploration with offline data, providing algorithms and matching upper and lower bounds on simple regret for fixed-design algorithms and also for adaptive algorithms under partial and uniform coverage of offline data.

arXiv cs.LG80

结合中期天气预报的扩散式降水生成

arXiv:2610.12976v1 公告类型:新论文 摘要:准确的日降水预报对天气敏感型决策至关重要。尽管近期天气基础模型(WFMs)取得了显著进展,但它们使用再分析数据进行训练,而这类数据存在随区域和降水强度而变化的偏差。由于基于观测目标重新训练 WFMs 的成本高昂,我们研究了 Dynamics-Precipitation Decoupling,这是一种即插即用框架,利用单独训练的生成器,将冻结天气模型输出的非降水预报映射为与观测一致的日降水。其关键挑战在于捕捉异质性降雨分布,并使日累计降水与随时空演变的大气信号相匹配。为应对这些挑战,我们以大气状态为条件,利用卫星与雨量计融合降水估计训练扩散生成器。多尺度循环编码器概括大气演变过程,交叉注意力则在去噪过程中对齐大气特征与降水特征。针对 Pangu-Weather、GraphCast 和 HRES 等多种骨干模型开展的全球及区域预报广泛评估表明,该方法提高了日降水预报准确性和事件检测能力,并能更好地表征局地累计降水分布。全球集合预报评估显示,在所有报告的预报时效下,该方法的 Brier 评分均低于 ENS 和从头训练的 GenCast,且在大多数比较中具有更低的 CRPS。值得注意的是,一项利用中国 2,288 个站点开展的强降雨个例研究,通过与地面观测对比,进一步证实了该方法在日累计降水分布表征和事件检测方面的改进。

arXiv cs.LG65

DataCupid: Stable Matching Repair for Training Data Acquisition with Delayed Feedback

arXiv:2610.12960v1 Announce Type: new Abstract: Matching repair after utility changes can reuse valid observations, but new evaluations must respect matching capacity and may return out of order. We study announced invalidations with bounded, possibly reward-dependent reporting delays and pending evaluations that do not occupy future matching capacity. Residualload repair (RLR) combines sound outcome certificates with batched matching schedules for unequal sample deficits. Its complete dyadic quota schedule admits a sharp issue-round bound with coefficient cb < 1.607 relative to the terminal residual endpoint load, together with separate delay and evidence-lifetime bounds. RLR-P monitors complete issue-order prefixes without changing the sampling plan before acceptance. With a deterministic pure oracle, a single-repair coupling from the same drained initial state guarantees no increase in certification time, issued observations, or time to a reusable checkpoint. We also derive an information-load lower bound. On fixed priority-star markets with unknown Bernoulli gaps, PrioritySearch attains the history-adjusted first-order constant for expected certification time as {\delta} {\sim} 0, under hypothesis-independent linear history scaling and sublogarithmic reporting delays. Paired simulations and measured-feedback replay quantify the trade-offs: in the replay, RLR-P has lower mean capped update-certification time than a monitored width-based baseline, at higher oracle CPU cost.

arXiv cs.LG70

Controlled-Minimax: Stabilizing Adversarial Optimization via Control Theory

arXiv:2610.12915v1 Announce Type: new Abstract: Minimax optimization is fundamental to adversarial learning yet suffers from severe stability challenges, often manifesting as persistent limit cycles or divergence. This paper investigates these instabilities through the lens of continuous-time dynamical systems. By modeling training dynamics as a second-order Ordinary Differential Equation (ODE), we rigorously identify the root cause of non-convergence: the problematic eigenstructure of the loss function's Hessian near equilibrium points. To address this, we introduce Controlled-Minimax, a novel stabilization framework that reformulates the optimization trajectory as a controllable physical system. Unlike heuristic regularization techniques, our approach derives a specific feedback controller that provides theoretical guarantees of local exponential stability, regardless of the underlying Hessian spectrum. Consequently, our method stabilizes general minimax problems where traditional gradient descent-ascent methods fail. Extensive experiments on pathological benchmarks including the Dirac-GAN, 2D Gaussian mixtures, and Swiss Roll datasets, as well as high-resolution image synthesis with StyleGAN2-ada, demonstrate that Controlled-Minimax effectively suppresses oscillatory behavior while ensuring fast and robust convergence.

arXiv cs.LG71

GIST: Generator-state-Informed Supervision for Tabular Foundation Models

arXiv:2610.12904v1 Announce Type: new Abstract: Tabular foundation models (TFMs) based on prior-data fitted networks typically use sampled hard query labels, leaving avoidable sampling variability in pretraining. A key observation is that intermediate generator states can reveal label uncertainty beyond a sampled outcome. We propose Generator-state-Informed Supervision for TFMs (GIST), which uses these states to average query losses over conditional label distributions. GIST jointly designs task generation and query supervision, using entropy-calibrated encodings to obtain tractable, input-consistent targets on supported label-dependent branches. We show that this averaging preserves the population objective and yields no larger gradient covariance at fixed parameters within each generating distribution. Under matched pretraining schedules, GIST improves mean rank across three architectures on both TabArena and TALENT without changing architecture or inference. Across these benchmarks and architectures, GIST wins 62.1% of paired dataset-level comparisons against the corresponding hard-label controls.

arXiv cs.LG65

Personalized Federated Direct Preference Optimization via LoRA Adapters

arXiv:2610.12885v1 Announce Type: new Abstract: Federated Direct Preference Optimization (FedDPO) enables privacy-preserving alignment of language models across distributed agents without centralizing sensitive preference data. However, existing FedDPO analysis assumes all clients converge to a single shared policy, which is fundamentally at odds with the heterogeneous preference distributions present in practice. We propose Personalized FedDPO (pFedDPO), in which each client maintains a small private LoRA adapter atop a shared global base model. In pFedDPO, the base model is synchronized via FedAvg each round, while LoRA adapters remain local and capture client-specific preference patterns. We prove that this decomposition replaces the gradient diversity $\kappa^2$ in the FedDPO bound with a residual heterogeneity $\kappa^2_{\mathrm{eff}}$, which is no larger than $\kappa^2$ when adapters absorb client-specific directions, at the cost of an additive adapter-bias term that vanishes as adapters converge. Experiments on preference datasets show that pFedDPO consistently attains a lower gradient norm than vanilla FedDPO while remaining competitive in training loss and per-client reward accuracy.

arXiv cs.LG64

CoGameReg: Cooperative-Game-Based Collaborative Semi-Supervised Regression for Seismic Reservoir Property Prediction

arXiv:2610.12884v1 Announce Type: new Abstract: Accurate estimation of reservoir properties from seismic data remains challenging when well-log labels are sparse but three-dimensional seismic observations are abundant. We present CoGameReg, a cooperative-game-based semi-supervised regression framework that uses contribution-weighted predictions of heterogeneous regressors to construct shared pseudo-labels. This design makes each regressor's influence explicit in the learning targets generated for unlabeled seismic samples under sparse well control. The Tri-Game Shapley mechanism combines a rank-MSE performance game with a pairwise consistency game to estimate nonnegative regressor weights from labeled training data. These weights remain fixed during collaborative training and are reused for final prediction fusion. Candidate pseudo-labels are screened using a common local k-nearest-neighbor error-reduction criterion before retraining. We evaluate CoGameReg on 10 public regression datasets, a synthetic three-dimensional P-wave velocity prediction task, and a field water-saturation application. The benchmark advantage is most pronounced at the 2.5% labeled-data ratio and decreases as supervision increases. CoGameReg obtains the best blind-well metrics in the synthetic experiment. In the field application, predictions clipped to 0-100% yield a pooled coefficient of determination of 0.5942, a root mean square error of 5.7441 percentage points, and a Pearson correlation of 0.8265 across three blind wells. CoGameReg has the highest Pearson correlation among the four field methods. The results support the use of CoGameReg to integrate sparse well labels with dense seismic attributes, with performance varying across metrics and well locations.

arXiv cs.LG75

HART:混合注意力与 RWKV 的时间序列基础模型

arXiv:2610.12883v1 公告类型:新发布 摘要:非自回归多步预测(NAF)使时间序列基础模型能够通过单次前向传播预测整个未来时域。尽管 RWKV-7 具有强大的序列建模能力,但在我们的 NAF 实验中,它在长预测时域下出现了性能退化。在这一设置中,未来的数据片段被表示为缺失输入,因此循环状态需要在没有新观测值的情况下沿预测时域演化。我们对 RWKV-7 广义 delta 规则的分析揭示了一种机制,我们称之为占位符空转衰减(PID):当占位符驱动的写入无法提供充分补偿时,反复的状态转移可能会削弱历史信息。逐层分析表明,大多数层的状态注入范数有所降低,而历史区间与预测区间的转移矩阵范数仍然相近,这与 PID 机制一致。基于这一分析,我们提出了 HART,这是一种混合注意力与 RWKV 的时间序列基础模型,交替使用注意力模块和 RWKV 模块,将对历史词元的直接访问与循环时序建模相结合。在 GIFT-Eval 和 TIME-Benchmark 上的受控实验表明,HART 在中长预测时域下优于纯 RWKV,同时在短预测时域下保持相当的性能,并且其综合性能优于规模相近的纯 Transformer。HART 仅有 3600 万个参数,在 GIFT-Eval 上实现了与 Toto-2.0-1B 相当的长时域预测 MASE(0.752 对 0.754),而参数量约为后者的 1/28。

arXiv cs.LG72

状态依赖重尾噪声下的鲁棒 Langevin 采样及其在 LLM 微调中的应用

arXiv:2610.12865v1 公告类型:新提交 摘要:现有针对随机梯度 Langevin 蒙特卡洛(SG-LMC)非对数凹采样的收敛性保证通常假设梯度全局 Lipschitz 连续且噪声方差一致有界。即使在基本的采样问题中,这两个假设也可能不成立,放宽它们能更好地反映大型语言模型(LLM)的贝叶斯微调等实际应用。我们通过广义光滑性以及所提出的具有有限 $p$ 阶矩($1<p\leq 2$)的状态依赖重尾预言机,放宽了这两个假设。该预言机允许在 $p<2$ 时方差无穷大,并涵盖 Blum-Gladyshev(BG-0)条件。我们分别及联合研究这两种放宽情形,并提出三种方法。在梯度全局 Lipschitz 连续的条件下,鲁棒 LMC(R-LMC)通过几何中位数中心化与残差裁剪来处理所提出的预言机。在使用精确梯度时,自适应 LMC(A-LMC)在广义光滑性条件下控制离散化误差。鲁棒自适应 LMC(RA-LMC)通过鲁棒估计和噪声感知步长,同时应对这两种放宽情形。对于这三种方法,我们均在不假设耗散性或函数不等式的情况下,建立了以 Fisher 信息(FI)衡量的非渐近收敛性保证。合成实验展示了 FI 收敛性以及各项机制的益处。针对 LLM 和视觉语言模型的贝叶斯微调实验支持了我们的预言机模型,并表明 RA-LMC 在准确率和不确定性量化方面达到或超越了当前最先进的基线方法。

arXiv cs.LG71

基于不动点流匹配的强化学习

arXiv:2610.12848v1 公告类型:新投稿 摘要:在最大熵强化学习和行为约束强化学习(RL)中,最优策略由 Boltzmann 分布给出。从该分布中生成样本可能颇具挑战性,因为它通常复杂且具有多模态特征。尽管通过流匹配训练的流策略为表示此类分布提供了强大的框架,但传统上它们依赖训练数据,而在线 RL 中通常无法获得这些数据。为解决这一局限,我们提出不动点流匹配(FFM),一种用于在 RL 中训练和微调流策略的无数据方法。FFM 利用目标流恒等式,将演员更新表述为不动点迭代,并采用流匹配式回归目标,其中评论家对动作的梯度作为回归目标值。我们证明,最优策略是该迭代的一个不动点。通过广泛的实验,我们表明,FFM 能够在离线到在线的设置下有效微调预训练流策略,并能在在线 RL 中从零开始高效、准确地训练流策略。这些结果确立了不动点流匹配作为一种具有理论依据的方法的地位,使其无需训练数据即可学习表达能力强的基于流的策略。

arXiv cs.LG74

当生成模型充当模拟器时

arXiv:2610.12847v1 公告类型:新论文 摘要:模拟器将环境概括为用户可以读取和修改的状态。生成模型无需显式指定即可学习世界的表示,但并未提供编辑该世界状态的内置方式。我们通过 $\textit{可编辑性}$ 来研究生成模型在何种条件下能够充当模拟器:改变模型的潜在表示,是否能产生与编辑模拟器状态后所得到的相同的下一步输出。简而言之,我们的目标是在生成模型的潜在状态与模拟器状态之间找到一种映射,其中模型仅基于观测进行训练,而映射则在之后学习。我们测试了两类映射,两者均从与模拟器状态配对的观测中学习。一类方法训练探针,从模型中解码模拟器状态,然后沿探针所指示的方向进行引导。另一类方法借助逆向映射直接学习这种映射,即使在引导失败的情况下也可能成功。我们在两个截然不同的环境中研究这两类方法:离散棋盘游戏 Othello,以及我们自行构建的 Rayworld——一个通过局部渲染来观测运动圆盘的连续世界,更接近物理场景和视频。我们发现,哪种映射能够成功高度依赖于环境,尽管在所有变体中都可以解码出状态。在 Othello 中,随着游戏被简化,可编辑性平滑下降,即使细微的规则变化也会产生重大影响。在 Rayworld 中,探针无法沿模拟器的连续状态进行引导,但逆向映射仍能根据该状态设定模型。因此,模型可以由一个它可能从未学会的模拟器来驱动。

arXiv cs.LG75

Cross-Behavior Policy Optimization:通过共享回归目标统一学习离线示范与同策略、异策略采样轨迹

arXiv:2610.12830v1 公告类型:新提交 摘要:推理后训练利用离线示范和策略生成的采样轨迹,但传统的先监督微调(SFT)、后强化学习(RL)的方法将两者的目标分离,限制了对已收集回答的持续复用。我们提出 \textbf{Cross-Behavior Policy Optimization}(CBPO),通过基于固定参考的目标,使这些数据源能够兼容同一个回归损失。由于该目标不涉及生成策略,正向和负向离线补全、参考回答,以及当前和历史采样轨迹可以共享统一的带奖励标签的记录格式和回放缓冲区。当新回答进入缓冲区时,优化过程仍保持相同的回归目标,无需协调独立的 SFT 和 RL 目标,也不需要存储行为策略概率、进行重要性加权或学习评价器。通过限制可用数据源,可以得到离线、同策略和异策略变体,而 CBPO-Unified 则将它们结合起来。在两种模型规模和四个数学推理基准上,CBPO 在不同基准和数据设置下均取得领先的 \texttt{avg@32} 和 \texttt{pass@32} 表现。在 Qwen3-1.7B 上,相较于 SFT DAPO,CBPO-Unified 在各基准上平均的 \texttt{avg@32} 和 \texttt{pass@32} 分别提高了 2.33 和 4.44 个百分点。在回放消融实验中,相较于不保留历史轨迹的配置,保留历史采样轨迹使这两项指标分别提高了 1.87 和 3.72 个百分点。

arXiv cs.LG76

TreeHCA:面向多轮 LLM 智能体的后见引导探索与信用分配

arXiv:2610.12828v1 公告类型:新论文 摘要:结果监督的组相对策略优化(GRPO)为 LLM 智能体轨迹中的每个词元分配相同的优势值,未能区分有益和无益的决策。后见信用分配(HCA)可以细化这一反馈,但其估计依赖于轨迹中在关键历史状态下包含具有信息价值的备选路径。我们提出 TreeHCA,将后见引导探索与后见信用分配相结合。TreeHCA 利用参考答案似然的变化,重新访问中间历史状态并采样替代性的后续轨迹,从而构建由推理和工具交互组成的树。由于这种探索改变了所采样后续轨迹的分布,我们推导出相应的逆后见重要性校正,其中包括一个精确的有限样本池恒等式。我们在归一化的树回传中使用这一结构,将终局结果传播为轮次级优势值。这种共享的后见结构是 TreeHCA 的核心设计原则,将具有信息价值的探索与后见引导的信用分配联系起来。在七个问答基准和三个 Qwen 骨干模型上,TreeHCA 的总体精确匹配率均优于所有基线,验证了将探索与信用分配配对用于多轮 LLM 智能体训练的有效性。我们的代码已发布于 github.com/TheBlackCat22/TreeHCA。

arXiv cs.LG67

Inverse Synthesis of Shock Time Series from Shock Response Spectra Using Ensembles of Conditional Variational Autoencoders

arXiv:2610.12820v1 Announce Type: new Abstract: The shock response spectrum (SRS) characterizes the peak responses of single-degree-of-freedom systems to transient acceleration. Because the mapping from an acceleration time series (waveform) to its SRS is many-to-one, inverse synthesis seeks candidate waveforms consistent with a prescribed spectrum. We extend an earlier conditional variational autoencoder (CVAE) framework with a recurrent decoder that predicts multiscale wavelet coefficients and reconstructs waveforms through an inverse discrete wavelet transform. Individual models exhibit limited variability under latent sampling, motivating an ensemble of independently trained models spanning different mother wavelets and decomposition levels. Each member produces a separate candidate waveform for the same target SRS. Evaluation on four held-out benchmark datasets shows lower mean and median spectral errors for the illustrated wavelet model than for the earlier direct time-domain CVAE and a sum-of-decayed-sinusoids (SDS) baseline. A conditioning study examines frequency-grid and encoding choices. Comparisons with published inverse-SRS methods demonstrate superior performance across four holdout benchmark datasets (two operational shock datasets, one earthquake dataset, and one synthetic shock dataset) and competitive performance on synthetic targets reported by Brake. Single-model inference requires approximately 0.1-0.6 ms per waveform during batched inference compared with 5-8 s for SDS, excluding offline training. This low inference cost enables rapid generation of distinct SRS-compatible candidate waveforms across the ensemble, although their physical suitability requires assessment beyond spectral agreement.

arXiv cs.LG80

为大规模推理和训练工作负载部署 RDMA

arXiv:2610.12815v1 公告类型:新论文 摘要:RDMA 可以降低分布式机器学习系统中的通信开销,但要在生产环境中实现这些收益,仅仅替换 TCP 还不够。大小可变的消息、并发连接、突发性的多对一流量以及服务故障带来了仅靠硬件卸载无法解决的挑战。本文介绍了我们在 ByteDance 基于参数服务器的推理和训练系统中部署 ByteExpress 的经验。ByteExpress 是构建于硬件 RDMA 之上的软件通信层。我们总结了此次部署的三点经验:高效的 RDMA 需要跨应用层和传输层进行缓冲区管理;可扩展的通信需要协调管理连接、接收容量和尚未完成传输的数据;服务可用性需要传输层之上的恢复机制。ByteExpress 结合了支持 RDMA 的内存分配器、零拷贝传输、QP 多路复用、基于接收信用的流量控制、软件发送窗口以及 RPC 辅助恢复机制。在生产环境的推理任务中,与内核 TCP 相比,ByteExpress 将模型参数拉取的平均延迟和 P99 延迟分别降低了 35.7% 和 43.7%,并将 CPU 总使用量降低了 10.69%。在训练任务中,与 TCP 相比,ByteExpress 将模型参数推送的 P99 延迟最多降低了 74.0%,且运行期间未观察到 PFC 暂停。

arXiv cs.LG68

The Expressive Power of Random-Feature MPNNs Across Graph Sizes

arXiv:2610.12809v1 Announce Type: new Abstract: Continuous random node features increase the expressivity of message-passing neural networks (MPNNs), but existing theoretical analyses are not fine-grained enough to capture their limitations, particularly in approximating functions across graph sizes or under a fixed number of random bits. We develop a population-level framework for size-uniform approximation. For MPNNs with a graph-size-independent Lipschitz bound in their initial features, including standard mean, max, and order-normalized sum architectures, we show that continuous random features can be quantized to finitely many colors with arbitrarily small uniform error in expected output. This reduction lets us transfer limitations on finite-color distinguishability to limitations on size-uniform approximation with continuous random features. We use this framework to prove that these MPNNs are not size-uniformly universal, analyze the expressivity gains from colors and independent trials, and establish incomparability between random-color MPNNs and higher-order Weisfeiler-Leman methods. We also give finite-color MPNN constructions for tree subgraph isomorphism via color coding and a local MST task.

arXiv cs.LG75

元反思进化有效提升生产环境 AI 系统性能

arXiv:2610.12794v1 公告类型:新论文 摘要:大语言模型(LLM)引导的进化优化已成功改进提示词、程序,乃至整个 AI 系统。大多数优化方法从搜索前沿选择一个候选方案,然后仅依据其失败情况,通过单次 LLM 调用对其进行变异。这种对搜索前沿的聚焦可能使生产环境 AI 系统的优化陷入停滞,因为孤立的失败很难揭示复杂的隐性交互。我们提出 CoCoEvolve(Cortex Code Evolve),这是一种围绕我们所称的“元反思”构建的进化优化器:智能体式变异器(Cortex Code)选择性地从整个进化历史中检索证据,并探查实际运行的系统,然后验证并提出变异方案。我们将 CoCoEvolve 应用于三类生产环境 AI 系统的优化:数据智能体、LLM 驱动的数据流水线和数据工程工作流。在 DABStep 上,CoCoEvolve 位居公开排行榜榜首,在留出的高难度测试集上达到 92.8% 的准确率,而在相同条件下,Meta-Harness 和 GEPA 的准确率分别为 78.4% 和 30.4%。在六项 LLM 数据流水线工作负载中,它均优于所有接受评估的基线。在 data-eng-bench 上,CoCoEvolve 将测试集求解率从 59.1% 提升至 77.4%,超过了最佳基线的 73.6%。它在 HotpotQA 上的准确率也比 GEPA 高出 11 个百分点,同时在成熟的数学和系统基准测试上仍可与专用优化器保持竞争力。同等条件下的消融实验确认,CoCoEvolve 的每个组件都对这些提升有所贡献,体现了元反思在优化生产环境 AI 系统方面的价值。

arXiv cs.LG74

无坐标自动微分

arXiv:2610.12776v1 公告类型:新论文 摘要:自动微分(AD)是现代机器学习的基石。由于所有 AD 框架都隐式地固定了欧几里得空间的一组基来定义梯度和雅可比矩阵,因此它们原生处理的数学对象都是由一组固定坐标构成的数组。这使得旋转、体积形式和曲率等对象难以表达,而这些对象会出现在例如人工智能应用于科学的场景中。为特定问题定制的变通方案缺乏可复用性。我们证明:(i)单个与基无关的导数算子可以表示多种几何微分运算;(ii)该算子可以完全用现有的 AD 原语来表达。这种可复用的变换可以与 JAX 的批处理、编译和硬件加速功能组合使用。几何结构以代数数据的形式引入。我们通过实验展示了该变换在不同流形算子之间的复用、在时空逆问题处理流程中的组合,以及在不同旋转子子群上的优化中的应用。