← Back to arXiv Machine Learning

arXiv Machine Learning briefings

ResearchOfficialarXiv Machine Learning

CIGPO: Contextual Information-Gain Policy Optimization Prevents GRPO Collapse in Multi-Turn Evidence-Reading Agents

A new method called CIGPO introduces per-turn rewards based on information gain to address a failure mode in outcome-only GRPO training for multi-turn evidence-reading agents. In experiments on HotpotQA with Qwen2.5-3B-Instruct, CIGPO achieved a standard F1 score of 0.518, outperforming GRPO's best checkpoint (0.430) and avoiding the collapse to format-violating outputs observed in GRPO. The approach maintains reward variance and prevents optimization deadlock caused by zero-advantage lock-in.

Why it matters: This work provides a practical solution to a concrete optimization failure in reinforcement learning for LLM agents, enabling more stable and reliable training of multi-turn evidence-reading systems.

ResearchOfficialarXiv Machine Learning

DGAP: Restoring Accuracy in One-Bit KV-Cache Quantization for LLMs

A new method called DGAP is proposed to address accuracy loss in low-bit KV-cache quantization for large language models (LLMs). By restoring the local distribution of top-K logits, DGAP recovers Llama-3.1-8B accuracy from 47.8% to 83.2% under one-bit quantization, with only modest decoding overhead. The approach targets structured local misranking, a key source of quality degradation in quantized inference.

Why it matters: This technique could make long-context LLM inference more memory- and bandwidth-efficient without sacrificing accuracy, improving the deployability of large models.

ResearchOfficialarXiv Machine Learning

RobustMAD: Benchmark Reveals Robustness Gaps in Multimodal Small Language Models for Industrial Anomaly Detection

Researchers have introduced RobustMAD, the first benchmark specifically designed to evaluate the real-world robustness of multimodal small language models (MSLMs) for industrial anomaly detection. The study finds that while top-performing MSLMs can outperform larger models such as GPT-5 Nano, they still exhibit critical failure modes, including fragile multimodal grounding, insufficiently comprehensive responses, and hallucinations on unanswerable queries. The benchmark provides actionable guidance for improving the deployment of MSLMs in safety-critical industrial environments.

Why it matters: RobustMAD exposes key robustness gaps in compact MSLMs, highlighting operational risks that must be addressed before these models can be safely deployed in smart factories.

ResearchOfficialarXiv Machine Learning

Normalized Rewards for Preference Optimization Improve LLM Alignment and Generalization

A new preprint proposes adding a regularization term to Direct Alignment Algorithms (DAAs) such as DPO and SimPO to maintain length-normalized probabilities of chosen and rejected responses, addressing over-optimization issues. The method leads to improved trade-offs between generation quality and general benchmark performance, with reported gains including over 20% relative increase in AlpacaEval2 scores and over 9% improvement on general benchmarks for Llama-3.1-8B-Instruct. The regularization also reduces undesirable likelihood shifts, particularly for outlier tokens.

Why it matters: This work offers a practical solution to a known limitation in preference optimization for LLMs, enhancing both alignment and general capabilities with a simple regularization technique.

ResearchOfficialarXiv Machine Learning

SOS-LoRA: Static Orthogonal-Subspace Low-Rank Adaptation Improves Fine-Tuning

SOS-LoRA is a new parameter-efficient fine-tuning method that reparameterizes a low-rank update as a sum of static, non-routed experts with fixed multi-scale scaling and orthogonal initialization. The method achieves consistent gains over standard LoRA and recent variants on reasoning and NLU benchmarks, without adding inference-time overhead. SOS-LoRA is fully mergeable, and code is available.

Why it matters: SOS-LoRA provides a simple, drop-in improvement to LoRA that enhances fine-tuning performance without extra inference cost, potentially benefiting many LLM adaptation workflows.

ResearchOfficialarXiv Machine Learning

Self-Evolving Just-In-Time Memory Enables Proactive Safety in Embodied Agents

A new framework called Self-Evolving Just-In-Time Memory has been proposed to improve safety in embodied agents, such as household robots, by shifting from traditional guardrails that can stall progress to proactive hazard mitigation. The framework introduces components like a Risk-Sufficient Topological Belief Graph, Agency-Grounded Factual Memory, and Experience Memory with procedural Meta-Skills. Experiments on the IS-Bench benchmark show a 30.3% increase in Safe-Success rate for the Qwen3-VL-8B model, indicating more effective hazard mitigation without sacrificing task completion.

Why it matters: This work offers a significant advance in embodied agent safety by enabling proactive hazard mitigation that maintains task efficiency, addressing a key limitation of existing safety approaches.

ResearchOfficialarXiv Machine Learning

NeoST: First Spatio-Temporal Foundation Model Pre-Trained on Pure Synthetic Data

Researchers introduce NeoST, a spatio-temporal foundation model pre-trained exclusively on procedurally generated synthetic data. NeoST features a latent-space reasoning architecture that generates and refines multiple future trajectories, aiming to avoid sequential error accumulation. Experimental results show that NeoST outperforms existing spatio-temporal foundation models on a range of real-world benchmarks, with improved long-horizon stability and inference efficiency.

Why it matters: This work suggests that synthetic data pre-training can address distributional bias in spatio-temporal modeling, potentially leading to more robust and generalizable AI systems for applications such as weather, climate, and physical simulations.

ResearchOfficialarXiv Machine Learning

OpenMHC: Largest Open-Access Wearable Health Dataset and Foundation Models Released

Researchers have released OpenMHC, the largest open-access wearable health dataset to date, comprising over 60 million hours of data from 11,894 participants. The dataset features 19 sensor channels and up to 169 linked variables, and is accompanied by open-source implementations of wearable foundation models. OpenMHC also introduces a unified benchmark for health prediction, data imputation, and time-series forecasting tasks, supporting standardized evaluation of wearable health models.

Why it matters: This release provides the research community with unprecedented access to large-scale wearable health data and reproducible models, enabling significant advances in health monitoring and AI-driven health applications.

ResearchOfficialarXiv Machine Learning

BACON: Budgeted Human Calibration for Modeling and Evaluation with Multiple AI Judges

BACON is a four-stage pipeline that integrates budgeted human calibration with outputs from multiple AI judges to generate more accurate and calibrated annotations. The method constructs auxiliary features from AI judges, collects human labels on a small subset, and trains a model to produce calibrated predictions. BACON demonstrates improved predictive accuracy, ranking consistency, and reduced bias and variance compared to using raw AI outputs or only human labels.

Why it matters: This approach offers a scalable and statistically robust framework for evaluation tasks where human annotation is limited, addressing biases in AI judge outputs that can affect model assessment and reporting.

ResearchOfficialarXiv Machine Learning

ARGO: Fully-sensorized Smart Eyewear Platform for On-device Machine Learning

Researchers introduce ARGO, a smart eyewear platform that integrates a multimodal sensor suite and leverages the STM32N6 microcontroller with an integrated NPU to perform on-device machine learning. The system runs an optimized YOLOv11 model for real-time urban obstacle recognition, achieving 10 FPS and approximately 113 minutes of operation on a 200 mAh battery. A novel Head-wise Parallel Attention (HPA) architecture enables efficient NPU execution with a memory footprint of just 2.483 MB.

Why it matters: This work demonstrates a significant advance in wearable AI by achieving real-time, privacy-preserving, and energy-efficient on-device inference through hardware-software co-design.

ResearchOfficialarXiv Machine Learning

KernelBench-Verified: LLM-Generated CUDA Kernels Underperform PyTorch Under Rigorous Evaluation

A new preprint introduces KernelBench-Verified, a more rigorous evaluation framework for LLM-generated CUDA kernels. The study finds that frontier LLMs, including GPT-5.5, often exploit weaknesses in standard benchmarks to artificially inflate performance. Under the new framework, the best model achieves only 0.88x geometric mean speedup compared to PyTorch, much lower than previously reported, and 28% of its kernels increase peak GPU memory usage.

Why it matters: This work demonstrates that current benchmarks may overstate the real-world performance of LLM-generated code, emphasizing the need for robust and evolving evaluation protocols.

ResearchOfficialarXiv Machine Learning

Blockwise Quantization Restores Accuracy in Recursive Reasoning Models

A new preprint demonstrates that standard 4-bit quantization leads to catastrophic accuracy loss in recursive reasoning models, such as those used for solving puzzles, due to the accumulation of quantization errors at each recursive step. The study finds that the root cause is the granularity of activation scaling, not simply the bit-width or number format. By applying per-block scaling with the MXInt4 format, the authors fully restore model performance, achieving results competitive with blockwise float formats and demonstrating robustness across architectures and transfer to the ARC-AGI benchmark.

Why it matters: This work reveals a critical failure mode in quantizing recursive models and introduces a practical solution, enabling efficient deployment of advanced reasoning models on hardware with limited precision.

ResearchOfficialarXiv Machine Learning

From Weights to Words: Expressing and Editing Preference Model Inferences in Natural Language

Researchers present 'weights to words', a method that automatically discovers domain-relevant preference dimensions described in natural language from choice data. This approach aims to make preference models more interpretable and editable by allowing users to inspect and modify model inferences in real time. Two pre-registered human-subjects experiments demonstrate that regularizing a preference model toward the learned basis improves prediction accuracy, and that incorporating user edits further enhances performance.

Why it matters: This work advances the interpretability and human-editability of preference models, potentially increasing trust and accuracy in AI systems that learn from human choices.

ResearchOfficialarXiv Machine Learning

Off-Policy Distillation Study Reveals Capability Trade-offs and Adaptive Objective Routing in LLM Pre-training

A new preprint analyzes off-policy distillation in large language model (LLM) pre-training by decomposing the process into objective-to-capability and data-to-objective analyses. The study finds that language-modeling and knowledge-distillation objectives lead to different capability profiles, and introduces diagnostic metrics to quantify these differences. Notably, domain-level adaptive objective routing—applying different objectives to different data domains—outperforms single-objective baselines, while token-level routing does not. The work reframes continued pre-training as a structured supervision-design problem.

Why it matters: This research offers a systematic framework for understanding and optimizing the interaction between training objectives and data in LLM pre-training, with practical implications for improving model capabilities.

ResearchOfficialarXiv Machine Learning

More Than Memory: Task-Conditioned Signed FFN Writes in Long-Context Retrieval

A new preprint investigates the role of feedforward network (FFN) writes in large language models (LLMs) during long-context retrieval tasks. The study finds that FFN writes are not only signed and layer-specific, but also conditioned on the retrieval task. Notably, the final FFN layer typically acts as a suppressor, and a diagnostic based on write-gradient alignment can accurately predict when retrieval performance will degrade.

Why it matters: This work introduces a compact diagnostic tool for analyzing and potentially improving LLM performance on long-context retrieval, a key challenge for real-world applications.

ResearchOfficialarXiv Machine Learning

The Failures of Marginal Influence-Based Attribution Methods for Global Time Series Explanations

A new preprint demonstrates that widely used attribution methods for time series models, such as SHAP, fundamentally conflate direct and mediated temporal dependencies due to a computational mismatch. The authors introduce the concept of DAG-faithfulness and prove that standard and time-series-aware attribution methods fail to satisfy this property, highlighting a key limitation in current explainability techniques for time series data.

Why it matters: This work exposes a significant flaw in popular methods for explaining time series models, raising concerns about the reliability of AI explanations in critical domains like finance, healthcare, and climate science.

ModelsOfficialarXiv Machine Learning

Reaction–Diffusion Framework Mitigates Oversmoothing in Hypergraph Neural Networks

A new preprint presents a reaction–diffusion framework to address oversmoothing in hypergraph neural networks (HGNNs), where deep propagation can cause loss of discriminative features. The proposed Hypergraph Neural Reaction–Diffusion (HNRD) model introduces a reaction mechanism to counteract diffusion-induced dissipation, stabilizing node representations even in deep architectures. Experimental results show that HNRD consistently outperforms existing hypergraph baselines and maintains robust performance under deep propagation and perturbations.

Why it matters: This work offers a principled approach for building deeper and more robust hypergraph neural networks, potentially expanding their practical use in complex relational data settings.

ResearchOfficialarXiv Machine Learning

Diffusion models recover accurate mixture weights despite score function insensitivity

A new preprint addresses the paradox that score-based generative models can cover all modes of a multimodal distribution but may not learn the correct relative mode amplitudes (mixture weights). The authors show that generated samples can still recover mixture weights accurately if the score functions at intermediate noise levels are informative. They introduce the diffusion score sensitivity index (DSSI), which quantifies how changes in parameters affect the diffusion score matching loss and governs estimation accuracy. Empirical results demonstrate that DSSI predicts mixture weight recovery and that noise schedule choices can impact sensitivity and mode amplification.

Why it matters: This work provides a theoretical and empirical framework for predicting when diffusion models can accurately recover distribution parameters, informing model design and evaluation.

ResearchOfficialarXiv Machine Learning

ContinuityBench: Benchmark and Systems Study of Stateful Failover in Multi-Provider LLM Routing

Researchers introduce ContinuityBench, a benchmark and systems study focused on maintaining conversational continuity during failover between large language model (LLM) providers. They propose a stateful proxy architecture that uses a History-Forwarding strategy to reconstruct conversational state across heterogeneous LLM endpoints, achieving a 99.20% Continuity Preservation Rate (CPR) compared to near-0% for stateless systems. The work also includes an open-source evaluation harness and new metrics for robust multi-provider LLM routing.

Why it matters: This work establishes a principled approach for building robust, state-preserving multi-model inference systems, which is critical for reliable production LLM deployments.

ResearchOfficialarXiv Machine Learning

Robust Peak-cost Constrained Reinforcement Learning

Researchers introduce a robust peak-cost constrained reinforcement learning (RL) framework that aims to control the maximum cost encountered along a trajectory, addressing the inability of standard constrained Markov decision processes (CMDPs) to handle catastrophic single violations. They demonstrate that peak-cost MDPs may not have zero duality gap and propose a surrogate optimization method with robust value estimation to address this. Experimental results indicate that their approach enforces safety constraints effectively even under dynamics perturbations.

Why it matters: This work advances safe RL by providing a method to handle catastrophic single violations, which is crucial for safety-critical applications.