AI/ML arXiv cs.AI

Decoupling Search from Reasoning: A Vendor-Agnostic Grounding Architecture for LLM Agents

This work introduces Decoupled Search Grounding (DSG), a vendor-agnostic architecture designed to separate the complex process of real-time search from the core reasoning capabilities of Large Language Model (LLM) agents. The primary contribution is a novel architectural boundary that treats search grounding as an optimizable interface rather than an intrinsic model feature. This research addresses a critical limitation in current LLM agent deployments: the tight coupling of retrieval policy, provider selection, evidence integration, cost, latency, and generation within a single model-provider abstraction. This fusion hinders inspection, tuning, reuse, and portability, and can lead to issues like Search-Induced Verbosity.

The implications of this decoupling are significant. It enables more robust, controllable, and cost-effective LLM agents, particularly in production environments requiring dynamic access to up-to-date information. The authors, Emmanuel Aboah Boateng, Kyle MacDonald, Amardeep Kumar, Siddharth Kodwani, and Sudeep Das from an unnamed institution, published this work on arXiv. The intended audience is software engineers and researchers working with LLM agents, who will benefit from a more modular and manageable approach to integrating external knowledge.

Key technical innovations include an MCP-compatible gateway that exposes fine-grained controls. This gateway facilitates vendor-agnostic provider routing, allowing agents to dynamically select optimal search services. Source-aware context rendering ensures that retrieved information is appropriately formatted for the LLM, preserving nuance and accuracy. Configured fallback mechanisms mitigate failures by gracefully transitioning to alternative retrieval strategies. Furthermore, explicit control over retrieval depth allows for precision in information gathering, and exact and semantic caching strategies drastically reduce redundant computations and latency.

The results presented demonstrate tangible benefits. On benchmark datasets like SimpleQA and HotpotQA, DSG achieves accuracy comparable to native search while significantly reducing search costs and latency, particularly when leveraging its caching mechanisms. Crucially, it preserves concise output contracts, a common challenge with traditional LLM grounding. In a large-scale e-commerce query-understanding workload, DSG matched native-search accuracy with over 98% cost reduction. This work enables future LLM agent architectures to be more flexible, efficient, and adaptable, potentially leading to a paradigm shift in how agents interact with the external world and reducing vendor lock-in. The findings, presented in an abstract only, highlight real-time grounding as a distinct, optimizable interface.

AI/ML arXiv cs.AI

User as Engram: Internalizing Per-User Memory as Local Parametric Edits

This work, "User as Engram: Internalizing Per-User Memory as Local Parametric Edits," authored by Bojie Li, addresses a fundamental challenge in personalizing large language models (LLMs): how to efficiently and effectively store and recall user-specific information without degrading general capabilities or requiring vast computational resources. Traditional approaches often store user data externally, such as in natural language files or retrieval indexes, or utilize methods like per-user LoRA adapters which conflate content with reasoning skills, leading to global weight updates that can contaminate unrelated knowledge and result in significant memory overhead. The core contribution is a novel architecture that disentangles per-user content from shared reasoning skills, drawing an analogy to biological memory systems.

The problem this paper solves is the inefficiency and potential for performance degradation associated with current LLM personalization methods. Existing techniques struggle to scale with a growing user base and can negatively impact a model's ability to generalize. This research fills the gap by proposing a memory system that more closely resembles how humans store episodic and semantic information, keeping them separate to preserve both specific memories and broad competencies.

The most important technical ideas include: 1) Hash-Keyed Memory Tables: Instead of directly modifying model weights globally, user facts are stored as sparse, local "engram" edits within a dedicated hash-keyed memory table. This ensures that updates for one user do not affect the representations of others or the base model's general knowledge. 2) Disentangled Content and Skill: The system separates user-specific content (facts) stored in the Engram table from the general reasoning skills, which are maintained in a shared adapter. This layering prevents user facts from overwriting or interfering with the model's core competencies. 3) Additive and Lossless Composition: Because different users' facts occupy distinct hash slots, their edits compose additively and losslessly within the shared table. This allows a single model instance to serve a large number of users concurrently without the compositional limitations of methods like per-user LoRA, which only supports one adapter at a time.

This research enables highly scalable and performant LLM personalization. By reducing the memory footprint by an estimated 33,000x compared to writing facts into weights and demonstrating significant improvements in indirect-reasoning accuracy (5.6x on average) while maintaining or improving performance on unrelated tasks, it paves the way for LLMs that can be truly personalized for millions or billions of users. The "glass box" nature of the edits, where each fact lookup is precisely controlled and other model parameters remain untouched, offers a transparent and robust mechanism for personalized interaction, potentially influencing future LLM architectures towards more modular and biologically inspired memory management. This work is published on arXiv and is primarily intended for software engineers and researchers working on LLM deployment, personalization, and efficiency. The abstract provided is the sole content available for this analysis.

AI/ML arXiv cs.AI

Breaking the Solver Bottleneck: Training Task Generators at the Learnable Frontier

This work, presented as a preprint on arXiv by Lorenz Wolf and colleagues from various institutions, introduces PROPEL, a novel framework designed to overcome a critical bottleneck in training advanced artificial intelligence agents, particularly those employing reinforcement learning (RL). The core contribution is a method for efficiently training task generators that produce an optimal supply of "frontier tasks" – those that are valid, solvable, and precisely challenging enough to facilitate agent learning. This addresses the growing problem where fixed task distributions become saturated as agent capabilities advance, and naive synthetic generation produces unusable tasks. PROPEL is of significant interest to researchers in machine learning, artificial intelligence, and natural language processing, especially those developing sophisticated reasoning and agentic models, and ultimately benefits the broader field by accelerating AI development.

Two crucial technical ideas underpin PROPEL. First, it leverages a "solver-amortized" approach. Instead of requiring computationally expensive repeated solver rollouts for every task generated during the generator's training, PROPEL trains a lightweight "activation probe." This probe, developed on a one-time labeled corpus of tasks and their outcomes, acts as a proxy for the true solve rate. It allows for the generator's optimization using only a single forward pass of the frozen reference solver, drastically reducing computational overhead. Second, PROPEL explicitly targets training task generators at the "learnable frontier," meaning tasks are generated with a specific, desirable solve rate. This is demonstrated by the framework's ability to shift task generation towards this targeted solve rate across diverse domains like mathematics, code generation, and software engineering. For instance, in coding tasks, PROPEL significantly increased the proportion of tasks generated at the learnable frontier from 10.1% to 20.0% for a Qwen2.5-3B-Instruct solver, and similarly for other model scales and domains, including a substantial improvement for Qwen3.5-27B on unseen software engineering repositories.

Looking ahead, PROPEL enables the scalable and efficient training of more capable AI agents by decoupling the expensive solver evaluation from the generator training process. This advancement is expected to accelerate research into complex AI systems, allowing for the exploration of larger and more intricate problem spaces. The methodology offers a promising direction for future work in procedural content generation for AI training, potentially influencing the design of benchmark environments and the development of more robust and generalizable agents. This paper content is an abstract only.

AI/ML arXiv cs.AI

Ghost Attractor Networks: Basin-Structured Dynamical Decoders for Closed-Loop Sequential Generation

Ghost Attractor Networks, as presented by Tianyu Wang, Ying Wang, Zhihao Liu, Xi Vincent Wang, and Lihui Wang on arXiv (cs.AI), introduce a novel dynamical decoder architecture designed to address the computational and memory inefficiencies of large-scale sequential generation models like Transformers and diffusion models. The core contribution is a theoretically grounded decoder that generates sequences by evolving its latent state within a learned potential field, inherently constructing a basin-attractor structure. This design directly tackles the problem of limited closed-loop control and phase-conditioned action generation in traditional, efficient feed-forward decoders, which often suffer from unstructured latent representations.

The significance of this work lies in bridging the gap between efficient, small decoders and the sophisticated control capabilities afforded by more complex models. The authors posit that the memory cost and iterative computation of Transformer and diffusion decoders become prohibitive for long sequences. Conversely, small feed-forward decoders, while efficient, lack the necessary latent geometry for tasks requiring stable basins for phase-conditioned action generation and cross-step latent carry-over. Ghost Attractor Networks offer a solution by providing both efficiency and structured latent dynamics.

The intended audience comprises software engineers and researchers working on generative models, robotics, and reinforcement learning, who stand to benefit from more efficient and controllable sequence generation. Three key technical ideas underpin this architecture. Firstly, the potential-drift form of the latent evolution is derived from three desiderata: multi-modality, single-pass switching at the decoder level, and constant memory usage. This form allows for emergent basin structures. Secondly, mode transitions are modeled as saddle-node bifurcations, enabling fluid switching between different generation modes, facilitated by what the authors term "ghost-attractor escape." Thirdly, a hierarchical phase-space decomposition is employed, separating the coarse convergence within a basin (first-order) from finer, proprioceptive refinement (second-order), leading to more nuanced control.

Empirically, the paper demonstrates that a Ghost network, trained end-to-end with behavioral cloning and contrastive objectives, exhibits the predicted gradient-flow contraction within its potential. It achieves significant improvements in parameter count and latency compared to Diffusion Transformers, while outperforming other small decoder baselines in offline accuracy. Crucially, on the LIBERO-10 closed-loop benchmark, phase conditioning on the Ghost's basin-structured latent led to a substantial success rate gain.

This work enables a new paradigm for efficient and controllable sequential generation. It suggests that by explicitly designing latent spaces with attractor dynamics, we can achieve the benefits of complex models with the efficiency of simpler ones. This could significantly influence fields like robotics, where real-time, closed-loop control of complex actions is paramount, and generative AI, by offering a more resource-efficient path to sophisticated sequence generation. This analysis is based on the abstract provided.

AI/ML arXiv cs.AI

SAE Interventions are Unreliable: Post-Intervention Recovery of Suppressed Behavior

This work, "SAE Interventions are Unreliable: Post-Intervention Recovery of Suppressed Behavior," authored by Mingyue Cui, Linghui Shen, and Xingyi Yang, investigates a critical vulnerability in the application of Sparse Autoencoders (SAEs) for controlling large language model (LLM) behavior. The core contribution is the demonstration that interventions targeting specific SAE features, designed to suppress undesirable model outputs, are not as robust as previously assumed. The paper reveals that even when an SAE feature is successfully clamped to prevent its activation, the underlying undesirable behavior can still be recovered. This is a significant finding because a growing body of research in LLM interpretability and safety relies on the premise that identified SAE features serve as reliable control handles. The problem this work addresses is the potential for a false sense of security when employing SAE-based interventions, leaving models susceptible to the very behaviors they were meant to prevent.

The research is particularly relevant to AI safety researchers, LLM developers, and practitioners working on model interpretability and alignment. The authors highlight a gap in understanding the relationship between feature-level control and complete behavioral control. The most important technical ideas revolve around the concept of "post-intervention recovery." This is framed as a constrained optimization problem in the residual space, where perturbations are sought to restore pre-intervention behavior while simultaneously preserving the clamped state of the targeted SAE features. This optimization is performed even under a strong threat model, simulating scenarios where the intervention remains active during the recovery process. Furthermore, the paper employs encoder-orthogonal updates and cross-layer Jacobian analysis to distinguish true recovery from merely undoing the intervention, validating their findings across various steering tasks, including truthfulness, unlearning, indirect object identification, and refusal steering. The results are stark, showing high recovery rates in safety-critical refusal steering scenarios, even with minimal drift in the defended SAE features. This implies that while SAE features can be causally linked to behaviors, manipulating these features does not guarantee comprehensive control over the model's emergent properties.

This research fundamentally challenges the reliability of current SAE-based intervention strategies, suggesting that a deeper understanding of the causal pathways and distributed representations within LLMs is necessary for robust safety mechanisms. It enables forward-looking work to develop more sophisticated intervention techniques that account for these recovery modes, potentially leading to more resilient LLM safety architectures and a more nuanced approach to interpretability research beyond simple feature clamping. The findings are published on arXiv and are a crucial read for anyone building or deploying LLMs with an emphasis on safety and predictable behavior. The content appears to be an abstract.

Cybersecurity arXiv cs.AI

SafeClawBench: Separating Semantic, Audit-Evidence, and Sandbox Harm in Tool-Using LLM Agents

This work introduces SafeClawBench, a novel benchmark designed to precisely evaluate the security vulnerabilities of large language model (LLM) agents that interact with external tools. Existing evaluation methods often conflate different stages of an attack, making it difficult to pinpoint the exact failure mode. SafeClawBench addresses this gap by separating observed harm into three distinct categories: semantic acceptance (whether the LLM's intent aligns with an attacker's goal), audit-visible evidence (whether the LLM generates output that could be used as proof of harm), and sandbox-observed harm (actual execution of harmful actions by tools or state modifications). This granular approach is crucial for understanding and mitigating risks beyond simple text-based prompt injection.

Developed by researchers from multiple institutions including Tsinghua University, the work was published on arXiv (cs.AI). SafeClawBench is primarily intended for LLM developers, security researchers, and researchers in agent-based AI. It provides a structured framework for rigorously testing the robustness of tool-using LLM agents, benefiting those who deploy or build such systems.

The benchmark's most significant technical contributions lie in its multi-stage evaluation framework and its comprehensive attack taxonomy. SafeClawBench comprises 600 adversarial tasks across six attack families, including prompt injection variations, memory manipulation, and ambiguity-driven inference. The key is its three distinct evaluation endpoints: semantic attack acceptance, audit-visible harm evidence, and sandbox-observed tool/state harm. This separation allows for a nuanced understanding of how LLM agents fail. For instance, the paper highlights that even when an LLM passes semantic checks, it can still produce observable harm in the sandbox, demonstrating that textual compliance does not guarantee safety. The influence of different prompt protection policies is also meticulously analyzed, revealing that their effectiveness varies significantly depending on the specific LLM agent and the evaluation protocol.

Going forward, SafeClawBench enables more targeted development of defenses for LLM agents. By providing a clear methodology to distinguish between different types of security failures, it will allow researchers to prioritize efforts and create more effective mitigation strategies. This work is likely to influence the field by establishing a new standard for agent security evaluation, encouraging a shift from monolithic attack success rates to a more analytical, staged assessment of LLM agent safety. The open-source nature of the dataset further facilitates community-driven research and development in this critical area. This abstract is based on the provided arXiv submission.

AI/ML arXiv cs.AI

From Specification to Execution: AI Assisted Scientific Workflow Management

This work introduces an AI-assisted methodology for managing scientific workflows, aiming to automate the complex process from initial specification to distributed execution. The core contribution is a framework that bridges the gap between high-level scientific intent and low-level computational requirements, reducing the manual burden on researchers. The problem it addresses is the significant expertise and time required to design, implement, and debug scientific workflows, which often hinders reproducibility and scalability in research. The researchers, Komal Thareja, Hamza Safri, Rajiv Mayani, Anirban Mandal, and Ewa Deelman, have published this work on arXiv, categorized under Software Engineering and Artificial Intelligence.

The intended audience comprises software engineers and researchers who utilize or develop scientific workflow management systems. The system benefits those who may lack deep expertise in WMS or coding, enabling them to construct sophisticated, expert-level workflows and accelerate their research.

Two crucial technical innovations stand out. Firstly, the introduction of a structured, specification-driven workflow generation process separates workflow intent, design, and implementation. This explicit separation allows for validation at a higher level before code synthesis, enhancing transparency and enabling easier debugging of conceptual errors. Secondly, an LLM-based debugging agent is developed to diagnose and resolve failures across various system layers, significantly reducing manual troubleshooting effort. The integration with Pegasus, a mature WMS, via a Model Context Protocol (MCP) layer for unified submission, monitoring, and control is also a key element, facilitating robust distributed execution and user interaction.

This research enables the creation of more accessible and efficient scientific computing platforms. It suggests a future where AI agents can manage a substantial portion of the scientific workflow lifecycle, from abstract conceptualization to execution on distributed resources. The approach holds the potential to democratize scientific computing, allowing researchers to focus more on scientific discovery rather than the intricate details of computational pipeline management, thereby influencing the broader field of computational science and research infrastructure development. The content provided is an abstract only.

AI/ML arXiv cs.AI

SFT Overtraining Predicts Rank Inversion via Entropy Collapse Under RLVR

This work, "SFT Overtraining Predicts Rank Inversion via Entropy Collapse Under RLVR" by Siddharth Aphale and Kelly Liu, submitted to the Deep Learning for Code (DL4C) Workshop at ICML 2026, addresses a critical failure mode in the training pipeline for large language models, particularly in code generation tasks. The core contribution is the identification and explanation of why standard practices for selecting Supervised Fine-Tuning (SFT) checkpoints can lead to suboptimal performance during Reinforcement Learning from Human Feedback (RLHF) stages, specifically when using Proximal Policy Optimization (PPO), often referred to as GRPO in this context. This problem arises because excessive SFT can compress the distribution of generated outputs, leading to a "rank inversion" where the model's learned preference during RLHF is misaligned with its actual performance.

The problem this paper solves is the common heuristic of selecting the SFT checkpoint that achieves the highest pass@1 accuracy before RLHF. This heuristic fails when SFT overtrains, causing the model to produce outputs that are very similar to each other, thus collapsing the reward signal necessary for effective RLHF. The gap filled is a predictive mechanism for this failure, rooted in information-theoretic principles and observable during training. The intended audience is researchers and engineers working on the development and fine-tuning of large language models, especially for code generation, who benefit from more robust and predictable training methodologies.

Two crucial technical ideas are presented. First, the concept of "entropy collapse" describes how over-SFT leads to a reduction in the output distribution's entropy. This is quantified by the expected within-group advantage variance for binary rewards, which becomes negligible when the probability of a positive reward ($p$) falls below a critical threshold ($p^(g)$), rendering RLHF ineffective. Second, the paper demonstrates that pre-RL pass@1 and pre-RL entropy are positively correlated with RLHF outcomes. Specifically, on Qwen2.5-Coder-3B, higher pre-RL entropy ($ \rho = +0.69 $) was associated with better GRPO performance, while overtraining led to a significant drop in peak GRPO pass@10. The failure mode is characterized as rank inversion when early GRPO effectively drives the reward probability below $p^(g)$, leading to a reversal of performance ranking, and compression rather than inversion when $p$ remains above $p^*(g)$.

This research enables a more diagnostic approach to SFT checkpoint selection. By monitoring pre-RL entropy and later introducing an early GRPO entropy monitor, practitioners can identify high-risk checkpoints and potentially halt failing training runs before significant computational resources are wasted. The findings suggest that simple regularization techniques like KL divergence or label smoothing are insufficient to mitigate this specific failure mode, pointing to a deeper issue in the SFT-to-RLHF transition. This could influence future research into more effective SFT strategies or novel RLHF alignment techniques that are more resilient to output distribution compression. This analysis is based on the provided abstract, as the full paper was not available.

AI/ML arXiv cs.AI

Sparsity Curse: Understanding RLVR Model Parameter Space from Model Merging

This research, "Sparsity Curse: Understanding RLVR Model Parameter Space from Model Merging" by Chenrui Wu, Zexi Li, Jiajun Bu, Jiangchuan Liu, and Haishuai Wang, addresses a fundamental challenge in aggregating reasoning capabilities from independently trained Reinforcement Learning with Verifiable Reward (RLVR) models. The core contribution is the identification and characterization of a "sparsity curse" that hinders traditional model merging techniques when applied to RLVR. Unlike Supervised Fine-Tuning (SFT) which results in dense, principal parameter updates that naturally coalesce, RLVR induces sparse, off-principal updates. The authors demonstrate that this sparsity leads to parameter updates that are not only few but also spread widely and diverge, forming near-orthogonal structures. This divergence makes standard model merging methods, which rely on parameter space convergence, brittle and prone to catastrophic performance degradation.

The problem this work solves is the lack of a scalable, training-free method to combine the distinct reasoning abilities learned by separate RLVR models. This gap is crucial as RLVR shows promise over SFT for eliciting complex reasoning and resisting forgetting, making it a desirable technique for enhancing model intelligence. The intended audience for this paper includes researchers and engineers working with large language models, particularly those involved in fine-tuning, model merging, and the application of RL-based alignment techniques.

Two of the most important technical insights are the characterization of RLVR parameter updates as sparse and divergent, forming "near-orthogonal shortcuts" in parameter space, and the discovery that this geometric property is the root cause of merging failures. A third key technical contribution is the proposed solution: Sensitivity-aware Resolving Merging (SAR-Merging). SAR-Merging addresses the sparsity curse by employing Fisher Information-based arbitration to resolve conflicts in overlapping update regions, followed by magnitude-aware sparsification and rescaling to preserve delicate reasoning pathways.

This work enables the practical aggregation of diverse reasoning skills from multiple RLVR models, which was previously intractable. By overcoming the merging barrier, it paves the way for more efficient development of highly capable, multi-skilled AI systems without the need for extensive retraining. The findings are likely to influence the field by driving research into new model merging strategies tailored to the specific inductive biases of various training paradigms, moving beyond assumptions of parameter space convergence. This paper's abstract was submitted to arXiv on June 16, 2026, and accepted by KDD 2026.

AI/ML arXiv cs.AI

Dual-Channel Grounded World Modeling (DCGWM): Structural Prevention of Objective Interference Collapse via Heterogeneous External Grounding with Inward-Only Gradient Flow

This work, "Dual-Channel Grounded World Modeling (DCGWM): Structural Prevention of Objective Interference Collapse via Heterogeneous External Grounding with Inward-Only Gradient Flow," by Akshay Hazare, addresses a critical failure mode in state-of-the-art world models, particularly Joint Embedding Predictive Architectures (JEPAs), when attempting to learn from multiple, qualitatively different data sources. The problem it solves is the "Objective Interference Collapse" (OIC), where one learning signal, typically stronger or more constraining, can overpower and degrade the representation learning of another. This gap exists because current methods often struggle to jointly learn from heterogeneous grounding signals, such as physical dynamics (sparse, constraint-driven) and social-behavioral dynamics (diffuse, distribution-matching), without one corrupting the other, even with loss weighting.

The core contribution of DCGWM is a novel architectural design that structurally prevents OIC. This is achieved through a partitioned latent space, separating physical (Z_p) and behavioral (Z_b) representations. Crucially, gradient flow is restricted to be "inward-only," meaning physical grounding updates only Z_p, and social-behavioral grounding updates only Z_b. An Inter-Channel Interface Module facilitates task-level coupling without propagating gradients between subspaces. This design is theoretically underpinned by three key results: the partition eliminates the direct gradient interference pathway responsible for OIC; each grounded subspace benefits from the inherent anti-collapse properties of its specific alignment objective (e.g., VICReg-style for physical, distribution-matching for behavioral); and the generative rendering layer is architecturally isolated from the latent world model to prevent interference with the learned representations.

This approach is intended for researchers and engineers working on advanced representation learning for AI systems, particularly those aiming to integrate diverse forms of knowledge and prediction. The intended beneficiaries include developers of embodied AI agents, autonomous systems requiring complex reasoning about both physical and social environments, and researchers pushing the boundaries of generative models.

Looking forward, DCGWM enables the development of more robust and capable world models that can effectively leverage heterogeneous data sources without interference. This could lead to significant advancements in areas requiring sophisticated understanding of complex systems, such as advanced robotics, simulated environments for training AI, and human-AI interaction. The theoretical guarantees and architectural separation offer a principled way to scale world modeling to more diverse and challenging real-world scenarios. This position paper establishes the architectural framework; experimental validation is pending.

AI/ML arXiv cs.AI

Spotlight: Synergizing Seed Exploration and Spot GPUs for DiT RL Post-Training

Ruiqi Lai and colleagues, affiliated with multiple institutions, present "Spotlight: Synergizing Seed Exploration and Spot GPUs for DiT RL Post-Training," a novel system designed to drastically reduce the prohibitive computational cost of reinforcement learning (RL) post-training for Diffusion Transformers (DiTs). This work addresses a critical bottleneck for researchers and engineers working with large-scale generative models, particularly those leveraging DiTs for tasks requiring fine-tuning through RL. The core contribution lies in an innovative approach to harness the cost-efficiency of spot GPUs by cleverly managing the exploration phase of RL post-training, a phase that typically requires extensive computational resources and has been a significant barrier to broader adoption and experimentation.

The problem Spotlight solves is the exorbitant expense associated with achieving high-quality results from DiT models when fine-tuned using RL. Traditional methods demand thousands of high-end GPUs, making advanced research and development inaccessible for many. Existing cost-reduction strategies have limitations: seed exploration, which selects informative samples to accelerate convergence, adds to the critical training path, while spot GPUs, offering substantial cost savings, are underutilized because DiT rollouts tend to finish simultaneously, preventing effective pipelining with training. Furthermore, spot GPU preemptions can disrupt crucial distributed training setups like Sequence Parallelism (SP), causing significant recovery overhead.

Spotlight introduces two central technical ideas. First, it demonstrates that exploration can tolerate stale model weights. By performing exploration using weights from the previous training iteration, the relative ranking of random seeds is preserved, enabling exploration to be effectively offloaded to idle spot GPUs without impacting the main training loop's criticality. This decouples exploration from the immediate state of the evolving model. Second, Spotlight enables efficient recovery from spot GPU preemptions by implementing a dynamic SP reconfiguration. This mechanism reuses on-node state, reducing recovery times from minutes to mere seconds by leveraging techniques like intra-node weight copying and persistent schedulers. These insights are operationalized through three key techniques: a bandit-based exploration planner to optimize reward variance within time constraints, elastic sequence parallelism for on-the-fly SP group adjustments, and a preemption-aware request scheduler to manage load and commit in-flight states upon preemption.

This research, published on arXiv, is primarily intended for ML engineers, distributed systems researchers, and AI scientists involved in training and fine-tuning large generative models, especially DiTs. The immediate benefit is a significant reduction in training costs and time, enabling more extensive hyperparameter tuning, broader experimentation with RL strategies, and potentially the deployment of more sophisticated DiT-based applications. Looking ahead, Spotlight's synergistic approach to resource utilization and fault tolerance in distributed deep learning training has the potential to influence the design of future training frameworks for large-scale AI models, making advanced capabilities more accessible and efficient across the field. This analysis is based on the provided abstract.

AI/ML arXiv cs.AI

Essential Subspace Merging for Multi-Task Learning

Longhua Li, Lei Qi, Xin Geng, and Qi Tian from an unspecified institution, published on arXiv under the Computer Science AI and Machine Learning categories, introduce Essential Subspace Merging (ESM). This work addresses the critical challenge of inter-task interference that arises when attempting to combine the capabilities of multiple models, each fine-tuned from a common pre-trained checkpoint, into a single unified model for multi-task learning. The core contribution is a novel, training-free method for merging these task-specific models.

The primary problem ESM solves is the degradation of performance that occurs when naively aggregating parameters from independently fine-tuned models. This interference stems from conflicting parameter updates across tasks. The researchers observe that the significant parameter changes induced by training on a specific task are largely concentrated within a low-dimensional subspace, termed the "essential subspace," defined by the principal directions of output shifts. Conversely, directions outside this subspace carry minimal task-specific energy but can accumulate noise and cause detrimental interference during merging.

Two key technical ideas underpin ESM. First, Essential Subspace Decomposition (ESD) is proposed, which analyzes task updates and decomposes them based on the principal components of their activation shifts. This decomposition allows for the identification and isolation of the essential subspace for each task. Second, the Essential Subspace Merging (ESM) method leverages this decomposition. It achieves merging by orthogonalizing and fusing these identified essential components, effectively creating a compact multi-task model that preserves task knowledge while minimizing interference. An extension, ESM++, further refines this by decomposing task-specific residuals into low-rank experts and employing prototype-based routing to dynamically select the most relevant expert during inference, offering a training-free dynamic merging approach.

This research enables more efficient and effective multi-task learning by providing a principled way to combine specialized models. It is particularly beneficial for researchers and engineers working on model compression, knowledge distillation, and efficient deployment of models across diverse applications. The demonstrated ability to reduce inter-task interference and preserve task knowledge suggests significant implications for scaling multi-task learning systems and developing more versatile AI agents. This work is presented as an abstract only.

AI/ML arXiv cs.AI

STARE: Surprisal-Guided Token-Level Advantage Reweighting for Policy Entropy Stability

STARE, or Surprisal-Guided Token-Level Advantage Reweighting for Policy Entropy Stability, addresses a critical challenge in training large language models (LLMs) with reinforcement learning (RL), specifically the prevalent issue of policy entropy collapse. This phenomenon hinders the model's ability to explore and adapt, leading to suboptimal performance. The work, authored by Haipeng Luo, Qingfeng Sun, Songli Wu, Can Xu, Wenfeng Deng, Han Hu, and Yansong Tang, was published on arXiv in Computer Science (cs.AI, cs.LG, cs.CL). It is targeted at researchers and engineers in machine learning and natural language processing who are involved in LLM fine-tuning and RL.

The core contribution of STARE is a novel method for stabilizing the training of RL algorithms like GRPO when applied to LLMs. The problem it solves is that existing methods often experience a premature reduction in policy entropy, effectively causing the model to become overly deterministic and cease exploration. STARE's innovation lies in a first-order gradient analysis that reveals a mismatch in token-level credit assignment. This analysis decomposes per-token entropy variation into the product of trajectory-level advantage and an "entropy sensitivity function" related to the next-token distribution, highlighting an "advantage-surprisal four-quadrant structure."

The most important technical ideas are: (1) Surprisal-guided token subset identification: STARE identifies subsets of tokens that are "entropy-critical" by analyzing batch-internal surprisal quantiles. This allows for targeted intervention rather than blanket regularization. (2) Selective advantage reweighting: The effective advantages of these identified critical tokens are reweighted, precisely adjusting their influence on the policy update. (3) Target-entropy closed-loop gate: A mechanism is incorporated to ensure stable entropy regulation by actively managing it towards a predefined target.

The results demonstrate that STARE effectively sustains stable RL training across various model scales (1.5B to 32B parameters) and task families (Short CoT, Long CoT, and Multi-Turn Tool Use) over thousands of training steps, keeping policy entropy within a desired band. The improvements in accuracy (4%-8% over baselines on AIME24 and AIME25) are attributed to a sustained balance between exploration and exploitation, as evidenced by the synchronized growth of reflection tokens and response length. This work enables more robust and effective RL fine-tuning of LLMs, potentially unlocking more complex reasoning capabilities. Going forward, STARE's insights into the relationship between advantage, surprisal, and entropy dynamics could influence the development of more sophisticated RL algorithms for generative models, leading to broader applications in areas requiring nuanced and exploratory language generation. The provided content is an abstract only.

AI/ML arXiv cs.AI

Explaining Attention with Program Synthesis

Amiri Hayes, Belinda Li, and Jacob Andreas from the Massachusetts Institute of Technology have introduced a novel method for explaining the internal workings of transformer language models by approximating attention heads with synthesized programs. This work addresses a critical gap in interpretable deep learning: the difficulty in translating complex, opaque neural computations into human-understandable symbolic descriptions. The core contribution is a scalable pipeline that reverse-engineers attention mechanisms into executable Python code.

The problem this research solves is the "black box" nature of deep learning models, particularly transformers, where understanding why a model makes a certain prediction is challenging. This lack of transparency hinders debugging, deployment in high-stakes applications, and fundamental research into model behavior. By generating symbolic programs, this approach moves towards "symbolic transparency."

The intended audience consists of software engineers and AI researchers who work with or study transformer models. The beneficiaries include anyone seeking to understand, debug, or verify the behavior of these powerful models, as well as those developing more interpretable AI systems.

Key technical ideas include:

  1. Attention Matrix Summarization and Prompting: The method first computes attention matrices for a given head across training data. Crucially, it then uses a language model, prompted with a summary of these attention patterns and the corresponding input text, to generate candidate Python programs. This leverages the generative capabilities of LLMs to hypothesize symbolic explanations for observed neural behavior.
  2. Program Synthesis and Re-ranking: Candidate programs are synthesized to reproduce attention patterns. These are then rigorously re-ranked based on their ability to predict attention behavior on held-out data, ensuring the generated programs are not just superficially similar but functionally representative.
  3. Programmatic Head Replacement: A significant result is demonstrating that these synthesized programs can replace actual attention heads within transformer architectures. This replacement incurs minimal performance degradation (e.g., a 16% average perplexity increase across tested models and benchmarks), validating the functional fidelity of the synthesized explanations.

This work enables a future where components of deep neural networks can be replaced with interpretable, executable code. It suggests a path toward building more transparent and verifiable AI systems, potentially influencing the development of hybrid neuro-symbolic architectures and formal methods for AI verification. The findings presented are based on an abstract.

AI/ML arXiv cs.AI

Towards Understanding What State Space Models Learn About Code

This research, "Towards Understanding What State Space Models Learn About Code" by Jiali Wu, Abhinav Anand, Shweta Verma, and Mira Mezini, published on arXiv (cs.AI), offers the first systematic analysis of how State Space Models (SSMs) process and learn from code, a critical domain for AI. The work addresses a significant gap: while SSMs have demonstrated competitive performance against Transformers in code understanding tasks, their internal learning mechanisms have remained opaque. This lack of transparency hinders further development and trust in these models for software engineering applications. The intended audience comprises AI researchers and software engineers working with or considering SSMs for code-related tasks, such as code generation, summarization, and vulnerability detection.

Two pivotal technical contributions stand out. First, the authors reveal that SSMs, during pre-training, appear to capture both syntactic and semantic code structures more effectively than Transformers. This suggests a potential advantage for SSMs in foundational code representation. Second, and perhaps more crucially, their analysis highlights a phenomenon where SSMs exhibit a "spectral shift toward short-range dependencies" during fine-tuning. This means that while SSMs start with a broad understanding, they can narrow their focus to local patterns when adapting to specific tasks, potentially leading to the forgetting of longer-range, more global program relationships. To facilitate this analysis, the researchers introduce SSM-Interpret, a novel frequency-domain framework that effectively probes and visualizes these internal learning dynamics.

The implications of this work are substantial. By demystifying SSM behavior in code, it provides a foundation for developing more robust and predictable code models. The identification of the spectral shift during fine-tuning directly informs architectural modifications. Indeed, the authors leverage these insights to propose design changes that yield up to a +6 MRR improvement on the NLCodeSearch benchmark. This strongly suggests that a deeper understanding of SSM learning dynamics, particularly in the context of complex data like code, can directly translate into tangible performance gains and guide the future evolution of SSM architectures for natural language processing and software engineering tasks. The analysis, based on an abstract, provides valuable insights into the model's learning process.

AI/ML arXiv cs.AI

LLM-Evolved Domain-Independent Heuristics for Symbolic AI Planning

This work, "LLM-Evolved Domain-Independent Heuristics for Symbolic AI Planning" by Elliot Gestrin and Jendrik Seipp, published on arXiv, introduces a novel method for automatically generating effective heuristics for symbolic AI planning systems. The core contribution is the creation of domain-independent heuristics that outperform, in certain aspects, decades of hand-engineered heuristics. This addresses the long-standing challenge of developing generalizable heuristics that can boost the performance of AI planners across a wide range of problem domains, rather than being specialized to a single one. The intended audience is researchers and engineers working in AI planning, automated reasoning, and related fields who can leverage these improved heuristics to build more efficient and capable planning systems.

Two critical technical innovations underpin this research. Firstly, the authors employ an evolutionary search process guided by a large language model (LLM). The LLM's role is to propose mutations to existing C++ heuristic code, fostering exploration of novel heuristic structures. This LLM-driven mutation mechanism is coupled with the MAP-Elites algorithm, a technique for maintaining a diverse archive of candidate solutions. The archive is indexed by key performance metrics, specifically "informedness" (how well the heuristic guides the search) and "speed" (the computational overhead of calculating the heuristic value). Secondly, fitness is evaluated by a blend of heuristic coverage (the proportion of tasks solved) and average solving time, allowing for a robust assessment of heuristic utility across various planning problems. A key finding is that starting the evolutionary process from a simple "blind" heuristic, rather than a more complex hand-engineered one, paradoxically leads to superior results, even when the evolved heuristic is an FF variant. The authors also observe that the LLM's reasoning effort impacts the compilation rate of generated code more significantly than the inherent quality of the successful compilations.

This research enables the creation of planners that can tackle a broader spectrum of tasks more efficiently, without requiring manual heuristic engineering for each new domain. By producing heuristics as plain C++ code, they integrate seamlessly into existing planners, inheriting their theoretical guarantees of soundness and completeness. This work signals a promising direction for the automated discovery of AI components, potentially accelerating progress in domains requiring sophisticated planning capabilities, such as robotics, logistics, and game playing. It suggests that LLMs, when combined with robust evolutionary search and appropriate performance metrics, can transcend their typical text-generation roles to contribute to algorithmic innovation.