Machine Learning
Breaking the Tokenizer Barrier: On-Policy Distillation across Model Families
Yifan Niu, Han Xiao, Dongyi Liu et al. (+4 more)
On-Policy
Distillation (OPD) has become a core technique in the post-training of Large
Language Models (LLMs) for transferring knowledge from domain experts to student models. However, existing OPD
distillation methods require teacher and student models to share the same tokenizer, restricting the applicability of OPD within the model series. Current mainstream practice typically employs Supervised Fine-Tuning (SFT) on teacher-generated responses for cross-tokenizer
distillation, which fails to capture the rich knowledge embedded in the teacher's probability distribution. In this work, we enable the standard on-policy
distillation method to operate across model families, ensuring that high-fidelity
token-level signals can propagate across different tokenizers with a precise
token-mapping
algorithm. Extensive experiments show that cross-tokenizer OPD is significantly more compute-efficient than baselines on various
benchmarks. Our results unlock a broader range of teacher-student pairs for OPD, opening up new avenues for adapting and enhancing interactions between LLMs.
Read the full abstract ▾
All authors Yifan Niu, Han Xiao, Dongyi Liu, Zelong Wang, Dihong Gong, Yasheng Wang, Jia Li
Machine Learning · Cross-listed from cs.RO
Dense Force Estimation with an Event-based Optical Tactile Sensor
Agis Politis, René Zurbrügg and Valentina Cavinato
Humans rely on spatially dense, geometry and force-aware tactile feedback at high temporal resolution for dexterous manipulation. While
vision-based tactile sensors enable dense force estimation, they are limited by camera frame rates, motion blur, and data bandwidth. Event-based optical tactile sensors offer an attractive alternative with microsecond temporal resolution and low motion blur, but existing methods are restricted to predicting only net forces. We introduce the first framework for dense 3D force field reconstruction using event-based optical tactile sensors. Our approach estimates 3D surface displacements from event data and maps them to forces via the inverse Finite Elements Method (iFEM). Shear displacements are recovered through the proposed event-based marker tracking
algorithm, while normal displacements are predicted by a convolutional neural network trained on a collected
dataset of synchronized force-displacement-event data. Experiments demonstrate accurate reconstruction of physically grounded forces, achieving a mean absolute error of (0.14 N, 0.10 N, 0.93 N) over force ranges up to (4 N, 4 N, 20 N), while operating at an average of 100 Hz. This work constitutes a first step toward enabling dense force feedback for high-frequency control in robotic grasping and dexterous manipulation.
Read the full abstract ▾
Machine Learning
A transition-density-based operator learning method for Fokker-Planck equations with various initial conditions
Li Zeng, Xiaoliang Wan, Yaobin Wang et al. (+2 more)
Solving Fokker-Planck equations (FPEs) for multiple initial conditions typically requires repeated computations, leading to substantial computational costs. In this work, we propose a transition-density-based operator learning method to efficiently approximate the solution operator of FPEs with various initial conditions. The core idea is to learn the transition probability density function (PDF) of the underlying
stochastic differential equation (SDE), from which the solution associated with a new initial distribution can be obtained through the Chapman-Kolmogorov equation without retraining the model. A major challenge in learning the transition PDF lies in the singular behavior induced by the Dirac initial condition. To address it, we introduce a conditional normalizing flow whose base distribution is given by the explicit transition PDF of a linearized SDE. This base distribution captures the short-time behavior of the target transition PDF and allows the normalizing flow to learn a near-identity transformation at small times. We further incorporate a time-weighted loss function to stabilize training near the initial time and develop an importance-sampling strategy for evaluating solutions associated with general initial conditions. A variety of numerical experiments are presented to illustrate the effectiveness and robustness of the proposed method.
Read the full abstract ▾
All authors Li Zeng, Xiaoliang Wan, Yaobin Wang, Fabio Nobile, Tao Zhou
Machine Learning
Graph Mamba Operator: A Latent Simulator for Interacting Particle Systems
Karn Tiwari, Niladri Dutta, N M Anoop Krishnan et al. (+1 more)
Modeling interacting dynamical systems requires capturing spatial interactions alongside long-range temporal dependencies. Graph neural networks (GNNs) provide a natural representation but typically rely on autoregressive rollouts and treat spatial and temporal dynamics separately, leading to error accumulation over long horizons. Existing approaches also focus on local interactions and short temporal contexts, limiting their ability to capture multi-hop dependencies and global structure. We introduce the Graph Mamba Operator (GraMO), a latent-space simulator that integrates state-space models with graph-based interaction learning. In contrast to prior work that sequences nodes or applies spatial and temporal updates in separate stages, GraMO couples graph-based interactions and temporal state updates within a single recurrence. The update is linear in the latent state, with input-dependent coefficients that adapt across regimes. We evaluate GraMO on N-body systems, motion capture, and robotics
datasets, achieving the lowest error across
benchmarks and the largest gains in long-horizon prediction.
Read the full abstract ▾
All authors Karn Tiwari, Niladri Dutta, N M Anoop Krishnan, Prathosh A P
Machine Learning
LargeMonitor: Monitoring Online Task-Free Continual Learning via Large Pretrained Models
Mingqi Yuan, Xiaoquan Sun, Shihao Luo et al. (+1 more)
Online task-free continual learning (TFCL) requires intelligent
agents to sequentially accumulate knowledge from an unbounded, non-stationary data stream under strict single-pass constraints and without any explicit task identifiers. Existing online TFCL paradigms primarily rely on parameter-efficient
prompt tuning or dynamic structure expansion driven by training-coupled optimization dynamics, such as empirical loss fluctuations or evolving latent distances. As a result, these training-coupled solvers remain agnostic to the structural origins of distribution drift, mechanically enforcing a fixed strategy across fundamentally distinct streaming variations. To address this gap, we propose LargeMonitor, a framework that leverages large pretrained foundation models to autonomously orchestrate task-free continuous adaptation. Specifically, LargeMonitor introduces a decoupled detection module utilizing the frozen, stable representation space of large
vision models (LVMs) to achieve robust, zero-shot drift detection without training-dependent interference or brittle threshold tuning. Upon a confirmed drift, the framework activates a context-aware diagnostic module driven by large
multimodal models (LMMs) to interpret the precise semantic etiologies of the stream variation (e.g., novel class emergence vs. environmental domain shift). This dual-stage capability empowers the continuous learner to dynamically deploy adaptive and shift-specific optimization strategies. Extensive experiments across multiple TFCL settings and
benchmarks demonstrate that LargeMonitor achieves precise, robust detection and diagnosis of complex data streams while consistently improving the performance of existing online TFCL
algorithms.
Read the full abstract ▾
All authors Mingqi Yuan, Xiaoquan Sun, Shihao Luo, Jiayu Chen
Machine Learning · Cross-listed from cs.CR
Now You (Still) See Me: Detecting Evasive Steganographic Payloads in LLMs
Charles Westphal, Timothy Douglas, Keivan Navaie et al. (+2 more)
Large
language models can be fine-tuned to encode
prompt-borne secrets into fluent, seemingly benign outputs. This creates a steganographic exfiltration risk that is difficult to detect with output-level steganalysis. Recent work proposes mechanistic detection using linear probes that recover the secret from internal activations. We show that this defense can be systematically evaded, but that detectability can be recovered through a targeted data-level intervention. First, we extend the detection setup to include a non-linear MLP probe. We then adversarially fine-tune steganographic trojans across five base models: Qwen3-8B, Llama-3.1-8B, Ministral-8B, Qwen3-14B, and Phi-4-14B. The resulting models retain $58$–$79\%$ exact-match secret recovery while evading both ridge and held-out MLP probes, with $1$–$8\%$ average capability degradation across six
benchmarks. We then give an information-theoretic characterization of this evasion. Successful evasion preserves recoverability while reducing low-order extractability of the secret from the content-aligned representation, forcing the payload into synergistic interaction with residual degrees of freedom. This motivates a recontextualization
dataset that restricts these residual degrees of freedom. On this distribution, both ridge and MLP detectability are restored across all five evasive trojans. Overall, our findings show that activation-based steganography detection is vulnerable to adaptive evasion, but also that theory-guided evaluation distributions can expose otherwise hidden payloads.
Read the full abstract ▾
All authors Charles Westphal, Timothy Douglas, Keivan Navaie, Tiago Pimentel, Fernando E. Rosas
Machine Learning · Cross-listed from cs.AI
Correct Looks Better: Pairwise Comparisons Reveal Accuracy Rankings
Mina Remeli and Moritz Hardt
Pairwise comparisons combined with aggregation methods like Elo have become central to evaluating generative models, yet concerns remain that they reward superficial stylistic cues or display judge biases. In a more positive turn, we show that model rankings from pairwise comparisons strongly agree with ground-truth-based accuracy rankings when such ground truth is available for comparison. By converting five well-known
benchmarks into free-form generative evaluations, we find that Elo rankings achieve a Spearman correlation above 0.9 with accuracy rankings and substantially outperform direct evaluation when the judge is weak. Furthermore, style and judge bias have only minor effects on model rankings, despite most judgments occurring on pairs where both candidate answers are correct (or incorrect). On such pairs, we find that repetition after the final answer (echo) is a causal driver of judge preference.
Read the full abstract ▾
Machine Learning · Cross-listed from stat.ML
SAILS: Surrogate-based Analysis of Interactions via Local Effect Smooths
Timo Heiß, Julia Herbinger, Bernd Bischl et al. (+1 more)
Feature interactions drive much of the predictive power of machine learning models, yet existing explanation methods only detect and quantify interactions without revealing their functional form, or visualize only restricted interaction types. We propose Surrogate-based Analysis of Interactions via Local Effect Smooths (SAILS), a model-agnostic framework that analyzes pairwise interactions through generalized additive model (GAM) surrogates fitted to the local effects of a black-box model. For each interval of a feature of interest, the surrogate smooth terms isolate the interaction components on derivative level, enabling (i) interaction detection through a heuristic derived from significance tests on smooth terms, (ii) interaction form categorization into linear, product-separable, and non-product-separable types, and (iii) tailored, interpretable visualizations for each interaction type. We empirically validate the framework through controlled simulations and a real-world task, showing its effectiveness for pairwise interactions, with limitations under strong feature correlations and higher-order interactions. SAILS fills a notable gap in the eXplainable AI (XAI) toolbox, going beyond detecting interactions alone to characterizing their functional form.
Read the full abstract ▾
All authors Timo Heiß, Julia Herbinger, Bernd Bischl, Giuseppe Casalicchio
Machine Learning
Benchmarking Empirical Privacy Protection for Adaptations of Large Language Models
Bartłomiej Marek, Lorenzo Rossi, Vincent Hanke et al. (+4 more)
Recent work has applied differential
privacy (DP) to adapt large
language models (LLMs) for sensitive applications, offering theoretical guarantees. However, its practical effectiveness remains unclear, partly due to LLM pretraining, where overlaps and interdependencies with adaptation data can undermine
privacy despite DP efforts. To analyze this issue in practice, we investigate
privacy risks under DP adaptations in LLMs using state-of-the-art attacks such as robust membership inference and canary data extraction. We
benchmark these risks by systematically varying the adaptation data distribution, from exact overlaps with pretraining data, through in-distribution (IID) cases, to entirely out-of-distribution (OOD) examples. Additionally, we evaluate how different adaptation methods and different
privacy regimes impact the vulnerability. Our results show that distribution shifts strongly influence
privacy vulnerability: the closer the adaptation data is to the pretraining distribution, the higher the practical
privacy risk at the same theoretical guarantee, even without direct data overlap. We find that parameter-efficient fine-tuning methods, such as LoRA, achieve the highest empirical
privacy protection for OOD data. Our
benchmark identifies key factors for achieving practical
privacy in DP LLM adaptation, providing actionable insights for deploying customized models in sensitive settings. Looking forward, we propose a structured framework for holistic
privacy assessment beyond adaptation
privacy, to identify and evaluate risks across the full pretrain-adapt pipeline of LLMs.
Read the full abstract ▾
All authors Bartłomiej Marek, Lorenzo Rossi, Vincent Hanke, Xun Wang, Michael Backes, Franziska Boenisch, Adam Dziedzic
Machine Learning · Cross-listed from cs.CL
PriFT: Guiding Supervised Fine-Tuning with Pretrained Priors
Ke Wang, Shuangqi Li, Mathieu Salzmann et al. (+1 more)
Supervised fine-tuning (SFT) is an efficient approach for downstream task adaptation and often serves as the initialization stage for reinforcement learning (RL), but it can show weaker generalization than RL. A key limitation is its off-policy objective: SFT fits fixed demonstrations
token by
token, including targets poorly aligned with the model's pretrained distribution, which can lead to overfitting. A recent line of work addresses this issue by assigning larger training weights to
tokens better aligned with the current model's predictive distribution, with the intuition that fitting these
tokens are less distortive to the model's knowledge and representations. However, computing the
token weights from the model that is currently fine-tuned entangles
token weights with the optimization trajectory, inducing a self-reinforcing loop which concentrates the supervision signal to a narrow set of initially preferred
tokens. To address this, we propose PriFT (Prior-support guided Fine-Tuning), which derives
token weights from a frozen pretrained reference to obtain a stable reweighting signal unaffected by fine-tuning. This signal estimates prior support: the extent to which each target
token is supported by the pretrained distribution. By reweighting
tokens according to their prior support, PriFT limits excessive adaptation to targets requiring substantial deviation from the pretrained distribution and improves SFT generalization. We introduce two instantiations: PriFT-prob uses pretrained target-
token probability, while PriFT-mass selects
tokens by cumulative probability mass under the pretrained distribution. Across mathematical
reasoning, code generation, and medical QA, PriFT achieves state-of-the-art results among SFT baselines, while also providing a stronger initialization for subsequent RL.
Read the full abstract ▾
All authors Ke Wang, Shuangqi Li, Mathieu Salzmann, Pascal Frossard
Machine Learning
Distilling Safe LLM Systems via Soft Prompts for On Device Settings
Motasem Alfarra, Cristina Pinneri, Dana Kianfar et al. (+2 more)
Deploying safe large
language models (LLMs) on resource-constrained edge devices presents a critical challenge: while dual-model systems combining LLMs with guard models provide effective safety guarantees, their substantial
memory and computational demands make them prohibitively expensive for on-device deployment. This paper presents a comprehensive study of parameter-efficient safety
alignment methods for resource-constrained settings. Through systematic evaluation across multiple LLM architectures, training objectives, and parameter-efficient fine-tuning approaches, we identify that soft
prompts combined with
distillation-based training consistently outperform alternative methods. We introduce
distillation frameworks based on total variation and KL divergence that effectively transfer safety behaviors from guard models into learned soft
prompts. Our evaluations on various
benchmarks demonstrate that this combination achieves superior safety-usefulness trade-offs compared to LoRA adapters, steering vectors, and direct optimization methods, while requiring minimal additional
memory and compute at inference time. These findings establish soft
prompt distillation as the preferred approach for safety
alignment in on-device LLM deployment.
Read the full abstract ▾
All authors Motasem Alfarra, Cristina Pinneri, Dana Kianfar, Mohammed Almousa, Christos Louizos
Machine Learning
Reasoning Arena: Trace Tournaments When Verifiable Rewards Fall Short
Han Zhou, Adam X. Yang, Laurence Aitchison et al. (+2 more)
Reinforcement learning with verifiable rewards (RLVR) has become a leading paradigm for improving the
reasoning ability of large
language models through outcome-based supervision. However, verifiable rewards frequently become uninformative at the group level: when all sampled traces of a given
prompt receive identical rewards, group-relative advantage estimation provides no gradient signal, even though the traces may differ substantially in
reasoning quality. We propose
Reasoning Arena, an adaptive training framework that routes such non-diverse reward groups to a judge system instead of discarding them. Beyond examining the final answer,
Reasoning Arena constructs trace tournaments, where
reasoning traces are compared head-to-head to expose finer-grained preferences within the group, converting
reasoning quality into rich relative reward signals. To make reward estimation efficient, rather than exhaustively comparing every pair, each new trace is evaluated against a small, dynamically updated pool of previously generated traces as anchors to efficiently establish a relative ranking. We then fit a Bradley-Terry model on the incomplete comparison graph, enabling scalable RL integration without quadratic pairwise comparisons. Empirical results demonstrate that
Reasoning Arena consistently outperforms the RLVR baseline by 7.6% on average in competition mathematics and coding
benchmarks. By converting otherwise wasted zero-advantage samples into useful gradient updates, our method accelerates training by 27% to 41%, saving nearly 50% of generation compute, and substantially improves overall
reasoning performance.
Read the full abstract ▾
All authors Han Zhou, Adam X. Yang, Laurence Aitchison, Anna Korhonen, Albert Q. Jiang
Machine Learning
Scaling Neural Network Verification with Tensor Parallelism and Fully Sharded Data Parallelism
Sergei Vorobyov and Eugene Ilyushin
Formal neural network verification – proving that a network satisfies safety properties for *all* inputs in a specified domain – is bounded in practice by GPU
memory: standard implementations of bound-propagation
algorithms (IBP, CROWN, $\alpha$-CROWN) require weight and relaxation-coefficient
matrices to reside entirely on one accelerator. We adapt two parallelism techniques originally developed for large-scale model training to the auto_LiRPA / $\alpha,\beta$-CROWN verification framework. Tensor Parallelism (TP) shards both weight and $A$-
matrices across GPUs, achieving ${\approx}2\times$ peak-
memory reduction at $P{=}2$; soundness is confirmed on VNN-COMP 2022 MNIST-FC
benchmarks, though bound tightness degrades with the number of sharded zones due to forced IBP substitution for intermediate bounds inside sharded zones. Fully Sharded Data Parallelism (FSDP) shards only weight
matrices with a per-layer AllGather, producing bounds that are bitwise identical to the single-GPU baseline: baseline
memory drops by 80–90%, peak
memory by 34–39% on wide MLPs. FSDP integrates cleanly with complete verification ($\beta$-CROWN + Branch-and-Bound) and with convolutional layers (BoundConv); a complete unsat result is obtained for CIFAR-100 ResNet-large (VNN-COMP 2024) under FSDP. Across all experiments the
memory bottleneck in $\alpha$-CROWN+BaB mode proves to be per-
neuron alpha tensors, not weight
matrices, pointing to the key direction for future work.
Read the full abstract ▾
Machine Learning · Cross-listed from cs.CV
Zero-Shot Semantic Re-Identification for Autonomous Driving: A VLM Baseline Study
Eduardo Borges, Manuel Abreu, Luís Garrote et al. (+1 more)
Re-Identification (ReID) in autonomous driving is typically formulated as a visual matching problem, where observations of vehicles, pedestrians, and cyclists are associated across time, frames, or camera views using learned appearance
embeddings, often complemented by motion,
geometric, or
multimodal cues. However, purely visual representations may be sensitive to viewpoint, occlusion, illumination, and sensor-domain variations, limiting their interpretability and robustness in complex driving scenes. We propose a baseline study of a zero-shot pipeline using
Vision-
Language Models (VLMs) to generate textual descriptions of detected traffic participants and evaluate whether these descriptions can support identity matching across observations. Instead of relying only on low-level visual similarity, the proposed formulation represents each object through structured semantic attributes, including category, color, shape, pose, visible parts, spatial context, and distinctive visual cues. This study provides an initial
benchmark for
language-based re-identification in autonomous-driving scenarios, discussing and evaluating the strengths and limitations of current VLMs for this task. Results demonstrate that zero-shot semantic descriptions can support effective object re-identification, achieving
retrieval performance comparable to a supervised CNN baseline while offering greater interpretability through explicit identity cues. However, the experiments also reveal important challenges, including attribute inconsistency across viewpoints and limited fine-grained discrimination between visually similar instances.
Read the full abstract ▾
All authors Eduardo Borges, Manuel Abreu, Luís Garrote, Urbano J. Nunes
Machine Learning
PBSD: Privileged Bayesian Self-Distillation for Long-Horizon Credit Assignment
Yang Tian, Rui Wang, Xumeng Wen et al. (+5 more)
Long-horizon agentic tasks pose a fundamental credit assignment challenge for outcome-base reinforcement learning: trajectory-level rewards verify final correctness but provide limited guidance on which intermediate
reasoning steps or tool interactions contribute to the outcome. The difficulty is especially pronounced in multi-turn search
agents, where successful trajectories may contain misleading actions and failed trajectories may contain valuable evidence-gathering steps. We propose PBSD (Privileged
Bayesian Self-
Distillation), a Bayes-calibrated self-
distillation method for fine-grained credit assignment under sparse final rewards. PBSD measures trajectory quality through the posterior-to-prior probability ratio of the verified answer and applies Bayes' rule to convert this hard-to-estimate answer-side ratio into a tractable likelihood ratio between a standard student model and a privileged answer-conditioned teacher model. Autoregressive decomposition of this
Bayesian evidence score yields turn-level signals that identify whether each intermediate turn supports or undermines the verified outcome. Consequently, PBSD provides a principled and elegant reweighting scheme that transforms sparse outcome supervision into Bayes-calibrated turn-level credit signals, while remaining fully compatible with standard policy optimization. Experiments demonstrate that PBSD consistently enhances performance across both in-domain and out-of-domain settings, and effectively transfers knowledge from short-context training to long-context inference, suggesting that its fine-grained credit assignment mechanism facilitates more effective policy learning and yields improved generalization.
Read the full abstract ▾
All authors Yang Tian, Rui Wang, Xumeng Wen, Junjie Li, Shizhao Sun, Lei Song, Jiang Bian, Bo Zhao
Machine Learning
Thresholded Local Hyper-Flow Diffusion
Meher Chaitanya, Sebastian Dalleiger and Luana Ruiz
Local Hyper-Flow
Diffusion (HFD) gives an edge-size-independent Cheeger-type guarantee for seeded clustering in general submodular hypergraphs, but existing HFD solvers do not keep intermediate computation local at every iteration. We introduce Thresholded Local HFD (TL-HFD), a first-order method that maintains an active region around the seeds, performs projected subgradient updates on that region and its immediate boundary, and expands via thresholded (top-k) boundary activation. We prove that the local update is exact: the degree-preconditioned projected subgradient step restricted to the active region and its boundary coincides with the unrestricted global update. We establish finite-time dual suboptimality for both exact and thresholded updates, treating the latter as inexact projected subgradient steps with explicit skipped-boundary error. We further derive an additive activated-volume bound controlled by realized local subgradient norms and the minimum boundary-push among newly activated vertices, and translate approximate dual optimality with localized support into a robust sweep-cut guarantee for early-stopped iterates. For general submodular cut-costs, each iteration is local in the scanned region and oracle-sensitive in the hyperedge primitive. Empirically, TL-HFD often matches or improves over HFD while activating less volume, with the largest gains on noisy instances where
diffusion tends to absorb non-target vertices.
Read the full abstract ▾
Machine Learning · Cross-listed from cs.MM
Conan-embedding-v3: Fusing Modality-Specific Models for Omni-Modal Embedding
Shiyu Li, Zhiyuan Hu, Yifan Wang et al. (+3 more)
Omni-modal
retrieval promises a single
embedding space for text, image,
video, document, and audio inputs, but building such a unified retriever is difficult since these modalities differ in data distribution, architecture, and optimization dynamics. In this work, we present Conan-
embedding-v3, a decouple–fuse–recover framework for omni-modal
retrieval. Conan-
embedding-v3 first trains modality specialists independently and fuses their task vectors into a single dense backbone, a strategy we call Decoupled Specialist Fusion. We show that this fusion composes visual,
video, and document
retrieval capabilities, but also exposes a failure mode for projector-based modalities: when audio is attached through an external encoder and projector, fusing the backbone leaves the projector calibrated to the audio-specialist backbone, causing a large audio
retrieval regression despite copying all audio-specific modules unchanged. We call this failure Projector Drift. To repair it, Conan-
embedding-v3 applies Projector Recovery (i.e., full-parameter fine-tuning of the projector while keeping the backbone frozen) followed by balanced multi-modal rehearsal. The resulting model supports these
retrieval pathways in one backbone, achieving 74.9 scores on MMEB while obtaining 55.61 on the 30-task MAEB audio suite.
Read the full abstract ▾
All authors Shiyu Li, Zhiyuan Hu, Yifan Wang, Peiming Li, Zheng Wei, Yang Tang
Machine Learning
A Universal Dense Football Event Representation Based on TabTransformer
Weiran Yang, Daniel Memmert and Maximilian Klemp-Weins
Football event data constitute a rich spatiotemporal source for quantitative analysis of player actions in team sports. These
datasets contain heterogeneous features, combining continuous location coordinates with categorical variables such as action type, action outcome, and body part. Such data have been applied in sports analytics for match outcome
forecasting, player evaluation, and tactical pattern recognition. However, existing approaches predominantly encode categorical features using one-hot or ordinal
embedding representations, overlooking the intrinsic semantics of action descriptors. The
Transformer is a deep neural network architecture based on self-
attention that captures dependencies between input features at arbitrary positions. We propose and implement a
Transformer-based model to learn latent dependencies among categorical event features and produce dense representations of football events. By encoding categorical features as learned
embedding vectors, sport-specific action semantics are captured during pretraining, enabling the representations to support downstream tasks such as action value estimation and play style recognition. Empirical evaluation shows that the
embedding representations yield superior probability calibration over task-specific baselines on the downstream prediction tasks, as measured by Brier score.
Read the full abstract ▾
Machine Learning
Machine-Learning Emulation of Satellite Greenhouse Gas Retrievals: Stability over Time
Nugzar Gognadze, Motonobu Kanagawa, Yu Someya et al. (+1 more)
Retrieval algorithms are used to estimate atmospheric concentrations of greenhouse gases (GHGs), such as carbon dioxide (CO2) and methane (CH4), by solving inverse problems from high-spectral-resolution satellite radiance measurements. However, these
algorithms are computationally expensive, which makes real-time estimation at scale difficult. Machine-learning models have therefore been proposed as fast emulators of
retrieval algorithms. Most existing studies, however, evaluate them only on test data from the same period as the training data. We study the stability over time of such emulators using data from the Greenhouse Gases Observing SATellite (GOSAT). We show that prediction accuracy generally deteriorates when the test period moves away from the training period. We also show that including time as an input feature substantially improves XCH4 prediction for Lasso and neural-network models. Among the methods considered, a simple Lasso model performs as well as or better than more complex methods such as neural networks, and yields more stable predictions over time. We further validate the results using the Total Carbon Column Observing Network (TCCON), a ground-based observation network. On the TCCON-matched
dataset, the time-augmented Lasso achieves errors against TCCON that are comparable to the disagreement between GOSAT and TCCON for both XCO2 and XCH4.
Read the full abstract ▾
All authors Nugzar Gognadze, Motonobu Kanagawa, Yu Someya, Hisashi Yashiro
Machine Learning
Toward Compiler World Models: Learning Latent Dynamics for Efficient Tensor Program Search
Haolin Pan, Lianghong Huang, Xvlin Zhou et al. (+2 more)
Tensor program optimization is essential for modern machine learning systems, but its search space is enormous. Existing auto-schedulers reduce measurement cost with learned cost models, yet they usually evaluate each candidate as a static code snapshot, ignoring the schedule trajectory that produced it. This makes them insensitive to action dependencies and vulnerable to superficial code variations. We propose a world-model-inspired evaluator that models schedule evaluation as action-conditioned latent dynamics over program states. Starting from the initial program, it rolls out scheduling actions in a continuous latent space with a lightweight transition model, avoiding expensive AST mutation and repeated code encoding. The final dynamic representation is combined with action and hardware features to rank candidates. Implemented in TVM AutoScheduler, our method improves representative-subgraph latency over Ansor by 1.37$\times$ on GPU and 1.54$\times$ on CPU under the same 64-trial budget. It also matches Ansor-10K within 2.2%
geometric mean using 10$\times$ fewer measurements, and accelerates full-model inference over PyTorch/PyTorch-opt(cuDNN) by 4.61$\times$/3.67$\times$
geometric mean.
Read the full abstract ▾
All authors Haolin Pan, Lianghong Huang, Xvlin Zhou, Mingjie Xing, Yanjun Wu
Machine Learning · Cross-listed from cs.CL
SG-OPD: Sign-Gated On-Policy Distillation via Sign-Consistency Gating and Phased Teacher Sampling
Haoran Xu, Hongyu Wang, Yifei Gao et al. (+3 more)
On-policy
distillation (OPD) trains a student on its own trajectories with dense per-
token supervision from a stronger teacher, and often outperforms off-policy
distillation and standard reinforcement learning. However, we find that its effectiveness implicitly relies on two assumptions that frequently break in practice: trajectory-level
alignment between the student and the teacher, and uniform
token-level reliability of the teacher's preferences. We therefore propose Sign-Gated On-Policy
Distillation (SG-OPD), which uses a binary verifier as a trust signal for the teacher at two complementary granularities: phased teacher sampling mixes in verifier-endorsed teacher rollouts at cold-start, and a sign-consistency gate extrapolates the
distillation update on
tokens where the teacher agrees with the verifier-correct direction and interpolates it where it disagrees. Experiments on competition-level mathematical
reasoning benchmarks show that SG-OPD consistently outperforms standard OPD, with average gains of 1.98 and 7.50 at the per-sample and per-question levels, respectively.
Read the full abstract ▾
All authors Haoran Xu, Hongyu Wang, Yifei Gao, Jiaze Li, Xiaofeng Zhang, Xiaosong Yuan
Machine Learning
PRISM: Topology-Aware Cross-Modal Imputation for Modality-Deficient Federated Graph Learning
Zekai Chen, Miao Zhang, Jiayang Xing et al. (+4 more)
Multimodal federated graph learning (MM-FGL) aims to collaboratively learn from decentralized graphs with text and images. However, real-world clients may not share a common modality basis: a visual-search client may contain image–interaction graphs but no seller descriptions, while a catalog client may provide text but no product images. We refer to this practical setting as client-level modality deficiency. Unlike random instance-wise missingness, a deficient client lacks the local semantic basis needed to reconstruct the absent modality. More importantly, in graph learning, incomplete representations initialize message passing, so imputation errors can be filtered, mixed, and amplified by the receiving
topology. To address this gap, we propose PRISM (Proactive Retrieval and Imputation via Structural Meta-prompting), a
topology-aware
federated cross-modal imputation framework. Rather than reconstructing the missing modality solely from local observations, PRISM recovers missing-modality semantics from the federation and introduces them into local graph propagation under
topology-aware control. Experiments on six
multimodal graph
datasets across graph-centric and modality-centric tasks show that PRISM consistently improves modality-deficient clients, outperforming state-of-the-art baselines by 4.48% on average.
Read the full abstract ▾
All authors Zekai Chen, Miao Zhang, Jiayang Xing, Xunkai Li, Xun Wu, Rong-Hua Li, Guoren Wang
Machine Learning
Intention Driven Identification of In-Possession Match Phases in Association Football through Temporal Graph Learning
Yuesen Li and Daniel Link
Recognising intentions from continuous movement trajectories is an important problem in spatiotemporal data mining. This becomes difficult when behaviours change over time and their boundaries cannot be directly observed. Football provides a useful setting for studying this problem because tactical intentions emerge from the coordinated movements of multiple players during possession. Existing methods for match phase recognition often classify individual frames, use predefined spatial rules, or identify only selected tactical patterns. They therefore cannot fully describe continuous possession sequences in which tactical intentions change over time. We address this gap by treating the identification of the in-possession match phase as a hierarchical sequence
segmentation task. The framework includes three tactical intentions (Invade Opponent Space, Keep Possession, and Scoring) and six phases (Build Up, Progression, Counter Attack, Maintenance, Sustained Threat, and Finishing). We propose a Temporal Graph
Attention Network (T-GAN) that combines player interaction graphs, contextual features, and
Transformer based temporal modelling. The framework was evaluated on seven Bundesliga matches using leave-one-match-out validation and compared with rule-based, Random Forest, GNN, and
Transformer baselines at both frame and sequence levels. The results show that temporal modelling is important for coherent phase
segmentation, while graph modelling provides additional value for transition phases such as Counter Attack. The framework provides an interpretable method for discovering intention based behavioural segments from continuous player tracking data and supports automated tactical analysis in football.
Read the full abstract ▾
Machine Learning
Trajectory Geometry of Transformer Representations Across Layers
Vishal Pandey, Gopal Singh and Yacine Mahdid
Understanding how
transformer representations evolve across layers, not merely what they encode, remains an open problem in mechanistic interpretability. We recast the
transformer forward pass as a discrete population trajectory through a high-dimensional representation
manifold, drawing on
geometric tools from computational neuroscience. Rather than probing for pre-specified features, we characterize trajectory geometry using five metrics computed directly in the ambient space: trajectory length,
curvature, a semantic convergence index, layerwise cosine similarity, and representational stability. Across three model families (GPT-2, TinyLlama, Qwen2.5) and five controlled
prompt families, we report four findings. First, semantically related
prompts converge significantly in middle-to-late layers (peak CI 0.41–0.58, p<0.001, Mann-Whitney U), consistent with attractor-like dynamics. Second,
reasoning tasks produce trajectories of greater
curvature than lexical variations (0.71–0.83 rad vs. 0.27–0.31 rad), suggesting
curvature encodes computational complexity. Third, ambiguous
tokens exhibit trajectory bifurcation with up to 5.6x representational separation by the final layer, absent in unambiguous controls. Fourth, layerwise cosine similarity reveals a universal three-phase structure: encoding, elaboration, and output preparation, consistent across all three architectures. All four effects vanish under shuffled-layer and random-
embedding controls. We release a fully open-source, model-agnostic pipeline and argue that trajectory geometry constitutes a principled, probe-free lens for mechanistic interpretability.
Read the full abstract ▾
Machine Learning
Internalizing Geometric Law: Learning from Solver Residuals for Precision-Critical Generation
Rafael Cabral, Pang Zixi, Ziyi Shou et al. (+1 more)
Large
Language Models frequently hallucinate in precision-critical domains such as technical diagramming and mechanical design, where outputs must satisfy strict
geometric constraints. We study open-ended
geometric synthesis from natural
language: translating free-form descriptions into precise constructions whose entities must simultaneously satisfy dozens of interacting constraints. To make this tractable, we release PyGeoX, a programmable
geometric DSL that compiles declarative constraints into a differentiable loss, and PyGeoX-Bench, a stratified suite of 300 problems with per-constraint verifiable rewards. Using PyGeoX as a verifier, we identify a failure mode we call Outlier Gradient Masking: under global-norm rewards (any scheme that aggregates residuals through a single norm, for example, $\exp(-\mathrm{MSE})$), a single outlier constraint can nullify the learning signal across all others. To address this, we propose Saturating Additive Rewards (SAR), which decompose the reward into bounded per-constraint terms, preserving partial progress and ensuring consistent gradients even under severe violations. Against MSE-based rewards, the natural baseline for geometry solvers, SAR improves the hard-tier solving rate by $2.3\times$, and the resulting 8B model is competitive with much larger frontier systems on this
benchmark. We release the engine,
benchmark, and data at https://github.com/Huawei-AI4Math/PyGeoX.
Read the full abstract ▾
All authors Rafael Cabral, Pang Zixi, Ziyi Shou, Shen Xin
Machine Learning
ERBench: A Benchmark and Testsuite for Equation Discovery Algorithms
Paul Kahlmeyer, Henrik Voigt, Michael Habeck et al. (+1 more)
Equation discovery aims to automate the discovery of scientific models in the form of mathematical equations from data. Technically, equation discovery is implemented by symbolic regression
algorithms. Performance of symbolic regression for equation discovery is measured along two dimensions: Prediction accuracy on test data, and recovery of known groundtruth formulas. For standard regression, accuracy is typically measured on in-domain test data, for instance, by splitting a data set randomly into training and test data. While this makes sense for in-domain interpolation, which is the common goal in ordinary regression, it can be a misleading proxy for true model discovery and generalization. The obvious alternative is to measure out-of-domain accuracy. However, obtaining challenging out-of-domain test data is a non-trivial problem. Therefore, we focus on equation recovery for evaluating symbolic regression
algorithms for equation discovery. The rationale is that symbolic regression
algorithms that perform well in recovering known groundtruth formulas are good candidates to perform well in unknown equation discovery. Existing
benchmarks for symbolic regression include equation recovery tasks, however, with only a small number of groundtruth formulas that are publicly known. Moreover, these
benchmarks place less emphasis on evaluating the robustness of
algorithms in terms of their behavior under changing dimensionality, sampling size, sampling distribution and sampling domain. This, however, is of central importance to practitioners wanting to discover equations for modeling natural phenomena, since data is almost certainly noisy and comes from diverse domains, distributions, and sample sizes. To fill this gap, we introduce the Equation Recovery
Benchmark (ERBench), a new evaluation framework designed to rigorously assess
algorithms explicitly targeting the task of equation discovery.
Read the full abstract ▾
All authors Paul Kahlmeyer, Henrik Voigt, Michael Habeck, Joachim Giesen
Machine Learning · Cross-listed from cs.SD
Multi-View Speech Representation Learning for Parkinson's Disease Detection Using Context-guided Cross-modal Attention
George Theodosiou, Loukas Ilias and Dimitris Askounis
Parkinson's disease (PD) is a progressive neurodegenerative disorder that frequently causes
speech impairments associated with hypokinetic dysarthria. As
speech production relies on the precise coordination of complex neuromuscular mechanisms,
speech analysis has emerged as a promising non-invasive and cost-effective biomarker for early PD detection. Recent deep learning approaches have shown encouraging results; however, most existing methods rely on a single
speech representation, potentially overlooking complementary pathological information encoded across different feature spaces. In this work, we propose a multi-branch deep learning framework for automatic PD detection from
speech. Each recording is segmented into 5-second chunks and represented using three complementary modalities: Log-Mel spectrograms, MFCCs, and HuBERT
embeddings extracted from raw waveforms. The spectrograms are processed using a pre-trained ResNet-18 encoder, MFCC sequences are modeled through a BiLSTM network, and raw
speech is encoded using a pre-trained HuBERT model. To effectively integrate these heterogeneous representations, we introduce a context-guided cross-modal
attention mechanism that dynamically weights temporal HuBERT
embeddings according to the global acoustic context derived from the spectrogram and MFCC branches. Experiments conducted on the publicly available Spanish PC-GITA corpus under strict speaker-independent 5-fold cross-validation demonstrate the effectiveness of the proposed approach. The proposed architecture achieves an accuracy of 91.51%, an F1-score of 91.24%, and an AUC of 95.97%. Furthermore, ablation studies confirm the contribution of both the proposed context-guided cross-modal
attention mechanism and the integration of complementary
speech representations. These findings highlight the potential of heterogeneous
speech modeling for robust and clinically reliable PD detection.
Read the full abstract ▾
Machine Learning
BSTabDiff: Block-Subunit Diffusion Priors for High-Dimensional Tabular Data Generation
Al Zadid Sultan Bin Habib, Md Younus Ahamed, Prashnna Gyawali et al. (+2 more)
High-Dimensional Low-Sample Size (HDLSS) tabular domains (e.g., omics) are characterized by $n \ll m$, where $n$ = number of samples, and $m$ = number of features. Such domains often exhibit strong local correlation groups, sparse cross-group dependencies, heavy-tailed non-Gaussian marginals, heteroscedastic noise, and structured missingness, making direct density learning in $\mathbb{R}^m$ ill-conditioned since $n \ll m$. We propose BSTabDiff, a block-subunit generative framework that partitions the $m$ observed features into $M$ latent blocks ($M \ll m$) and generates each block via a shared low-dimensional subunit variable, concentrating global dependence learning in the compact block-latent space $\mathbb{R}^M$ while decoding to the full feature space with copula-driven dependence, flexible per-feature marginals, and explicit missingness mechanisms. BSTabDiff supports modern deep priors on block latents, including
diffusion and normalizing flows, enabling stable synthesis and controllable
benchmark generation in the HDLSS regime. Empirically, BSTabDiff produces more realistic and stable high-dimensional synthetic data when compared with unstructured tabular generators on HDLSS data.
Read the full abstract ▾
All authors Al Zadid Sultan Bin Habib, Md Younus Ahamed, Prashnna Gyawali, Gianfranco Doretto, Donald A. Adjeroh
Machine Learning
Orange Lab: Lowering Barriers to Data Mining through Embedded Interactive Workflows
Matej Bevec, Aleš Erjavec, Vesna Tanko et al. (+5 more)
While visual programming of data analysis workflows has become an important vehicle for the democratization of data science, such systems remain largely confined to standalone applications and offer limited support for transitioning their visual analytics solutions into interactive web environments. As a result, data analysis pipelines are difficult to share, embed, and adapt into user-facing analytical tools. We present Orange Lab, a web-based collaborative environment for visual data analytics. At its core, Orange Lab enables users to visually construct machine learning workflows from modular components, where interactions in any component propagate seamlessly through the workflow, turning static pipelines into dynamic, reactive systems that support exploration and data-driven storytelling. Our key contribution is component exposition, a paradigm that allows authors to embed selected workflow components, or parts of their interfaces, into arbitrary web contexts, creating synchronized, interactive interfaces while hiding underlying workflow complexity. This enables the development of tailored analytical views and narrative-driven experiences that integrate data analysis directly into online materials. We demonstrate the approach through deployments in data literacy education, where embedded components guide students in hands-on exploration of machine learning concepts without requiring knowledge of the underlying system, showing that Orange Lab effectively lowers barriers to entry and supports the democratization of data science.
Read the full abstract ▾
All authors Matej Bevec, Aleš Erjavec, Vesna Tanko, Lena Trnovec, Lan Žagar, Ana Farič, Janez Demšar, Blaž Zupan
Machine Learning · Cross-listed from cs.PL
SNN-MLIR: An MLIR Dialect for Compiling Neuromorphic SNNs from NIR to Bare-Metal C
Alejandro García Gener and Alvaro Rollón de Pinedo
Spiking neural networks (SNNs) are increasingly trained in a wide range of frameworks (SnnTorch, Lava, Norse, and others) each with its own model format. The Neuromorphic Intermediate Representation (NIR) addresses this fragmentation by providing a common, framework-independent format for exchanging trained SNN models. NIR solves the exchange problem, but it stops there. It provides a description of a network, not a path to running one. Each backend is still left to implement deployment on its own, with no shared, transformable compiler representation in between. This paper presents snn-mlir, an outof-tree MLIR dialect for SNNs together with a NIR-MLIR-C compilation bridge. The dialect provides a small set of typepolymorphic operations that work identically on floating-point (f32/f64) and quantized data, so a single intermediate representation serves both simulation and hardware-oriented deployment. A Python front end reads any NIR file and emits dialect IR, automatically inserting rescaling operations to keep quantization scales consistent across layers. A reference lowering pass converts the dialect to standard linalg and arith operations, from which the toolchain produces self-contained, dependency free C11 code that compiles and runs on any C-capable CPU or embedded target. We evaluate numerical fidelity against reference outputs, portability across CPU targets, and the cost of quantization. The current scope is feedforward, fully-connected networks with a CPU backend. snn-mlir is released as open source under the Apache-2.0 license with LLVM-exception and it is already available on Github.
Read the full abstract ▾