Machine Learning
Graph Computation Meets Circuit Algebra: A Task-Aligned Analysis of Graph Neural Networks for Electronic Design Automation
Hyunmog Kim
EDA problems are graph-structured, but not all graph-structured problems call for the same GNN computation. We argue that successful GNN-for-EDA methods are those whose propagation, aggregation, and supervision align with the native
algebra of the target task. Concretely: static timing analysis is a max-plus/min-plus recurrence on a topologically ordered DAG, structurally aligned with asynchronous DAG-GNNs; placement is governed by hypergraph wirelength and density penalties and is exploited by differentiable placers rather than by message-passing GNNs alone; routing congestion is a sparse demand-supply field over a layout grid; switching-activity propagation is a probabilistic recurrence on a directed netlist; IR drop is a linear system on the power-delivery network; and analog
symmetry extraction is a discrete constraint-prediction problem on schematic graphs. Through these task-by-task
alignments we (i) review the GNN architectural toolkit relevant to
circuits, (ii) formalize how
circuit graphs differ from generic graphs (directed, heterogeneous, multi-scale, with sequential and clock structure), (iii) characterize where current methods succeed and where the
algebra-architecture mismatch limits them, and (iv) identify failure modes–stage leakage, proxy-to-signoff gap, calibration, and design-distribution shift–that we believe are likely to dominate the next phase of work. We position the paper as a GNN-for-EDA, task-aligned analysis rather than a comprehensive AI-for-chip-design survey. Continuous SE(3)-equivariant
geometric GNNs are usually mismatched to Manhattan digital layout, and LLM-for-RTL, HLS, and RL/
diffusion-based
topology generation are outside our scope.
Read the full abstract ▾
Machine Learning
Toward Optimal Regret in Robust Pricing: Decoupling Corruption and Time
Kalana Kalupahana, Francesco Emanuele Stradi, Matteo Castiglioni et al. (+1 more)
We design the first regret guarantees for robust dynamic pricing that decouple the dependence on the corruption $C$ and the time horizon $T$. In dynamic pricing, a seller with unlimited supply of a good interacts with a stream of buyers over \( T \) rounds, with the goal of maximizing revenue. At each round $t$, the seller posts a price $p_t$, and the buyer purchases the good only if their unknown valuation $v^\star$ exceeds this price. The seller observes only the binary feedback $\mathbb{I} \left\{ p_t \leq v^\star \right\}$, indicating whether a sale occurred. In the robust pricing setting, a malicious adversary is allowed to corrupt this feedback in at most $C$ rounds. Even if the learner knows the corruption $C$, the best known regret bound is $\mathcal{O}(C\log\log T)$ by Gupta et al. [2025]. This leaves as an open problem to “decouple” the dependence on $C$ and $T$. In this work, we resolve this open problem. In particular, we develop a robust variant of binary search that achieves regret $\mathcal{O}(C+\log T)$ when the corruption $C$ is known and $\mathcal{O}(C+\log^2 T)$ when the corruption is unknown.
Read the full abstract ▾
All authors Kalana Kalupahana, Francesco Emanuele Stradi, Matteo Castiglioni, Alberto Marchesi
Machine Learning
What If We Let Forecasting Forget? A Sparse Bottleneck for Cross-Variable Dependencies
Fan Zhang, Shiming Fan and Hua Wang
Multivariate time series
forecasting is critical in many real-world systems, and thus modeling cross-channel dependencies is essential. Although existing methods improve overall accuracy by enhancing representations and cross-channel interactions, it remains challenging to reliably capture inter-variable dependencies under specific conditions. We observe that dependencies in real data are often state-dependent and noisy; in such cases, dense interactions can amplify spurious correlations and lead to representation over-smoothing, which may yield unreliable predictions in certain scenarios. Motivated by this, we propose MS-FLOW, a sparse-bottleneck framework that explicitly models inter-variable interaction as capacity-limited information flow. Specifically, MS-FLOW replaces fully connected communication with selective sparse routing, retaining only a few critical dependency paths and injecting cross-variable signals under a strict communication budget, thereby suppressing redundant connections and spurious-correlation propagation. Extensive experiments demonstrate that MS-FLOW learns more reliable multivariate correlations, achieving state-of-the-art
forecasting accuracy on 12 real-world
benchmarks while producing fewer yet more reliable dependencies, shifting multivariate
forecasting from "more interaction" to "more effective interaction".
Read the full abstract ▾
Machine Learning
UMEDA: Unified Multi-modal Efficient Data Fusion for Privacy-Preserving Graph Federated Learning via Spectral-Gated Attention and Diffusion-Based Operator Alignment
Shih-Yu Lai, Hirozumi Yamaguchi, Shang-Tse Chen et al. (+2 more)
Device-free localization trains models from heterogeneous wireless and visual sensors (e.g., Wi-Fi,
LiDAR) distributed across edge devices.
Federated learning offers a
privacy-respecting framework, but is brittle when clients differ in sensor modality and resolution, when their data distributions drift, and when
privacy noise destroys the structural signal needed for localization. We propose UMEDA, a graph
federated learning framework in which clients form nodes of a global graph that share a continuous integral operator, and aggregation is reformulated as spectral signal processing on this operator. Each client encodes its local sensors with a linear-
attention layer whose kernel spectrum is low-rank filtered, suppressing modality-specific residuals so clients with different sensors align in a common low-rank subspace. The server then aggregates client updates via a
diffusion model over the kernel's spectral coefficients, treating updates as discretizations of a shared operator rather than
topology-bound weights – this absorbs varying graph sizes and missing modalities without node-wise correspondence. To balance
privacy and utility, we add an anisotropic differential-
privacy mechanism that projects noise preferentially into the null space of the signal subspace, preserving dominant eigendirections while ensuring formal $(\epsilon, \delta)$-DP under gradient clipping. On MM-Fi and the RELI11D out-of-distribution
benchmark, UMEDA outperforms state-of-the-art
federated baselines in accuracy, convergence, and communication efficiency, particularly under high modality heterogeneity and tight
privacy budgets.
Read the full abstract ▾
All authors Shih-Yu Lai, Hirozumi Yamaguchi, Shang-Tse Chen, Yu-Lun Liu, Bing-Yu Chen
Machine Learning
Multi-Armed Bandits With Best-Action Queries
Francesco Bacchiocchi, Matteo Castiglioni, Alberto Marchesi et al. (+1 more)
We study multi-armed bandits (MABs) augmented with best-action queries, in which the learner may additionally query an oracle that reveals the best arm in the current round. This setting was recently characterized by Russo et al. [2024] in the full-feedback model, where the learner observes the rewards of all arms after each round. They show that, in both
stochastic and
adversarial environments, $k$ best-action queries reduce the optimal $\widetilde{\mathcal{O}}(\sqrt{T})$ regret to $\widetilde{\mathcal{O}}(\min\{T/k,\sqrt{T}\})$. Whether this improvement extends to the more realistic bandit-feedback model – where the learner observes only the reward of the played arm – was left as an open problem. We fully resolve this question. When rewards are
stochastic but correlated among arms, we show that the full-feedback result does not carry over: any
algorithm must incur regret at least $\Omega(\sqrt{T-k})$. This lower bound directly extends to
adversarial environments. On the positive side, we show that $\widetilde{\mathcal{O}}(\min\{T/k,\sqrt{T-k}\})$ regret is still achievable when rewards are
stochastic and i.i.d., and establish a matching lower bound, up to logarithmic factors. Together, these results provide a complete characterization of the benefits of best-action queries in the bandit-feedback model.
Read the full abstract ▾
All authors Francesco Bacchiocchi, Matteo Castiglioni, Alberto Marchesi, Francesco Emanuele Stradi
Machine Learning
Diagnosing Spectral Ceilings in Equivariant Neural Force Fields
Hyunmog Kim
We introduce a spectral-injection diagnostic for measuring which angular frequencies a trained equivariant force-field backbone preserves: inject a controlled angular-frequency perturbation into a molecular force field, attach a lightweight Spectral Prediction Network (SPN) to the frozen backbone, and read off which frequencies are recoverable. On aspirin, a quadratic SPN attached to an L = 2 NequIP backbone recovers the boundary signal at l = 4 but collapses at l = 5: a 11.7x cliff at the predicted drL boundary, with p dropping from 0.913 to 0.078. The same boundary-vs-above contrast persists across n = 4 independently trained backbones (raw-gain delta contrast, hierarchical cluster bootstrap) and is corroborated by a denominator-free injected-residual metric (R2_inj(4) = 0.374 versus R2_inj(5) = 0.006). A finite-degree span
theorem calibrates the diagnostic: for a single marked direction, degree-d
polynomials of degree-L spherical-harmonic features span exactly H less than or equal to dL with multiplicity-one saturation at the boundary (scoped to single-direction degree-bounded probes, not a function-class upper bound on multi-atom MPNNs). A synthetic C5 calibration plus capacity, activation, and cross-architecture controls rule out parameter count alone as the explanation.
Read the full abstract ▾
Machine Learning
Exactness Matters for Physical Rule Enforcement
Bum Jun Kim
Autoregressive scientific forecasters often enforce physical or structural constraints by repairing each predicted state before feeding it back into the model. However, it remains unclear when stronger physical rule enforcement becomes reliable and when it becomes a source of distribution shift. We study this question through operator exactness, meaning whether the repair map is the identity on the target
manifold and is aligned with the target geometry. We compare raw
forecasting, post hoc repair, and in-loop repair across periodic incompressible Navier–Stokes, non-periodic CFDBench flows, and a hierarchical-
forecasting support task. In the exact periodic regime,
Fourier projection substantially improves rollout accuracy. On the NS-128
benchmark, a strong Raw-FNO has a final-step rollout MSE at horizon 100 of $(9.390 \pm 6.290)\times 10^{-5}$, and post hoc and in-loop projection reduce it to $(1.130 \pm 0.165)\times 10^{-6}$ and $(5.370 \pm 0.113)\times 10^{-7}$. However, once an exact projection is unavailable and only approximate boundary-preserving cleanup is available, the ordering changes. Across cavity, tube, dam, and cylinder flow, stronger Poisson-based cleanup can reduce divergence while worsening rollout error; target-distortion MSE predicts this harm far better than a linear-system residual. Controlled mismatch, screened cleanup, adaptive gating, and external-backbone checks show that the best approximate-regime operating point can be raw or near-identity. Hierarchical
forecasting gives the same broader pattern. Exact forecast reconciliation is a stable baseline, whereas blended top-down repair, a validation-tuned interpolation toward historical-proportion top-down reconciliation, is
dataset-dependent. Thus, constraint enforcement should be benchmarked by operator–data
alignment before enforcement strength.
Read the full abstract ▾
Machine Learning
HTPO: Towards Exploration-Exploitation Balanced Policy Optimization via Hierarchical Token-level Objective Control
Xincheng Yao, Ruoqi Li, Cheng Chen et al. (+4 more)
Reinforcement Learning with Verifiable Rewards (RLVR) has emerged as a pivotal technique for enhancing the
reasoning capabilities of Large
Language Models (LLMs). However, the de facto practice of mainstream RL
algorithms is to treat all
tokens of one response equally and assign the same optimization objective to each
token, failing to provide granular guidance for the
reasoning process. While in Chain-of-Thought (CoT)
reasoning, different
tokens usually play distinct roles. Therefore, the current RL
algorithms lack an effective mechanism to dynamically balance the exploration-exploitation trade-off during learning. To this end, we propose Hierarchical
Token-level Objective Control Policy Optimization (HTPO), a novel RL
algorithm that takes the divide-and-conquer idea to hierarchically partition the response
tokens into specific functional groups from three aspects (i.e.,
prompt difficulty, answer correctness, and
token entropy). Within each group, according to the contributions to exploration or exploitation, we design specialized optimization objectives to facilitate the effective execution of each
token's expected functionality. In this way, HTPO can achieve a more balanced exploration-exploitation trade-off. Extensive experiments on challenging
reasoning benchmarks validate the superiority of our HTPO
algorithm, which significantly outperforms the strong DAPO baseline (e.g., +8.6% and +6.7% on AIME'24 and AIME'25, respectively). When scaling test-time compute, the HTPO-trained model maintains a consistent performance advantage over the DAPO baseline, and the gap widens as the sampling budget increases, validating that our adaptive
token-level control method fosters effective exploration without sacrificing exploitation performance. Code will be at https://github.com/xcyao00/HTPO.
Read the full abstract ▾
All authors Xincheng Yao, Ruoqi Li, Cheng Chen, Daoxin Zhang, Yi Wu, Yao Hu, Chongyang Zhang
Machine Learning
Beyond the False Trade-off: Adaptive EWC for Stealthy and Generalizable T2I Backdoors
Lu Bowen, Xinyu Tang, Yin Yin Low et al. (+1 more)
Preserving model fidelity is essential for stealthy text-to-image (T2I)
backdoor attacks. Existing methods such as Learning without Forgetting (LwF) rely on output-based
distillation, which provides limited regularization. We introduce Elastic Weight Consolidation (EWC) as a parameter-based alternative for preserving fidelity in
backdoor learning. While stronger in principle, we show that standard static EWC with a fixed regularization weight lambda and mean-squared utility loss creates an artificial trade-off between attack success rate (ASR) and fidelity, particularly degrading performance on weak triggers. To address this, we propose Cosine-Aware Adaptive EWC, which dynamically adjusts EWC regularization using a cosine-based semantic utility and adaptive scheduling. This approach transforms EWC from a fixed penalty into a context-sensitive constraint, maintaining high ASR while preserving model fidelity. Experiments demonstrate improved ASR-fidelity balance and enhanced robustness on out-of-domain (OOD)
datasets compared to existing baselines.
Read the full abstract ▾
All authors Lu Bowen, Xinyu Tang, Yin Yin Low, Shu-Min Leong
Machine Learning
LaWM: Least Action World Models for Long-Horizon Physical Consistency from Visual Observations
Qixin Xiao and Maani Ghaffari
Learning predictive world models from visual observations is a core problem in
embodied AI, with applications to model-based reinforcement learning and robotic planning. Existing latent world models typically generate future states with unconstrained neural transition functions, while modern
video generation systems often prioritize perceptual plausibility or introduce physical structure through auxiliary losses, external guidance, or separate dynamics modules. As a result, long-horizon rollouts can remain weakly grounded in the physical principles that govern real dynamics, leading to compounding error, energy drift, and physically inconsistent futures. We propose Least Action World Models (LaWM), a latent world-modeling framework that operationalizes the Principle of Least Action in learned visual latent space: future rollouts are governed by a learned Lagrangian action functional rather than produced only by an unconstrained transition predictor. Our main technical realization is a latent variational integrator: LaWM encodes observations into learned generalized coordinates, learns a latent discrete Lagrangian over consecutive latent states, constructs a discrete action functional, and advances prediction by solving the corresponding discrete integration condition. Thus, physical structure is not merely used to score, regularize, or constrain a completed trajectory; it defines the latent transition rule itself. Because the transition is induced by a discrete variational principle, LaWM provides a structure-preserving bias for long-horizon visual prediction. Across physics-clean synthetic dynamics and
embodied robot interaction
benchmarks, LaWM improves physical invariance, background consistency, motion smoothness, and appearance and
geometric prediction metrics over
video-generation and world-model baselines.
Read the full abstract ▾
Machine Learning
Trapping Attacker in Dilemma: Examining Internal Correlations and External Influences of Trigger for Defending GNN Backdoors
Fan Yang, Binyan Xu, Di Tang et al. (+1 more)
GNNs have become a standard tool for learning on relational data, yet they remain highly vulnerable to
backdoor attacks. Prior defenses often depend on inspecting specific subgraph patterns or node features, and thus can be circumvented by adaptive attackers. We propose PRAETORIAN, a new defense that targets intrinsic requirements of effective GNN
backdoors rather than surface-level cues. Our key observation is that flipping a victim node's prediction requires substantial influence on the victim: attackers tend to either inject many trigger nodes or rely on a small set of highly influential ones. Building on this observation, PRAETORIAN (i) analyzes internal correlations within potential trigger subgraphs to detect abnormally large injected structures, and (ii) quantifies external node influence to identify triggers with disproportionate impact. Across our evaluations, PRAETORIAN reduces the average attack success rate (ASR) to 0.55% with only a 0.62% drop in clean accuracy (CA), whereas state-of-the-art defenses still yield an average ASR of >20% and a CA drop of >3% under the same conditions. Moreover, PRAETORIAN remains effective against a range of adaptive attacks, forcing adversaries to either inject many trigger nodes to achieve high ASR (>80%), which incurs a >10% CA drop, or preserve CA at the cost of limiting ASR to 18.1%. Overall, PRAETORIAN constrains attackers to an unfavorable trade-off between efficacy and detectability.
Read the full abstract ▾
All authors Fan Yang, Binyan Xu, Di Tang, Kehuan Zhang
Machine Learning
Qianru Zhang, Xinyi Gao, Alexander Zhou et al. (+3 more)
Accurate traffic prediction is essential for optimizing transportation systems, enhancing resource allocation, and improving overall urban administration. Spatio-temporal graph neural networks (GNNs) have achieved state-of-the-art performance and have been widely used in various spatio-temporal prediction scenarios. However, these prediction methods often exhibit low generalization ability, struggling with distribution shifts caused by spatio-temporal dynamics. To address this challenge, we propose an approach to enhance the generalization and adaptation of spatio-temporal GNNs through efficient prompting. Specifically, we introduce a lightweight and model-agnostic
prompt tuning framework for spatio-temporal GNNs, named SimpleST. It facilitates adapting pre-trained spatio-temporal GNNs to novel distributions while keeping the model parameters fixed. This
prompt mechanism reduces the overhead and complexity of adaptation, enabling efficient utilization of pre-trained models for out-of-distribution generalization. Extensive experiments conducted on five real-world urban spatio-temporal
datasets demonstrate the superiority of our approach in terms of prediction accuracy and computational efficiency.
Read the full abstract ▾
All authors Qianru Zhang, Xinyi Gao, Alexander Zhou, Reynold Cheng, Siu-Ming Yiu, Hongzhi Yin
Machine Learning · Cross-listed from cs.CV
Normalizing Trajectory Models
Jiatao Gu, Tianrong Chen, Ying Shen et al. (+3 more)
Diffusion-based models decompose sampling into many small Gaussian denoising steps – an assumption that breaks down when generation is compressed to a few coarse transitions. Existing few-step methods address this through
distillation, consistency training, or
adversarial objectives, but sacrifice the likelihood framework in the process. We introduce Normalizing Trajectory Models (NTM), which models each reverse step as an expressive conditional normalizing flow with exact likelihood training. Architecturally, NTM combines shallow invertible blocks within each step with a deep parallel predictor across the trajectory, forming an end-to-end network trainable from scratch or initializable from pretrained flow-matching models. Its exact trajectory likelihood further enables self-
distillation: a lightweight denoiser trained on the model's own score produces high-quality samples in four steps. On text-to-image
benchmarks, NTM matches or outperforms strong image generation baselines in just four sampling steps while uniquely retaining exact likelihood over the generative trajectory.
Read the full abstract ▾
All authors Jiatao Gu, Tianrong Chen, Ying Shen, David Berthelot, Shuangfei Zhai, Josh Susskind
Machine Learning
Zero-Shot Imagined Speech Decoding via Imagined-to-Listened MEG Mapping
Maryam Maghsoudi and Shihab Shamma
Decoding imagined
speech from non-invasive
brain recordings is challenging because imagined
datasets are scarce and difficult to align temporally across subjects and sessions In this work, we propose a new approach to the decoding of imagined
speech that leverages the richer and more reliably labeled recordings during listening to
speech. We collected paired listened and imagined MEG recordings to rhythmic melodic and spoken stimuli from trained musicians. Using trained musicians helped improve temporal
alignment across conditions. We then developed a three-stage decoding pipeline that revealed consistent and meaningful relationships between neural activity evoked by imagining and listening to the same stimuli. First, we trained six linear and neural models to map imagined MEG responses to listened responses. We evaluated these models against a null baseline from unseen subjects to validate that the predicted-listening responses preserve stimulus-specific information. In the second stage, we trained a contrastive word decoder exclusively on the listened MEG responses, and evaluated it using four
embedding strategies including semantic, acoustic, and phonetic representations. In the third stage, we process the imagined MEG responses from held-out subjects through the mapping pipeline to compute the corresponding listening responses that are then decoded by the listened decoder. Using rank-based analysis, we show that the imagined words are decodable significantly above chance. We shall report here the results of a proof-of-concept implementation to decode imagined
speech, where all evaluations are performed on held-out subjects. We also demonstrate that performance improves with training data size, suggesting that this approach is scalable and can directly be made applicable to realistic
brain-computer interface scenarios.
Read the full abstract ▾
Machine Learning
GRAPHLCP: Structure-Aware Localized Conformal Prediction on Graphs
Peyman Baghershahi, Fangxin Wang, Debmalya Mandal et al. (+1 more)
Conformal prediction (CP) provides a distribution-free approach to uncertainty quantification with finite-sample guarantees. However, applying CP to graph neural networks (GNNs) remains challenging as the combinatorial nature of graphs often leads to insufficiently certain predictions and indiscriminative
embeddings. Existing methods primarily rely on
embedding-space proximity for localization, which can be unreliable for graphs and yield inefficient prediction sets. We propose GRAPHLCP, a proximity-based localized CP framework that explicitly incorporates graph
topology and inter-node dependencies into localization and weighting. Our approach introduces a feature-aware densification step to mitigate locality bias in sparse graphs, followed by a Personalized PageRank-based kernel computation to model structural proximity. This enables
topology-dependent anchor sampling and calibration weighting that captures both local and long-range dependencies. Extensive experiments on several regression and classification
datasets demonstrate that GRAPHLCP guarantees marginal coverage with finite samples while efficiently attaining favorable test conditional coverage across various conditioning scenarios.
Read the full abstract ▾
All authors Peyman Baghershahi, Fangxin Wang, Debmalya Mandal, Sourav Medya
Machine Learning · Cross-listed from stat.ML
A Note on Non-Negative $L_1$-Approximating Polynomials
Jane H. Lee, Anay Mehrotra and Manolis Zampetakis
$L_1$-Approximating
polynomials, i.e.,
polynomials that approximate indicator functions in $L_1$-norm under certain distributions, are widely used in computational learning theory. We study the existence of non-negative $L_1$-approximating
polynomials with respect to Gaussian distributions. This is a stronger requirement than $L_1$-approximation but weaker than sandwiching
polynomials (which themselves have many applications). These non-negative approximating
polynomials have recently found uses in smoothed learning from positive-only examples. In this short note, we prove that every class of sets with Gaussian surface area (GSA) at most $\Gamma$ under the standard Gaussian admits degree-$k$ non-negative
polynomials that $\eps$-approximate its indicator functions in $L_1$-norm, for $k=\tilde{O}(\Gamma^2/\varepsilon^2)$. Equivalently, finite GSA implies $L_1$-approximation with the stronger pointwise guarantee that the approximating
polynomial has range contained in $[0,\infty)$. Up to a constant-factor, this matches the degree of the best currently known Gaussian $L_1$-approximation degree bound without the non-negativity constraint.
Read the full abstract ▾
Machine Learning
Reinforcement Learning for Exponential Utility: Algorithms and Convergence in Discounted MDPs
Gugan Thoppe, L. A. Prashanth, Ankur Naskar et al. (+1 more)
Reinforcement learning (RL) for exponential-utility optimization in discounted Markov decision processes (MDPs) lacks principled value-based
algorithms. We address this gap in the fixed risk-aversion setting. Building on the Bellman-type equation for exponential utility studied in \cite{porteus1975optimality}, we derive two Q-value-style extensions and show that the associated operators are contractions in the $L_\infty$ and sup-log/Thompson metrics, respectively. We characterize their fixed points and prove that the induced greedy stationary policy is optimal for the exponential-utility objective among stationary policies. These structural results lead to two model-free
algorithms: a two-timescale Q-learning–style
algorithm, for which we establish almost-sure convergence and provide finite-time convergence rates via timescale separation, and a one-timescale
algorithm governed by a sublinear power-law operator. Since the latter does not admit a global contraction in standard metrics, we prove its convergence using delicate arguments based on local Lipschitzness, monotonicity, homogeneity, and Dini derivatives, and provide a scalar finite-time analysis that highlights the challenges in obtaining convergence rates in the vector case. Our work provides a foundation for value-based RL under exponential-utility objectives.
Read the full abstract ▾
All authors Gugan Thoppe, L. A. Prashanth, Ankur Naskar, Sanjay Bhat
Machine Learning · Cross-listed from cs.CL
Julie Kallini, Artidoro Pagnoni, Tomasz Limisiewicz et al. (+5 more)
Recent byte-level
language models (LMs) match the performance of
token-level models without relying on subword vocabularies, yet their utility is limited by slow, byte-by-byte autoregressive generation. We address this bottleneck in the Byte Latent
Transformer (BLT) through new training and generation techniques. First, we introduce BLT
Diffusion (BLT-D), a new model and our fastest BLT variant, trained with an auxiliary block-wise
diffusion objective alongside the standard next-byte prediction loss. This enables an inference procedure that generates multiple bytes in parallel per decoding step, substantially reducing the number of forward passes required to generate a sequence. Second, we propose two extensions inspired by speculative decoding that trade some of this speed for higher generation quality: BLT Self-speculation (BLT-S), in which BLT's local decoder continues generating past its normal patch boundaries to draft bytes, which are then verified with a single full-model forward pass; and BLT
Diffusion+Verification (BLT-DV), which augments BLT-D with an autoregressive verification step after
diffusion-based generation. All methods may achieve an estimated
memory-bandwidth cost over 50% lower than BLT on generation tasks. Each approach offers its own unique advantages, together removing key barriers to the practical use of byte-level LMs.
Read the full abstract ▾
All authors Julie Kallini, Artidoro Pagnoni, Tomasz Limisiewicz, Gargi Ghosh, Luke Zettlemoyer, Christopher Potts, Xiaochuang Han, Srinivasan Iyer
Machine Learning
Beyond Pairs: Your Language Model is Secretly Optimizing a Preference Graph
Ning Liu, Chuanneng Sun, Kristina Klinkner et al. (+1 more)
Direct Preference Optimization (DPO) aligns
language models using pairwise preference comparisons, offering a simple and effective alternative to Reinforcement Learning (RL) from human feedback. However, in many practical settings, training data consists of multiple rollouts per
prompt, inducing rich preference structure that pairwise DPO fails to exploit. Collapsing such data into independent pairs discards transitivity, introduces redundant or conflicting supervision, and can lead to unstable optimization. We propose Graph Direct Preference Optimization (GraphDPO), a structured generalization of DPO that operates over directed acyclic preference graphs induced by rollout rankings. GraphDPO encodes dominance relations as edges and optimizes a graph-structured Plackett–Luce-inspired objective that aggregates supervision over dominated neighborhoods, propagating the transitive structure of the input preferences while recovering standard DPO as a special case. GraphDPO naturally handles tied and incomparable responses through an equivalence-class construction: responses with identical preference signals form graph layers, incomparable responses share a layer without an edge, and absent edges contribute zero loss, preventing the spurious gradients that strict total-order objectives incur. Under this layered construction, GraphDPO maintains linear per-
prompt complexity via efficient log-sum-exp aggregation. We further incorporate optional ground-truth anchoring by inserting verified solutions as dominant nodes and applying an annealed schedule that stabilizes early training while gradually relaxing oracle supervision. Experiments on
reasoning and program synthesis tasks demonstrate superior performance, suggesting that graph-structured preference modeling is a scalable and robust alternative to pairwise and listwise
alignment objectives.
Read the full abstract ▾
All authors Ning Liu, Chuanneng Sun, Kristina Klinkner, Shervin Malmasi
Machine Learning
Don't Get Your Kroneckers in a Twist: Gaussian Processes on High-Dimensional Incomplete Grids
Mads Greisen Højlund, August Smart Lykke-Møller, Henry Moss et al. (+1 more)
We introduce CUTS-GPR, a new method for performing numerically exact GPR in high-dimensional settings. The key component of CUTS-GPR is an extremely fast kernel
matrix-vector product, which exhibits near-linear or even linear scaling with the amount of training data, $N$, and low-order
polynomial scaling with dimensionality, $D$. This is obtained by combining an additive kernel with an incomplete grid and exploiting the resulting structure of the kernel
matrix. The scalability of the
matrix-vector product is verified by
benchmarks with billions of data points and thousands of dimensions. We demonstrate the end-to-end scalability of CUTS-GPR by running full GPR calculations, including hyperparameter optimization, on synthetic
datasets with up to $N = 4\,494\,001$ and $D = 500$. As a realistic and challenging test, we finally apply CUTS-GPR to a set of ten potential energy surfaces (PESs) with $N = 447\,265$ and $D = 24$. The calculations are completed in a matter of hours, showing that CUTS-GPR enables
Bayesian modelling of high-dimensional PESs - a longstanding challenge in computational chemistry.
Read the full abstract ▾
All authors Mads Greisen Højlund, August Smart Lykke-Møller, Henry Moss, Ove Christiansen
Machine Learning · Cross-listed from eess.SP
PropSplat: Map-Free RF Field Reconstruction via 3D Gaussian Propagation Splatting
William Bjorndahl, Maninder Pal Singh, Farhad Nouri et al. (+1 more)
Building a site-specific propagation model typically requires either ray-tracing over detailed 3D maps or dense measurement campaigns. Both approaches are expensive and often infeasible for rapid deployments where geographic data is unavailable or outdated. We present PropSplat, a map-free propagation modeling method that reconstructs
radio frequency (RF) fields using 3D anisotropic Gaussian primitives. Each Gaussian encodes a scalar path loss offset relative to an explicit baseline path loss model with a learnable path loss exponent. Gaussians are initialized along observed transmitter–receiver paths and optimized end-to-end to learn the propagation environment without external information like floor plans, terrain databases, or clutter data. We evaluate PropSplat against wireless radiance field methods NeRF$^2$, GSRF, and WRF-GS+ on two real-world
datasets. On large-scale outdoor drive-tests spanning multiple topographical regions at six sub-6 GHz frequencies, PropSplat achieves 5.38 dB RMSE when training measurements are spaced 300m apart and outperforms WRF-GS+ (5.87 dB), GSRF (7.46 dB), and NeRF$^2$ (14.76 dB). On indoor Bluetooth Low Energy measurements, PropSplat achieves 0.19m mean localization error, an order of magnitude better than NeRF$^2$ (1.84m), while achieving near-identical received signal strength prediction accuracy. These results show that accurate site-specific propagation reconstruction is achievable from sparse RF-native measurements. The need for geographic data as a prerequisite for scalable RF environment modeling is reduced.
Read the full abstract ▾
All authors William Bjorndahl, Maninder Pal Singh, Farhad Nouri, Joseph Camp
Machine Learning · Cross-listed from stat.ML
Semiparametric Efficient Test for Interpretable Distributional Treatment Effects
Houssam Zenati and Arthur Gretton
Distributional treatment effects can be invisible to means: a treatment may preserve average outcomes while changing tails, modes, dispersion, or rare-event probabilities. Kernel tests can detect discrepancies between interventional outcome laws, but global tests do not reveal where the laws differ. We propose DR-ME, to our knowledge the first semiparametrically efficient finite-location test for interpretable distributional treatment effects. DR-ME evaluates an interventional kernel witness at learned outcome locations, returning causal-discrepancy coordinates rather than only a global rejection. From observational data, we derive orthogonal doubly robust kernel features whose centered oracle form is the canonical gradient of this finite witness. For fixed locations, we characterize the local testing limit: DR-ME is chi-square calibrated under the null, has noncentral chi-square local power, and uses the covariance whitening that optimizes local signal-to-noise for discrepancies visible through the selected coordinates. This efficient local-power geometry yields a principled location-learning criterion, with sample splitting preserving post-selection validity. Experiments show near-nominal type-I error, competitive power against global doubly robust kernel tests, and interpretable learned locations that localize distributional effects in a semi-synthetic medical-imaging study.
Read the full abstract ▾
Machine Learning · Cross-listed from cs.CV
PET-Adapter: Test-Time Domain Adaptation for Full and Limited-Angle PET Image Reconstruction
Rüveyda Yilmaz, Yuli Wu, Johannes Stegmaier et al. (+1 more)
Positron Emission
Tomography (PET) image reconstruction is inherently challenged by Poisson noise and physical degradation factors, which are further exacerbated in limited-angle acquisitions. While deep learning methods demonstrate promising performance, their generalization to unseen clinical data distributions remains limited without extensive retraining. We propose PET-Adapter, a test-time domain adaptation framework for generative PET reconstruction models pretrained solely on phantom data. Our method enables adaptation to clinical
datasets with varying anatomies, tracers, and scanner configurations without requiring paired ground truth. PET-Adapter introduces layer-wise low-rank anatomical conditioning during adaptation and Ordered Subset Expectation Maximization-based warm-starting that initializes the generation from physics-informed reconstructions, reducing
diffusion steps from 50 to 2 without compromising quality. Experiments across multiple clinical
datasets demonstrate superior 3D reconstruction performance in both full-angle and limited-angle settings, highlighting the clinical feasibility and computational efficiency of the proposed approach.
Read the full abstract ▾
All authors Rüveyda Yilmaz, Yuli Wu, Johannes Stegmaier, Volkmar Schulz
Machine Learning · Cross-listed from cs.CV
STARFlow2: Bridging Language Models and Normalizing Flows for Unified Multimodal Generation
Ying Shen, Tianrong Chen, Yuan Gao et al. (+6 more)
Deep generative models have advanced rapidly across text and
vision, motivating unified
multimodal systems that can understand, reason over, and generate interleaved text-image sequences. Most existing approaches combine autoregressive
language modeling with
diffusion-based image generators, inheriting a structural mismatch between causal text generation and iterative visual denoising. We observe that autoregressive normalizing flows are autoregressive
Transformers–sharing the same causal mask, KV-cache mechanism, and left-to-right structure as LLMs–making them the most natural paradigm for true unified
multimodal generation. We present STARFlow2, built on the Pretzel architecture that vertically interleaves a pretrained VLM stream with a TarFlow stream via residual skip connections, both operating under the same causal mask. Combined with a deep-shallow flow design and a unified FAE latent space, STARFlow2 enables cache-friendly interleaved generation where both text and visual outputs directly enter the KV-cache without re-encoding. Experiments demonstrate strong performance across image generation and
multimodal understanding
benchmarks, validating autoregressive flows as a viable foundation for unified
multimodal modeling.
Read the full abstract ▾
All authors Ying Shen, Tianrong Chen, Yuan Gao, Yizhe Zhang, Yuyang Wang, Miguel Ángel Bautista, Shuangfei Zhai, Joshua M. Susskind, Jiatao Gu
Machine Learning
Observation-Aligned Two-Stage Domain Decomposition for Physics-Informed Traffic State Estimation with Sparse Fixed Sensors
Eunhan Ka, Ludovic Leclercq and Satish V. Ukkusuri
Traffic state estimation from sparse fixed sensors is challenging because physics-informed neural networks (PINNs) tend to over-smooth sharp transitions admitted by the Lighthill-Whitham–Richards (LWR) model. This study proposes Two-Stage Domain Decomposition Physics-Informed Neural Networks (TSDD-PINN), an observation-aligned framework for LWR-based offline speed-field reconstruction. The framework supports spatial, temporal, and space–time refinement. Matched direction analysis shows that spatial refinement has the lowest mean error and less than half the training time of space–time refinement in the tested setting, while temporal refinement is faster. A global parent PINN is first trained. In the controlled spatial implementation, its residual profile guides a deterministic partition for warm-started child networks. An optional operational safeguard retains Stage~1 when the prespecified screen does not activate. The primary I-24 MOTION evaluation spans five days, five sensor configurations, and ten seeds per configuration, yielding 1{,}500 runs. Controlled TSDD-PINN attains the lowest relative $L_2$ error in 18 of 25 configurations and 14 of 15 sparse-sensing cases, while training 2.4 times faster than the extended PINN (XPINN) baseline under the evaluated implementations and training budgets. Non-neural comparisons show that the advantage over interpolation is concentrated under sparse sensing, whereas dense sensing often favors interpolation. A separate 250-run operational evaluation finds infrequent activation and motivates the Stage-1-preserving safeguard. The residual is treated as an indicator of model difficulty rather than a validated shock detector. The evidence supports a sensing-density-dependent operating range rather than uniform improvement.
Read the full abstract ▾
Machine Learning · Cross-listed from cs.NE
Convex Optimization of Spiking Neural Networks with Global and Approximation Guarantees
Himanshu Udupi, Calvin Ang, Sungyoon Kim et al. (+2 more)
Spiking neural networks (SNNs) have been proposed as energy-efficient alternatives to deep neural networks. Their training is non-convex and involves a non-differentiable spike function, so existing methods offer no guarantee of reaching a global optimum. We address this issue by formulating SNN optimization as an equivalent convex optimization problem. Specifically, we show that the weight-decay-regularized training problem for a class of threshold RNNs, including leaky integrate-and-fire SNNs, has zero duality gap and the same optimal value as a convex program. Since the resulting convex programs can have exponentially many variables in the amount of data, we introduce a practical randomized surrogate and establish a high-probability guarantee on its approximation quality. Empirically, the best convex variant outperforms surrogate-gradient training on addition and first-last-XOR, matches its performance on sequential MNIST, and trails a reservoir baseline on base-3 and base-5 addition.
Read the full abstract ▾
All authors Himanshu Udupi, Calvin Ang, Sungyoon Kim, Xiaocong Yang, ChengXiang Zhai
Machine Learning
Position: Mechanistic Interpretability Must Disclose Identification Assumptions for Causal Claims
Zezheng Lin and Fengming Liu
Mechanistic interpretability papers increasingly use causal vocabulary:
circuits, mediators, causal abstraction, monosemanticity. Such claims require explicit identification assumptions. A purposive audit of 10 papers across four methodological strands finds no dedicated identification-assumptions section and a recurring pattern: validation metrics such as faithfulness, completeness, monosemanticity,
alignment, or ablation effects are reported as causal support without stating the assumptions that make them identifying. A two-human-coder audit on $n=30$ reproduces the direction of the main finding: dedicated identification sections are absent, and validation-metric substitution is common, though exact Dim B/D counts are coding-rule sensitive. The paper proposes a disclosure norm: state whether the claim is causal, name the identification strategy, enumerate assumptions, stress at least one, and explain how conclusions shift if assumptions fail. Validation is not identification.
Read the full abstract ▾
Machine Learning
Interpreting Reinforcement Learning Agents with Susceptibilities
Chris Elliott, Einar Urdshals, David Quarel et al. (+1 more)
Susceptibilities are a technique for neural network interpretability that studies the response of posterior expectation values of observables to perturbations of the loss. We generalize this construction to the setting of the regret in deep reinforcement learning and investigate the utility of susceptibilities in a simple gridworld model that nevertheless exhibits non-trivial stagewise development. We argue that susceptibilities reveal internal features of the development of the model in parameter space that one cannot detect purely by studying the development of the learned policy. We validate these results with activation-steering, and discuss the framework's extension to RLHF post-training.
Read the full abstract ▾
All authors Chris Elliott, Einar Urdshals, David Quarel, Daniel Murfet
Machine Learning · Cross-listed from math.OC
Penalty-Based First-Order Methods for Bilevel Optimization with Minimax and Constrained Lower-Level Problems
Yiyang Shen, Yutian He, Weiran Wang et al. (+1 more)
We study a class of bilevel optimization problems in which both the upper- and lower-level problems have minimax structures. This setting captures a broad range of emerging applications. Despite the extensive literature on bilevel optimization and minimax optimization separately, existing methods mainly focus on bilevel optimization with lower-level minimization problems, often under strong convexity assumptions, and are not directly applicable to the minimax lower-level setting considered here. To address this gap, we develop penalty-based first-order methods for bilevel minimax optimization without requiring strong convexity of the lower-level problem. In the deterministic setting, we establish that the proposed method finds an $\epsilon$-KKT point with $\tilde{O}(\epsilon^{-4})$ oracle complexity. We further show that bilevel problems with convex constrained lower-level minimization can be reformulated as special cases of our framework via Lagrangian duality, leading to an $\tilde{O}(\epsilon^{-4})$ complexity bound that improves upon the existing $\tilde{O}(\epsilon^{-7})$ result. Finally, we extend our approach to the
stochastic setting, where only
stochastic gradient oracles are available, and prove that the proposed
stochastic method finds a nearly $\epsilon$-KKT point with $\tilde{O}(\epsilon^{-9})$ oracle complexity.
Read the full abstract ▾
All authors Yiyang Shen, Yutian He, Weiran Wang, Qihang Lin
Machine Learning
STEPS: A Temporal Smooth Error Propagation Solver on the Manifolds for Test-Time Adaptation in Time Series Forecasting
Jiaqi Liu, Yifan Ouyang, Zhifei Song et al. (+2 more)
Test-Time Adaptation (TTA) aims to improve time series
forecasting under distribution shifts by using limited observations revealed during inference. However,
forecasting TTA must operate in a source-free online setting, where the adaptation signal is short, temporally correlated, and potentially noisy. Existing methods can therefore suffer from weak identifiability, error accumulation, and unstable long-horizon corrections when the revealed prefix is sparse or contaminated. To address these issues, we propose STEPS, a Smooth Temporal Error Propagation Solver for TTA in time-series
forecasting. STEPS reformulates
forecasting TTA as a Dirichlet Boundary Value Problem on a temporal
manifold, where the revealed prefix error serves as the boundary condition for the unknown future error field. Then, STEPS solves a smooth and bounded correction field in prediction space: a Local Solver propagates prefix errors under temporal smoothness, a Global Solver retrieves stable cross-window error
memory and Spatiotemporal
Manifold Fusion (SMF) integrates both solutions into the final correction. Across six standard
benchmarks and four frozen backbones, STEPS achieves an average relative MSE reduction of 26.82% over the zero-shot backbone, exceeding the strongest compared TTA baseline by 12.77%. Additional sparse prefix and contamination tests confirm the robustness of STEPS under limited and noisy prefixes.
Read the full abstract ▾
All authors Jiaqi Liu, Yifan Ouyang, Zhifei Song, Sim Kuan Goh, Ashwaq Qasem