Machine Learning
SAFE Quantum Machine Learning with Variational Quantum Classifiers
Ying Chen, Paolo Giudici, Vasily Kolesnikov et al. (+1 more)
We propose a variational
quantum classifier operating on high dimensional deep representations via amplitude encoding, stabilized by a learnable classical pre encoding layer.By combining normalized amplitude
embeddings with bounded
quantum observables, the resulting model induces a structured and smooth hypothesis class with controlled sensitivity to input variations. Model reliability is assessed using SAFE-AI metrics derived from the Cramer von Mises divergence, enabling consistent evaluation across accuracy, robustness, and explainability dimensions. Empirical results show that the proposed
quantum model provides competitive predictive performance compared with strong classical baselines while exhibiting a more balanced SAFE reliability profile, with improved robustness to noise and stability under structured feature removal. These findings suggest that variational
quantum circuits offer a principled mechanism for stability oriented SAFE learning in safety critical settings.
Read the full abstract ▾
All authors Ying Chen, Paolo Giudici, Vasily Kolesnikov, Paolo Recchia
Machine Learning
Ada-Diffuser: Latent-Aware Adaptive Diffusion for Decision-Making
Fan Feng, Selena Ge, Minghao Fu et al. (+6 more)
Recent work has framed decision-making as a sequence modeling problem using generative models such as
diffusion models. Although promising, these approaches often overlook latent factors that exhibit evolving dynamics, elements that are fundamental to environment transitions, reward structures, and high-level
agent behavior. Explicitly modeling these hidden processes is essential for both precise dynamics modeling and effective decision-making. In this paper, we propose a unified framework that explicitly incorporates latent dynamic inference into generative decision-making from minimal yet sufficient observations. We theoretically show that under mild conditions, the latent process can be identified from small temporal blocks of observations. Building on this insight, we introduce Ada-Diffuser, a causal
diffusion model that learns the temporal structure of observed interactions and the underlying latent dynamics simultaneously, and furthermore, leverages them for planning and control. With a modular design, Ada-Diffuser supports both planning and policy learning tasks, enabling adaptation to latent variations in dynamics, rewards, and latent actions. Experiments on simulated control and robotic
benchmarks demonstrate its effectiveness in accurate latent inference and adaptive policy learning.
Read the full abstract ▾
All authors Fan Feng, Selena Ge, Minghao Fu, Zijian Li, Yujia Zheng, Zeyu Tang, Yingyao Hu, Biwei Huang, Kun Zhang
Machine Learning
Reshape and Recur: Improving SSMs with Input Reshaping and Depth Recurrence
Mónika Farsang, Ramin Hasani, Daniela Rus et al. (+1 more)
State Space Models (SSMs) are increasingly deployed in the Edge because they offer, at comparable performance, a smaller
memory/training/inference footprint, compared to Large
Language Models (LLMs). These three advantages are a direct consequence of the time recurrence inherent in the SSMs architecture. Here, we further improve this recurrent architecture by positively answering two previously underexplored, orthogonal questions: (1) Can we reduce SSMs
memory-footprint without any performance penalty, by also employing depth recurrence? (2) Can we increase SSMs performance by using a fixed and consistent time-granularity across all tasks? The first question is somewhat unexpected, given that SSMs are already recurrent. However, the orthogonal depth recurrence further decreases SSMs
memory footprint. We show that a looped SSM with $k$ parameters adaptively iterated $M$ times, achieves a performance comparable to a standard SSM with $k \cdot L$ independent parameters, where $M \leq L$. The second question is also unexpected given the time-recurrent nature of the SSMs architecture. However, it makes perfect sense for the time-parallel training of SSMs on the entire input sequence. We show that concatenating time steps for lower-dimensional sequence elements, or flattening and re-chunking the joint feature-time dimension for high-dimensional ones, can improve the baseline by enhancing the way information is presented to the model. Our results for both extensions lead to consistent benefits across four representative SSM architectures: LRU, S5, LinOSS, LrcSSM.
Read the full abstract ▾
All authors Mónika Farsang, Ramin Hasani, Daniela Rus, Radu Grosu
Machine Learning · Cross-listed from cs.CL
RecMem: Recurrence-based Memory Consolidation for Efficient and Effective Long-Running LLM Agents
Zijie Dai, Shiyuan Deng, Sheng Guan et al. (+4 more)
Memory systems often organize user-
agent interactions as retrievable external
memory and are crucial for long-running
agents by overcoming the limited context windows of LLMs. However, existing
memory systems invoke LLMs to process every incoming interaction for
memory extraction, and such an eager
memory consolidation scheme leads to substantial
token consumption. To tackle this problem, we propose RecMem by rethinking when
memory consolidation should be conducted. RecMem stores incoming interactions in a subconscious
memory layer and encode them using lightweight
embedding models for
retrieval. LLMs are only invoked to extract episodic and semantic
memory when sustained recurrence are observed for semantically similar interactions. Such recurrence-based consolidation works because these interactions correspond to a semantic cluster with rich information and thus are worth extraction and summarization. To improve accuracy, RecMem also incorporates a semantic refinement mechanism that recovers the fine-grained facts omitted by
memory extraction. Experiments show that RecMem reduces the
memory construction
token cost of three SOTA
memory systems by up to 87% while exceeding their accuracy.
Read the full abstract ▾
All authors Zijie Dai, Shiyuan Deng, Sheng Guan, Yizhou Tian, Xin Yao, Xiao Yan, James Cheng
Machine Learning · Cross-listed from stat.ML
Explainable AI Isn't Enough! Rethinking Algorithmic Contestability
Timo Freiesleben, Kristof Meding and Gunnar König
Machine learning systems increasingly make life-changing decisions about individuals, such as loan approvals, hiring, and cheating detection, raising a pressing question: how can individuals respond to negative decisions made by these opaque systems? While explainable artificial intelligence (XAI) has largely focused on algorithmic recourse – helping individuals change their features to obtain a desired outcome – the parallel problem of algorithmic contestability – helping individuals review and correct erroneous algorithmic decisions – has received far less
attention, despite its central ethical and legal importance. We trace this neglect to the absence of clear formal definitions and a systematic operationalization of contestability as an algorithmic problem. To address it, we propose an operational definition of contestability as a natural complement to recourse: contestability starts from the presumption that a decision may be incorrect and focuses on identifying evidence to challenge and potentially overturn it, whereas recourse assumes the decision is valid and instead provides pathways for changing it. We show that standard XAI explanations, such as counterfactuals, LIME, or Anchors, even when combined with human intuitions about decision continuity or monotonicity, reveal only errors in the neighborhood of the individual, but provide insufficient grounds for overturning the decision at hand. Going thus beyond traditional XAI, we identify three types of evidence warranting reversal according to the decision maker's own ethical standards: predictive multiplicity, incorrect feature values, and neglected overruling evidence. We argue that these render decisions normatively indefensible and thus successfully contestable. Finally, we analyze how existing EU legislation connects to our framework and argue that individuals already hold some legal rights to these forms of evidence.
Read the full abstract ▾
Machine Learning
Mind Dreamer: Untethering Imagination via Active Causal Intervention on Latent Manifolds
Shaojun Xu, Xiaoling Zhou, Yihan Lin et al. (+4 more)
Model-Based Reinforcement Learning yields sample efficiency via latent imagination, yet remains constrained by Historical Tethering: imagination is typically initialized from observed states. This creates a learning asymmetry, where the world model's
manifold discovery outpaces the policy's sparse-reward optimization. We propose Mind Dreamer (MD), a framework that instantiates Active Causal Intervention to transcend Markovian continuity. MD reformulates discovery as the minimization of a global Relay Expected Free Energy. Instead of initializing from historical data, it draws initial states from an
adversarial generator $s_0 \sim p_{gen}(\cdot)$, creating non-continuous latent jumps to epistemic blind spots that are physically plausible yet cognitively challenging. We derive Relay Value Function and Relay Uncertainty Function to resolve the credit assignment paradox across these spatial ruptures. Treating synthesized anchors as interventional intermediary states, these potentials propagate pragmatic and epistemic value through Bellman-style backups. Notably, we prove that uncertainty propagation across discontinuities necessitates a quadratic discount $\gamma^2$, establishing a formal epistemic horizon. Theoretically, MD approximates a variance-minimizing importance sampler that expands the
manifold's spectral gap, reducing the hitting time to critical bottleneck states. Empirically, MD achieves a 1.67$\times$ average speedup over DreamerV3 on DeepMind Control Suite, reaching 8.8$\times$ in sparse-reward tasks.
Read the full abstract ▾
All authors Shaojun Xu, Xiaoling Zhou, Yihan Lin, Yapeng Meng, Xinglong Ji, Luping Shi, Rong Zhao
Machine Learning · Cross-listed from cs.CL
Judge Circuits Explain Format-Induced Inconsistency in LLM-as-a-Judge
Nils Feldhus, Tanja Baeumel, Elena Golimblevskaia et al. (+10 more)
LLM-as-a-judge has become the dominant paradigm for grading model outputs at scale, yet the same model assigns systematically different scores when its output format changes (e.g., a 1-5 rating vs. a True/False label). Existing diagnoses of these format-induced inconsistencies stop at the input-output level. Using Position-aware Edge Attribution Patching (PEAP), we causally investigate the internal mechanism in five open-weight instruction-tuned models (Gemma-3, Qwen2.5, Llama-3.1) across five judgment tasks. We find that judgments across structured understanding and open-ended preference tasks share a sparse Latent Evaluator sub-graph in the mid-to-late layers; zero-ablating it collapses judgment while damaging knowledge probes substantially less than a random ablation of equal size in architecturally modular models. By structurally decoupling abstract judging from output formatting, we provide a mechanistic account of format-induced inconsistency on the open-weight models we study: a continuous judgment signal computed in the shared trunk is mapped through fragile, format-specific terminal branches. The judgment itself can therefore be read out independently of the requested output format. Our findings imply that
benchmark comparisons of judge reliability across formats partly measure the fragile formatting stage, and can understate the quality of the underlying evaluation.
Read the full abstract ▾
All authors Nils Feldhus, Tanja Baeumel, Elena Golimblevskaia, Qianli Wang, Van Bach Nguyen, Aaron Louis Eidt, Selin Kahvecioglu, Christopher Ebert, Wojciech Samek, Jing Yang, Vera Schmitt, Sebastian Möller, Simon Ostermann
Machine Learning
Variational Autoregressive Networks with probability priors
Piotr Białas, Piotr Korcyl, Tomasz Stebel et al. (+1 more)
Monte Carlo methods are essential across diverse scientific fields, yet their efficiency is frequently hampered by critical slowing down-a sharp increase in autocorrelation times near phase transitions. Although deep learning approaches, such as neural-network-based samplers, have been proposed to alleviate this issue, they face another serious problem: the difficulty of training the models. This difficulty partially stems from the overly general nature of original machine-learning architectures, which often ignore underlying physical
symmetries and force networks to relearn them from scratch. In this paper, we demonstrate that incorporating physical priors into the model significantly enhances performance. Building upon existing strategies that integrate
spin-
spin interactions, we propose a framework that utilizes a prior probability distribution as a starting point for training. Our results for the Ising model, as well as for the Edwards-Anderson
spin glass model, suggest that moving away from `blank slate' models in favor of physics-informed priors reduces the training burden and facilitates the simulation of larger system sizes in discrete
spin models.
Read the full abstract ▾
All authors Piotr Białas, Piotr Korcyl, Tomasz Stebel, Dawid Zapolski
Machine Learning
Accelerated Gradient Descent for Faster Convergence with Minimal Overhead
Manuel Graca, L. Miguel Silveira, Arlindo Oliveira et al. (+1 more)
In this paper, we present CT-AGD (
Curvature-Tuned Accelerated Gradient Descent), an optimization method for non-convex optimization problems in deep learning training tasks. CT-AGD is a general boosting procedure that accelerates first-order methods by explicitly capturing the local
curvature using finite-difference quotients, and the development of heuristics aimed at mitigating noise and bias introduced by
stochastic mini-batch training. CT-AGD has a comparable storage and computational overhead as adaptive gradient methods such as Adam. Our extensive experiments demonstrate that CT-AGD achieves the same level of accuracy as the baseline first-order methods, yet reduces the required training epochs by 33% on average.
Read the full abstract ▾
All authors Manuel Graca, L. Miguel Silveira, Arlindo Oliveira, Frank Liu
Machine Learning · Cross-listed from cs.RO
Adaptive Outer-Loop Control of Quadrotors via Reinforcement Learning
Vishnu Saj, Sushil Vemuri, Dileep Kalathil et al. (+1 more)
Deep Reinforcement Learning (DRL) for quadrotor flight control typically relies on Domain Randomization (DR) for sim-to-real transfer, resulting in overly conservative policies that struggle with dynamic disturbances. To overcome this, we propose a novel adaptive control architecture that actively perceives and reacts to instantaneous perturbations. First, we train an optimal outer-loop policy, then replace its reliance on ground-truth disturbance data with a Residual Dynamics Predictor (RDP). The RDP estimates the external forces and moments acting on the aircraft in flight online using only the history of states and control actions. For seamless hardware transfer, we introduce a data-efficient linear calibration bridge and an online thrust correction mechanism that align the simulated latent space with reality using mere seconds of flight data. Real-world validations on a Crazyflie micro-quadrotor demonstrate that our adaptive controller significantly outperforms baselines, maintaining precise trajectory tracking under severe uncertainties including mass variations, asymmetric payloads, and dynamic slung loads
Read the full abstract ▾
All authors Vishnu Saj, Sushil Vemuri, Dileep Kalathil, Moble Benedict
Machine Learning · Cross-listed from stat.ML
Testing properties of trees in graphical models with covariance queries
Sofiya Burova, Francisco Calvillo, Gábor Lugosi et al. (+1 more)
We consider the problem of testing properties of graphs underlying high-dimensional graphical models. We adopt the model of covariance queries introduced by Lugosi, Truszkowski, Velona, and Zwiernik (2021). We study the case when the underlying graph is a tree. The main results of the paper show that, while reconstructing the entire tree may be costly, certain global structural properties can be tested efficiently. In particular, we design randomized tests for global structural properties that use a sub-quadratic number of queries. We develop testing procedures for several fundamental properties, including the number of leaves, the maximum degree, the typical distance, and the diameter of the tree. For each property, we obtain explicit query complexity bounds that depend on the target threshold and tolerance parameters.
Read the full abstract ▾
All authors Sofiya Burova, Francisco Calvillo, Gábor Lugosi, Piotr Zwiernik
Machine Learning
Constrained latent state modeling: A unifying perspective on representation learning under competing constraints
Gwenolé Quellec
Learning latent representations from temporal,
multimodal, and partially observed data requires specifying what information a latent state should retain, discard, and organize. Existing approaches encode these requirements through heterogeneous objectives, making methods difficult to compare and learned representations difficult to interpret. We propose Constrained Latent State Modeling (CLSM), a conceptual framework that characterizes latent states through six complementary properties: predictive sufficiency, minimality, temporal coherence, observation compatibility, invariance to nuisance factors, and structural constraints. CLSM separates these properties from the surrogate objectives used to induce them and from the diagnostics used to evaluate them, and clarifies how combinations of constraints can improve identifiability by restricting the space of admissible representations. We reinterpret major representation-learning families through this common design space and illustrate the framework with a controlled synthetic
benchmark. The experiments show how objectives produce distinct latent organizations and empirical trade-offs depending on the prediction target, surrogate formulation, parameterization, and optimization. Companion repository containing the reference implementation, reproducible experiments, documentation, and model cards: https://github.com/gwenole-quellec/clsm
Read the full abstract ▾
Machine Learning
Entropy-Based Characterisation of the Polarised Regime in Latent Variable Models
Peter Clapham, Lisa Bonheme and Marek Grzes
Variational Autoencoders (VAEs) often exhibit a polarised regime in which latent variables separate into active, passive, and mixed subsets. Existing criteria for identifying active dimensions depend on a Gaussian prior, limiting their applicability to variational models and specific priors. We propose a simple information-theoretic classification of the polarised regime based on the
entropy of the mean representation. We show theoretically how this
entropy couples to KL minimisation through
entropy–variance bounds, and we relate the resulting criterion to Bonheme's active/passive conditions. We also clarify a key limitation:
entropy of the mean alone cannot reliably distinguish active from mixed dimensions without additional signals from the variance representation. Empirically, we evaluate the
entropy criterion on $\beta$-VAEs, identifiable VAEs, Least-Volume Autoencoders, and L2-regularised autoencoders, and find that it consistently recovers a polarised regime when such a regime is present across the model classes studied. Finally, we show that passive dimensions can yield small but consistent improvements on downstream tasks when latent codes are appropriately normalised, suggesting that collapse is often a matter of scale rather than absolute information removal.
Read the full abstract ▾
Machine Learning · Cross-listed from cs.AI
Imperfect World Models are Exploitable
Logan Mondal Bhamidipaty, Esmeralda S. Whitammer, David Abel et al. (+2 more)
We propose a novel definition of model exploitation in reinforcement learning. Informally, a world model is exploitable if it implies that one policy should be strictly preferred over another while the environment's true transition model implies the reverse. We analogize our definition with a prior characterization of reward hacking but show that the associated proof of inevitability does not transfer to exploitation. To overcome this obstruction, we develop a general theory of reward hacking and model exploitation that proves that exploitation is essentially unavoidable on large policy sets and yields the corresponding claim for hacking as a special case. Unfortunately, we also find that the conditions that guarantee unhackability in finite policy sets have no counterpart that precludes exploitation. Consequently, we introduce a relaxed notion of exploitation and derive a safe horizon within which it can be avoided. Taken together, our results establish a formal bridge between reward hacking and model exploitation and elucidate the limits of safe planning in world models.
Read the full abstract ▾
All authors Logan Mondal Bhamidipaty, Esmeralda S. Whitammer, David Abel, Mykel J. Kochenderfer, Subramanian Ramamoorthy
Machine Learning
When and Why Adversarial Training Improves PINNs: A Neural Tangent Kernel Perspective
Yuan-dong Cao, Chi Chiu SO, Jun-Min Wang et al. (+1 more)
Physics-informed neural networks (PINNs) are powerful surrogates for differential equations but are notoriously difficult to train due to spectral bias, stiffness, and poor accuracy on high-frequency or multiscale solutions.
Adversarial training based on generative
adversarial networks (GANs) has recently gained surprisingly strong empirical results in improving training, but the underlying mechanisms remain elusive. To this end, we propose a new analysis framework for adversarially trained PINNs, based on the key observation of how the discriminator in GANs can influence the training dynamics of PINNs. The framework first provides a much needed theoretical grounding to why and when
adversarial training is effective in PINNs, then presents a unified analysis of GANs variants in such training, and finally leads to a new, practical, efficient training
algorithm for PINNs. Empirical results demonstrate that our method can significantly reduce the pathology of PINNs training, thereby providing better models with superior performances, often several magnitudes more accurate than alternative methods.
Read the full abstract ▾
All authors Yuan-dong Cao, Chi Chiu SO, Jun-Min Wang, He Wang
Machine Learning · Cross-listed from cs.RO
FocalPolicy: Frequency-Optimized Chunking and Locally Anchored Flow Matching for Coherent Visuomotor Policy
Qian He, Zhenshuo Yang, Wenqi Liang et al. (+3 more)
Visuomotor policies aim to learn complex manipulation tasks from expert demonstrations. However, generating smooth and coherent trajectories remains challenging, as it requires balancing proximal precision with distal foresight. Existing approaches typically focus on optimizing intra-chunk action distributions, often neglecting the inter-chunk coherence. Consequently, inter-chunk discontinuities significantly impede the learning of coherent long-horizon actions. To overcome this limitation and achieve a synergetic balance between precision and foresight, we propose FocalPolicy, a foresight-aware visuomotor policy that combines Frequency-Optimized Chunking with Locally Anchored flow matching. We introduce a foresight composite objective that supervises time-domain
alignment within the proximal actions while regularizing frequency-domain structure over multiple future action chunks to improve cross-chunk coherence. To efficiently learn complex action distributions, we design locally anchored sampling to enhance target signal propagation efficiency during consistency flow matching training. Extensive experiments demonstrate that FocalPolicy outperforms existing approaches and confirm the generalizability of our modules to other baselines. Project website: https://focalpolicy.github.io/
Read the full abstract ▾
All authors Qian He, Zhenshuo Yang, Wenqi Liang, Chunhui Hao, Nicu Sebe, Jiandong Tian
Machine Learning · Cross-listed from physics.bio-ph
Clock-state olfactory search in turbulent flows using Q-learning: The geometry of plume recovery
Marco Rando, Robin A. Heinonen, Yujia Qi et al. (+1 more)
Finding an odor source in a turbulent flow requires effectively leveraging the history of olfactory observations into a robust
navigation strategy. In this work, we use tabular Q-learning to train an olfactory search
agent with a minimal
memory of past observations: only a running clock since the last whiff. This
agent learns an interpretable strategy to recover the plume which combines well-known behaviors observed in insects: surging, casting, and a return downwind. While achieving good performance on data from direct numerical simulations of
turbulence, the
agent is limited by an inability to adapt its strategy to the local intermittency level; we show that providing more flexibility improves robustness.
Read the full abstract ▾
All authors Marco Rando, Robin A. Heinonen, Yujia Qi, Agnese Seminara
Machine Learning
A Retrieval-Enhanced Transformer for Multi-Step Port-of-Call Sequence Prediction in Global Liner Shipping
Yanzhao Su, Fang He and Yineng Wang
Accurate multi-step port-of-call sequence prediction is vital for tactical resource orchestration and logistical efficiency. However, existing methods struggle with unreliable voyage schedules and the inability of AIS data to provide visibility beyond the immediate next port. To address this, this study proposes a Connectivity-Constrained and
Retrieval-Enhanced (CCRE) deep learning framework. Inspired by
Retrieval-Augmented Generation, CCRE introduces a
retrieval-enhanced historical encoder that queries a global maritime database for contextually similar navigational precedents. Transforming these scenarios into candidate-level semantic representations compensates for data sparsity in long-tail routes and resolves routing ambiguities. Integrating this with a
Transformer-based trajectory encoder, the architecture executes adaptive "middle fusion" via cross-
attention. This dynamically shifts predictive reliance from real-time kinematics for short-term accuracy to historical context for long-term strategic stability. To ensure sequence-level coherence,
forecasting is formulated as a joint sequence generation problem using an autoregressive
Transformer decoder enriched with Scheduled Sampling and Gumbel-Softmax relaxation. This mitigates error accumulation, while
topology masks strictly enforce maritime network reachability to eliminate operationally infeasible routes. Evaluated on a global
dataset, CCRE achieves a 72.3% first-destination accuracy and a 61.4% average three-step accuracy, outperforming baselines like CatBoost and LSTM by average margins of 12.6% and 11.3%, respectively. Case studies further corroborate the model's scalability and ability to capture complex routing patterns across diverse international trade lanes.
Read the full abstract ▾
Machine Learning · Cross-listed from stat.ML
Unsupervised Domain Shift Detection with Interpretable Subspace Attribution
Sebastian Springer and Alessandro Laio
We developed a tool for detecting domain shifts, namely subtle differences in the probability distributions of
datasets. We identify these shifts using an
algorithm designed to detect localised density anomalies in high-dimensional feature spaces. If an anomaly is present, we then identify the feature subspace in which the anomaly is most pronounced. This allows us to trace the domain shift to a small set of features, making the shift interpretable. Moreover, we provide a protocol for compensating domain shifts by extracting, from two unlabelled
datasets, subsets of samples with no detectable residual distributional difference. We validate the framework on controlled 20-dimensional
benchmarks with known ground truth, recovering both broad and localized shifts together with their supporting feature subspaces. We then apply it to healthy electrocardiogram (ECG) recordings represented by 782 features. In age- and sex-matched cohort comparisons differing in measurement-device composition, the method detects device-induced shifts, extracts representative subsets enriched in the imbalanced device components, and identifies ECG features associated with the acquisition contrast. These results suggest that density-shift detection and subspace attribution provide a practical framework for uncovering hidden cohort biases before downstream modelling.
Read the full abstract ▾
Machine Learning
LoCO: Low-rank Compositional Rotation Fine-tuning
An Nguyen, Jaesik Choi and Anh Tong
Parameter-efficient fine-tuning (PEFT) has emerged as an critical technique for adapting large-scale foundation models across natural
language processing and computer
vision. While existing methods such as low-rank adaptations achieve parameter efficiency via low-rank weight updates, they are limited in their ability to preserve the
geometric structure of pretrained representations. We introduce Low-rank Compositional Orthogonal fine-tuning (LoCO), a novel PEFT method that constructs orthogonal transformations through low-rank skew-symmetric
matrices and compositional rotation chains. We propose an approximation scheme that enables fully parallel computation of compositional rotations, making the approach practical for high-dimensional feature spaces. Our method maintains low computational complexity while maintaining orthogonality with controlled approximation error. We validate LoCO across diverse domains, including
diffusion transformer fine-tuning,
vision transformer adaptation, and
language model adaptation. Our method demonstrates superior or competitive performance compared to both existing orthogonal and non-orthogonal methods.
Read the full abstract ▾
Machine Learning
Context-aware Entity-Relation Extraction for Threat Intelligence Knowledge Graphs
Inoussa Mouiche and sherif Saad
Cybersecurity Knowledge Graphs (CKGs) unify diverse Cyber Threat Intelligence (CTI) sources into structured, queryable formats, offering scalable solutions for automating proactive and real-time security responses. Their increasing adoption has significantly enhanced the workflow and decision-making efficiency of security professionals. However, constructing CKGs requires extracting entity-relation triples from unstructured CTI reports, a task hindered by complex report structure, domain-specific
language, and semantic ambiguity. As a result, existing pipeline-based approaches often suffer from error propagation, reducing extraction accuracy and limiting generalizability. This paper introduces the Context-aware Threat Intelligence Knowledge Graph (CTiKG) framework, a pipeline architecture designed to accurately extract and classify threat entities and their relationships from CTI reports. CTiKG incorporates hybrid NLP models that leverage SecureBERT+ contextual
embeddings and expert knowledge from a domain ontology to reduce misclassifications and mitigate cascading errors. Experiments on the DNRTI-AUG-STIX2
dataset, which comprises 21 entity types aligned with STIX 2.1, demonstrate significant improvements over state-of-the-art baselines, yielding 3-4% gains in NER and up to 8% in RE performance, based on precision, recall, and F1-score. Additional validation on DNRTI and STUCCO
benchmarks confirms the framework's robustness and practical applicability. All
datasets, including the curated DNRTI-AUG-STIX2, are released on GitHub to foster reproducibility and further research.
Read the full abstract ▾
Machine Learning
From Layers to Networks: Comparing Neural Representations via Diffusion Geometry
Atharva Khandait and Jan E. Gerken
Diffusion geometry is a
manifold learning framework that uses random walks defined by Markov transition
matrices to characterize the geometry of a
dataset at multiple scales. We use
diffusion geometry for neural representations, incorporating tools from multi-view learning into this field for the first time. Our key technical observation is that a broad class of similarity measures based on representational similarity
matrices (RSMs) admits a closed-form equivalent formulation in terms of row-
stochastic Markov
matrices, opening the door to manipulations from
diffusion geometry. As a first application, we develop multi-scale variants of Centered Kernel
Alignment and Distance Correlation, which utilise the $t^{th}$ power of the underlying transition
matrix to probe the data geometry at adjustable
diffusion scales. Going further, we introduce variants of these measures which fuse the Markov
matrices of several layers via alternating
diffusion into a single operator that captures the network's joint sample geometry, allowing similarity to be computed across multiple layers and shifting the comparison from layer-to-layer to network-to-network. We perform extensive numerical experiments, evaluating our measures on the Representational Similarity (ReSi)
benchmark comprising 14 architectures trained on 7
datasets across three different domains. Our methods achieve SoTA results in accuracy and output correlation for both
language and
vision tasks across different models. We furthermore show SoTA performance on an additional
benchmark evaluating on out-of-distribution data.
Read the full abstract ▾
Machine Learning · Cross-listed from cs.CR
A Multi-Layer Cloud-IDS Pipeline with LLM and Adaptive Q-Learning Calibration
Syed Waqas Ali, Ibrar Ali Shah, Farzana Zahid et al. (+2 more)
Security in cloud computing has become a major concern due to several factors such as layered cloud architectures, dynamic environments, and exposure to unseen or zero-day attacks. Moreover, intrusion detection systems (IDS) typically operate at specific layers and rely heavily on machine learning models, which often perform well in experimental settings but fail to sustain performance in real cloud deployments. In this work, we implement a confidence-aware multilevel intrusion detection system using reinforcement learning tailored for cloud environments. The system secures three distinct layers: network, host, and hypervisor. Machine learning models at each layer detect known attack patterns, while prediction confidence distinguishes reliable decisions from uncertain outcomes. Within the multi-gate flow, low-confidence events pass through a learned-threshold confidence gate (Gate-1), followed by a Chroma
memory-matching gate (Gate-2), with unresolved events escalated to a large
language model (LLM) for semantic analysis and explanation. Final attack promotion at Gate-3 uses calibrated LLM confidence or weighted-fusion fallback, while uncertain events are retained in a review bucket to avoid forced classification. Generated explanations and confirmed knowledge are stored in ChromaDB to support future analysis and retraining. The approach is first evaluated using static thresholds, establishing a baseline for comparison. Results show that the proposed system learns adaptive thresholds and reduces LLM escalation by 58.78%, lowering cost while maintaining strong performance (88.68% accuracy, 85.29% precision, 84.72% recall, 85.00% F1). The network and hypervisor layers achieve 98.02% and 97.08% accuracy, demonstrating a balanced and efficient detection system.
Read the full abstract ▾
All authors Syed Waqas Ali, Ibrar Ali Shah, Farzana Zahid, Daniyal Munir, Hans D. Schotten
Machine Learning
CHoE: Cross-Domain Heterogeneous Graph Prompt Learning via Structure-Conditioned Experts
Peiyuan Li, Yongqi Huang, Jitao Zhao et al. (+3 more)
Heterogeneous Graph
Prompt Learning (HGPL)has emerged as a promising paradigm for bridging the gap between the objectives of pre-training foundation models and their downstream applications in heterogeneous graph settings. However, existing HGPL methods are primarily designed for in-domain scenarios, whereas real-world deployments often span multiple domains, and the data used for pre-training and downstream tasks may originate from different distributions. Consequently, the applicability of current HGPL approaches is limited to in-domain settings, and their performance typically degrades when application domains shift. To address this serious limitation, we develop CHoE, a cross-domain HGPL method built upon an expert network. During pre-training, we introduce and train structure-conditioned experts, and during
prompt tuning, we adopt a structure-aware expert routing and load balancing mechanism to select structurally compatible experts for each meta-path view. In addition, we design a
prompt-based semantic fusion module to integrate representations across multiple views for downstream prediction. Extensive experiments show that CHoE consistently improves performance in few-shot cross-domain applications, outperforming all baseline approaches.
Read the full abstract ▾
All authors Peiyuan Li, Yongqi Huang, Jitao Zhao, Dongxiao He, Di Jin, Weixiong Zhang
Machine Learning
Practical Validity Conditions for Byzantine-Tolerant Federated Learning
Mélanie Cambus, Darya Melnyk, Tijana Milentijević et al. (+1 more)
Robust aggregation is the core operation in Byzantine-tolerant
federated learning. To ensure the quality of aggregation independently of data distribution or attacks, validity conditions are needed. They provide
geometric guarantees of where the output of the aggregation must lie. The widespread convex validity requires the output to lie in the convex hull of the honest vectors. Although this guarantee is strong in theory, it is poorly suited to modern
federated learning systems, as it has dimension-dependent resilience and excludes many practical aggregation rules. We introduce the minimum enclosing ball (MEB) validity condition for robust aggregation, as well as its multiplicative relaxation, $c$-MEB validity, where $c$ is a constant. We show that exact MEB validity still suffers from limited resilience, while relaxed $c$-MEB validity is achievable if a majority of clients is honest, i.e. $n>2t$. We give an optimal MinMax-MEB rule for the relaxed condition with the bound $c<\sqrt{2}$ and prove explicit relaxed-MEB guarantees for standard aggregators including minimum-diameter averaging, medoid and
geometric median. Finally, we relate MEB validity to convex, relaxed-convex and box validity studied in prior literature, thus providing a systematic map of
geometric validity conditions for Byzantine-robust aggregation. Our results show that relaxed MEB validity connects validity conditions in distributed computing and Byzantine-tolerant aggregation rules, and offers a practical alternative to convex validity.
Read the full abstract ▾
All authors Mélanie Cambus, Darya Melnyk, Tijana Milentijević, Stefan Schmid
Machine Learning
Shapley Neuron Values for Continual Learning: Which Neurons Matter Most?
Mohammad Ali Vahedifar, Abhisek Ray and Qi Zhang
Continual learning enables neural networks to learn tasks sequentially without forgetting previously acquired knowledge. However, neural networks suffer from catastrophic forgetting, where learning new tasks degrades performance on earlier ones. We address this problem with Shapley
Neuron Valuation (SNV), a principled framework that quantifies
Neuron importance in continual learning, grounded in cooperative game theory. SNV selectively freezes important
Neurons while keeping others plastic, enabling buffer-free continual learning without expanding architecture. Experiments on ImageNet-1k show that SNV consistently outperforms existing buffer-free methods. In particular, SNV improves accuracy by +2.88% in the class incremental learning and +6.46% in the task incremental learning scenarios compared to the second baseline.
Read the full abstract ▾
Machine Learning
Ti-iLSTM: A TinyDL Approach for Logic-Level Anomaly Detection in Industrial Water Treatment Systems
Mandar Joshi, Farzana Zahid, Judy Bowen et al. (+3 more)
Industrial Water Treatment Systems (IWTS) are safety critical cyber-physical infrastructures and due to increased connectivity, these systems are exposed to cyber threats that can manipulate process behaviour without creating obvious devices outliers. In particular, logic-layer deception anomalies can preserve numerically plausible measurements while breaking expected cause-and-effect relationships in the control process. These attacks are difficult to detect using threshold-based monitoring or require heavy server-oriented anomaly detection models. This paper explores the potential of Tiny Deep Learning (TinyDL) to provide lightweight on-device logic-level anomaly detection for resource constrained Programmable Logic Controllers (PLCs). We propose a novel framework, TinyDL-based incremental LSTM (Ti-iLSTM) which optimises the
memory and space foot print of Long Short-Term
Memory (LSTM), to detect logic-layer inconsistencies in Programmable Logic Controller (PLC) based Industrial Water Treatment Systems (IWTS). Experiments on the publicly available SWaT
dataset show that the optimised model achieves high detection performance (F1-score=0.983 and ROC-AUC=0.998). A deployment-style validation on the WADI
dataset confirms that the proposed light-weight framework remains applicable beyond a single
dataset. The research demonstrates that combining logic-aware supervision with Tiny Deep Learning (TinyDL) sequence learning creates an efficient and accurate anomaly detection suitable for resource constrained Programmable Logic Controllers (PLCs) in industrial environments.
Read the full abstract ▾
All authors Mandar Joshi, Farzana Zahid, Judy Bowen, Matthew M.Y. Kuo, Valeriy Vyatkin, Emil Karlsson
Machine Learning
From Observed Viability to Internal Predictive Approximation: A Single-Subject Latent-Space Analysis of Gait Dynamics Under Occlusal Constraint
Jacques Raynal, Pierre Slangen, Elsa Raynal et al. (+1 more)
Understanding adaptive biomechanical systems requires distinguishing observable performance, static multivariate representation, longitudinal displacement, and internal approximation of observed change. This study introduces Level 5, which examines whether the M1-M2 transformation observed in a single-subject gait
dataset can be approximated within a selected PCA representation. Gait was recorded with instrumented insoles in a participant with Parkinson's disease under six occlusal observational probes during two sessions eleven weeks apart. A simplified feed-forward neural network was trained to approximate M2 PC1-PC2 coordinates from M1 coordinates, occlusal-probe descriptors, and the longitudinal-transition indicator. In the core analysis aligned with Level 4, the model preserved the Euclidean centroid-displacement hierarchy dOC3 < dONL < dOC2.5. In the extended six-probe analysis, it preserved the broad structure of the exploratory ordering. Held-out M2 and leave-condition-out analyses provided internal tests beyond the full-
dataset fit, while a within-session analysis described probe positions relative to ONL. The term predictive is used only in a restricted methodological sense. The model does not provide prospective clinical prediction, patient-level
forecasting, or generalization to unseen individuals. Occlusal conditions are treated as observational probes applied during measurement, not as continuous causal drivers of longitudinal
evolution. The findings are exploratory, retrospective, representation dependent, and non causal. They do not establish causal occlusal effects, validated viability thresholds, therapeutic superiority, distinct physiological states, or generalizable predictive validity.
Read the full abstract ▾
All authors Jacques Raynal, Pierre Slangen, Elsa Raynal, Jacques Margerit
Machine Learning · Cross-listed from cs.DS
Complexity of Non-Log-Concave Sampling in Fisher Information
Sinho Chewi and Andre Wibisono
We study the query complexity of obtaining a relative Fisher information guarantee for sampling from a log-smooth non-log-concave distribution; this is a sampling analog of finding an approximate stationary point in optimization. Our
algorithm is based on the proximal sampler, which is an implicit discretization of the Langevin
diffusion, and requires an implementation of the backward step known as the restricted Gaussian oracle (RGO). We show that by leveraging the recent results for log-concave sampling with high-accuracy guarantees in Rényi divergence, we can obtain an approximate RGO implementation that – when used with the proximal sampler – yields a complexity guarantee in relative Fisher information that inherits the same dimension dependence as log-concave sampling, and improves upon prior work for non-log-concave sampling. We also show a converse reduction that any improvement in the dimension dependence in relative Fisher information for non-log-concave sampling will yield an improved dimension dependence for high-accuracy log-concave sampling.
Read the full abstract ▾
Machine Learning · Cross-listed from cs.PF
Heuristic-Based Merging of HPC Traces to Extend Hardware Counter Coverage
Júlia Orteu Aubach, Fabio Banchelli, Marc Clascà Ramírez et al. (+1 more)
This work extends a framework for predicting the performance of High-Performance Computing (HPC) workloads using Machine Learning (ML). A common limitation in performance modeling is the restricted number of hardware counters that can be collected simultaneously. To address this, we propose a heuristic-based methodology to merge execution traces from multiple runs, each instrumented with a different set of hardware counters. Our approach matches computation bursts across executions by analyzing MPI structure, timing, and communication patterns. This process enables the construction of a unified
dataset that includes a wider set of hardware features without relying on multiplexing. The output is a new synthetic trace with all merged counters, which can be used both for HPC performance prediction and for conventional performance analysis. The methodology has been validated on MareNostrum5 machine with a range of kernels and real applications. Results show that the merged counters maintain acceptable accuracy depending on the application, and can be directly used to train ML models on a richer feature space without prior counter selection.
Read the full abstract ▾
All authors Júlia Orteu Aubach, Fabio Banchelli, Marc Clascà Ramírez, Marta Garcia-Gasulla