全部动态

本页 40 条
10月9日 周五
arXiv cs.LG78

面向分布内数据获取的共形数据污染检验

arXiv:2507.13835v2 公告类型:replace-cross 摘要:在许多机器学习任务中,高质量数据的数量受限于数据所有者本地可获得的数据。通过与外部数据代理进行交易或共享,可以扩展高质量数据集。然而,外部数据可能受到污染,或引入不理想的样本多样性,从而降低个性化机器学习任务的性能,例如罕见疾病诊断或推荐系统。因此,数据购买者在获取数据之前需要质量保证。以往研究主要依赖关于不同代理所持数据的分布假设,将质量检查 relegated to 事后步骤,并涉及成本高昂的数据估值流程。我们提出一种无分布假设且具备污染感知能力的数据获取框架,该框架仅通过检查少量数据,便可识别出对模型个性化最有价值的外部数据代理。为实现这一目标,我们引入了新颖的双样本检验程序,在完整获取数据之前执行,并以关于共形异常值检测的严谨理论基础为依据,用于确定某个代理的数据是否超过污染阈值。所提出的检验称为共形数据污染检验,在任意污染水平下均保持有效;此外,新颖的 Storey-type 检验可通过 Benjamini-Hochberg 程序,在有限样本条件下实现可证明的错误发现率控制。在多种协作学习场景中的实证评估表明,我们的方法具有稳健性和有效性。总体而言,共形数据污染检验是一种通用的数据聚合程序,能够提供具有严格统计保证的数据质量保障。

arXiv cs.LG77

异构模态无监督域适应:通过潜在空间桥接

arXiv:2506.15971v2 公告类型:替换交叉版本 摘要:无监督域适应(UDA)能够有效弥合有标注源域与无标注目标域之间的域差异,但其前提是假设两个域具有相同的模态。异构域适应(HDA)则处理跨域的不同特征空间,但要求目标域样本带有标注,或存在连接源域与目标域的配对数据。当有标注源域和完全无标注目标域分别具有完全不同的模态(例如二维图像和三维点云)时,上述两种方法均不适用。为解决这一局限,我们提出了一种新的设置,称为异构模态无监督域适应(HMUDA)。该设置通过一个无标注桥接域跨模态迁移知识,其中包含来自两种模态的配对观测,且其分布可能偏离源域和目标域的分布。为在HMUDA设置下进行学习,我们提出了Latent Space Bridging(LSB),这是一种双分支框架:通过对桥接域配对样本施加特征一致性损失来缩小模态差异,并通过类别质心对齐损失来减小源域与目标域之间的差异。在涵盖二维到三维和三维到二维迁移的八种基准设置上进行的大量实验表明,LSB达到了最先进的性能。

arXiv cs.LG71

Delphos:用于辅助离散选择模型设定的强化学习框架

arXiv:2506.06410v4 公告类型:替换交叉版本 摘要:我们介绍了 Delphos,这是一个用于辅助离散选择模型设定过程的深度强化学习框架。Delphos 旨在通过提供自动化、数据驱动的模型设定建议来支持建模者,从而减少开发和完善效用函数所需的工作量。Delphos 将模型设定视为一个序贯决策问题,其设计灵感来自人类离散选择建模者通过一系列有依据的设定决策迭代构建模型的方式。在这一设定下,智能体通过选择一系列建模操作来学习设定候选模型,例如添加备选项特定常数、同时纳入通用和备选项特定的偏好参数、对属性应用非线性变换,以及加入与协变量的交互项。每个生成的候选模型都会依据由建模者定义的奖励函数进行估计和评估,该奖励函数既可以反映统计模型拟合优度,也可以体现行为预期。具体而言,Delphos 使用 Deep Q-Network 来学习各项模型设定决策如何影响最终生成模型的质量,以及哪些建模决策序列倾向于产生表现良好的候选模型。我们使用不同的奖励函数,在模拟数据集和实证数据集上对 Delphos 进行了评估。在模拟案例中,学习曲线、Q 值模式和性能指标表明,Delphos 只探索了可行建模空间中的很小一部分,就学会了有效的模型设定策略。我们进一步将该框架应用于两个实证数据集,以对其进行基准测试并展示其实际用途。这些实验表明,Delphos 能够生成具有竞争力且符合行为规律的模型,并凸显了这一自适应学习型框架辅助模型设定过程的潜力。

arXiv cs.LG61

Equilibrium Distribution for t-Distributed Stochastic Neighbor Embedding with Generalized Kernels

arXiv:2505.24311v3 Announce Type: replace-cross Abstract: We study the large-sample variational problem for t-distributed stochastic neighbor embedding with a class of input and output kernels. The input law has compact support and a density continuous on that support. An entropy equation determines the scale parameter in the input kernel, and we prove that this parameter exists and is unique at interior points of positive density. We then give sufficient conditions for solutions to exist and be uniformly bounded on the entire support. Under these conditions and a decay assumption on the output kernel, the discrete optimal values converge to a continuum minimum. Empirical measures of approximate minimizers are tight after translation; every subsequential limit is a compactly supported minimizer satisfying the equilibrium equation. The admissible output kernels include Gaussian kernels and, in output dimension two, the Cauchy kernel. Numerical examples compare the two-dimensional representations obtained with different kernels.

arXiv cs.LG63

A Survey on Archetypal Analysis

arXiv:2504.12392v3 Announce Type: replace-cross Abstract: Archetypal analysis (AA) was originally proposed in 1994 by Adele Cutler and Leo Breiman as a computational procedure for extracting distinct aspects, so-called archetypes, from observations, with each observational record approximated as a mixture (i.e., convex combination) of these archetypes. AA thereby provides straightforward, interpretable, and explainable representations for feature extraction and dimensionality reduction, facilitating the understanding of the structure of high-dimensional data and enabling wide applications across the sciences. However, AA also faces challenges, particularly as the associated optimization problem is nonconvex. This is the first survey that provides researchers and data mining practitioners with an overview of the methodologies and opportunities that AA offers, surveying the many applications of AA across disparate fields of science, as well as best practices for modeling data with AA and its limitations. The survey concludes by explaining crucial future research directions concerning AA.

arXiv cs.LG65

Networks with Finite VC Dimension: Pro and Contra

arXiv:2502.02679v3 Announce Type: replace-cross Abstract: Approximation and learning of classifiers of large data sets by neural networks in terms of high-dimensional geometry and statistical learning theory are investigated. The influence of the VC dimension of sets of input-output functions of networks on approximation capabilities is compared with its influence on consistency in learning from samples of data. It is shown that, whereas finite VC dimension is desirable for uniform convergence of empirical errors, it may not be desirable for approximation of functions drawn from a probability distribution modeling the likelihood that they occur in a given type of application. Based on the concentration-of-measure properties of high dimensional geometry, it is proven that both errors in approximation and empirical errors behave almost deterministically for networks implementing sets of input-output functions with finite VC dimensions in processing large data sets. Practical limitations of the universal approximation property, the trade-offs between the accuracy of approximation and consistency in learning from data, and the influence of depth of networks with ReLU units on their accuracy and consistency are discussed.

arXiv cs.LG71

Foundations of Large Language Models

arXiv:2501.09223v4 Announce Type: replace-cross Abstract: This is a book about large language models. As indicated by the title, it primarily focuses on foundational concepts rather than comprehensive coverage of all cutting-edge technologies. The book is structured into six main chapters, each exploring a key area: pre-training, generative models, prompting, alignment, inference, and reasoning. It is intended for college students, professionals, and practitioners in natural language processing and related fields, and can serve as a reference for anyone interested in large language models.

arXiv cs.LG

Graphons of Line Graphs

arXiv:2409.01656v4 Announce Type: replace-cross Abstract: We consider the problem of estimating graph limits, known as graphons, from observations of sequences of sparse finite graphs. In this paper we show a simple method that can shed light on a subset of sparse graphs. The method involves mapping the original graphs to their line graphs. We show that graphs satisfying a particular property, which we call the square-degree property are sparse, but give rise to dense line graphs. This enables the use of results on graph limits of dense graphs to derive convergence. In particular, star graphs satisfy the square-degree property resulting in dense line graphs and non-zero graphons of line graphs. We demonstrate empirically that we can distinguish different numbers of stars (which are sparse) by the graphons of their corresponding line graphs. Whereas in the original graphs, the different number of stars all converge to the zero graphon due to sparsity. Similarly, superlinear preferential attachment graphs give rise to dense line graphs almost surely. In contrast, dense graphs, including Erdos-Renyi graphs make the line graphs sparse, resulting in the zero graphon.

arXiv cs.LG74

生成流的普适性与收敛性

arXiv:2610.05490v2 公告类型:替换 摘要:生成流通过训练流使其达到平衡,从而从未归一化目标分布中采样,而训练损失正是实践者所关注的信号。我们探讨这一信号究竟有多大价值:较小的损失是否能保证采样器准确,损失能否降至零,以及梯度下降达到这一目标的速度。损失决定了第一个问题的答案。通过差值比较平衡两侧的损失,能够以总变差距离界定流所隐含采样器的误差,并且界中的常数是显式的,不涉及策略;而通过比值进行比较的流匹配损失则不存在这样的界,即使在单个环上,只要其生成器在平衡点处连续,也已经如此。在图上,反向策略决定了另外两个问题的答案。一旦将其固定,平衡就等价于反向链下的不变性,因此在有限图上存在性是自动满足的;而一个常数——该反向链的格林算子范数,它所发挥的作用类似于谱隙的倒数——决定了平衡流附近损失曲率的上下阶,并为训练在其附近的收敛速度设定下限。其作用机制在于,梯度下降沿着反向策略对流进行扩散。对于详细平衡和轨迹平衡的平方对数生成器,在任意有限、路径连通的图上,从任意正初始化出发,在状态上训练平衡损失都能全局收敛。当反向轨迹的平均长度较短时,该常数仍可能为无穷大,此时精确流匹配可能失败。我们通过在可枚举状态空间上的精确计算检验了这些界和收敛速率,并且每个定理都带有由 Lean~4 开发计算出的认证状态。

arXiv cs.LG75

人口规模扩展还是数据稀释?去中心化学习中局部拓扑演化的动态

arXiv:2610.05476v2 公告类型:替换 摘要:去中心化学习的规模扩展改变的不仅是客户端数量 $N$,还包括信息传播与共识的动态。我们认为,增加 $N$ 的影响无法孤立地理解,因为数据分配、依赖拓扑的混合过程以及通信容量可能会同时发生变化。我们在 CIFAR-10 上研究了这些耦合效应,取 $N\in\{10,50,100,200\}$,比较了二度 Ring、Static Random 图以及 Local-First Heuristic Evolution (LFHE)——一种基于“朋友的朋友”发现机制的局部自适应拓扑过程。Ring 提供了一种在解析上清晰的失效模式:其 Metropolis 谱隙按 $\Theta(N^{-2})$ 衰减,这意味着随着群体规模扩大,模型分歧的收缩会逐渐变慢。实验表明,当保持名义上的本地数据集大小不变时,将固定总数据集划分给更多客户端所导致的表面上的规模惩罚会显著减小。剩余的性能下降高度依赖通信结构:Ring 会进入高分歧状态,而 Static Random 和 LFHE 则仍接近共识。进一步提高 LFHE 的度阈值可以改善准确率和共识程度,但模型传输成本也会大幅增加。这些结果表明,去中心化扩展由学习与通信的耦合动态所支配,而不仅仅取决于客户端数量。

arXiv cs.LG70

Measuring Learned Monotone Temporal Aggregation at Matched Admissibility

arXiv:2610.05196v2 Announce Type: replace Abstract: Risk regulation imposes directional constraints on scores; we adopt their strict per-input form -- the score monotone non-decreasing in every exposure input -- as a normative commitment. Deployed pipelines -- monotone hand-crafted aggregates feeding sign-constrained gradient boosting -- already satisfy it by composition, so constrained-versus-unconstrained comparisons price a guarantee the incumbent has for free. We instead hold admissibility fixed on both sides and measure what learning the aggregation is worth. Our instrument is a recurrent network whose state is classical risk statistics (an exponentially weighted moving average and a high-water mark with learned transforms), monotone by construction in every input and per MC-dropout sample. The central finding, by functional regression, is a subsumption boundary: a learned monotone channel reproduces the geometrically weighted separable family of hand-crafted statistics, one channel per member, to Spearman $\rho \ge 0.996$, approximates window statistics with measurable ceilings, and fails at consecutivity ($\rho = 0.924$) and time localization (0.628), both structural, and at the exposure floor (0.829), a learnability boundary. One explicit admissible basis repairs each failure (rank correlation 1.000). In or near the separable family, learned and engineered aggregation are substitutes, and the learned channel is never statistically behind at full sample size and specified capacity. Its advantages are incumbent-specific: a committed grid pays up to 0.019 AUC in decay regions it leaves uncovered (the learned channel stays within 0.004 of the strongest engineered consumer at every swept point); the highest-dimensional comparator degrades fastest with scarce data; and beyond the training support, grid-fed tree-ensemble scores go flat while a strictly increasing head keeps ranking. No single incumbent is dominated on all three axes.

arXiv cs.LG75

How Long, Not How Close: A Learned Temporal Metric for Planning in Latent World Models

arXiv:2610.04988v2 Announce Type: replace Abstract: Latent world models plan by rolling a frozen predictor forward under candidate action sequences and ranking the candidates by the latent distance between their imagined end state and the goal. However, this ranking breaks down when the goal lies several plans away, because the latent distance measures how closely an end state resembles the goal rather than how far it remains from reaching it. To address this, we propose TEMPO, a temporal-distance planning objective that leaves the world model untouched, learns only from the recorded trajectories already used to train it, and adds negligible cost to the planner's search. TEMPO learns a small map of the frozen latent in which the distance between two states of an episode reflects the number of environment steps between them, and blends this distance into the planner's cost. It requires no rewards, policies or success labels and, being a cost rather than a model, applies to frozen world models with one latent vector per state that plan by a latent distance. We evaluate TEMPO on eleven simulated environments (e.g., maze navigation, tabletop pushing, robotic arm control and three-dimensional manipulation) with the LeWM and PLDM planners. With a small MLP that adds at most 0.3% to a plan's arithmetic, TEMPO improves both planners at every goal distance, including the one-plan setting of their evaluations, raises LeWM from 36% to 99% on TwoRoom three plans from the goal, and remains competitive on a broad range of 2D and 3D navigation, reaching and manipulation tasks.

arXiv cs.LG67

MA-JEPA: Joint-Embedding World Models for Multi-Agent Reinforcement Learning

arXiv:2609.33563v2 Announce Type: replace Abstract: World models improve sample efficiency by training policies on imagined trajectories, but their usefulness depends on learning representations that capture the information needed for future control. We study whether self-supervised joint-embedding prediction (JEPA) can provide this learning signal for multi-agent reinforcement learning. We introduce MA-JEPA, a stochastic world model that replaces observation reconstruction with prediction of target representations, enabling model-based multi-agent reinforcement learning with centralized training and decentralized execution. A categorical latent state and a causal Transformer are trained with posterior and action-conditioned dynamics prediction objectives and are then used for actor-critic learning from latent imagination. A training-only joint predictor conditions on all agents' local states and actions to predict each agent's next local observation embedding. These predictions are passed through the same local posterior used during real interaction with a centralized critic that is used only for value learning, with execution remaining decentralized. Our experiments show that this architecture performs strongly on SMAC, matching or exceeding the strongest reported comparator mean win rate on four of eight evaluated maps.

arXiv cs.LG71

SMAT: Simple and Efficient Merge-Aware Training

arXiv:2609.33437v2 Announce Type: replace Abstract: Model merging integrates the capabilities of multiple experts without joint retraining, but standard expert training optimizes task loss alone and does not guarantee good performance after merging. Merge-aware training (MAT) aims to improve merged performance, but existing methods do not fully account for common merging operations and add training cost. We observe that, from an expert's perspective, common merging methods can be described by three operations: Scale reweights its own update, Mask removes selected coordinates, and Perturb adds updates from other experts. Based on this view, we introduce SMAT (Simple MAT), which jointly optimizes expert loss and expected loss at simulated merged parameters generated by sampling scaling coefficients, masks, and additive noise. We further introduce periodic scheduling, kernel fusion, and parameter storage switching to make SMAT efficient, with one forward and one backward pass per step. Across four language and vision-language backbones, SMAT improves the mean score across five merging methods by 1.07-2.16 points over the strongest baseline for each backbone, with less than 2% training-time overhead over standard fine-tuning.

arXiv cs.LG80

何时驱逐,而非保留什么:Draft-Guided Eviction for Training-Free KV-Cache Compression

arXiv:2609.33334v2 公告类型:替换 摘要:SnapKV、H2O 和 PyramidKV 等无需训练的 KV-cache 压缩方法会在预填充结束时驱逐 token,旨在保留未来查询预计会使用的注意力权重总量,即优化“保留什么”。我们发现,这一目标存在两种截然不同的失效方式。(1) 补偿:恢复被驱逐的注意力权重总量,可以恢复注意力层面的目标,却无法恢复任务质量。(2) 选择:即使覆盖了更多真实解码查询的注意力权重总量,当恢复的权重被分散而非集中于连贯的片段时,也可能损害质量。这些失效有一个共同原因:驱逐发生在决定答案轨迹的查询出现之前。我们提出 Draft-Guided Eviction (DGE),利用完整缓存先生成前 k=2 个答案 token,再延后执行驱逐——这仅比预填充多一步解码。由于草稿是根据答案自身的前缀生成的,因此在这一轨迹信号可用之前,不会丢弃任何缓存条目。每个注意力头的缓存预算保持不变,并且无需修改 SnapKV、PyramidKV、H2O 和 StreamingLLM 的驱逐评分,即可直接应用 DGE。不同于额外遍历方法,DGE 改变的是驱逐发生的时机,而不是选择哪些缓存条目。大量实验表明,在六个经过指令调优的骨干模型中的五个上,DGE 在所有评估预算下都优于先前方法;在 LongBench 上达到 44.2,几乎与 FullKV 的 44.3 持平。仅改变时机的对照方法 DGE-W 达到了相同分数,表明性能提升来自驱逐发生的时机,而非所选择的内容——我们将这一效应称为轨迹锚定。

arXiv cs.LG75

后处理公平性约束何时有帮助、何时有害:来自八项跨领域评估的证据

arXiv:2609.26955v3 公告类型:替换 摘要:生产环境中的 ML 公平性审计通常只进行一次,即在部署时针对单一领域开展。实践表明,这两点都存在问题:重新训练或用户群体变化后,公平性可能发生偏移;而在单个数据集上验证过的干预措施,很少在组织实际部署所涉及的异质领域中接受测试。我们提出 FAPE(生产环境公平性审计),这是一个四阶段框架,用于在八项领域评估中评估单一的后处理干预措施 Fairlearn 的 ThresholdOptimizer。这八个领域包括刑事司法、收入预测、法律录取、信贷放贷、农业贷款、多领域基准语料库、医疗保健和教育。每项评估均依据人口统计学均等和均等机会差异进行评分,并在可计算时加入差别影响比和准确率成本。干预效果与基线差异的幅度相关:在模型—领域配对中,该约束在14个高差异案例中的9个改善了差异,却在4个接近公平的案例中的3个使差异恶化。5个高差异例外中的每一个,都会在两项测量检查中的任一项下逆转:最小群体规模检查,或使用留出数据拟合阈值。部署时启动的 CUSUM 监测器在模拟偏移测试中,能够区分从未达到0.1平等标准的受约束模型,与曾达到该标准但后来退化的模型。因此,单次部署时审计并不能可靠地提供指导,这说明有必要进行基线差异筛查和持续监测。

arXiv cs.LG76

MemCalib:LLM Agents中的记忆使用基准测试与优化

arXiv:2609.24259v3 公告类型:替换 摘要:智能体记忆的有效性最终取决于底层LLM是否能让上下文中的每条记忆对其响应产生适当程度的影响。然而,这种能力一直在很大程度上被忽视。为评估这一能力,我们引入了MemCalib,这是一个基于真实记忆系统场景的基准,用于评估记忆使用并推动优化算法的发展。MemCalib测试集上的结果表明,前沿开源和闭源模型都难以恰当地使用记忆。它们经常过度使用或使用不足,而不是让每条命题的实际使用程度与其目标水平相匹配,从而导致响应存在偏差且质量低下。对包括组相对策略优化和在策略自蒸馏在内的常见后训练算法进行实验进一步揭示了明显的方向性偏差:模型经过训练后,会在一个方向上有所改善,却在另一个方向上恶化。因此,我们提出MemCalib-RL,这是一种有序的双向反事实信用分配算法,通过精确的原子消融将过度使用和使用不足的信号分离开来,并将其信用归因定位到响应词元。跨模型系列和规模(Qwen3-8B、Ministral-3-8B-Instruct和Qwen3.5-35B-A3B)的结果表明,MemCalib-RL取得了最佳的总体性能,同时更好地平衡了过度使用和使用不足;在外部基准评估中,其收益也能泛化到MemCalib之外。进一步实验支持了其设计选择和稳健性,并为其训练动态提供了深入见解。

arXiv cs.LG67

CTRL: Control-Based Time Series Forecasting with LLM-Guided Residual Learning

arXiv:2609.23257v2 Announce Type: replace Abstract: Time series forecasting underpins critical decision-making across diverse domains. While large language models (LLMs) offer promising reasoning capabilities, existing LLM-based time series forecasting approaches either reduce them to numerical predictors that bypass their strengths, or allow direct forecast generation that destabilizes predictions in non-stationary settings. We introduce CTRL, a framework that decouples semantic reasoning from quantitative prediction. A frozen backbone generates base forecasts, while specialized LLM agents function as controllers that analyze backbone prediction errors through decomposed trend, seasonal, and irregular components, grounding reasoning in interpretable temporal structure. Each agent outputs compact control signals that a lightweight residual decoder translates into forecast corrections. CTRL incorporates label-free test-time adaptation that detects distribution shift from input statistics alone and readapts control signals with only 3-24 LLM calls via caching. CTRL is explicitly designed to improve robustness under non-stationary temporal dynamics and distribution shift, while remaining competitive on highly stationary time series where adaptive correction provides limited additional benefit.

arXiv cs.LG76

无标签引导:将测试时强化学习压缩到仅偏置子空间中

arXiv:2609.18587v3 公告类型:替换 摘要:测试时强化学习(TTRL)使模型无需依赖带标签的训练数据即可提升推理能力,但现有方法通常会优化模型中相当大比例的参数。这引出了一个自然问题:当奖励信号和优化空间都受到严格限制时,是否仍能实现有效的测试时适应?我们通过无标签的仅偏置 TTRL 回答了这一问题。该方法使用多数投票伪标签作为奖励,仅优化约 100K 个偏置参数,同时冻结预训练骨干网络。在 MATH-500 上,我们的方法使用 Qwen2.5-7B 达到 76.67% 的准确率,略高于我们对有标签偏置引导方法的复现结果,同时相比全参数 TTRL 优化的参数数量减少了 76,000 倍。同一训练流程还提升了模型在视觉-语言和音频推理任务上的表现,包括 MathVista、AI2D、LogicVista 和 MMAU。我们进一步表明,学习到的引导向量能够迁移到 4,500 个留出的 MATH 问题上,说明这种适应并不局限于测试时优化所使用的问题。最后,我们分析了这种高度受限的适应为何能够奏效,发现多数投票的可靠性会随着 rollout 一致性的提高而提升,而具有更高可访问梯度能量的偏置子空间则表现出更强的下游可训练性。这些结果表明,仅使用完全无标签的奖励来优化一个极小的仅偏置子空间,也能实现显著的测试时适应。

arXiv cs.LG82

无数据 on-policy 蒸馏:在不使用外部数据的情况下我们能走多远?

arXiv:2609.14193v4 公告类型:替换 摘要:On-policy 蒸馏(OPD)正日益应用于前沿基础模型的后训练。该领域以往的工作主要聚焦于算法改进,但 OPD 在多大程度上依赖其训练问题,尤其是这种依赖可以降低到什么程度,仍不明确。在两个具有代表性的单教师 OPD 设置中,我们发现,使用 8 个真实提示词进行训练所取得的性能,与使用 17k 个问题进行训练相当;而在测得的难度和初始蒸馏差距上存在显著差异的数据集,也会产生相近的结果。我们的分析提出了两种互补的解释:重复采样可能使少量提示词也能揭示教师模型的大量监督信号;同时,OPD 能够迁移超越数据集特定知识的、可泛化的推理能力。基于这些观察,我们进一步提出无数据 on-policy 蒸馏(DF-OPD)设置,以研究系统能否自行提供训练问题,从而消除对外部数据的需求。在不使用种子示例的情况下,通过 64 个自行生成的问题,DF-OPD 在两个单教师设置中都取得了与完整数据 OPD 相当的性能。这一发现同样适用于多教师 OPD:在数学、代码和指令遵循任务中,1k 个生成问题取得了与使用约 7k 个真实后训练示例进行训练相当的性能。我们还进一步探索了 OPD 是否能够在没有明确训练问题的情况下运行。实验表明,这仅在有限情况下有效:学生模型会出乎意料地生成并回答自己的问题,从而使这一过程退化为一种隐式的 DF-OPD。总体而言,这些发现促使我们重新审视训练数据在 on-policy 蒸馏中的作用。代码可见于 https://github.com/Ryuki661/DF-OPD

arXiv cs.LG77

真相从未消失:合规上下文真值探针中的完美别名

arXiv:2609.10739v2 公告类型:替换 摘要:有人提出使用从语言模型激活中解码真相的线性探针来监测欺骗;而当一个经过欺骗训练的模型上的真值探针得分低于随机水平时,人们自然会将其解读为模型隐藏了真相,或不再表征真相。我们表明,拟合标签歧义也能产生相同的读出结果。在一个受控博弈中,模型应向盟友报告一个秘密比特,向对手报告其补码。在合规(盟友)上下文中,真实比特与任务规定的答案是相同的标签,因此,在这些上下文上拟合的探针无法判断自己测量的是二者中的哪一个。我们将这种完全一致称为完美别名。在对手上下文中,这些标签互为补码,因此,一个在盟友上下文上拟合的探针分别针对这两个标签进行评分时,其对手 AUROC 之和为 1。将盟友和对手上下文结合起来进行混合拟合,会使两个标签产生差异;随机化的输出编码表也会使任务规定的答案与输出字母解耦。对于一个经过奖励训练、在所有评估的对手试验中都给出虚假答案的 Gemma-2-9B 策略,在最终层上,盟友拟合的探针 AUROC 为 $0.006 \pm 0.005$(三个 RL 训练随机种子上的均值 $\pm$ 样本标准差),而在相同留出激活上,混合拟合的探针得分为 1.000。混合拟合还使用了更多样本和带标签的对手数据,因此这表明真实比特仍可被线性恢复,而不是说明仅仅分离标签就能解释性能提升。在遵循指令的 Llama-3.1-8B 中,在一个提示变体上重新拟合的探针,以及一个从参考变体中获得并冻结的探针,在留出的盟友数据上准确率都为 1.000,但在相同试验上的对手真值 AUROC 分别为 0.080 和 0.986。我们的主要实验使用一种单一令牌的博弈,其中提示词明确给出了该比特,因此恢复出的方向可能读取的是模型保留的一份副本;而在模型必须推断该比特的情况下,没有任何受测实验条件能稳定地产生欺骗。我们研究的是探针测量了什么,而不是模型是否使用了这些信息。

arXiv cs.LG80

MoEMB:使用高效混合专家模型扩展通用多模态嵌入

arXiv:2609.08663v2 公告类型:替换 摘要:通用多模态嵌入(UME)对编码器处理更广泛任务和模态的能力提出了日益增长的需求,同时任务和模态的复杂性也在不断提升。以往的扩展方法要么增加表示规模,要么增加检索开销,要么将编码器扩展为庞大的多模态 LLM。近期工作(如 Think-Then-Embed(TTE))探索了通过推理 token 进行扩展。然而,嵌入模型难以扩展:直接增加参数量会与对比学习所需的大训练批次大小形成权衡,而检索服务则必须满足严格的延迟要求。此外,UME 任务的复杂程度各不相同,扩展嵌入器可能带来大量冗余计算。在本文中,我们提出 MOEMB,通过混合专家(MoE)沿专家轴扩展 UME,在保持单向量、非自回归编码的同时提升编码器容量。通过系统研究基于 MoE 的 UME 设计空间和训练方案,MoEMB 在使用公开 MMEB 系列数据训练的模型中,于 MMEB-V2 和 MRMR 上均取得了新的最佳性能:仅使用 3B 激活参数,MoEMB 就超越了激活参数量超过其 4 倍的基于 TTE 的方法,同时显著减少了计算量。为进一步提升可扩展性和效率,我们首次全面研究了基于 MoE 的嵌入的自适应计算,涵盖训练方法和仅推理方法中的多种策略。综合来看,这些结果表明,专家扩展是 UME 的一种高效有效方向;自适应计算还能进一步提升基于 MLLM 的嵌入模型的效率,使其更适用于大规模检索和推荐系统。

arXiv cs.LG68

Risk-Conditioned Fine-Tuning of Large Language Models

arXiv:2609.08064v3 Announce Type: replace Abstract: Large Language Models (LLMs) are increasingly deployed in settings where rare but severe harmful generations can have significant consequences. Existing Risk-Averse RLHF addresses this issue by optimizing Conditional Value-at-Risk (CVaR), but it trains policies for fixed risk levels and therefore cannot adjust the desired degree of risk aversion at inference time. In this paper, we propose risk-conditioned RLHF, a framework that trains a single policy that provides a continuous risk-control interface, enabling users to select different degrees of risk aversion without retraining or deploying multiple risk-specific models. Experiments across multiple benchmarks demonstrate that a single risk-conditioned policy can adapt to different risk levels at inference time, enabling more flexible and risk-aware LLM deployment.

arXiv cs.LG40

K\"ahler landscapes for complex neural network descents and guarantees including a search and destroy of the Calabi-Yau manifold

arXiv:2608.19584v3 Announce Type: replace Abstract: We study landscapes for complex-parameterized networks. Our approach is motivated with an information-theoretic manifold perspective of the parameter and via classical optimization guarantees although of complex geometric variety such as through Dolbeault asymptotics. The descent path admits a K\"ahler information metric under a cross-entropy via the Wirtinger Hessian on the log-likelihood potential. We restrict attention to a descent update rule with natural gradient descent via a differentiated loss scaled by the inverse metric, so the descent path remains in the holomorphic tangent bundle. We emphasize Calabi-Yau information manifolds which profane theoretical guarantees via an ill-curvature-conditioned landscape. We focus on Calabi-Yau metrics specifically in a non-compact setting with a global potential, so defined geometrically rather than invoking the topological requirements of the Calabi conjecture. In non-compact settings, we can write the metric determinant with respect to a background in terms of a pluriharmonic or real-valued function. Under bounded, nonuniform, and almost low-rank assumptions, we get a partial eigenvalue blow-up effect. In an empirical setting, a Ricci-flat metric will not form, but the blow-up effect is a local condition and can partially hold empirically on open sets. We isolate the Calabi-Yau case in a theoretical setting, and we counteract the corrupted geometries under regularization. Moreover, it has been discovered that negative curvature subverts the loss landscape, specifically sectional curvature, so we expand on this and draw interconnections to negative-definite Ricci curvature. Our arguments primarily exist via geometric analysis, although we establish roots in deep learning theory such as through asymptotics at initialization and connections through failure modes of neural network guarantees under vanishing and negative Ricci curvature.

arXiv cs.LG65

Quantum Multi-Armed Bandits and Linear Bandits: Lower Bounds and Algorithms

arXiv:2608.14319v2 Announce Type: replace Abstract: We study quantum multi-armed bandits (QMAB) and quantum linear bandits (QLB), where the learner queries each arm or action through a quantum reward oracle or its inverse. Prior work gives algorithms over horizon $T$ with regret $O(K\log T)$ for QMAB with $K$ arms and $O(d^2\operatorname{polylog} T)$ for $d$-dimensional QLB. This leaves open the optimal dependence on $K$ and $T$ and whether the dependence on $d$ can be further improved. In this work, we prove the first tight minimax regret bound of $\Theta(K\log(1+T/K))$ for QMAB and the first lower bound of $\Omega(d\log(1+T/d))$ for finite-action QLB, ruling out regret independent of $T$. Our lower bounds rely on a high-confidence single-arm quantum testing lower bound for distinguishing a fixed reward mean from an interval of alternatives. A bandit-to-testing reduction then lifts it to the QMAB lower bound, while a linear embedding gives the finite-action QLB lower bound. The matching QMAB upper bound is obtained using a tail bound for the Quantum Monte Carlo (QMC) estimator. For finite-action QLB, we propose a phased elimination algorithm that combines a low-bias low-variance quantum mean estimator with a small-support $G$-optimal design through a query allocation matched to the design weights. When the action set has size $\operatorname{poly}(d)$, its regret is nearly linear in $d$ and matches our lower bound up to polylogarithmic factors.

arXiv cs.LG66

Robust and Efficient Noisy-Label Time-Series Classification via Dynamic Time Warping Based Granular Ball Computing

arXiv:2608.11704v2 Announce Type: replace Abstract: Dynamic Time Warping (DTW)-based Nearest-Neighbor (NN) classifiers are effective for time-series classification but are vulnerable to mislabeled training samples and require numerous DTW computations during inference. We propose DTW-based Granular Ball Computing (DTW-GBC), which organizes temporally similar training samples into granular balls and performs classification at the granule level. We further develop two granular-ball construction strategies for DTW-GBC. Experiments on four benchmark datasets with symmetric label noise show that the two DTW-GBC variants generally mitigate the performance degradation caused by label noise while requiring substantially fewer comparisons than DTW-based 1-NN during inference. These findings suggest that DTW-GBC provides a favorable balance between classification robustness and inference efficiency.

arXiv cs.LG77

自适应子模型联邦学习中的容量混杂与覆盖保证

arXiv:2608.07157v2 公告类型:替换 摘要:子模型联邦学习允许资源受限的客户端训练全局模型的缩窄版本,但现有方法仅依据设备资源分配容量。近期方法开始利用源自训练的信号确定子模型规模,这启发了一个自然的下一步:根据从服务器已观测到的更新所估计的每个客户端数据异质性来分配容量。我们以自适应容量分配框架 HAS-FL 为案例,探究这一步是否可行。首先,在可复现的数据划分上,以标签分布散度的真实值为基准进行验证后发现,客户端异质性的更新散度估计主要受容量而非数据支配:在两个图像基准和每个随机种子下,这些估计都与设备容量呈强负相关;控制容量后,它们与数据异质性的关联接近于零或为负。任何从子模型更新中估计客户端统计量的方法,都会受到这一此前未被记录的混杂因素影响。其次,自适应分配存在一种隐藏的失效模式:当每个客户端的容量上限都低于完整宽度时,未被覆盖的参数会停留在随机初始化状态,并逐步破坏全局模型。一个简单的覆盖保证即可消除该失效,也解释了均匀分配为何会崩溃。第三,通过预算匹配的对照实验可以确定自适应策略的实际贡献:在两个图像基准上,将相同平均预算随机分配所得的结果与自适应策略相当,差异处于不同随机种子之间的波动范围内;而在自然划分的文本基准上,自适应策略是三种策略中最弱的,同时消耗的容量最多。子模型训练能够以按平方级降低的成本接纳受限客户端,但与全模型训练相比会牺牲相当大的准确率;真正保护这种准确率的是参数覆盖,而不是分配策略的智能程度。

arXiv cs.LG76

消息传递究竟有多重要?面向神经网络图回归的 GNN 层直接替换研究

arXiv:2607.26404v2 公告类型:替换 摘要:图神经网络(GNN)被广泛用作回归器,例如用于预测神经网络架构的准确率。然而,新消息传递(MP)层的开发和基准测试几乎完全围绕分类任务展开,而回归流程通常只采用一种 MP 层,且不进行消融实验。我们探究了 MP 层的选择对图级回归究竟有多大影响。在保持四种现有 GNN 回归器的架构、损失函数和训练方案不变的情况下,我们替换了十种 MP 配置,涵盖卷积式、基于同构和基于注意力的设计。我们在十一个神经网络图数据集上对其进行评估;这些数据集的单图节点数从不足十个到超过一千个不等,样本数从几百个到超过四十万个不等。评估指标包括秩相关性、预测误差、top-$k$ 检索、延迟和内存占用。MP 层的选择会显著改变结果,我们发现,最佳选择取决于图规模、训练集大小和回归目标。在小型架构图上,GEN、$k$-GNN 和 PNA 等经典层以更低的成本达到或超越基于注意力的层的表现,而 GATv2 在仅包含少量大型图的数据集上表现良好。

arXiv cs.LG65

Do Sheaf Neural Networks Use Holonomy? A Measure--Intervene--Control Study

arXiv:2607.19514v3 Announce Type: replace Abstract: Geometric architectures are often motivated by internal mechanisms, but accuracy alone does not show whether predictions use them. In Sheaf Neural Networks (SNNs), edge transports form a connection whose cycle products define holonomy. We ask whether training changes triangle holonomy, whether predictions rely on the learned connection, and whether holonomy drives triangle counting. We use basis-independent loop readouts with identity interventions and shortcut controls. On high-homophily GraphUniverse graphs, triangle counting increases the mean SO(2) triangle rotation in Neural Sheaf Propagation (NSP) from 0.010 to 0.388 radians, while community detection ends at 0.029 radians. With more data, learned SO(2)--NSP outperforms Identity NSP, and replacing its transports after training increases error further. However, ridge regression is more accurate, diagonal maps improve without continuous rotation, and fixed-degree models develop rotation without improved counting. Thus, NSP can learn and rely on a nontrivial connection, but our experiments do not show that triangle holonomy drives its predictions.

arXiv cs.LG78

超越欧氏裁剪:通过黎曼等距策略优化克服 LLM 强化学习中的探索坍缩

arXiv:2607.10169v2 发布类型:替换 摘要:强化学习(RL)已成为提升 LLM 推理能力的主流范式。然而,采用 PPO-Clip 的强化学习算法受探索坍缩的固有限制。后续工作仍主要依赖启发式方法,未能找出 PPO-Clip 失效的根本原因。本文揭示了 PPO-Clip 的基本缺陷:它使用欧氏度量隐式衡量策略差异,这在理论上与策略黎曼流形的内在几何结构不一致。这种几何不匹配导致算法在低概率区域的更新过于保守,而在高概率区域的更新过于激进,最终使探索能力坍缩。为纠正这一几何缺陷,我们提出黎曼等距策略优化(Riemannian Isometric Policy Optimization,RIPO),保证在黎曼流形上进行等距策略更新,从而有效平衡探索与利用。我们进一步证明,RIPO 实现了良好的偏差—方差权衡,有助于稳定优化过程。大量实验表明,在七个竞赛级基准测试中,RIPO 的表现显著优于现有的 LLM 强化学习算法;在 AIME24 上,其性能较 GRPO 提升最高达 60%。

arXiv cs.LG71

System-Prompt Conditioning and Hidden-State Geometry in Four Open-Weight Models: Corrections and What Survives

arXiv:2607.09842v3 Announce Type: replace Abstract: Versions 1 and 2 of this preprint reported that an identity-specifying system prompt leaves a geometric fingerprint in the final-layer hidden-state trajectories of four open-weight language models, and that instruction tuning moves this fingerprint from the direction to the magnitude of the hidden-state vector. An audit of their code and data found the following. The curvature statistic described as Ollivier-Ricci curvature on Euclidean k-NN graphs was a non-standard Forman-type edge statistic on graphs built from temporal and cosine k-NN edges. Its released test permuted pooled edges instead of trajectories, and the published p-values came from unreleased code. The quantity reported as the norm of the first generated state is the state at the last prompt position, from which the first output token is predicted. The generic control prompt was matched to the identity prompt in characters, not in tokens. This version corrects the methods, withdraws the regime-specific claims (one model per regime) and the direction-to-magnitude claim, and re-analyzes the data with added controls. What survives is narrower. Centroid distance, maximum mean discrepancy and a linear probe separate every pair of prompt conditions in every model, while the curvature statistic exceeds its split-half noise floor in only four of twelve comparisons. In Gemma-4-E4B-it this state has a lower norm under the identity prompt than under a token-length-matched generic prompt (138.1 vs. 216.5; Cohen's d = -5.45; n = 20). Its direction also separates the conditions, and the effect fits the state's role in planning the output: the identity prompt instructs a pause before every answer, and the model opens 98 of 100 responses with a pause marker. When the first token is fixed, the norm ordering reverses. The base model continues the prompt template instead of answering. A redesigned follow-up study is in preparation.

arXiv cs.LG77

群不变谱嵌入

arXiv:2607.08987v2 公告类型:替换 摘要:谱嵌入方法广泛用于对具有内在低维结构的高维数据集进行降维和聚类。尽管许多具有实际应用价值的数据集在旋转等对称变换下表现出不变性,标准谱嵌入方法却未考虑这一点,而是将具有对称关系的数据点视为彼此无关。我们针对这一问题的方法,是将对称性直接纳入谱嵌入所使用的相似性核中。我们分析了具有黎曼数据流形 $M$ 的情形,其中的对称性由紧李群~$G$ 给出,并证明在适当条件下,由三类不变核构造的图拉普拉斯算子逐点收敛于商空间 $M/G$ 上的显式二阶微分算子。我们的分析表明,该方法具有更高的收敛速率,因为有效维数会根据群的维数而降低。我们在具有 $\mathrm{SO}(2)$ 或 $\mathrm{SO}(3)$ 对称性的数据集上验证了该方法,并表明 $G$-不变谱嵌入能够恢复数据的内在几何结构;相比之下,即使在数据量趋于无穷时,标准谱嵌入也无法做到这一点。

arXiv cs.LG76

直接优化非线性回归的平均人口统计均等性

arXiv:2607.05098v2 发布类型:替换 摘要:我们关注这样的回归场景:公平性目标是使敏感属性不同取值对应的平均预测值相等,这一标准称为平均人口统计均等性。直接优化这一标准很困难,因为它依赖于一个未知且会在训练过程中发生变化的条件均值。常见的依赖性惩罚方法和对抗方法并不估计该条件均值,而是将预测结果推向完全独立。即使平均预测值已经相等,这种更强的约束也可能降低准确率。现有的条件均值方法仅适用于线性预测器或低维敏感属性。我们通过使用 DPVar 实现对平均人口统计均等性的直接优化;DPVar 是一种公平性度量,定义为条件均值预测的方差。由于必须随着预测器的变化估计条件均值,优化 DPVar 会形成一个函数型双层问题。我们开发了两种求解器:FBO 使用闭式超梯度,而迭代微分(ITD)求解器则通过对条件均值估计器的更新过程进行微分来求解。不同于以往的条件均值方法,我们的方法适用于非线性预测器和高维连续敏感属性。在一个由 21 个表格回归数据集和 Communities & Crime 数据构建的半合成基准测试中,FBO 和 ITD 获得了不逊于或优于现有方法的准确率-DPVar 权衡。

arXiv cs.LG74

奖励可观测性与 RSSM 世界模型中离线检查点选择的局限性

arXiv:2607.01736v3 公告类型:替换 摘要:我们研究了在 Gymnasium 的 LunarLander-v3 中使用人类演示训练的循环状态空间模型(RSSM)世界模型的闭环特性。我们使用训练好的世界模型进行零样本 CEM 模型预测控制(MPC),并在想象环境中进行演员-评论家(A2C)训练。在 100 个留出的回合上进行评测后,所选的基于模型的 A2C 策略(在世界模型检查点 280 上训练)达到了 +189.5 的平均回报,与表现最佳的无模型 A2C 检查点(+183.7;回合上限分别为 600 步和 1000 步)相当,而实际训练交互次数减少了约 65 倍。我们还将世界模型 MPC 与基于成功演示训练的行为克隆(BC)策略进行了比较。只有在随机动作选择下,BC 策略的平均回报才与 MPC 相当;在相同的 20 个回合中,BC 策略出现了 1 个灾难性回合,而 MPC 没有。随后,我们引入奖励可观测性比例(Reward Observability Fraction,ROF):它表示在线性化潜在动力学的可观测子空间中,奖励梯度的欧几里得比例。我们证明,接下来的 H 个观测包含关于该子空间中每个方向的 Fisher 信息,而不包含关于与该子空间正交方向的信息。然而,ROF 本身取决于潜变量的缩放方式:重新缩放潜变量会改变 ROF,却不会改变模型,因此不同模型之间的原始 ROF 水平不可比较。将奖励头强制连接到经过后验校正的潜变量 z 上会提高 ROF,并且在我们测试的两组运行上,这一提升通过了坐标不变性检验。最后,我们检验 ROF 或其他离线指标能否预测 MPC 的闭环崩溃。即使数据和配置完全相同,不同训练运行之间也会出现不同程度的崩溃。ROF 无法预测崩溃;我们筛选的 108 个离线汇总指标均未通过置换检验,最佳候选指标在新的运行上也会失效。仅凭模型和记录数据从离线结果预测崩溃仍是一个开放问题。

arXiv cs.LG76

Red Queen G\"odel Machine:共同进化的智能体及其评估器

arXiv:2606.26294v3 公告类型:替换 摘要:自我改进型智能体在智能体编码基准测试中已达到最先进水平,但其搜索方法假设评估标准是静态不变的。这忽略了进化的一个核心特征:物种会随着环境的变化而适应。我们提出 Red Queen G\"odel Machine(RQGM),一种针对非平稳效用进行递归自我改进的进化框架。该框架使学习型评估器能够与其引导的智能体共同改进。在 DeepSWE 上,RQGM 通过加入互补的“智能体作为评审者”代码审查信号,超越了固定评估器基线:共同进化的评审者对编码器生成的补丁进行评分,以引导搜索。在较低推理投入下,RQGM 编码器在留出任务上的通过率达到 82.1%,而基线为 75.0%;在较高投入下,其表现几乎达到 GPT-6 Astra 模型的水平。在科学论文写作与评审,以及奥林匹克竞赛级别的证明写作与评分任务中,共同进化的评估器能够提供评估标准。以人类 IMO 评分为锚点,一名共同进化的评分器自行撰写阶段性评分标准,并以低 3 倍的搜索成本超越静态基线,推动证明器取得最高平均分。由于 RQGM 能够跨轮次修改搜索目标,因此可以对搜索过程进行正则化。例如,RQGM 通过加入额外的对抗性目标,发现对 AI 作品和人类作品同样严格的评审者,从而减少自我偏好偏差。在这些经过校准的评审者引导下,共同进化的写作者在“智能体作为评审者”评审面板下的接受率达到基线的 1.78 至 1.86 倍。RQGM 使自我改进系统成为可能:智能体与评估器能够在递归过程中相互促进,突破静态评估的限制。

arXiv cs.LG72

Neural Conjugate Aggregation:异质传感器偏差下可辨识的无监督多传感器回归

arXiv:2606.22200v2 发布类型:替换 摘要:我们研究不确定性下基于回归的数据融合问题:有多个带噪声且存在偏差的测量源可供使用,但训练期间缺少真实标签。这一场景常见于传感器网络、仿真集成以及监督成本高昂或难以实现的科学监测系统。我们提出 Neural Conjugate Aggregation Model(NCAM),这是一个将神经网络与共轭高斯推断相结合、用于无监督多源融合的层次贝叶斯框架。NCAM 根据上下文协变量学习各数据源特有的偏差和可靠性,从而针对潜在目标变量得到具有解析可处理性的后验分布,并分解认知不确定性与偶然不确定性。通过传感器锚定和方差正则化,我们解决了结构性不可辨识问题,实现了稳定且可解释的后验聚合。为利用有限样本保证性补充贝叶斯不确定性,我们整合了局部自适应蒙特卡洛保形预测,在可交换性假设下生成具有覆盖率保证的异方差预测区间。在合成数据集和真实世界空气质量数据集上的实验表明,与包括均值聚合、概率 PCA 和 Kalman filtering 在内的无监督基线方法相比,该方法具有更高的预测准确性和校准良好的不确定性。

arXiv cs.LG77

从自身解中学习:面向可验证奖励强化学习的自条件信用分配

arXiv:2606.18810v2 发布类型:替换 摘要:带有可验证奖励的强化学习(RLVR)推动了用于推理任务的 LLMs 训练取得重大进展,但 GRPO 等代表性方法在所有 token 间分配均等的信用,将梯度浪费在常规 token 上,同时对关键推理步骤的信用分配不足。现有的 token 级信用分配方法需要模型自身 rollout 之外的资源。GRPO 的变体依赖过程奖励模型或真实答案。知识蒸馏通过每个 token 的散度分配信用,但需要外部教师(On-Policy Distillation)或特权信息(On-Policy Self Distillation)。然而,这些依赖限制了其在纯 RLVR 环境中的适用性。我们观察到,以模型自身经过验证的轨迹为条件,会在原始分布与条件分布之间诱导出可测量的逐 token KL 散度;我们还证明,当存在多条经过验证的轨迹时,从由这些轨迹构建的自教师进行蒸馏会导致不可行的加权平均解。我们提出 SC-GRPO(Self-Conditioned GRPO),将前述 KL 散度作为乘法权重作用于 GRPO 梯度。在涵盖数学、代码和智能体任务的五个基准测试中,SC-GRPO 始终优于 GRPO 8.1%,优于 DAPO 5.9%,并展现出更强的 OOD 性能。此外,SC-GRPO 的性能高于 OPD。

arXiv cs.LG72

S4oP:面向资源受限设备的结构化状态空间模型算子级剪枝

arXiv:2606.18096v2 公告类型:替换 摘要:结构化状态空间模型(SSM),包括 S4 和 S4D 架构,近年来已成为基于注意力的模型的有力替代方案,能够捕捉序列数据中的长程依赖关系。尽管这些模型在经验性能方面表现出色,但由于计算和内存开销较大,在时间和资源受限的环境中部署仍然面临挑战。本文提出了一种针对基于 S4 和 S4D 的模型的新型增量式算子级剪枝方法,在保持预测性能的同时显著降低推理成本。据我们所知,这是首项系统研究针对 SSM 的结构化算子剪枝工作的研究。我们的方法通过交替执行结构化掩码和微调,逐步剪除模型算子,同时联合监测准确率和推理延迟。我们在统一的训练与评估框架中实现了该方法,从而能够系统地探索效率与准确率之间的权衡。在多个基准数据集上的实验表明,在大多数情况下,剪除多达 70% 的模型算子仍能保持原始模型的性能,同时显著降低推理延迟。这些结果表明,结构化算子剪枝是提升 SSM 效率的一种有效且此前尚未被探索的策略,有助于推动其在实际资源受限场景中的部署。

arXiv cs.LG78

GRASP:面向可扩展预训练数据归因的几何感知残差对齐

arXiv:2606.06892v3 公告类型:替换 摘要:可扩展的数据归因方法通常为单个训练样本分配彼此孤立的效用分数。这一普遍的加性假设无法从根本上捕捉关键的子集动态,包括数据冗余和互补覆盖。在本文中,我们将归因重新定义为子集级反事实效用预测,并提出具备交互感知能力的代理模型 GRASP。基于理论上的平滑性下界,GRASP 通过二次几何惩罚显式建模子集交互。为在不依赖隐藏的预言机调优的情况下实现预训练规模的高效性,我们将低维特征摘要与严格有限的下置信界选择协议相结合。广泛的子集重训练评估表明,GRASP 明显优于现有的可扩展基线方法。在反事实子集保真度方面,它使任务级秩相关性提升至原来的两倍以上,同时将前期工件构建成本降低了近一个数量级。下游诊断进一步表明,该评分机制能够迁移到语言模型数据筛选和跨领域视觉数据选择中,为优化大规模预训练语料库奠定了稳健基础。

arXiv cs.LG77

在生成空间中学习隐式偏差以加速蛋白质动力学模拟

arXiv:2606.01833v2 公告类型:替换 摘要:蛋白质动力学的生成式模拟器能够以分子动力学一小部分的成本生成合理的轨迹,但它们会继承训练分布,并且在长时域外推过程中往往会重复访问已知状态,而不是到达稀有状态。受经典增强采样的启发,我们在预训练模拟器的生成空间中引入一种隐式的、依赖历史的偏差。具体而言,历史感知评分估计器通过距离加权偏差增强冻结的模拟器,引导反向时间采样远离此前生成的结构,并通过环境支持项进行正则化。为在长时域内保持结构有效性,我们采用基于评分的细化步骤,利用冻结的模拟器将漂移后的样本重新投影到数据流形上。实验表明,该方法:(i)在 DynamicPDB-80 上将多样性提升了 $35\%$;(ii)对于 $12$ 个零样本 Fast-Folding 蛋白,仅使用学习到的偏差,就能以高达约 ${\sim}15\times$ 的速度达到无偏模拟器的覆盖范围;将其与细化步骤结合后,达到该覆盖范围的速度最高提升至约 ${\sim}37\times$,同时覆盖的低能量状态数量增加至约 ${\sim}3\times$。代码即将发布。