Machine Learning
Improved Confidence Estimates for Black-Box Large Language Models
Sokhna Diarra Mbacke, Mouloud Belbahri and Gabriel Loaiza-Ganem
Uncertainty quantification (UQ) is essential for the safe deployment of large
language models (LLMs). Existing methods, from verbalized confidence to ones requiring multiple generations, are often zero-shot and produce scores quantifying uncertainty without the need for labelled data. Nonetheless, in practice one must always evaluate their performance on a
dataset of interest before deployment. In this work we show that, by leveraging this
dataset, we consistently outperform these existing scores. Specifically, we build simple classifiers that predict LLM response correctness by using these scores and the correctness of similar queries as features. Our method produces minimal computational overhead, making it a cheap and straightforward enhancement for UQ in LLMs for real-world applications.
Read the full abstract ▾
Machine Learning · Cross-listed from quant-ph
Quantum Gaussian processes for prediction of channel observations
Jonas Jäger, Yaroslav Khmelnitskiy, Paolo Braccia et al. (+4 more)
Given a set of input states, we consider the task of predicting the expectation value of a Pauli observable at the output of an unknown
quantum evolution, using only a limited number of measurements. Recently,
quantum Gaussian process (QGP) regression was introduced for this task across various classes of
unitary evolution. Here, we extend the QGP framework beyond
unitary dynamics. In particular, we prove convergence of the channel's outputs to a QGP and derive the associated closed-form kernel under a uniform (Lebesgue measure) prior over
quantum channels. The kernel's dimensional factor, however, dictates the required observation precision. While manageable when the channel and observable are restricted to small subsystems, exponential suppression precludes learning when the subsystem grows extensively with the system size. Since the Lebesgue prior is overly broad for many applications, we propose an empirical Bayes heuristic that replaces the dimensional factor with a learnable scale parameter while retaining the kernel's state-overlap correlation structure. In numerical simulations of up to 64
qubits, channel QGP regression with the Lebesgue kernel exhibits a strong inductive bias for local channels, enabling faithful extrapolation. For global 64-
qubit channels, the rescaled kernel restores learnability, with predictions improving systematically with the shot budget. Results from a noisy
quantum computer further demonstrate the robustness of QGP regression under experimental conditions. Beyond regression, we validate QGPs as
Bayesian-optimization surrogates for state preparation under noisy XXZ dynamics.
Read the full abstract ▾
All authors Jonas Jäger, Yaroslav Khmelnitskiy, Paolo Braccia, Artur Miroszewski, Diego García-Martín, M. Cerezo, Piotr Czarnik
Machine Learning
Quantum Kernel Estimation for the Discovery of Early Lung Cancer Detection
Hamed Javidi, Alex Zajichek, Hakan Doga et al. (+3 more)
Lung cancer screening with low-dose chest computed
tomography reduces mortality, but its impact is limited by uptake, adherence, and management challenges. Blood-based
cell-free DNA (cfDNA) biomarkers offer a complementary approach, although early detection remains difficult because of lung cancer heterogeneity and high-dimensional, nonlinear molecular signals. We evaluated
quantum-classical hybrid machine learning for lung cancer detection using DNA fragmentomics and DNA methylation. After feature selection, models were trained using 20- and 40-feature subsets. Features were encoded into
quantum Hilbert space using angle and dense-angle feature maps with multiple
entanglement strategies. Fidelity-based
quantum kernels were computed with exact statevector simulation and integrated with precomputed-kernel SVM and kernel-PCA logistic regression and compared with an SVM model trained on the original features. This framework enabled systematic evaluation of how encoding and
entanglement design affect classification. Across repeated held-out evaluations,
quantum-kernel models achieved competitive performance on both
datasets. For fragmentomics, several 20-feature configurations improved AUC relative to a classical SVM baseline, suggesting effective capture of nonlinear cfDNA fragmentation structure. For methylation, the classical SVM achieved the highest AUC, although selected
quantum models remained competitive and improved specificity in some cases. Increasing features from 20 to 40 did not consistently improve performance and often increased variability. Overall, these results support
quantum kernel methods as a promising approach for cfDNA-based lung cancer detection.
Read the full abstract ▾
All authors Hamed Javidi, Alex Zajichek, Hakan Doga, Laxmi Parida, Filippo Utro, Peter J. Mazzone
Machine Learning
Holtercare-Bench: A Multimodal Benchmark for Evaluating Long-Term Dynamic ECG Analysis
Yihan Xie, Hanwen Cui, Runze Ye et al. (+8 more)
While
multimodal large
language models (MLLMs) excel in medical applications, most of them favor static images or short-term signals. In the critical field of dynamic electrocardiograms (ECG), models struggle with complex temporal
reasoning and diagnostic report generation due to a lack of high-quality
datasets and
benchmarks. To address this, we introduce (i) Holtercare-23K, a large-scale
multimodal dynamic ECG
dataset comprising 22,980 QA pairs derived from 788 clinical Holter records and featuring a novel signal-
video-text tri-modal
alignment. Based on this
dataset, we present (ii) Holtercare-Bench, a
multimodal benchmark that evaluates models on temporal localization, clinical diagnosis, and global summarization. Zero-shot evaluations of leading MLLMs reveal a significant performance gap in processing ultra-long pathological sequences. However, fine-tuning representative models yields substantial improvements. This work illuminates the limitations of current MLLMs in electrophysiology and provides a foundational
benchmark for long-term medical MLLMs. Our project is available at https://github.com/ZJU4HealthCare/Holtercare-Bench.
Read the full abstract ▾
All authors Yihan Xie, Hanwen Cui, Runze Ye, Juekai Lin, Haoyang Wang, Jinhao Mao, Bo Zhang, Wenqiao Zhang, Xiaogang Guo, Jun Xiao, Lei Zhang
Machine Learning · Cross-listed from cs.CV
Clustering and Token Denoising for Faster and More Robust VLMs
Baptiste Rossigneux, Inna Kucher, Vincent Lorrain et al. (+1 more)
Recent Visual-
Language Models (VLMs) have enhanced the capabilities of pre-trained LLMs by adding
vision tokens alongside text, with approaches like LLaVA showing impressive results. However, the computational burden of processing up to 576 or 729 visual
tokens makes edge deployment challenging. While various
token pruning techniques require retraining, some are training-free and thus can easily adapt to architecture changes. We introduce ClustRS, a two-part, training-free
algorithm for robust
token pruning. Its first component is an
attention-weighted, clustering
algorithm that selects representative
tokens from each semantic cluster. The second component, Residual Shrinkage, is a one-pass denoising step on the selected
tokens. These training-free lightweight steps make LLaVA ready for real-world data, improving robustness to a wide range of image-noise types and intensities. Experimental results on the ScienceQA-IMG and MM-VET
benchmarks show our method outperforms
attention- and diversity-based methods by up to 20% under extreme noise and
token conditions (reducing
tokens by 97%, down to 16
tokens) on LLaVA 1.5 7b and achieves exceptional results on LLaVA-OneVision, where we match baseline performance with fewer than one-third of their
tokens under mild noise conditions. Our study demonstrates a simple yet powerful alternative to both score-only and diversity-only
pruning rules, paving the way for compute-efficient and noise-resilient VLM deployment.
Read the full abstract ▾
All authors Baptiste Rossigneux, Inna Kucher, Vincent Lorrain, Emmanuel Casseau
Machine Learning · Cross-listed from stat.ME
Recovering Nonlinear Functions of Latent Variables: A Plausible-Value Neural Network Framework
Eunjeong Song and Sehee Hong
When factor scores replace true latent scores in nonlinear prediction, measurement error attenuates the recoverable variance of any $k$th-order component of the regression function by $\rho^k$ – the $k$th power of the score's coefficient of determination – for any linear score type. This study derives the bound via Hermite
polynomial expansion and proposes PV-ANN – plausible values (posterior draws preserving latent variance) combined with artificial neural networks (learning functional form without prespecification). The bound governs recovery of the latent-scale function, not prediction of the outcome from observed indicators, for which factor scores are already sufficient; the two metrics are therefore predicted to dissociate. An 18-condition simulation supports both predictions: in the nonlinear low-reliability conditions PV-ANN closes about four fifths of the function-shape recovery gap between a factor-score learner and one given the true latent values, and the margin widens as reliability falls, while predictive accuracy is not improved, as the theory requires. A Big Five application illustrates the intended exploratory workflow and delineates boundary conditions under weak signal and measurement model misspecification.
Read the full abstract ▾
Machine Learning
Beyond Teacher Likelihood: Group-Calibrated On-Policy Distillation for Long-Context Reasoning
Zhu Zhang, Jixun Wang, Xiaoang Xu et al. (+6 more)
On-policy
distillation (OPD) trains a student on its own responses using dense
token-level guidance from a stronger teacher. In long-context tasks, however,
token-level teacher support can favor locally plausible responses that omit evidence distributed across the input or violate global task constraints. Task-specific verifiers, in contrast, evaluate task completion at the response level and may return graded rewards that reflect partial success. We diagnose this mismatch on fixed responses from two representative long-context evidence-aggregation tasks. Across longer input ranges, trajectory-level OPD scores become progressively less aligned with verifier rewards, indicating teacher-verifier disagreement. Motivated by this observation, we introduce Group-Calibrated On-Policy
Distillation (GC-OPD). GC-OPD separately normalizes verifier rewards and trajectory-level OPD scores within each rollout group and uses their difference as a signed teacher-verifier disagreement residual. Relative-advantage-based credit assignment (RACA) distributes this trajectory-level residual across
tokens according to their relative OPD advantages while preserving the original OPD signal. Across five long-context
benchmarks, post-training with GC-OPD raises the five-
benchmark averages of the official Qwen3-4B and Qwen3-8B checkpoints from 29.08 to 40.47 and from 35.12 to 44.65, respectively. Vanilla OPD reaches 39.31 and 43.56 under the same setup. Controlled ablations show that the signed residual is more effective than either an additional OPD-derived term or direct group-normalized verifier reward addition, while RACA further improves over uniform
token allocation. Together, these results demonstrate that group-relative residual calibration can incorporate verifier outcomes without discarding dense
token-level guidance. Code is available at https://github.com/SolereZhang/GC-OPD.
Read the full abstract ▾
All authors Zhu Zhang, Jixun Wang, Xiaoang Xu, Xiaorong Wang, Zihan Zhou, Zhiyuan Wang, Shuo Wang, Chaojun Xiao, Yuezhi Zhou
Machine Learning
Lévy Attention: Single-Pass Predictive Uncertainty for Continuous-Time Attention
Sotirios P. Chatzis and Loukas Papadoulas
Deep models for irregularly-sampled time series answer queries at arbitrary continuous timestamps, yet report nothing about how far each answer should be trusted. We show the
attention layer itself can close that gap: with the right
stochastic formulation, the pass that makes each prediction also reports, in closed form and at no extra cost, how far it should be trusted. We introduce Lévy
Attention, a cross-
attention operator whose output is a
stochastic integral against an inhomogeneous Poisson random measure: query-key compatibilities assemble an intensity over a continuous (time x channel) index space, the measure scatters atoms under it, and the output averages an interpolated value field at those atoms. In expectation it reduces to a mollified cosine-kernel
attention, so it replaces a softmax layer and trains with exact gradients. What softmax discards, the Poisson construction preserves in closed form: the evidence $\Lambda_q$ (total compatibility mass) and the disagreement $\mathrm{tr}\,\Sigma_V(q)$ (value spread). An exact variance identity makes their combination $\hat\sigma(q)=\sqrt{\mathrm{tr}\,\Sigma_V(q)\,\varphi(\Lambda_q)}$ the root-mean-square deviation of the sampled operator, emitted by the deterministic pass with no trained head. Empirically, disagreement carries the signal, while the evidence factor swings from uninformative on dense data to strongly informative on sparse. On t-PatchGNN the operator swap costs at most 5.6% accuracy against a matched control and nothing on the sparsest
dataset. The free disagreement signal improves on 20-pass MC dropout across matched five-seed suites, and $\hat\sigma$ scales a calibrated Gaussian whose zero-sample CRPS beats a fifty-draw sampler; a split-conformal wrapper reaches nominal coverage at every level, and one pass ranks 3,383 unseen patients by trust in 1.4 seconds.
Read the full abstract ▾
Machine Learning
Learned, Then Lost: A Measured Single-Example Counterfactual in Pre-training
Zachary Speck and Asa Shepard
A single training example's contribution to a finished model is normally estimated rather than measured, because measuring it takes two expensive full pre-training runs that differ in one row of one batch. We ran that counterfactual 24 times at a small scale. We trained 32 GPT-2 models at 124M parameters from scratch on OpenWebText, over four conditions and eight seeds. At step 200 of 9,536, at peak learning rate, we replaced one row of a 256-row batch with a fixed context injection carrying a 194-
token passage. The three injected conditions are: 1. fluent prose with a corpus-attested subject, 2. fluent prose with a fabricated subject matched to it within 0.14% on full-batch gradient delta, and 3. random keyboard characters. The fourth condition is an uninjected twin. The passage is learned from one exposure and then decays. Fifty steps after injection, the arm that saw a passage predicts it better than the arm that did not by 0.039 and 0.044 nats of cross-
entropy on the passage, at eight of eight seeds with p < 0.0001. At the final step we do not detect that difference for either passage, at p = 0.25 and p = 0.71, against minimum detectable effects of 0.025 and 0.079 nats, nor between the two passages, at p = 0.54. Every
geometric measure we report is taken after that decay. Our pre-registered contrast on interpolation loss barrier is +0.0068 with p = 0.509, against a minimum detectable effect of 0.032 barrier units. Held-out cross-
entropy is -0.00044 with p = 0.310. Per-layer centered kernel
alignment does not detectably separate any condition at any layer. Weight displacement reaches 44.1% of the seed-to-seed Euclidean distance and is 92% settled by the midpoint of training, while the barrier reaches 3.0% of the seed-to-seed barrier. Those two figures sit roughly 15 times apart, and that is a lower bound. The injection relocates the model within its basin without moving it out.
Read the full abstract ▾
Machine Learning
Continuous-Time Reinforcement Learning for Controlled Hawkes Jump-Diffusions
Tomasz R. Bielecki, Thibaut Mastrolia and Haoze Yan
We study
stochastic control of multivariate Hawkes-driven
stochastic differential equations with machine learning
algorithms in a non-Markovian setting. Due to the path dependence of the
memory of the Hawkes intensity, this problem does not fall within classical
stochastic control theory outside particular Markovian kernels. We first develop a finite-dimensional Markovianization procedure and
algorithm to approximate multivariate Hawkes processes with mixtures of exponential kernels. We prove the convergence of the Markovianized approximation of the Hawkes process, its intensity, and the value of the problem to the original non-Markovian processes and the value of the primal problem. We then formulate continuous-time deterministic policy gradient learning on the Markovianized approximation of the problem, called Hawkes-CT DDPG. We propose a model-free
algorithm to solve the non-Markovian Hawkes-driven optimization by observing only the event times of the process, the realization of the solution to the SDE, and a chosen set of decay filters, while the Hawkes kernel coefficients remain unknown. We compare our continuous time reinforcement learning Hawkes-CT DDPG method with discrete time reinforcement learning techniques under three different types of kernels: simple exponential, Erlang, and power-law kernels.
Read the full abstract ▾
Machine Learning · Cross-listed from cs.SD
Geometric Iterative Retrieval for Neural Audio Codec Resynthesis
Leo Schmidt-Traub, Frédéric Berdoz, Luca A. Lanzendörfer et al. (+1 more)
Neural audio codecs based on Residual Vector Quantization (RVQ) have become the dominant discrete representation for
token-based general audio generation, yet resynthesizing high-quality audio from coarse codec
tokens remains an open problem and bounds the fidelity of every system that generates them. Prior work has framed resynthesis as a choice between discrete
token prediction and continuous regression. We argue that this dichotomy is incomplete and introduce
geometric iterative
retrieval, a paradigm that uses the RVQ layer hierarchy itself as a natural iterative decomposition in continuous codebook space. Rather than classifying over discrete vocabularies or regressing to a single target vector, our method performs contrastive
retrieval in the codebook's
geometric space. We evaluate our method on codec restoration tasks across
speech and music, and show improvements over both single-pass
token prediction and one-step regression baselines.
Read the full abstract ▾
All authors Leo Schmidt-Traub, Frédéric Berdoz, Luca A. Lanzendörfer, Roger Wattenhofer
Machine Learning · Cross-listed from cs.AI
Grouping the Stochastic Machine: Precision, Not Capability, as the Frontier Metric for AI Systems
George Andrikopoulos
Frontier
language models are compared, marketed, and benchmarked on capability – what their best or average output can achieve. I argue this measures the wrong axis. The models have saturated accuracy: their mean output lands on the target. What now separates one system from another in practice is precision: how tightly concentrated their outputs are around that target across repeated, identical requests. Borrowing the marksman's distinction, capability is where the average shot lands; reliability is the size of the group. I make three claims. First, precision, not capability, is the frontier differentiator between systems, and
benchmark culture systematically fails to measure it, reporting central tendency rather than spread. Second, precision is measurable, cheaply and without circularity, by running a fixed suite of deterministically scored tasks many times at fixed temperature and computing the per-task consistency of outcomes – no model-in-the-loop grader required. Third, the measurement is not merely descriptive but decision-guiding: it separates consistent failures (a tight group off-centre, correctable by the operating discipline of Paper 1 – a sight adjustment) from scattered failures (a wide group, correctable only by changing the model or its sampling – a rifle problem). I define a grouping metric, specify a harness, and show how tracking a human-AI pair's grouping over time yields the compounding signal that Paper 1's field study requires. A first real run, since replicated, illustrates both the method and its most important limit: one measured gap was closed completely by a single rule (0/5 -> 5/5), while a suite of tasks authored from the rules themselves found no value, because a frontier model already embodies explicit good practice – establishing that a discipline's worth is found by measurement on real work, not constructed from its own rulebook.
Read the full abstract ▾
Machine Learning
SCORE: Subject Coordinate Recovery for Label-Free Cross-Subject EEG-to-Image Retrieval
Zhenyao Cui, Siyuan Kan, Siyang Li et al. (+2 more)
Accurate visual decoding can reveal how the
brain represents visual information and recover perceived content from neural signals such as electroencephalography (
EEG), with potential for neural communication. However, current
EEG-to-image
retrieval methods perform far below their within-subject counterparts for new users without labeled calibration, limiting real-world deployment. To understand this gap, we analyze
EEG features across subjects and find that different subjects preserve similar relationships among concepts but express them along different coordinate directions. We therefore propose Subject Coordinate Recovery (SCORE), a target label-free framework combining recovery-aware source training with coordinate
alignment at deployment. During training, SCORE aligns source subject
EEG with a common image space and simulates unseen-subject recovery through source-only episodes. At deployment, with both encoders frozen, SCORE selects reliable
EEG-image landmarks through hubness-corrected matching and estimates an orthogonal transformation to recover target
EEG coordinates without source data or target labels. In 200-way
retrieval on two public
benchmarks, SCORE outperforms the unadapted baseline for every target subject and achieves the best overall accuracy. It reaches 53.23%/83.55% and 12.01%/32.16% Top-1/Top-5 on THINGS-
EEG2 and Alljoined-1.6M, respectively, surpassing the strongest baselines by 17.45/15.70 and 3.08/4.62 percentage points. Without target labels or encoder updates, SCORE brings
brain-based visual decoding closer to robust, practical, low-latency deployment across users.
Read the full abstract ▾
All authors Zhenyao Cui, Siyuan Kan, Siyang Li, Ziwei Wang, Dongrui Wu
Machine Learning
Beyond Trial Averaging: Anchoring Neural and Visual Representations for Few-Repetition Brain-to-Image Retrieval
Zhenyao Cui, Siyuan Kan, Dingkun Liu et al. (+1 more)
Decoding visual information from
brain signals probes neural representations and enables neuro-rehabilitation and dream decoding. Recent
brain-to-image
retrieval approaches have achieved promising performance, typically by averaging many (up to 80) neural trials per image, requiring repeated stimulus presentation that increases latency, cost, and user burden. When only one or a few repetitions are available, the
retrieval accuracy drops sharply. This drop is commonly attributed to query noise because averaging suppresses noise and increases signal stability. However, we find a non-transitive
alignment pattern: the low-repetition query signal and the image representation each align with the high-repetition center, but not directly with each other. This pattern shows that query noise is only part of the problem and that gallery placement also affects
retrieval. We therefore propose a neural-anchor-based
retrieval (NEAR) framework that treats the high-repetition center as an anchor and approaches it from both sides: a denoiser pulls the noisy query toward the true anchor, and a small network predicts each candidate's pseudo anchor from its image and pulls the image toward it. Across four
datasets spanning
EEG, MEG and fMRI, NEAR consistently improved
retrieval in the few-repetition regime. On THINGS-
EEG2, it improved 200-way Top-1 accuracy by 5.7 and 9.3 percentage points respectively, when averaging one and four repetitions. By anchoring neural and visual representations, NEAR reduces reliance on repeated acquisition and brings neural
retrieval closer to real-world deployment.
Read the full abstract ▾
All authors Zhenyao Cui, Siyuan Kan, Dingkun Liu, Dongrui Wu
Machine Learning
Leaf Values as Coordinates: Exact Contrastive Explanation for Gradient-Boosted Ensembles
Emanuele Luzio
A gradient-boosted ensemble predicts by summing one leaf value per tree. Read those values as coordinates rather than as intermediate results, and every instance becomes a point in R^M on which the model acts linearly: the score is the sum of the coordinates. This small change of view makes contrastive explanation exact. The difference between two instances is a vector that is identically zero wherever they share a leaf, so the gap between a rejected applicant and an accepted one is carried by a handful of coordinates, each traceable to a real split in a real tree. Nothing is fitted, sampled, or assumed additive in features – the additivity is already there, in the right space. We build a recourse method on this representation and evaluate it on five tabular
datasets under repeated cross-validation. Its recommendation reconstructs the model's own decision to 6.2 x 10^-15, so an auditor can re-check the arithmetic without the model. On the credit
datasets it is Pareto-non-dominated on effort against realism. And when recommendations are restricted to changes the subject could actually make – not their age, not a settled delinquency – it retains 58% of its validity where the strongest baseline retains 41%, a distinction the standard evaluation cannot see because it never asks whether a recommendation can be carried out.
Read the full abstract ▾
Machine Learning
PGFS++: Molecular Property Improvement under Synthesis and Diversity Constraints
Boqiao Zhang, Godbless James, Sai Krishna Gottipati et al. (+1 more)
Improving molecular properties, such as drug-likeness or binding affinity, is a recurring task in early-stage drug discovery. However, molecules optimized in an unconstrained chemical space have limited practical value if they cannot be synthesized. Policy Gradient for Forward Synthesis (PGFS) is a synthesis-aware reinforcement learning method for molecular improvement, but its use of reactant
embedding prediction makes reactant selection indirect, which, as we show, limits learning effectiveness. We first develop PGFS+, in which reaction templates and second reactants are represented by trainable
embedding lookup tables. Combined with a more effective scoring function and RL
algorithm, PGFS+ significantly improves the desired property. However, it exposes a reward-hacking failure mode: a powerful reactant search can map diverse input molecules to the same high-reward magnet molecule, improving the reward while collapsing the output diversity. We therefore introduce PGFS++, a synthesis-aware reinforcement learning framework for input-specific molecular improvement. Given an input molecule, PGFS++ treats it as the start of a forward-synthesis trajectory, applies learned reaction templates with compatible in-stock building blocks, and produces a molecule with improved target properties, an explicit synthesis route, and structural similarity to the input. Experiments on molecular improvement tasks show that PGFS++ improves target properties while preserving high output diversity.
Read the full abstract ▾
All authors Boqiao Zhang, Godbless James, Sai Krishna Gottipati, Andrew Fitzgibbon
Machine Learning
Discretizing Continuous Time Series for Imputation with Masked Diffusion Training
Dongbin Kim, Seungyun Lee, Geonwoo Shin et al. (+1 more)
Time series imputation is a crucial area for reliable time series analysis, yet it remains challenging due to the complex temporal dynamics and noise of real-world data. Existing approaches, however, exhibit two limitations: missing and observed values are embedded within the same representation space without explicit structural separation, and continuous
diffusion-based methods are trained to predict added noise rather than the original signal. To address these, we propose the Masked
Diffusion Time-series Imputation Model (MDTIM), which leverages the training paradigm of masked
diffusion model for imputation tasks. The MASK
token is structurally orthogonal to valid observations, and the model directly predicts the original values, naturally aligning both the representation and the learning objective with the imputation task. To bridge the gap between discrete masked
diffusion and the continuous, ordinal nature of time series, we further introduce
Stochastic Discretization, which maps continuous values to ordinal-aware
tokens while preserving continuous dynamics. Our experiments on diverse
benchmarks confirm that MDTIM achieves superior robustness and scalability, consistently outperforming state-of-the-art deterministic and generative baselines across various missing scenarios.
Read the full abstract ▾
All authors Dongbin Kim, Seungyun Lee, Geonwoo Shin, Jaewook Lee
Machine Learning
Enhancing EBSD throughput of battery electrode materials using super-resolution generative adversarial networks
John Mangum, Andrew Glaws, Francois Usseglio-Viretta et al. (+2 more)
Quantitative microstructural characterization of Li-ion battery electrode materials using electron backscatter diffraction (EBSD) has been proven as a critical method for optimizing
cell performance. However, the inherently slow nature of EBSD can hinder the throughput of analyses needed for statistical representation of a material microstructure being developed. This work demonstrates a machine learning super-resolution framework using a generative
adversarial network (SRGAN) to significantly increase EBSD throughput. The SRGAN model was trained on EBSD data of LiNixMnyCozO2 (NMC) cathode particles to computationally enhance low-resolution
datasets and its performance is compared against classical interpolation methods across various upscaling factors (2x to 12x). Both qualitative image metrics and quantitative microstructural analysis verified that the SRGAN systematically outperformed classical methods, particularly in preserving small grains and maintaining realistic grain boundaries. We demonstrate that a 5x upscaling factor, corresponding to a 25x speed-up in acquisition time or a 25x larger field of view, is practical while maintaining acceptable accuracy in key metrics like grain size and shape. For instance, at 5x upscaling, relative errors were +5.7%, +8.2%, and -14.6% on grain area-equivalent diameter, grain maximum sphere-inscribed diameter, and grain boundary length, respectively. The SRGAN methodology developed in this work significantly enhances the efficiency of EBSD acquisition for more statistically robust microstructural
dataset, enabling EBSD as a high-throughput characterization tool for materials research and industrial process development.
Read the full abstract ▾
All authors John Mangum, Andrew Glaws, Francois Usseglio-Viretta, Steven Spurgeon, Donal Finegan
Machine Learning
Pretraining Reusable Inference Across Views with Synthetic Task Priors
Jielong Lu, Zhihao Wu, Jiajun Yu et al. (+2 more)
Modern pretrained encoders make representations from heterogeneous views increasingly reusable, but the procedure that determines view utility and combines evidence is still relearned for each downstream task. Consequently, knowledge about view relevance, complementarity, reliability, and missingness is repeatedly discarded rather than transferred across tasks. We therefore reformulate multi-view learning as learning a reusable, task-conditioned inference procedure rather than a fixed fusion function. Based on this perspective, we propose SIMPLE, a prior-fitted multi-view in-context learner that predicts query labels by conditioning on a small labeled support set. Since existing real-world
datasets cover only a limited range of view configurations and task structures, we construct a controllable synthetic task prior in
embedding space. It generates diverse support-query episodes with varying class structures, shared and view-specific factors, representation geometries, cross-view dependencies, reliability levels, missingness patterns, and distribution shifts. A hierarchical inference architecture then performs
reasoning within views, across views, and across support and query samples. Experiments on multi-view and multi-omics
benchmarks demonstrate that the frozen variant of SIMPLE achieves competitive performance without updating the inference backbone, while lightweight adapter calibration attains leading performance on most evaluated
datasets. Together, the results under frozen, one-shot, and missing-view settings support the central hypothesis that multi-view
reasoning itself can be pretrained and reused, while lightweight adapter calibration provides task-specific
alignment when needed.
Read the full abstract ▾
All authors Jielong Lu, Zhihao Wu, Jiajun Yu, Zhaoliang Chen, Haishuai Wang
Machine Learning
Open-MOPD: Diagnosing and Fixing Capability Imbalance in Multi-Teacher On-Policy Distillation
Huan-ang Gao, Haohan Chi, Yong Yan et al. (+7 more)
Multi-teacher on-policy
distillation (M-OPD) has emerged as a promising paradigm for consolidating domain-specialized reinforcement learning (RL) experts into a single generalist student via dense,
token-level reward supervision. Despite its practical success, the optimization dynamics governing multi-teacher capability integration remain poorly understood, and open, rigorously reproducible recipes are conspicuously lacking. In this work, we establish a controlled M-OPD
benchmark on SmolLM3-3B-Base with oracle routing, isolating capability integration from routing ambiguity. Our investigation reveals a pronounced capability integration gap: standard M-OPD captures only 35.6% of the available headroom relative to a domain-routed oracle ensemble, with concise tasks such as instruction following suffering severe degradation and premature stagnation. Crucially, we show that this failure stems not from gradient conflict, but from a severe misallocation of the
token-level optimization budget. This pathology is driven by three orthogonal factors: structural sequence-length disparities across domains, dynamic convergence drift due to non-uniform learning rates, and multi-step reward staleness from asynchronous policy updates. To resolve these imbalances, we introduce Open-MOPD, a principled framework incorporating
token-share balancing, gap-aware dynamic budget allocation, and student reward refresh. Together, these mechanisms systematically restore cross-domain balance, elevating headroom recovery from 35.6% to 83.4% in a single deployable student. We fully open-source our end-to-end post-training recipe, training trajectories, and evaluation suites on an academically accessible hardware budget.
Read the full abstract ▾
All authors Huan-ang Gao, Haohan Chi, Yong Yan, Shiyuan Feng, Hanlin Wu, Zheng Jiang, Bingxiang He, Wei-Ying Ma, Ya-Qin Zhang, Hao Zhou
Machine Learning
Cluster Assignments in Soft Targets Shape Speech Representations: Evidence from S-JEPA
Wenxuan He, Yunpeng Li, Zewei Li et al. (+4 more)
Cluster-based prediction is widely used in self-supervised
speech learning. A soft target preserves a distribution over clusters rather than a single label. This distribution specifies both the probability values and which clusters receive them. Comparisons between soft targets and hard labels do not separate the contributions of these two aspects to the learned representation. We study this in S-JEPA, a recent high-performing self-supervised
speech model trained with soft Gaussian mixture model (GMM) targets. We compare its original targets with counterfactual targets that preserve the most likely cluster and all probability values but change which remaining clusters receive the other probabilities. Across three training seeds, the original soft distribution is recovered more accurately from Encoders trained with the original than counterfactual targets. Because this could reflect target matching alone, we also test low-level acoustic and phonetic information. Both are more accessible from Encoders trained with the original targets. This suggests that cluster assignments affect acoustic and phonetic properties of the learned representation, not just recovery of the training target.
Read the full abstract ▾
All authors Wenxuan He, Yunpeng Li, Zewei Li, Yongke Yang, Yuze Li, Yin Cao, Shan Liang
Machine Learning · Cross-listed from stat.ML
Learning Random Geometric Graphs Drawn in Probabilistic Metric Spaces
Dalia Chakrabarty, Kangrui Wang, Chuqiao Zhang et al. (+1 more)
We present a new data-driven learning of a Random
Geometric Graph (RGG) of a multivariate
dataset, where the graph is drawn in a probabilistic metric space. This graph learning works for generic
datasets, irrespective of the type of the observables; their probability distributions; or size of the data. We identify a metric of the space that the graph is drawn in, as a probability distribution of a random variable that we introduce, namely, a variable that represents the disparity between the connectedness of two vertices of the graph, and the correlation between the two random variables that are attached to the respective vertex. It is the closed-form {\it{cdf}} of this disparity variable that we advance as the distance function of the host space of the learnt RGG, such that the edge exists between any two nodes, if this inter-nodal distance falls short of a chosen cutoff probability. Drawing the RGG in this probabilistic space leads to the graph being an Soft RGG, such that any edge - if it exists - exists with an identified probability. We forward a simple Rejection Sampling-based technique for learning the probability of any edge. The expected degree distribution of a vertex of this RGG is identified as local, and dependent on the inter-observable correlation
matrix. If said correlation
matrix is not known, it can be learnt given the data, using its closed-form posterior probability density function, that we forward. We illustrate our graph learning method by learning multiple RGGs of highly multivariate real
datasets.
Read the full abstract ▾
All authors Dalia Chakrabarty, Kangrui Wang, Chuqiao Zhang, Ye Liu
Machine Learning · Cross-listed from cs.CV
SPK: Eliciting Structured Prior Knowledge for Interpretable Out-of-Distribution Detection in Real-Time Object Detection
Changshun Wu, Weicheng He, Xiaowei Huang et al. (+1 more)
Object detectors often produce over-confident predictions for objects outside their training categories, leading to so-called out-of-distribution (OoD)
hallucinations. Existing approaches for detecting or mitigating such
hallucinations typically either construct scoring functions directly over learned object detector representations or modify the object detector itself to suppress
hallucination emergence. However, the latent priors implicitly encoded in these representations remain largely unexplored and have not been explicitly decoded for OoD detection. To uncover and exploit these latent priors, we propose Structured Prior Knowledge (SPK), a
hallucination-oriented framework that explicitly elicits OoD-relevant priors from pretrained object detectors. Specifically, SPK leverages in-distribution data and
hallucination-inducing samples as diagnostic supervision to elicit part-level semantic concepts underlying object detector decision-making, rather than using them merely for rejection or object detector adaptation. The elicited semantic priors are further integrated with
geometric and contextual priors to form a compact five-dimensional SPK representation for OoD detection. Extensive experiments across diverse object detector architectures and multiple OoD
benchmarks demonstrate that SPK achieves state-of-the-art OoD detection. Our findings reveal that pretrained object detectors already encode substantially richer latent knowledge than is typically exploited for OoD detection. More importantly, this knowledge can be explicitly elicited and organized into a compact, structured, and interpretable knowledge space for prediction reliability analysis. This suggests a promising proactive route for improving object detector reliability by explicitly uncovering and leveraging latent priors. Code and data are available at: https://gricad-gitlab.univ-grenoble-alpes.fr/dnn-safety/spk
Read the full abstract ▾
All authors Changshun Wu, Weicheng He, Xiaowei Huang, Saddek Bensalem
Machine Learning · Cross-listed from cs.AI
Robust Risk Under Evolving Uncertainty: A Wasserstein Counterpart of the Entropic Value-at-Risk
Deep Kumar Ganguly and Jan Křetínský
An
agent still learning its environment should be cautious while ignorant and bold once confident. The entropic value-at-risk captures this through a robust-optimization identity—a confidence level fixes the radius of a relative-
entropy ball of alternative models—but that ball cannot reach catastrophes the nominal deems impossible, precisely what a safe
agent must hedge. We instead use an optimal-transport ball and study the coherent risk measure it induces, the Wasserstein entropic value-at-risk. It has a variational dual mirroring the entropic formula (an inverse temperature becomes a transport price), occupies a definite place in the risk hierarchy, and provably accounts for the reachable catastrophes the entropic measure ignores; we verify both dualities numerically. Driving the transport radius by belief
entropy then yields a closed-form robust dynamic-programming operator whose caution contracts as the belief sharpens, with a certified safety sandwich and a sharp safety switch.
Read the full abstract ▾
Machine Learning · Cross-listed from cs.AI
What is Missing from AI Post-Training AI: An Empirical Analysis
Joy Jia Yin Lim, Xin Huang, Hao Peng et al. (+5 more)
Large
language model (LLM)
agents can now post-train an LLM end-to-end, raising the prospect of recursive self-improvement (RSI). Yet this progress is measured by aggregate
benchmark scores, which cannot tell whether an
agent executes a fixed plan well or strategically revises the plan when it fails. We separate these two capabilities: execution-level capability, iterating within an established training strategy, and strategy-level capability, revising that strategy as experimental evidence accumulates. Analyzing 1,338 post-training trajectories of frontier
agents, we find that
agents reliably execute post-training but lock into a default strategy, which follows the
agent rather than the task, and only 2.1% of transitions between adjacent training runs ever change strategy. We then test whether the
agent lacks experience,
reasoning, or the decision to switch. (1) Experience improves execution but not the strategy. (2) Additional
reasoning compute yields front-loaded gains on easier tasks but refines, rather than revises, the committed strategy. (3) Human review before training changes which strategy the
agent locks into, not whether it locks in, whereas a single mid-run instruction outperforms the
agent's own continuation by up to 17.44 points under the same budget. In conclusion, what the
agent lacks is the decision to reopen a committed strategy and try another one. Realizing RSI therefore calls for interaction protocols and training signals that make strategy revision an explicit, rewarded decision.
Read the full abstract ▾
All authors Joy Jia Yin Lim, Xin Huang, Hao Peng, Yaxi Lu, Xin Cong, Zhong Zhang, Maosong Sun, Yankai Lin
Machine Learning · Cross-listed from stat.ML
Diffusion Models for High-Dimensional Clustered Data: Intrinsic-Dimension Adaptivity via Bayesian Classification
Yuga Iguchi and Paul Fearnhead
The empirical success of
diffusion models in generative modelling has motivated theoretical work, including quantitative error bounds and qualitative analyses that characterise the different phases of denoising. We bring these two areas together by studying the adaptivity of
diffusion models to the structured geometry of
multimodal high-dimensional data that consists of multiple clusters in $\mathbb{R}^D$, each with its own low-dimensional structure, and inter-cluster separation depending on $D$. We employ $K$-mixture Gaussian distributions as a canonical framework to capture this geometry and establish two theoretical results. First, we interpret denoising as a dynamical
Bayesian classifier: the mixture score is a posterior-weighted average of cluster-wise scores, and we show that, with high probability, the posterior class probabilities concentrate on a single cluster once the signal-to-noise ratio reaches the scale $\Theta (\log (KD)/D)$. Second, by separately analysing the denoising process in its mixing and cluster-commitment phases, we prove that the KL error bound depends linearly on the maximum intrinsic dimension of a cluster, up to a logarithmic factor, even when $K$ grows polynomially with $D$. This improves on ambient-dimensional bounds and extends existing low-dimensional adaptivity analyses to
multimodal distributions with heterogeneous, approximately low-rank covariances.
Read the full abstract ▾
Machine Learning
Multi-Agent Off-Policy Deep Reinforcement Learning for Smart Campus Coverage
Omar Rady, Mohamed Ayman, Ali Arafa et al. (+1 more)
Deep reinforcement learning (DRL) has recently gained a great
attention due to its real-time adaptation and effectiveness in complex optimization problems. This paper investigates the optimal deployment of millimeter-wave (mmWave) base stations (BSs) in a realistic, non-convex campus
topology. The optimization problem is NP-hard, due to the non-convex, non-smooth nature of the max-min
fairness objective. To overcome these constraints, we formulate the BS placement as a Markov Decision Process (MDP) and systematically
benchmark four DRL schemes: a discrete single-
agent Deep Q-Network (DQN), a spatially partitioned Multi-
Agent DQN, a continuous single-
agent Deep Deterministic Policy Gradient (DDPG), and a geographically partitioned multi-
agent DDPG framework. Numerical evaluations reveal that the multi-
agent DDPG approach substantially outperforms single-
agent in dense scenarios. Additionally full coverage is achieved, and a
fairness Jain's index of 0.94 is obtained. Finally, the multi-
agent demonstrates highly efficient computational convergence of dense scenarios with $400$ users.
Read the full abstract ▾
All authors Omar Rady, Mohamed Ayman, Ali Arafa, Mohamed Shalma
Machine Learning · Cross-listed from quant-ph
Bernstein-Vazirani Networks: Quantum Machine Learning by Interference
Natacha Kuete Meli, Tolga Birdal, Prayag Tiwari et al. (+2 more)
We introduce Bernstein-Vazirani Networks (BVNs), a non-variational
quantum machine learning framework that leverages
quantum interference for supervised learning, demonstrated on
vision and representation learning tasks. In their standard form, BVNs follow the principle of
quantum Fourier sampling: labelled data are placed in superposition and interfered in the
Fourier basis to extract globally informative features. We then define generalised BVNs that enable interference in problem-adapted bases, yielding more expressive models under the same measurement budget as in the standard setting. BVNs achieve universal function approximation through (over)complete interference bases, while training of BVNs is gradient-free. Experiments on synthetic and real-world classification tasks, as well as implicit image representation, show strong generalisation capabilities and competitive performance with classical and
quantum baselines.
Read the full abstract ▾
All authors Natacha Kuete Meli, Tolga Birdal, Prayag Tiwari, Vladislav Golyanik, Michael Moeller
Machine Learning
Harness Continual Learning: Continual Adaptation Beyond Model Parameters
Borui Kang, Jinrui Gu, Junhan Lv et al. (+3 more)
Continual learning has largely been model-centric, treating model parameters as the state that changes with sequential experience. Modern
agents can also adapt through a harness of
prompts,
memories, tools, skills, and routing rules. Because these contents jointly shape later execution, a harness update can disrupt previously reliable behavior even when the model is frozen. This raises a new question: how can an
agent continually improve its state outside the model while retaining behavior acquired earlier? We formulate Harness Continual Learning (HCL), a new continual learning paradigm in which the harness evolves around a frozen foundation model, and define the resulting loss of earlier behavior as harness-level forgetting. We instantiate HCL with four execution-facing components: the Task Interface, Experience
Memory, Capability Map, and Adaptive Router. We further introduce guarded harness
evolution to separate update generation from state commitment. A Continual Optimizer proposes candidate harnesses from post-execution feedback, and a Continual Evaluator commits the resulting candidate harness only after checking current improvement, historical retention, and validity. Experiments on textual
reasoning,
multimodal perception, and open-world interaction demonstrate capability accumulation and failure recovery, with relative gains exceeding 10% over corresponding baselines in multiple settings. Component ablations assess the contribution of each harness component, while controlled retention sweeps reveal measurable harness-level forgetting and show that the stability–plasticity trade-off can be explicitly adjusted.
Read the full abstract ▾
All authors Borui Kang, Jinrui Gu, Junhan Lv, Wenbin Li, Lei Wang, Yang Gao
Machine Learning
Monroe: A Molecular Foundation Model for In-Context Probabilistic Inference
Blazej Banaszewski and Andrew W. Fitzgibbon
Bioassay activity prediction is often data-limited because drug-discovery
datasets rely on time-consuming and expensive wet-lab experiments for data generation and evaluation. This challenge has inspired recent research into molecular foundation models (MFMs), which aim to encode general-purpose chemical knowledge into molecular representations that generalize well in data-constrained scenarios. This paper presents Monroe, a new MFM with several innovations over the existing state of the art: increased scale allowing pre-training on over 81 million molecules from the PM6
quantum chemistry
dataset; improved graph representation of stereochemistry; improved training losses including conformer denoising and
embedding decorrelation; improved multi-task learning; and the use of a prior-data-fitted model (TabPFN) for downstream in-context prediction. Our evaluations use a principled pairwise comparison framework that measures statistically significant performance differences. Across established Polaris
benchmarks, Monroe matches or exceeds existing MFMs, while on activity cliff
benchmarks, designed to assess utility for molecular discovery, it achieves significant improvements over prior methods. Finally, ablation and transfer experiments show that PFN-based downstream predictors also substantially improve two leading existing models, MiniMol and CheMeleon, yielding new state-of-the-art variants we call MiniMol_PFN and CheMeleon_PFN, suggesting that our downstream adaptation strategy generalizes beyond Monroe. Source code is at github.com/blazejba/monroe.
Read the full abstract ▾