AI/ML arXiv cs.AI

Autonomous Topology Mutation: Safe Runtime Restructuring for Multi-Agent LLM Systems with Capability, State, and Shadow Invariants

Bronislav Sidik, Chaya Levi, and Nizzan Kimhi of the Technion—Israel Institute of Technology present Autonomous Topology Mutation (ATM), a runtime mechanism for restructuring multi-agent Large Language Model (LLM) systems. The core contribution is a safe, automated method to dynamically alter the internal architecture of these systems when individual agents experience performance degradation or overload, addressing a critical limitation of existing frameworks that fix topology at boot time. This work is crucial for building more robust, scalable, and adaptable LLM-powered applications, particularly those with complex workflows or unpredictable task demands. The intended audience includes software engineers and researchers working on multi-agent LLM systems, distributed AI, and runtime adaptation, who will benefit from systems that can self-optimize under stress.

Two paramount technical ideas underpin ATM. First, a telemetry-driven Bottleneck Index, comprising six signals like queue depth, context thrash, and tool error rates, serves as the detection mechanism for agent overload. This index is calibrated against system warmup to avoid false positives. Second, three safety invariants—capability monotonicity, state-routing completeness, and shadow-before-live validation—ensure that any structural change is performed without compromising system integrity or data privacy. Capability monotonicity ensures that specialized sub-agents collectively retain or expand the parent agent's capabilities. State-routing completeness mandates that all necessary state is transferred correctly, with privacy-level-aware routing preventing unauthorized data exposure. Shadow-before-live validation involves testing the new topology in a simulated environment before routing live traffic, preventing unforeseen disruptions. The research demonstrates significant improvements in task success rates, from 3.3% to 61.7% in specific scenarios, and dramatically reduces high-privacy memory exposure. The overhead introduced by ATM's invariants is minimal, less than 500 microseconds at the 99th percentile on the agent hot path.

This work enables the development of LLM systems that can dynamically adapt to varying workloads and agent performance, fostering greater resilience and efficiency. Going forward, ATM could influence the design of future multi-agent LLM frameworks, promoting a shift towards more emergent and self-optimizing architectures. The publication is on arXiv, under cs.AI. This analysis is based on the provided abstract as a full paper was not available.

AI/ML arXiv cs.AI

MiniCache: Reusable Program Caching with Small Model Interfaces for Efficient LLM Inference

MiniCache, presented by researchers Jingquan Chen, Jinghua Piao, Jie Feng, Shaogang Hu, and Yong Li on arXiv, introduces a reusable program caching framework designed to significantly reduce the inference cost of large language models (LLMs) in program-aided reasoning and structured task execution. The core contribution is the transformation of Program-of-Thought (PoT) outputs into parameterized cache objects, allowing for the reuse of computations across structurally similar requests. This work addresses the substantial inference overhead inherent in LLM applications, particularly those requiring complex reasoning or multi-step task execution, by filling the gap in efficient, reusable computation for LLM-driven programs. The intended audience includes software engineers and researchers working with LLMs in areas like agentic systems, code generation, and structured data manipulation, who stand to benefit from reduced latency and increased throughput.

Key technical ideas underpinning MiniCache include the parameterization of PoT programs into cache objects, enabling direct lookups for identical or highly similar program structures. Furthermore, MiniCache leverages a small, lightweight model not to replace the target LLM but to act as an efficient interface. This small model is responsible for semantically extracting variables from incoming requests, facilitating cache hit identification, and for speculative drafting during target-LLM generation on cache misses. This dual role of the small model is crucial for both efficiently identifying reusable computations and for priming the larger LLM, thereby minimizing expensive target-LLM invocations while maintaining task quality. Experimental results on datasets like WebShop, Formula, and CodeTAT-QA demonstrate substantial improvements, including up to 3.1x lower latency and 2.8x higher throughput under parallel serving.

This research enables a new paradigm for efficient LLM deployment where computational primitives are cached and reused, akin to traditional software caching mechanisms. It suggests that small models are most effective not as standalone replacements but as specialized interface components that unlock the power of reusable caching for complex LLM tasks. Going forward, MiniCache's approach could influence the design of LLM-powered applications, promoting architectures that prioritize inference efficiency through intelligent caching and the strategic deployment of smaller, specialized models. This abstract-only submission highlights the potential for significant performance gains by abstracting and reusing LLM-generated program logic.

AI/ML arXiv cs.AI

Autonomous disproofs of the sum-product conjecture over $\mathbb R$ with GPT-5.5 Pro

The autonomous generation of mathematical proofs has advanced with the development of a problem-agnostic agent built on GPT-5.5 Pro, which successfully disproved the Erdős–Szemerédi sum-product conjecture over $\mathbb R$. Authored by Yichen Huang and published on arXiv (cs.AI), this research introduces a system capable of independently producing diverse, mathematically rigorous disproofs of a fundamental conjecture in additive combinatorics. The work targets AI researchers, mathematicians, and computer scientists focusing on automated reasoning, formal verification, and LLM-based theorem proving.

This development addresses a critical bottleneck in artificial intelligence: transitioning large language models from text synthesis and code generation to advanced, multi-step mathematical reasoning without human-in-the-loop intervention. Architecturally, the agent relies on a three-stage prompting pipeline consisting of proof-plan proposal, proof construction, and critical review. Operating over an average of 132.4k reasoning tokens per trial, the system successfully generated valid disproofs in seven out of eight independent trials. In the single unsuccessful run, the agent's internal review mechanism correctly identified an unresolved logical gap rather than hallucinating a false proof.

The resulting proofs demonstrate significant mathematical diversity, showcasing the agent's ability to explore distinct mathematical landscapes. Some generated arguments utilize constructions close to existing unit-based methods, while others formulate entirely different approaches using $L^p$-type regions of algebraic integers. Because the human disproof of the sum-product conjecture over $\mathbb R$ was a very recent breakthrough, this benchmark serves as a clean, data-contamination-free case study for testing the true zero-shot reasoning bounds of frontier models.

Going forward, this methodology demonstrates that structured, agentic pipelines utilizing dense reasoning tokens can autonomously tackle open mathematical questions. By proving capable of discovering multiple, structurally distinct solutions, such agentic frameworks could soon act as collaborative discovery engines in pure mathematics, accelerating the identification of counterexamples and novel mathematical structures. Note that this analysis is based on the paper's published abstract and metadata.

AI/ML arXiv cs.AI

NVIDIA-labs OO Agents: Native Python Object-Oriented Agents

The development of autonomous AI agents has historically been fragmented, split across disconnected abstractions like prompt templates, JSON tool schemas, callback systems, and complex workflow graphs. To bridge this gap, a research team from NVIDIA-labs, led by Paul Furgale and colleagues, introduced NVIDIA Object-Oriented Agents (NOOA). Published on arXiv in July 2026, NOOA is a model-agnostic Python framework designed for software engineers and AI researchers that unifies agent architecture under a single, native programming paradigm: the Python class. By representing an agent directly as a standard Python object, NOOA allows developers to write, test, trace, and refactor agentic systems using established software engineering methodologies rather than specialized domain-specific languages.

At the core of NOOA is the "agent-as-a-Python-object" model. An agent's fields define its state, its docstrings act as natural-language prompts, and its type annotations establish execution contracts. The framework elegantly bridges deterministic and non-deterministic execution: any method written with a standard code body remains classical, deterministic Python, whereas any method defined with a simple ellipsis (...) is dynamically resolved and completed at runtime by an LLM-driven agent loop. Mechanistically, NOOA is the first framework to integrate six distinct model-facing abstractions into a unified interface. These include strictly typed input/output, pass-by-reference execution over live Python objects, treating code itself as an action, programmable loop engineering, explicit object state representation, and model-callable harness APIs for event and context management.

Evaluation of the framework demonstrates that contemporary language models navigate this native Python interface effectively. NOOA-based agents achieved robust performance on highly demanding agentic and reasoning benchmarks, including SWE-bench Verified, Terminal-Bench 2.0, and ARC-AGI-3. By treating LLM calls not as external API integrations but as native runtime method resolutions, NOOA positions agent development to align seamlessly with traditional software development pipelines. This paradigm shift could fundamentally influence the field by establishing pythonic object-oriented design as the standard interface for agent-computer interaction, simplifying debugging and accelerating the deployment of reliable, enterprise-grade AI systems. Note that the analysis presented here is based on the published abstract of the paper.

AI/ML arXiv cs.AI

Profiling Lightweight Large Language Models

Deploying lightweight large language models (LLMs) on resource-constrained edge devices, mobile platforms, and personal computers requires a delicate balance between computational efficiency and task accuracy. Traditional evaluations often rely on static proxy metrics, such as parameter count or floating-point operations (FLOPs), which fail to capture actual hardware behavior or relate directly to model precision. To address this gap, researchers Tomohiro Harada, Enrique Alba, and Gabriel Luque introduced a novel experimental framework in their paper "Profiling Lightweight Large Language Models," published on arXiv in July 2026. This framework directly measures hardware-level performance across four key dimensions: Precision, execution Time, peak Memory usage, and Energy consumption (PTME). By linking physical resource utilization directly to task accuracy, the PTME framework enables a more realistic assessment of model viability in edge environments.

The authors validated their methodology by executing a representative suite of lightweight LLMs on a controlled desktop platform configured to simulate edge-class resource envelopes, testing them on benchmarks for code generation, mathematical reasoning, and multi-task understanding. The empirical results reveal three critical insights. First, while static proxy descriptors like parameter size correlate reasonably well with physical inference costs, they are entirely decoupled from and fail to predict actual task precision. Second, restricting the hardware resource envelope escalates inference costs without yielding any improvement in precision, with execution latency suffering a much sharper penalty than energy consumption—a penalty that disproportionately affects larger model configurations. Finally, the research demonstrates that no single lightweight model dominates across all PTME dimensions. Through Pareto front analysis, the authors identified optimal, non-dominated configurations that would otherwise remain hidden under traditional, single-metric (accuracy-only or efficiency-only) selection processes.

This work is highly relevant for software engineers, systems architects, and machine learning researchers tasked with optimizing and deploying models in hardware-constrained environments. By demonstrating that selecting models based purely on size, latency, or standalone accuracy can lead to suboptimal deployment decisions, this profiling methodology establishes a new paradigm for edge-native AI benchmarking. Moving forward, the PTME approach is poised to influence the field by shifting the optimization focus from purely theoretical metrics to multi-objective hardware-in-the-loop evaluations, paving the way for automated, hardware-aware model selection engines. This analysis is based on the published abstract of the research paper.

AI/ML arXiv cs.AI

Is Deep Research Reliable? Misleading Knowledge Induces False Conclusions

This research, authored by Pengyu Zhu, Lijun Li, Longju Yang, and Sen Su, and published on arXiv, introduces a significant concern regarding the reliability of "Deep Research" agents. These agents, designed to extend Large Language Model (LLM) assistants into complex, long-horizon workflows encompassing planning, information retrieval, evidence synthesis, and report generation, are shown to be vulnerable to misinformation. The core contribution is the identification and empirical demonstration of a failure mode where seemingly credible yet factually misleading knowledge can propagate through these intricate workflows, leading to the adoption of false conclusions in final reports. This work addresses a critical gap in the current understanding and development of AI agents for sophisticated research tasks, particularly in open information environments where the veracity of data is not guaranteed.

The paper introduces "MisKnow-Agent," a novel framework for constructing and validating misleading knowledge instances tailored for Deep Research tasks. This framework allows for the generation of misleading data with controllable authority levels and stylistic characteristics, enabling the creation of a dataset of 5,933 quality-controlled instances. Crucially, the authors demonstrate through extensive experiments with both open-source and closed-source Deep Research agents that even limited exposure to this misleading knowledge can significantly induce the adoption of false conclusions. A key technical finding is the divergence between focused corpus validation, where search-enabled verifier models successfully identify misleading instances, and their actual adoption during long-horizon research workflows. This highlights a disconnect between superficial vetting and the practical integration of evidence in complex reasoning processes. The authors also evaluate pre- and post-research defense mechanisms, concluding that while these strategies mitigate, they do not fully prevent false conclusion adoption.

The implications of this work are profound for software engineers and researchers developing and deploying advanced AI agents. It suggests that achieving reliable Deep Research necessitates not just improvements in core LLM capabilities like planning, retrieval, evidence integration, or report generation, but also robust evidence verification and correction mechanisms embedded within the entire framework. This research, derived from an abstract only, thus points towards a future where AI agents require a deeper understanding of knowledge provenance and a more sophisticated approach to critical evaluation to be truly trustworthy for complex research endeavors.

AI/ML arXiv cs.AI

Naju: A Native Discrete State-Space Model with Independent Retention and Writing for Long-Sequence Memory

Hyuk Lim and Seunghyun Yoon, affiliated with an unspecified institution and presenting their work on arXiv, introduce Naju, a Native Adaptive Junction Unit. This represents a novel discrete state-space model (SSM) specifically designed for efficient long-sequence memory processing. The core contribution lies in its architectural innovation that allows for independent control over memory retention and writing, addressing a fundamental trade-off that limits current efficient baselines. Existing models, including continuous-time SSMs like Mamba which rely on zero-order-hold discretization, often excel at either retaining information or actively overwriting stale data, but not both simultaneously. Naju circumvents this by directly parameterizing the discrete recurrence relation, obviating the need for discretization.

The problem Naju addresses is the inherent difficulty in maintaining a stable and accessible memory state over extended sequences. This is crucial for tasks requiring long-range dependencies, such as language modeling or complex reasoning. The gap filled is the lack of models that can achieve both near-lossless retention of historical information and the ability to rapidly update or discard outdated information without sacrificing performance. The intended audience is software engineers and researchers working with recurrent neural networks, sequence modeling, and artificial intelligence more broadly, particularly those dealing with computationally constrained environments or very long input sequences.

Two principal technical ideas underpin Naju. Firstly, it factorizes the recurrent update into an explicit discrete pole, effectively a learned forget gate ($f_n$), an independent write gain ($i_n$), and input-dependent write/read maps. The sigmoid-parameterized pole ensures that $0 < f_n < 1$, making each local coordinate Schur-stable by construction. This architecture is designed to satisfy fading-memory and BIBO bounds under uniform boundedness, without requiring stability regularizers. Secondly, Naju tackles the structural limitation of coupled designs, where a single gate often enforces a constraint like $|r| + w \le 1$ for retention ($r$) and write gain ($w$). This means strong retention necessitates weak writing. By decoupling the forget gate ($f_n$) from the write gain ($i_n$), Naju removes this constraint, enabling simultaneous strong retention and writing.

Empirically, Naju demonstrates remarkable performance on a diagnostic suite, maintaining strength in both retention and overwriting at four times the training sequence length. It also shows competitive or superior results on standard benchmarks like WikiText-103, the Long Range Arena, and multi-query associative recall, outperforming Mamba baselines and remaining competitive with Transformers while preserving linear-time and linear-memory scaling. Going forward, Naju's independently controllable retention and writing mechanisms present a promising new paradigm for long-sequence memory in neural networks. This could lead to more efficient and powerful models for a wide range of applications requiring long-term context understanding, potentially influencing the design of future recurrent architectures and memory systems. This analysis is based on the abstract provided.

Hardware/Chips arXiv cs.AI

Identifying Good Rules for Efficient SAT Encodings of Single-Constant Multiplication Using Machine Learning

The optimization of Single Constant Multiplication (SCM)—an NP-hard hardware design task that decomposes a fixed constant into additions, subtractions, and bit-shifts—faces a bottleneck: existing dynamic programming methods generate near-optimal Boolean satisfiability (SAT) encodings but suffer from prohibitive computational costs when scaling to large constants. To resolve this, Chufeng Jiang and Neng-Fa Zhou from the Graduate Center of The City University of New York developed a novel neuro-symbolic framework. Published in the Proceedings of the International Conference on Logic Programming (ICLP 2026), this work targets hardware design engineers, Electronic Design Automation (EDA) tool developers, and SAT practitioners seeking highly efficient multiplication logic on chips.

The core of this framework is a hybrid approach that integrates machine learning into a symbolic search. Specifically, the method uses a Graph Neural Network (GNN) to predict the most promising operator types directly from constant decompositions. The confidence scores generated by the GNN are then utilized as heuristics to prune suboptimal search paths in the symbolic search space. When tested on unseen 17-to-32-bit constants, this learning-guided strategy achieved one to two orders of magnitude reduction in SAT encoding time and reduced memory consumption by more than 97%. Furthermore, the framework reduced search branching by an order of magnitude while maintaining a near-optimal count of addition operations.

This research demonstrates how deep learning can be successfully married to exact symbolic solvers to mitigate the state-explosion problem in hardware synthesis. By significantly accelerating SCM encoding, this methodology paves the way for faster, more scalable EDA pipelines and suggests that similar learning-guided heuristics can be applied to other complex logic synthesis tasks. Please note that this analysis is based on the published abstract and metadata of the research paper.

AI/ML arXiv cs.AI

ICAE-Bench: Evaluating Coding Agents as Interactive Project Builders

The research introduces ICAE-Bench, a novel benchmark designed to evaluate coding agents in dynamic, interactive project-building scenarios. This work addresses a critical gap in existing benchmarks, which typically assess agents on static, fully specified tasks. The core contribution lies in simulating a realistic "vibe-coding" workflow, where agents must translate fuzzy product requirements into functional software by orchestrating a suite of capabilities including planning, requirement clarification, tool utilization, debugging, and repository-level construction. The work is authored by Zhongyuan Peng and a team of eleven researchers from multiple institutions, and has been submitted to arXiv in the Computer Science, Artificial Intelligence category. ICAE-Bench is intended for researchers and developers in the field of AI for software engineering, particularly those focused on developing advanced coding agents.

The benchmark's realism and evaluability are grounded in three key technical designs. First, task ambiguity is derived from precise, executable behaviors of real open-source repositories, ensuring that requirements are grounded and not arbitrary. Second, a User Agent simulates interaction, guided by User Agent Data that reveals implicit constraints without inventing new requirements or exposing implementation details, thus ensuring high-quality and reproducible user feedback. Third, evaluation employs standardized black-box tests combined with multi-dimensional diagnostics. These diagnostics assess not only functional correctness but also semantic and API similarity to the original project, structural fidelity, design quality, and the overall quality of the agent's interaction.

ICAE-Bench enables a more accurate assessment of coding agents' readiness for complex, real-world software development tasks. By shifting evaluation from isolated code completion to iterative project building, it promotes the development of agents capable of nuanced problem-solving and adaptive behavior. This likely influences the field by pushing research towards agents that can effectively collaborate with human developers in dynamic environments and handle the inherent uncertainties of software projects. The provided content is an abstract, not the full research paper.

AI/ML arXiv cs.AI

SPORD: A Simulation-Propose-then-OR-Dispose Approach for Supply Chain Planning

The Simulation-Propose-then-OR-Dispose (SPORD) framework, developed by Jiayin He, Yutong Pan, Sen Yang, Ningxuan Kang, Yongzhi Qi, Jianshen Zhang, Wei Qi, and Zuo-Jun Max Shen and published in arXiv cs.AI, addresses the fundamental challenges of operational fragmentation, computational intractability, and implementation hurdles in supply chain planning. Historically, enterprise logistics has relied on isolated, bespoke models that require weeks of manual design and fail to scale when managing millions of stock-keeping units (SKUs) across complex fulfillment networks. SPORD solves this by decoupling the planning process into two distinct stages: a simulation engine that proposes operationally valid candidate paths, and an operations research (OR) integer program that disposes of non-optimal choices to select the global optimum.

Designed for software engineers, operations researchers, and system architects, SPORD has been deployed within the NetSim platform to optimize large-scale end-to-end services. The architecture relies on three primary technical mechanisms. First, a matrix-vectorized CPU/GPU accelerated simulation architecture achieves a 10-to-100-fold speedup over serial simulation methods, allowing the system to absorb complex, idiosyncratic business rules directly within the simulation layer. Second, a specialized list scheduling algorithm slashes coupled-order processing times from hours to minutes. Finally, an intelligent diagnosis engine forms a closed loop that generates transparent, verifiable outputs. Empirically, the framework has optimized operations for over 20,000 suppliers, reducing the cross-regional fulfillment rate from 6.1% to 4.9% and cutting monthly carbon emissions by approximately 5,745 tCO2e.

Going forward, SPORD redefines the role of simulation from passive post-hoc monitoring to active, upstream planning. By decoupling highly specific business logic from the mathematical optimization solver, it establishes a modular blueprint where new planning tasks require simple configuration rather than complete code reconstruction. This modularity dramatically lowers the engineering overhead of deploying and maintaining large-scale optimization models in production. Please note that this analysis is based on the published abstract of the research paper.

AI/ML arXiv cs.AI

Agentic coding without the cloud: evaluating open-weight large language models on longitudinal data preparation tasks

The emergence of local, open-weight large language models (LLMs) offers a viable pathway for automated software engineering in privacy-restricted domains. In the paper "Agentic coding without the cloud: evaluating open-weight large language models on longitudinal data preparation tasks," published on arXiv (cs.AI) in July 2026, authors Mack Nixon, Liam Wright, Yevgeniya Kovalchuk, Alison Fang-Wei Wu, Martin Danka, Andy Boyd, and David Bann introduce an open-source evaluation framework designed to benchmark local AI agents on complex data preparation tasks. This work addresses a critical bottleneck in longitudinal population studies: the labor-intensive process of data cleaning and harmonization. While cloud-based LLMs are frequently used for code generation, strict data governance policies prohibit sending sensitive personal data to third-party cloud services. By targeting local, consumer-grade hardware deployment, this research provides data engineers and clinical researchers with a secure alternative that maintains compliance while leveraging agentic code generation.

The technical core of the framework consists of three key components: a curated ground-truth dataset comprising R cleaning scripts from six sweeps of a British cohort study, clearly defined task categories such as category harmonization and multi-wave merging, and automated testing routines that execute the LLM-generated R code to validate the resulting datasets. The researchers tested the framework across 20 distinct data preparation tasks involving the creation of 102 variables. The benchmark results reveal that current-generation open-weight models in the 31-35 billion parameter range, when run locally on consumer-grade hardware, are highly capable, achieving an average task completion rate of up to 87.9%. This demonstrates that mid-sized local models can reliably handle complex longitudinal data transformations without cloud telemetry.

This work signals a shift toward decentralized, secure AI engineering workflows. By proving that sub-40B parameter models can nearly saturate rigorous data preparation benchmarks, the framework enables institutions handling highly sensitive healthcare, demographic, or financial data to adopt agentic workflows safely. This is poised to accelerate longitudinal research timelines and establish standardized, reproducible local pipelines for data curation. Note that this analysis is based on the published abstract and metadata of the research paper.

AI/ML arXiv cs.AI

Break Through the Compression Bottleneck: From Theory to Practice

This research, "Break Through the Compression Bottleneck: From Theory to Practice," by Xiusheng Huang, Lu Wang, Yequan Wang, Jun Zhao, and Kang Liu, addresses the critical challenge of scaling large language models (LLMs). As LLMs grow in parameter count, their computational and memory demands become prohibitive. Existing compression techniques, primarily low-rank decomposition and quantization, offer significant reductions but encounter a performance bottleneck: further compression leads to substantial accuracy degradation. This work, submitted to arXiv and relevant to the Computer Science AI and Computation and Language communities, bridges the gap between theoretical understanding and practical application for engineers and researchers grappling with LLM efficiency.

The core contribution is the mathematical proof and experimental validation that low-rank decomposition and quantization are non-orthogonal compression methods. This is a crucial insight because many assumed their combination would simply compound individual error, rather than interact in a more complex, performance-detrimental way. The research demonstrates that the interaction between these two techniques is non-trivial, leading to significant performance degradation when naively combined. To overcome this, the authors propose the Diagonal Adhesive Method (DAM), a novel approach designed to mitigate the performance loss incurred by co-applying low-rank decomposition and quantization.

This work is significant because it moves beyond existing assumptions in model compression and provides a clear explanation for why simple combinations of powerful techniques fail. The intended audience includes machine learning engineers deploying LLMs and researchers developing new compression algorithms. By demonstrating the non-orthogonality and offering a practical solution, this paper enables more aggressive yet effective LLM compression. Going forward, this research could influence the development of new, synergistic compression techniques and inform best practices for deploying LLMs in resource-constrained environments. The findings lay a solid theoretical and empirical foundation for future investigations into optimizing the trade-offs between model size, inference speed, and accuracy. This analysis is based on the provided abstract.

AI/ML arXiv cs.AI

The Active Ingredient in Muon's Grokking

This work by Yufeng Wang, published on arXiv, pinpoints the critical component responsible for the Muon optimizer's accelerated grokking performance in modular arithmetic tasks. The core contribution is the isolation and identification of orthogonalization, specifically through the Newton-Schulz iteration, as the primary driver of Muon's speed advantage over optimizers like AdamW. This research addresses the gap in understanding the precise mechanisms behind Muon's superior grokking, moving beyond general attributions to spectral-norm constraints and orthogonalized momentum.

The implications are significant for researchers and engineers working on neural network optimization, particularly in scenarios where rapid generalization and the avoidance of brittle solutions are paramount. The key technical insights reveal that orthogonalization alone replicates Muon's speedup; spectral-norm constraints provide little benefit and introduce unreliability. Mechanistically, orthogonalizers achieve generalization at approximately one-third the spectral norm of other methods, settling into lower-norm solutions. Furthermore, the study demonstrates a trade-off with the number of Newton-Schulz iterations: reducing them accelerates initial grokking but can lead to fragile solutions that degrade with higher learning rates. Five iterations, the canonical choice, offer learning-rate robustness. The research also shows spectral scaling can be omitted without performance degradation.

Moving forward, this analysis provides a clearer path for designing more efficient optimizers, suggesting that focusing on orthogonalization techniques could yield substantial gains in generalization speed. It also highlights the importance of a stability-aware metric for evaluating grokking claims, advocating for reporting both first-crossing and sustained-grok times to provide a more comprehensive understanding of optimizer performance. The release of the training and analysis code promotes reproducibility and further investigation into these findings, potentially influencing the development of future optimization algorithms for deep learning. This work is based on an abstract.

AI/ML arXiv cs.AI

Codec-Gauge: Learning Compression-Friendly Gauges for Transformer KV Caches

The work presented in "Codec-Gauge: Learning Compression-Friendly Gauges for Transformer KV Caches" introduces a novel post-training layer designed to improve the efficiency of Key-Value (KV) cache compression in Transformer models, particularly for long-context inference. This research addresses the significant memory and latency overheads associated with KV caches, which are a major bottleneck for deploying large Transformer models. By learning optimal channel transformations, Codec-Gauge enables more faithful reconstruction of KV cache data after compression and decompression, thereby allowing for higher compression ratios without substantial quality degradation.

The core contribution lies in the development of a trainable "gauge" layer that precedes existing compression and quantization backends. This gauge learns to reorient the channel basis of KV vectors to be more amenable to compression. This is crucial because the inherent structure of KV cache data can significantly impact compression fidelity; standard bases might not align well with the underlying signal distribution. Codec-Gauge fills this gap by adapting the data representation to the compression codec, rather than relying on generic transformations.

The authors, Yitao Jiang, Yaoqing Yang, Luyang Zhao, Muhao Chen, and Devin Balkcom, published this work on arXiv in the computer science AI category. The intended audience is software engineers and researchers working with large language models and other Transformer architectures, especially those concerned with optimizing inference performance and memory usage. Practitioners deploying models in resource-constrained environments or requiring low-latency inference will benefit directly from this technique.

Two key technical ideas drive Codec-Gauge's effectiveness. First, the frequency-distribution objective leverages a token-channel Discrete Cosine Transform (DCT) spectral-centroid loss. This encourages the energy of the KV data to be concentrated in lower-frequency components within the transformed basis, which are typically easier to compress with common codecs like zfp and block-uniform quantization. Second, a smooth rate proxy is incorporated into the objective to balance compression efficiency with fidelity. This mechanism guides the gauge to learn transformations that not only make the data spectrally favorable but also implicitly consider the expected compression bitrates. The learned gauges are small orthogonal transforms, meaning they do not alter the model's weights or attention semantics, and can be applied post-training without retraining the entire model.

This work enables a practical method to enhance KV-cache compression without architectural modifications or extensive retraining. It establishes cache-coordinate geometry as a tunable parameter for compression optimization. Going forward, Codec-Gauge could pave the way for more efficient inference strategies for extremely long contexts, potentially leading to wider accessibility of powerful language models. The learned gauges could also be integrated directly into model architectures or compression libraries, influencing the design of future efficient Transformer implementations and compression algorithms. The provided content is an abstract only.

AI/ML arXiv cs.AI

SOAP, Muon, and Beyond: Pushing LLM Pretraining Scales

This work, originating from researchers at institutions including NVIDIA and published on arXiv, addresses the significant challenge of scaling large language model (LLM) pretraining by improving optimizer performance. The core contribution is the practical adaptation and enhancement of higher-order optimizers, specifically SOAP and Muon, to overcome limitations that have historically confined their use to smaller-scale training compared to the ubiquitous AdamW. The problem solved is the trade-off between the faster convergence of advanced optimizers and their computational expense, numerical instability, and implementation complexity at the massive scales now prevalent in LLM pretraining. This research is intended for software engineers and researchers involved in deep learning optimization and large-scale model training, aiming to provide them with stable and more efficient pretraining methodologies.

Two crucial technical ideas underpin this contribution. First, the researchers introduce algorithmic modifications to address SOAP's instability at large batch sizes. These include per-step QR orthogonalization and refined preconditioning strategies, which effectively eliminate loss spikes and enable stable training in regimes where SOAP would otherwise fail. Second, they present a layer-wise distributed optimizer implemented to be compatible with frameworks like Megatron-LM. This novel implementation balances memory usage and communication overhead while crucially avoiding approximations to the optimizer's core computations, thereby preserving the theoretical convergence benefits of SOAP and Muon. Empirically, the work demonstrates that SOAP and Muon, when compared to AdamW using update-RMS matching for fair learning rate transfer, consistently outperform AdamW on multi-billion-parameter models trained on trillions of tokens, particularly at extreme batch sizes of up to 100 million tokens, where AdamW performance degrades.

This research enables a significant step forward in LLM pretraining efficiency and quality. By demonstrating the stable and superior performance of advanced optimizers at unprecedented scales, it opens avenues for faster convergence, potentially leading to reduced training costs and the ability to train even larger and more capable models. The release of a codebase containing these emerging optimization algorithms further democratizes access to these advanced techniques, likely influencing the field by encouraging wider adoption of higher-order optimizers and spurring further research into their theoretical underpinnings and practical applications in LLM development. The presented findings are based on the abstract provided.

AI/ML arXiv cs.AI

When Does Recurrence Become an Algorithm? Convergence Selection in Weight-Tied Looped Transformers

This research paper, "When Does Recurrence Become an Algorithm? Convergence Selection in Weight-Tied Looped Transformers," by Tong Zhang, Junhao Hu, Yun Peng, and Tao Xie, published on arXiv, investigates the emergent algorithmic capabilities of weight-tied looped transformers. The core contribution is identifying the precise conditions under which repeatedly applying a single, weight-tied transformer block (a looped transformer) transitions from a generic computational process to implementing specific algorithms. The work addresses a critical gap in understanding how simple recurrent structures within neural networks can learn and execute complex reasoning, particularly when compared to standard, non-recurrent transformer architectures. This is of significant interest to researchers and engineers working on sequence modeling, neural algorithm learning, and the interpretability of deep neural networks.

Key technical findings include the "budget law," which quantifies a linear relationship between training data and the computational speed achievable per loop, with speed scaling approximately as $v \sim n_{train}/T_{train}$. This suggests that the training contract, defined by the amount of data and training steps, dictates the network's ability to process information per recurrence step. Another crucial insight is that the choice of algorithm (parallel scan vs. serial frontier) is dictated by architectural priors like weight tying, rather than solely by the model's inherent expressivity or the availability of positional addressing mechanisms. The paper also challenges conventional complexity barriers, demonstrating that seemingly difficult tasks like NC1-completeness are learned efficiently, while other problems, like group order, become bottlenecks, a barrier that can be mitigated by curriculum learning. Finally, the authors highlight the portability of learned mechanisms, showing that learned algorithms can be transferred across different training budgets via warm-starting, unlike attempts to force seriality through input schedules. This work enables a more principled approach to designing and analyzing recurrent neural networks for algorithmic tasks, potentially leading to more efficient and interpretable models. The findings, particularly the focus on convergence-time scaling and the validation via "damage cones," offer new diagnostic tools for understanding model behavior beyond traditional fixed-point metrics. These insights could profoundly influence the development of neural architectures capable of systematic generalization and algorithmic reasoning. This analysis is based on the provided abstract only.

AI/ML arXiv cs.AI

Scaling Interpretable Transformers with Parity Bottleneck Layers

This research, published on arXiv by authors including Andrew Mack and colleagues from unspecified institutions, introduces the ParityTransformer, a GPT-2 scale architecture designed to improve the interpretability of large language models by construction. The core contribution is the development of a "Deep Parity Bottleneck" (DPB) mechanism that replaces the computationally expensive, over-complete learned bottlenecks typically used with sparse autoencoders (SAEs) for post-hoc interpretability. The problem this work addresses is the prohibitive cost of per-layer interpretability in large Transformer models, which stems from the need for wide, sparse representations that are difficult to train efficiently. The ParityTransformer aims to bridge this gap by integrating interpretability directly into the model's forward pass.

Two pivotal technical ideas underpin the ParityTransformer. First, the Deep Parity Bottleneck (DPB) employs a parameter-free algebraic dictionary and a hierarchical mixture-of-experts approach to enforce sparsity efficiently. This design provides a deterministic incoherence guarantee, meaning features are guaranteed to be distinct, and crucially, eliminates the memory overhead associated with learned over-complete bases. Second, by ensuring that subsequent computations only act on features that have passed through this sparse bottleneck, the ParityTransformer ensures that its learned features are intrinsically utilized by the model during its forward pass. This directly tackles the question of whether post-hoc interpretability methods like SAEs recover features that the model actually uses. Empirically, the authors demonstrate that ParityTransformers match or exceed the performance of post-hoc SAEs on sparse probing tasks and offer superior performance in feature absorption, steering effectiveness, and causal interventions. This work is intended for researchers and software engineers working on Transformer architectures, especially those focused on model interpretability and understanding the internal mechanisms of large language models.

The ParityTransformer's success in integrating interpretability by design, rather than as an afterthought, holds significant implications. It enables the training of more interpretable large-scale models with a manageable "interpretability tax," potentially making internal model workings more accessible for debugging, verification, and fine-grained control. This approach could influence the development of future LLMs, shifting the paradigm towards inherently interpretable architectures and fostering greater trust and understanding in these powerful systems. The abstract does not contain the full paper content.