Papers for

machine learning operations teams

Papers whose findings have a practical use for this group, as judged from the abstract. Open a paper to read what it means in practice.

Fine-tuning language models safely with replay and distillation methods

Preserving Safety During Sequential Fine-Tuning: A Factorial and Geometric Analysis of Replay

Abstract: Benign fine-tuning can erode safety alignment in large language models, and sequential adaptation substantially compounds this risk. Prior work mitigates this erosion by replaying harmful-prompt refusals and benign-compliance examples, while other work instead matches base-model logits via distillation. However, the individual contributions of supervised exposure and distillation to safety behavior and refusal geometry during sequential adaptation remain unclear. We factor these two strategies apart and study their behavioral and geometric effects during sequential LoRA fine-tuning of Mistral-7B and Llama-3-8B across diverse task sequences (code generation, German-language summarization, and science question-answering). Behaviorally, supervised replay exposure yields the largest reduction in HarmBench attack success rate (ASR) on Mistral, averaging 51.1 points below base, but at a substantial over-refusal cost, averaging 53.4 points above base on OR-Bench-hard. Logit distillation, without any supervised replay loss, largely prevents safety erosion at a fraction of this cost: an average 5.4-point ASR reduction against only a 5.1-point rise in over-refusal. Llama shows the same asymmetry between the two strategies. Combining exposure and distillation moderates exposure's over-refusal while leaving downstream task scores largely unaffected. Geometrically, the two models diverge: on Mistral, distillation preserves the base refusal direction more closely than exposure does, whereas on Llama, refusal behavior after exposure largely survives ablation of the extracted refusal direction. This dissociation between models indicates that preserving the base refusal direction is not a reliable proxy for behavioral safety.

Tue 27 OctMachine Learning
The gist
Fine-tuning large language models to improve them can accidentally cause them to become less safe by responding in harmful or unwanted ways. The authors study two techniques—replaying examples of safe and unsafe prompts and copying model output patterns (distillation)—to keep safety during fine-tuning. They find that replaying examples reduces harmful responses but also makes the model refuse too many harmless requests. Distillation mainly keeps safety without causing many unnecessary refusals. Combining both methods balances safety and helpfulness better. They also show that measuring safety by how the model refuses prompts geometrically is not always reliable.
Open → 2610.38001v1

Language helps vision models learn better and resist errors

Language Guidance for Supervised Vision Training: An Empirical Study of Generalization

Abstract: Deep neural networks have achieved remarkable success on vision benchmarks, yet they continue to struggle with many generalization challenges. Supervised vision training relies on one-hot labels, which provide limited information about semantic structure and shared attributes between classes. This limited supervision can leave visual representations vulnerable to distribution shifts, spurious correlations, texture bias, adversarial perturbations, and forgetting in sequential learning settings. We study whether pretrained language models can serve as lightweight auxiliary supervision for vision training without requiring paired image-text data, prompt engineering, or contrastive objectives. Specifically, we evaluate two forms of language guidance, Explicit Language Guidance (ExLG) and Implicit Language Guidance (ImLG). We conduct a comprehensive evaluation across six generalization regimes, including in-distribution, out-of-distribution generalization, shortcut and spurious correlation resistance, texture and shape bias, adversarial robustness, and continual learning. Our analyses show that the two mechanisms have complementary strengths, with explicit guidance consistently benefiting in-distribution, low-data performance, and continual learning retention, while implicit guidance is often more useful in shortcut-sensitive settings and under stronger adversarial perturbations. Importantly, both are lightweight and add minimal parameters and training overhead. The analyses characterize when language-derived structure helps supervised vision training and provides a practical roadmap for using off-the-shelf pretrained models from another modality as auxiliary supervision.

Mon 19 OctComputer Vision and Pattern RecognitionArtificial Intelligence
The gist
Vision models usually learn from simple labels that say what an image is but don’t explain how different classes relate. This paper studies how adding guidance from language models can improve these vision models without needing extra paired data or complicated setups. The authors test two ways to add language help—one that adds explicit language info and another that does so more indirectly. They find that these language guides improve how models handle new or tricky situations, resist being fooled by shortcuts or attacks, and remember older tasks better when learning new ones.
Open → 2610.27208v1

Large language model uncertainty measured by internal mechanism stability

Measuring Mechanistic Uncertainty in Large Language Models via Causal Robustness

Abstract: Large language models (LLMs) are increasingly used in critical domains, where their susceptibility to hallucinations makes reliable and interpretable uncertainty quantification essential. We introduce the concept of mechanistic uncertainty, which characterizes uncertainty through the stability of the internal mechanisms supporting a model's response. We propose the causal robustness margin, defined as the minimum internal intervention required to deviate the model's output, and operationalize it using Jacobian Lens (Jlens) concept steering vectors. Across three open-source models and five question-answering datasets, we show incorrect and uncertain responses exhibit smaller margins, indicating greater sensitivity to internal interventions, and the causal robustness margin complements conventional output-based uncertainty methods. Finally, in a controlled knowledge injection experiment, correctly learned facts show increased margins after training, suggesting that causal robustness provides a mechanistically grounded signal of uncertainty.

Thu 15 OctComputation and LanguageArtificial Intelligence
The gist
Large language models can sometimes give wrong or made-up answers, which is a problem for important tasks. The authors introduce a way to measure how certain a model is by checking how stable its inner workings are when giving a response. They find that when a model is uncertain or wrong, small changes inside can cause its answer to change, showing less stability. Their method adds new insight beyond just looking at the model’s final answer and can help understand when the model really knows something.
Open → 2610.22636v1

Forgetting past work boosts automated discovery with AI agents

Scaling Discovery Requires Forgetting

Abstract: Automated research with LLM agents has shown promising results in various domains. Through iterative refinement and scored verifiers, LLM agents are capable of discovering new ideas. There are different ways to scale automated research: let one agent iterate for longer, run several agents in parallel, or further let the parallel agents communicate and build on one another's candidates. However, we find that all three strategies eventually plateau as the agents accumulate history. Each agent becomes anchored to its own past attempts, making fewer and smaller edits. Beyond this point, additional budget brings little further improvement. In this paper, we show that the key to scaling discovery further is to forget. Periodically discarding the search state, including every agent's session and the candidate history, and restarting from the best candidate found so far, unblocks the search and pushes the same budget further. We implement forgetting in a team of agents with communication and evaluate it on four tasks spanning heuristic programming, kernel optimization and model training. The results show up to 15% improvement compared to parallel agents, and up to 39% compared to a single agent. Our results suggest that, for open-ended discovery, what an agent is allowed to remember matters as much as how much compute it is given.

Wed 14 OctArtificial IntelligenceComputation and Language
The gist
Automated research uses AI agents to come up with new ideas, but these agents tend to get stuck by relying too much on their past attempts. The authors show that if the agents periodically forget everything except their best current idea and start fresh, they can discover better solutions. This approach improves results by up to 39% compared to a single agent and up to 15% over multiple agents working in parallel. It suggests that letting AI agents forget old work can be as important as giving them more computing power.
Open → 2610.20563v1

Discriminative models reveal a new way to generate and edit images

Discriminative Score Function: Turning Pretrained Models into Functional Generative Priors

Abstract: Discriminative models are trained with task-specific objectives such as classification, detection, or representation learning, yet their practical utility depends on generalizing beyond the training task to unseen samples from the same domain. This suggests that a discriminative model may retain an implicit trace of the data manifold beyond its task labels. We ask whether this trace can be exposed as a generative signal without task-specific guidance. We introduce the Discriminative Score Function (DSF), which extracts an image-space generative field from the loss-gradient geometry of a pretrained discriminative model. DSF provides a training-free update rule that moves images toward model-consistent visual regions without retraining or task-specific targets. We apply DSF to diverse pretrained discriminative models, including ResNet classifiers, DETR detectors, and DINOv2 ViT encoders. DSF generates unconditional images from noise and supports conditional guidance, editing, inpainting, and explanation through the same field. DSF-generated samples also provide effective zero-shot quantization calibration when real images or labels are unavailable, preserving feature geometry better than data-free calibration baselines. These results show that discriminative training leaves a generative functional signal for synthesis and calibration. Project page: https://junhoo.me/dsf.

Wed 14 OctComputer Vision and Pattern RecognitionMachine Learning
The gist
Deep learning models trained to recognize images also contain hidden clues about how real images look. The authors show a method to turn these clues into a tool that can create new images, change parts of images, or fill gaps without needing extra training. This tool works with various popular pretrained models and helps adjust systems without needing real data or labels. Their approach uncovers a surprising connection between models built for understanding images and the ability to generate them.
Open → 2610.19743v1

Adaptive methods detect shifts in generative AI data faster and more accurately

Learning to Detect Shifts in Unstructured Data Streams via Monitoring-by-Betting

Abstract: Continuous monitoring of generative AI systems requires detecting distribution shifts in the unstructured data they ingest and generate. Existing monitoring methods typically control false alarms by requiring the discrepancy measure used to detect shifts to be *pre-specified*. However, for unstructured data, pre-specification creates a sensitivity--breadth trade-off: a narrowly defined measure is sensitive to the anticipated shift but may miss unexpected changes, whereas a broadly defined measure may detect a wider range of shifts but with reduced sensitivity. Although adaptively hypothesizing the shift direction can overcome this limitation, such data-dependent adaptation generally invalidates conventional false-alarm guarantees. We reconcile adaptivity with statistical validity through testing-by-betting, reframing sequential changepoint detection as a betting game in which the bettor continually updates its hypothesized shift direction, or witness function, as new observations arrive. We introduce *asymptotic e-detectors*, which approximate the pre-change distribution using a calibration set of size $m$, and prove that they asymptotically control the average run length to a false alarm as $m$ increases. Empirically, calibration sets as small as $m=30$ provide effective control. We then develop adaptive monitoring strategies ranging from black-box bettors that learn from embeddings to interpretable bettors based on LLM-proposed concepts. Across three real-world datasets, including data from a clinical AI scribe deployment, adaptive monitors detect shifts faster than fixed monitors at the same false-alarm level. Further, when the true shifts are sparse and interpretable, LLM-concept monitors provide the fastest and most interpretable detection, while also guiding root-cause diagnosis for complex shifts.

Mon 12 OctMachine LearningArtificial Intelligence
The gist
Detecting when the data used by AI changes unexpectedly is hard, especially when the data is complex like text or images. The authors propose a new way to monitor these changes by imagining it as a betting game where the system guesses the kind of change and adapts as new data comes in. This approach keeps false alarms low while detecting shifts more quickly than fixed methods. They show that even with small setups, their method works well and can also suggest explanations for the changes, helping to figure out their causes.
Open → 2610.17399v1

Small language models analyzed for understanding machine learning steps

Using Small Language Models to Reverse-Engineer Machine Learning Pipelines Structures

Abstract: Context: Once defined a taxonomy of stages structuring Machine Learning (ML) pipelines (e.g. Data Preprocessing, Modeling...), extracting these stages from source code is key for better understanding ML practices. However, the diversity caused by the constant evolution of ML (e.g., algorithms, datasets) makes this task challenging. Existing approaches either rely on non-scalable manual labeling or on classifiers that do not properly support domain's diversity. These limitations call for more reliable solutions. Objective: We evaluate whether Small Language Models (SLMs) can leverage their code understanding and classification abilities to address these limitations, and enhance our understanding of practices in ML. Method: We conduct a confirmatory study based on two relevant reference works representing current limitations in the state-of-the-art. We first compare several SLMs using Cochran's Q test, then evaluate the best-performing model against reference studies via two McNemar's tests. An additional Cochran's Q test examines how taxonomy definition variations affect the SLM performance. Finally, goodness-of-fit tests compare ML practice insights from SLM classification with those from prior studies. Results: First, we found that the taxonomy wording significantly impacts classification performance. Second, the best performing SLM yielded good results, yet, without outperforming other classifiers. Third, the three classification methods led to significantly different insights, with varying effect sizes, when exploring practices of data scientists. Conclusions: Limitations of existing classification methods bias our understanding of ML practices. While current SLMs show promising results without prior fine-tuning, they still exhibit common limitations, in addition to inference high costs challenging their applicability in large-scale studies.

Wed 7 OctSoftware EngineeringMachine Learning
The gist
It can be hard to figure out how people build machine learning programs because they involve many different steps, like cleaning data and creating models. The authors studied small language models to see if they can recognize these steps from code better than older methods. They found that while these models do well, they don’t outperform existing techniques and have some practical challenges. Also, how the steps are described can change how well the models work. This study helps us understand the limits and potential of using language tools for analyzing machine learning work.
Open → 2610.10261v1

NeMo-DCR speeds up trillion-parameter model updates across regions

NeMo-DCR: Bit-Exact Delta-Compressed Refit for Scalable Agentic RL at Trillion-Parameter Scale

Abstract: Agentic reinforcement learning (RL) disaggregates training from rollout, so each policy update must reach the rollout clusters before the next batch. Transferring a full 1T checkpoint for such weight synchronization (refit) takes 87.5 min between two AWS regions. Measurements of BF16 training show that about 1% of weights change their stored values per step. Recent systems exploit this sparsity but fall short on placement, exactness, or efficiency: they reimplement placement rules, assemble full tensors, rebuild values arithmetically, or use a cross-cluster collective, and none fully recovers from mid-refit failures. We present NeMo-DCR (Delta-Compressed Refit), which sends only changes yet is bit-exact: receivers obtain the same parameter and buffer bits as a dense refit. For placement, fixed affine mappings project changes from training shards into the checkpoint's canonical coordinates, residual conversion covers the other changes, and the serving runtime's native loader places all changes in receiver storage. For exactness, compressible XOR masks carry affine changes whose projection and loader preserve stored bits, and overwrites carry the others. Receivers apply both in place, retries overwrite partial writes, and a joint commit binds the policy to the baseline for the next delta. For efficiency, object storage or a relay tree streams payloads during delta construction, without a cross-cluster collective. Even at 3% and 5% change rates, NeMo-DCR refits of 30B-1T models are 12-40$\times$ faster than a transport-only full-checkpoint reference. A 1T relay-tree refit at 3% takes 150 s instead of 87.5 min, making refits practical for cross-cluster agentic RL at trillion-parameter scale.

Tue 6 OctDistributed, Parallel, and Cluster ComputingArtificial Intelligence
The gist
When huge AI models learn and update, moving the entire updated model across computers can take a very long time. The authors present NeMo-DCR, a method that only sends the small parts of the model that changed but still guarantees the exact same result as sending the whole model. This makes updating trillion-parameter models across different data centers much faster and more practical, cutting hours down to minutes. Their approach also handles errors during transfer and works efficiently without complex coordination between computers.
Open → 2610.08430v1

Self feedback loops undermine test-time training in language models

Self-Generated Feedback Destabilizes Test-Time Training: A Causal Decomposition of Long-Horizon Adaptation

Abstract: Test-time training (TTT) lets a model store information in its weights during inference. When the model learns from its own output, however, each update also changes the model that generates the next training example. Across 128K-token streams, retaining generated-text updates worsens prediction on independent human-written text with three TTT-E2E model configurations (labeled 125M, 760M, and 3B). The same failure occurs when Adam updates Qwen3-4B's existing weights. The same update mechanisms can improve on real text, so writing itself is not the failure. Three matched comparisons trace the causal pathway. Fixed Generation removes over 98% of the damage at 125M and 760M by using a frozen model to generate training chunks. Recorded Replay separates the loss caused by reading degraded text from the additional loss stored by updating on it. A paired one-update comparison then shows the local conflict: an update predicts its source better but new real text worse. This cost grows after Closed Loop adaptation, with a few trajectories accounting for most large failures. Finally, Settlement evaluates the candidate state on independent real text before commitment. It leaves mean endpoint gaps of 0.07 and -0.02 nats at 125M and 760M while retaining real-text adaptation. These results motivate checking prediction on independent evidence before retaining an update.

Sun 4 OctComputation and LanguageArtificial Intelligence
The gist
When a language model learns from its own writing during use, changes it makes can cause it to perform worse on real human text later. The authors show that this happens because the model’s updates affect the very text it uses to train itself, creating a feedback loop that degrades performance over time. They test different setups and find ways to reduce the problem by checking new model states on independent, real text before keeping updates. This suggests it’s important to verify improvements on fresh evidence before letting a model adapt.
Open → 2610.05076v1

High quality data can still introduce safety risks in language models

High-quality Data Do not Mean Safe! Poisoning LLMs after Data Selection

Abstract: Safety-aligned Large Language Models remain vulnerable to fine-tuning on small sets of harmful or benign-looking samples. However, prior studies typically assume that poisoned samples directly enter downstream fine-tuning, overlooking quality-based selection in practical training pipelines. To fill this gap, we systematically evaluate both the filtering effects against poisoning and the downstream safety impact of retained data. The results reveal that selection removes many overtly harmful samples, yet some retained high-quality samples can still degrade model safety alignment possibly due to their harmful-like training-update patterns at the layer-wise gradient level. Together, these findings expose a practical vulnerability: safety-degrading influence can pass through quality-based selection via retained high-quality samples. To examine its systematic exploitability, we propose Bi-Stage Quality-Constrained Safety-Degradation Text Optimization (Bi-QSTO), which optimizes poisoned samples under an explicit quality constraint to survive selection while preserving their safety-degrading influence. Across poisoning settings, target models, and filtering rates, Bi-QSTO maintains attack effectiveness before and after selection. Even at 90% filtering, harmful-seeded samples achieve a Poisoning Retention Rate above 90% and Harmful Score of 3.30--4.01. Their attack effectiveness strongly transfers across models and their retention advantage generalizes to additional selection methods.

Thu 1 OctCryptography and Security
The gist
Large language models (LLMs) can be fine-tuned in ways that make them unsafe, even if the training data looks good. The authors found that selecting high-quality data to train these models doesn't fully block harmful influences. Some harmful samples that appear high-quality can sneak through and still make the model unsafe by affecting its internal learning patterns. They created a method called Bi-QSTO to show these harmful samples can be designed to survive data filtering and still degrade model safety.
Open → 2610.01367v1

Disentangled diffusion autoencoders speed counterfactual edits in visual models

Towards Fast and Disentangled Counterfactuals for Visual Foundation Models

Abstract: Foundation models remain vulnerable to spurious correlations and ``Clever Hans'' strategies. Explainable machine learning can find and remove such strategies for classifiers without metadata. For foundation models, no such option exists yet. We propose Disentangled Diffusion Autoencoders (DiDAE). DiDAE wraps a frozen foundation model in a conditional diffusion decoder. A counterfactual is one closed-form edit along a direction of a disentangled dictionary, followed by decoding. The dictionary can be supervised (Procrustes) or unsupervised (Singular Value Decomposition, Sparse Autoencoders). No gradients are needed, so DiDAE is up to 2000 times faster than the state of the art. We evaluate on six datasets, two synthetic and four real-world. In a desiderata-driven benchmark on three of them, its counterfactuals are on par with or better than the state of the art, and they repair downstream classifiers through Counterfactual Knowledge Distillation (CFKD), where they beat metadata-based correction. The same machinery can rank a pretrained dictionary against a trained classifier. It returns the few directions the classifier actually reads, each causally verified by a counterfactual that flips the decision, and repairs the classifier along those a teacher marks spurious. The workflow is plug-and-play in our open-source Peal library we publish alongside the paper. With a public dictionary and a pretrained decoder, all that remains is a cheap linear distillation of the classifier and its own fine-tuning.

Thu 1 OctMachine LearningComputer Vision and Pattern Recognition
The gist
Computer vision models sometimes make decisions based on tricks or errors hidden inside their reasoning. The authors introduce a new method called Disentangled Diffusion Autoencoders (DiDAE) that helps find simple, clear changes to images which reveal these hidden tricks. This method is much faster than previous techniques because it doesn't rely on slow gradient calculations. It can also improve other models by teaching them to avoid these errors, making AI decisions more reliable.
Open → 2610.00895v1

Training seed effects outweigh compression in speech recognition fairness

Fairness Beyond a Single Run: Training-Seed Variability in Speech LLM Adaptation

Abstract: Demographic fairness gaps in automatic speech recognition are almost always reported from a single training run. We fine-tune the Q-former projector and LoRA adapters of a speech LLM at five audio compression factors and six random seeds, holding the encoder, base decoder, data and decoding fixed, and evaluate every run on Common Voice and Fair-Speech. At 460 h of clean LibriSpeech, the seed moves fairness metrics more than compression does on most demographic axes. A balanced 3x3 decomposition attributes 85.3% of the variation in Fair-Speech ethnicity normalized gap to the seed against 8.3% to compression (p = 0.009), though compression explains more on age and gender. Held-out LibriSpeech word error rate spreads by 0.04 points across those seeds while Common Voice spreads by 8.57, so these are not failed runs, and the effect survives controlling for accuracy and dropout. Scaling and diversifying the adaptation set to 960 h damps the effect but does not remove it. On Fair-Speech ethnicity, two single-run systems must differ by more than 0.30 in normalized gap to exceed seed variability.

Wed 30 SeptComputation and Language
The gist
Fairness differences in speech recognition systems are often measured using a single training run, but this can be misleading. The authors found that changing the random starting conditions (training seeds) causes bigger changes in how fair the system is across different ethnicities than changing how much the audio is compressed. Even when using more data, these training seed effects remain important. This means that assessing fairness should consider variability across multiple training runs to get a clearer picture.
Open → 2609.38976v1

Unlearningsoup cuts tuning time for safer large language models

UnlearningSoup: Is Repeated Tuning Necessary for Large Language Model Unlearning?

Abstract: Large language models trained on vast corpora inherently risk memorizing harmful content that may later re-emerge in their outputs. To mitigate this issue, existing unlearning methods typically rely on training-based parameter updates, such as gradient ascent and its variants, to delete targeted content while preserving other knowledge. However, balancing the competing goals of forgetting and retention makes hyperparameter choices for these methods particularly difficult, often requiring repeated tuning to obtain a strong model that still leaves substantial room for improvement and transfers poorly across models and datasets. To address this challenge, we investigate whether unlearning runs exhibit exploitable structure in weight space, and observe that models from different runs still lie in a shared evaluation-performance basin. This suggests that stronger models may be recovered through an unlearning-tailored soup strategy, reducing the need for repeated tuning for further improvement or new settings. Motivated by this, we propose UnlearningSoup, a unified framework that provides two strategies: EfficientSoup uses binary-search-based interpolation to quickly discover a well-performing model in the early stage, where repeated tuning would otherwise make strong model selection costly. PerformanceSoup uses reweighted souping to efficiently unlock the remaining performance potential in the later stage, where repeated tuning becomes increasingly inefficient. Extensive experiments across diverse datasets and models show that UnlearningSoup delivers 2.4x to 3.3x efficiency gains in hyperparameter selection, while consistently improving performance across settings.

Tue 29 SeptMachine Learning
The gist
Large language models sometimes memorize and repeat harmful or unwanted information. To fix this, people try to make the models "unlearn" certain content, but the ways to do this require lots of trial and error to balance forgetting bad stuff while keeping good knowledge. The authors found that different attempts at unlearning share common ground, so they created UnlearningSoup, a method that mixes these attempts to quickly find better models without repeated searching. This approach saves time and makes unlearning more effective across different cases.
Open → 2609.37076v1

Detecting early false confidence in large language models reasoning

When Confidence Rises Too Early: Detecting Shortcut Reasoning via Premature Answer Commitment

Abstract: The reasoning trajectory of a Large Language Model (LLM) is often treated as a verbalized description of its internal reasoning. However, such trajectories can be unfaithful: a model may rely on shortcuts to reach an answer and then post-rationalize the decision with a seemingly coherent chain of thought. Detecting this shortcut reasoning is challenging because existing monitors and verifiers mainly inspect textual traces or final outcomes, rather than how the model's belief in its answer develops during generation. We introduce ConfLens, a framework that tracks the evolution of confidence in the final answer throughout reasoning. Across three shortcut reasoning settings, we observe a common pattern of premature confidence, where shortcut samples become highly confident in the final answer at early reasoning stages. Existing confidence estimation methods, however, show limited generalizability, reliability, or efficiency for detecting this behavior. We therefore propose the Distributional Answer Commitment Score (DACS), a distributional confidence estimator that measures the entropy of the model's probability distribution over answer commitment at each reasoning step. DACS captures how concentrated the model's answer belief is without requiring ground-truth answers or task-specific verifiers. We further convert ConfLens detection results into interpretable signals for reward models to reduce their preference for shortcut reasoning. Experiments on mathematical and code reasoning tasks show that ConfLens with DACS improves shortcut reasoning detection by over 4.3% F1 compared with strong baselines and reduces the mismatch between faithfulness and correctness in reward model preferences.

Mon 28 SeptComputation and Language
The gist
Large language models sometimes jump to answers too quickly by using shortcuts rather than careful thinking. This can make their explanations seem plausible but actually be unfaithful. The authors introduce a new way to watch how confident the model is at each step of its thinking. They find that models using shortcuts become overly confident very early. Their method, called Distributional Answer Commitment Score (DACS), helps spot this early overconfidence without needing the right answer in advance. This improves detecting shortcut reasoning in math and code problems and helps preference systems avoid it.
Open → 2609.35074v1

Dual perspective test input prioritization improves dnn fault detection

When Ambiguity Meets Atypicality: Dual-Perspective Test Input Prioritization for DNNs

Abstract: While Deep Neural Networks (DNNs) have achieved remarkable progress in cutting-edge domains, their inherent brittleness has become a growing concern. To ensure the reliability and safety of DNN-enabled software, DNN testing has emerged as an indispensable practice. Within this context, test input prioritization is essential for early fault detection and reducing labeling costs. However, it remains challenging to accurately identify failure-inducing inputs. Although decision ambiguity and distributional atypicality are two widely adopted perspectives for characterizing inter-class competition and intra-class typicality respectively, relying on either perspective in isolation inevitably introduces blind spots. In this paper, we propose DuFP (Dual perspective Feature space Prioritization), a KNN density-based test input prioritization approach for DNNs that jointly incorporates both inter-class and intra-class perspectives. The prioritization framework of DuFP is built upon class-conditional density estimation. Based on the estimation results, prediction correctness is characterized by an ambiguity score and an atypicality score, with the former reflecting decision ambiguity and the latter quantifying distributional atypicality. A hybrid uncertainty score is then constructed by integrating both scores to guide the final prioritization. We evaluate DuFP on prioritization and selection tasks across image and text datasets under clean, corrupted, and adversarial scenarios. Experimental results demonstrate that DuFP effectively and efficiently prioritizes fault-inducing inputs and outperforms state-of-the-art approaches.

Mon 28 SeptSoftware Engineering
The gist
Deep neural networks can sometimes make unexpected errors, which is a problem for software relying on them. The authors propose a new way to find test cases that are likely to cause errors by combining two ideas: how unsure the network is about its decision and how unusual the input is compared to what it normally sees. Their method, called DuFP, uses these ideas together to better spot tricky inputs that might cause failures. They tested DuFP on images and text under different conditions and found it works better than other similar methods.
Open → 2609.34703v1

Attention reveals key model circuits in one pass

Attention as a Routing Graph: Live Circuit Extraction from a Single Forward Pass

Abstract: Finding circuits in language models usually means running many careful interventions. We try something simpler: treat attention as a routing map from one forward pass, keep a small set of routes that point toward the answer, and ask whether those routes actually matter. They often do. On induction and IOI (tasks where the "right" circuit is already known), ablating our extracted edges hurts the model much more than ablating a random set of the same size. We evaluate n=100 prompts per cell on GPT-2 Small, GPT-2 Medium, and Pythia-410M, with paired gap tests and bootstrap confidence intervals. The extract step costs one forward; a head-by-head patch sweep costs about two orders of magnitude more. We are not claiming a complete circuit atlas. We are claiming a cheap sketch that carries real causal signal on known tasks, with clear failure modes when it does not. Code and evaluation artifacts are at https://github.com/Aquinf03/live-circuit-routing.

Mon 21 SeptArtificial IntelligenceMachine Learning
The gist
Language models solve problems using internal circuits, which are usually found by testing many changes step-by-step. The authors tried a simpler way by looking at attention in just one run to find important routes that lead to the answers. They showed that removing these routes harms the model more than removing random ones, meaning these routes really matter. Their method is much faster than previous approaches but doesn’t find all circuits perfectly.
Open → 2609.25285v1

Llm agents benchmarked for trusted post-training model delivery

Trains but Doesn't Learn: A Post-Training Delivery Benchmark for LLM Agents as Forward-Deployed Engineers

Abstract: Post-training is becoming a service (PTaaS): a customer hands an operator data and a goal, and a forward-deployed engineer (FDE) returns a fine-tuned, evaluated, and deployed model under a budget, a human-approval gate, and reproducibility requirements. Seating an LLM agent in the FDE seat raises a question existing benchmarks cannot answer: not whether an agent can raise a metric, but whether it can be trusted to deliver. We answer it on a governed delivery plane, where an agent drives ten stages and an oracle scores each stage from platform-recorded facts. The central silent failure is the run that trains but does not learn (TBDL): loss falls, every signal stays green, and the delivered model is no better than the base. An operator-run acceptance gate catches every such run before payment, and a detector calibrated on known-corrupted runs flags severe corruption mid-run. We ran four frontier agents (Claude Opus 5, GPT-5.6-luna, Gemini 3.7 Flash, DeepSeek V4-Pro) end to end on metered L40S, A100, and H200 GPUs across 8B to 70B open bases, certifying every scenario before scoring. We also ran a human FDE arm under the same oracle and compare every agent against it.

Mon 21 SeptMachine LearningArtificial IntelligenceComputation and Language
The gist
Training large language models (LLMs) after deployment is now offered as a service, where an engineer tunes and delivers a model based on user data and goals. The authors created a new benchmark that tests whether an LLM acting as this engineer can be trusted to truly improve the model, not just reduce training loss without learning. They ran four advanced AI agents and compared them to human engineers using a detailed scoring process. The study found that some runs look successful but actually produce no improvement, and they developed techniques to detect and block such failures.
Open → 2609.25237v1

Neural spectral capacity predicts and optimizes model design efficiently

Neural Spectral Capacity: Measuring and Designing Architectures from Network Specification Alone

Abstract: Modern Transformer design and compression both reduce to allocating capacity under a budget. The standard scalars for these decisions, #Params and #FLOPs, capture size and compute but not architectural structure: two architectures with identical parameter budgets but different depth-width, head, or FFN allocations receive identical scores yet behave differently. We propose Neural Spectral Capacity (NSC), a closed-form scalar grounded in the singular-value spectrum of each weight matrix. Under standard random initialization, the Marchenko-Pastur law renders NSC computable from the architectural specification alone, with no model instantiation, data, or gradients. Its layer-wise additive structure admits NSC-DP, an exact dynamic-programming solver returning the architecture globally maximizing NSC under resource constraints in seconds on a CPU -- a guarantee that black-box search over existing training-free proxies cannot provide. Empirically, NSC outperforms #Params, #FLOPs, and representative training-free proxies in ranking across seven Transformer and CNN families (on FlexiBERT, $τ= 0.505$ on pairs differing in #Params by less than 10%, where #Params collapses to 0.082); NSC-DP discovers a Transformer-XL architecture on WikiText-103 that beats the human-designed baseline in 2 seconds; and prunes LLaMA-7B to the best 5.7B model across eight commonsense reasoning tasks without any calibration data, about 5900x faster than the strongest training-free proxy baseline.

Sat 19 SeptMachine LearningArtificial Intelligence
The gist
Choosing the right structure for AI models often involves balancing size and computing power, but common measurements miss key details about design. The authors introduce Neural Spectral Capacity (NSC), a quick way to predict a model's potential by looking at its weight matrices without needing to train it or use data. They also provide NSC-DP, a method that quickly finds the best architecture layout under limits like size or compute. Their method ranks models more accurately than traditional measures and speeds up tasks like pruning large models without extra data.
Open → 2609.23087v1

Coding agent leaderboards struggle to rank top systems accurately

Coding Agents Have Converged: Why the SWE-bench Leaderboard Can No Longer Order Its Top Entries, and What to Measure Instead

Abstract: Small differences on coding-agent leaderboards are often read as an ordering of systems. We audit whether the published verdicts support this reading, using 254 SWE-bench submissions across four splits without running models. On Verified, the leading two entries each resolve 396 of 500 instances. The top ten share 285 successes and 51 failures, leaving 164 instances that distinguish their outcomes. Frontier solution sets have median nesting 0.935 against a score-implied baseline of 0.774, indicating strongly shared successes. Scores also depend on the evaluated model-scaffold pair: observed within-model scaffold ranges reach 29.8 percentage points, compared with the 8.8-point spread of the top thirty. Six of nine cell-mean interaction tests remain significant after Holm correction, although this observational design does not identify causal scaffold effects. Exact paired McNemar tests separate none of the 29 adjacent Verified top-thirty pairs at alpha=0.05, while the larger Test split separates 14 of 23. A stated leader-based rule yields three descriptive tiers, or two after Holm correction; non-rejection does not establish equivalence. We release the partition and a five-step audit protocol that profiles shared outcomes, tests paired differences, reports grouping sensitivity, and estimates the instance budget needed for resolution. The results motivate reporting comparison-set-specific resolution and model-scaffold provenance instead of interpreting small aggregate gaps as established rank differences.

Tue 15 SeptSoftware EngineeringArtificial Intelligence
The gist
Leaderboards rank computer programs that write code, but very close scores don’t always mean clear performance differences. The authors found that many top-ranking systems share most successes and failures, making it hard to say which is truly better. They suggest new ways to compare these systems by focusing on shared outcomes and reliability of differences instead of small score gaps.
Open → 2609.17394v1

Failure guided method improves prompts and training data together

Failure-Guided Co-Evolution of Prompts and Training Data

Abstract: Automatic prompt optimization (APO) improves language-model programs by revising prompts from task feedback, yet it typically holds its training data fixed. Repeatedly optimizing against the same instances confines feedback to weaknesses already represented in those data, leaving related failure conditions unexplored. We therefore view each failure as a dual signal: it indicates both how the prompt should be revised and what new training evidence should be synthesized. We introduce FORGE, a failure-guided framework that co-evolves prompts and training data. FORGE abstracts imperfect executions into reusable failure modes and synthesizes new training data through four complementary mutation strategies. Verified instances are fed back into prompt search, allowing updated prompts to expose the next data needs. Across eight heterogeneous benchmarks, FORGE improves the aggregate score over the unoptimized baseline by 16.52 percentage points and outperforms all evaluated APO baselines. The synthesized data also transfer beyond FORGE: in a transfer study, they improve all nine APO comparisons by 2--9 points and all three GRPO comparisons by 4--8 points under matched optimization budgets. These results establish failures as a shared interface between prompt optimization and data synthesis, and show the benefit of jointly adapting what a model is instructed to do and what it learns from.

Mon 14 SeptSoftware EngineeringArtificial Intelligence
The gist
Improving computer programs that use language models often involves making better prompts, but this usually relies on the same examples over and over, causing limited improvement. The authors introduce a new way to fix this by treating mistakes as clues not only to change the prompt but also to create new example data. Their method, FORGE, cycles between updating prompts and generating fresh training cases based on where the model fails. This approach consistently improved performance across several tests and helped different prompt tuning methods work better when reusing the new data.
Open → 2609.15209v1

Decoder cosine similarity struggles to predict model feature interactions

Where Decoder Cosine Similarity Fails for SAE Feature Flow Discovery

Abstract: Foundation models are increasingly adapted through fine-tuning, model editing, and alignment procedures while retaining previously acquired capabilities. Understanding the internal computations that support these adaptations is therefore becoming increasingly important for continual model evolution. Sparse autoencoders (SAEs) provide interpretable feature dictionaries for residual-stream activations and sublayer outputs, but it remains unclear how state features and update features interact to produce downstream residual features. In this work, we focus on MLP updates as a first test case. We construct a transition atlas of triples $s_k + u_j \rightarrow t_\ell$, where a residual-state feature and an MLP-update feature jointly predict a target residual feature, and validate candidate triples by ablating the decoded update feature. In a 20M-token Pythia-160M $L_7 \rightarrow L_8$ run, we find 38,125 strong ablation-effect transitions, but 88.0% have both state-target and update-target decoder cosine similarity below 0.7. As a preliminary cross-model check, a run of 20M-token Gemma-3-4B $L_{21} \rightarrow L_{22}$ causally validates only the top 30,000 ranked candidate triples by ablating the decoded update feature, and 53.6% of strong-effect triples have both state-target and update-target decoder cosine similarity below 0.7. The Gemma result is directionally consistent with Pythia, but weaker, since update-target cosine recovers many of the strongest Gemma effects and the run is not a full-atlas causal validation. Ultimately, our results suggest that feature flow atlases can serve as diagnostics of representation-update mechanisms and thereby inform tools for steering model updates. Future work will validate more complex patterns across layers, models, and SAE families.

Fri 11 SeptMachine Learning
The gist
The paper looks at how parts of large AI models work together when they learn new things but keep old knowledge. The authors study small building blocks inside these models called sparse autoencoders and how they predict changes in the model’s internal signals. They find that a common technique called decoder cosine similarity often fails to spot important interactions between features that help the model update itself. This suggests that new tools are needed to better understand and guide how AI models change and learn over time.
Open → 2609.12591v1

Layer selective unlearning improves privacy and robustness in large language models

Forgetting Only What Matters: Layer-Selective Unlearning toward Robust LLMs

Abstract: Large Language Models (LLMs) can memorize and reproduce sensitive, copyrighted, or otherwise undesirable training content, creating privacy, safety, and regulatory concerns. Machine unlearning offers a practical alternative to full retraining, but many existing methods apply broad or fixed parameter updates that can degrade utility and remain brittle under deployment changes such as post-training quantization, where forgotten knowledge may partially re-emerge. We propose Forgetting Only What Matters via Unlearning Layers (FOM-UL), a layer-level unlearning framework that selects transformer layers using a forget-to-retain significance score. This score identifies layers with high influence on the forget set and low sensitivity to the retain set, allowing FOM-UL to concentrate updates where they are most effective while leaving most of the model unchanged. This targeted update strategy improves the forgetting-utility trade-off and provides an empirical path toward quantization-resilient unlearning by reducing the chance that small, diffuse updates are erased by low-bit rounding. Across TOFU, KnowUnDo, and MUSE-style evaluations, FOM-UL reduces residual memorization compared with strong GA, NPO, KLD, SURE, ReLearn, and LUNAR-based baselines while preserving retain-set utility close to the vanilla model. Under 8-bit and 4-bit post-training quantization, FOM-UL maintains stronger memorization suppression and utility preservation than competing methods, and adversarial prompt evaluations show lower recovery of forgotten content. Overall, FOM-UL provides an efficient unlearning strategy that improves targeted forgetting, utility preservation, and deployment robustness without claiming formal guarantees of erasure.

Wed 9 SeptMachine LearningArtificial Intelligence
The gist
Large language models sometimes remember sensitive or unwanted information they were trained on, which can cause privacy and safety problems. The authors propose a new way to selectively erase only the important parts of these models that hold this sensitive data, without messing up the parts that are still useful. This method focuses on updating specific layers in the model to forget unwanted content, making the model safer and more reliable even after compressing it for deployment. Their tests show this approach works better than previous methods at forgetting the right information while keeping the model useful.
Open → 2609.10439v1