SearcharxivSearch

arXiv subjects

Corrado Rainone

Publications and source records attributed to Corrado Rainone.

At least 19 recordsLinked to original sources

When Cloud Agents Meet Device Agents: Lessons from Hybrid Multi-Agent Systems

The design space of agentic AI inference spans two extremes: frontier large language models (LLMs), typically hosted in the cloud and offering strong performance across a wide range of tasks at substantially high cost, and more cost-efficient small language models (SLMs), which are amenable to on-device inference. Hybrid multi-agent systems (MASs) combining on-device and cloud models offer a promising middle ground, but they also introduce a complex and poorly understood design space in which task accuracy, monetary cost, and edge energy consumption are tightly coupled; in the absence of general design principles, hybrid components, although not the most prevalent choice, are typically introduced through ad hoc decisions tailored to specific domains. In this work, we examine this design space more systematically. We adapt two representative MAS architectures to support hybrid inference and study how individual design choices shift the operating point along the Pareto frontier of power, cost, and performance. Our findings paint a nuanced picture of hybrid MAS design: while SLMs can effectively benefit from LLM assistance, the optimal architecture is highly task-dependent, and greater frontier-level compute does not consistently translate to better performance.

cs.MA

Efficient Reasoning on the Edge

Large language models (LLMs) with chain-of-thought reasoning achieve state-of-the-art performance across complex problem-solving tasks, but their verbose reasoning traces and large context requirements make them impractical for edge deployment. These challenges include high token generation costs, large KV-cache footprints, and inefficiencies when distilling reasoning capabilities into smaller models for mobile devices. Existing approaches often rely on distilling reasoning traces from larger models into smaller models, which are verbose and stylistically redundant, undesirable for on-device inference. In this work, we propose a lightweight approach to enable reasoning in small LLMs using LoRA adapters combined with supervised fine-tuning. We further introduce budget forcing via reinforcement learning on these adapters, significantly reducing response length with minimal accuracy loss. To address memory-bound decoding, we exploit parallel test-time scaling, improving accuracy at minor latency increase. Finally, we present a dynamic adapter-switching mechanism that activates reasoning only when needed and a KV-cache sharing strategy during prompt encoding, reducing time-to-first-token for on-device inference. Experiments on Qwen2.5-7B demonstrate that our method achieves efficient, accurate reasoning under strict resource constraints, making LLM reasoning practical for mobile scenarios. Videos demonstrating our solution running on mobile devices are available on our project page.

cs.LG

Replacing thinking with tool usage enables reasoning in small language models

Recent advances have established a new machine learning paradigm based on scaling up compute at inference time as well as at training time. In that line of work, a combination of Supervised Fine-Tuning (SFT) on synthetic demonstrations and Reinforcement Learning with Verifiable Rewards (RLVR) is used for training Large Language Models to expend extra compute during inference in the form of "thoughts" expressed in natural language. In this paper, we propose to instead format these tokens as a multi-turn interaction trace with a stateful tool. At each turn, the new state of the tool is appended to the context of the model, whose job is to generate the tokens necessary to control the tool via a custom DSL. We benchmark this approach on the problem of repairing malfunctioning Python code, and show that this constrained setup allows for faster sampling of experience and a denser reward signal, allowing even models of size up to 3B parameters to learn how to proficiently expend additional compute on the task.

cs.LG

CodeIt: Self-Improving Language Models with Prioritized Hindsight Replay

Large language models are increasingly solving tasks that are commonly believed to require human-level reasoning ability. However, these models still perform very poorly on benchmarks of general intelligence such as the Abstraction and Reasoning Corpus (ARC). In this paper, we approach ARC as a programming-by-examples problem, and introduce a novel and scalable method for language model self-improvement called Code Iteration (CodeIt). Our method iterates between 1) program sampling and hindsight relabeling, and 2) learning from prioritized experience replay. By relabeling the goal of an episode (i.e., the target program output given input) to the realized output produced by the sampled program, our method effectively deals with the extreme sparsity of rewards in program synthesis. Applying CodeIt to the ARC dataset, we demonstrate that prioritized hindsight replay, along with pre-training and data-augmentation, leads to successful inter-task generalization. CodeIt is the first neuro-symbolic approach that scales to the full ARC evaluation dataset. Our method solves 15% of ARC evaluation tasks, achieving state-of-the-art performance and outperforming existing neural and symbolic baselines. Our code is available at https://github.com/Qualcomm-AI-research/codeit .

cs.AI

Learning Lattice Quantum Field Theories with Equivariant Continuous Flows

We propose a novel machine learning method for sampling from the high-dimensional probability distributions of Lattice Field Theories, which is based on a single neural ODE layer and incorporates the full symmetries of the problem. We test our model on the $ϕ^4$ theory, showing that it systematically outperforms previously proposed flow-based methods in sampling efficiency, and the improvement is especially pronounced for larger lattices. Furthermore, we demonstrate that our model can learn a continuous family of theories at once, and the results of learning can be transferred to larger lattices. Such generalizations further accentuate the advantages of machine learning methods.

hep-lat

Robust Scheduling with GFlowNets

Finding the best way to schedule operations in a computation graph is a classical NP-hard problem which is central to compiler optimization. However, evaluating the goodness of a schedule on the target hardware can be very time-consuming. Traditional approaches as well as previous machine learning ones typically optimize proxy metrics, which are fast to evaluate but can lead to bad schedules when tested on the target hardware. In this work, we propose a new approach to scheduling by sampling proportionally to the proxy metric using a novel GFlowNet method. We introduce a technique to control the trade-off between diversity and goodness of the proposed schedules at inference time and demonstrate empirically that the pure optimization baselines can lead to subpar performance with respect to our approach when tested on a target model. Furthermore, we show that conditioning the GFlowNet on the computation graph enables generalization to unseen scheduling problems for both synthetic and real-world compiler datasets.

cs.AI

Neural Topological Ordering for Computation Graphs

Recent works on machine learning for combinatorial optimization have shown that learning based approaches can outperform heuristic methods in terms of speed and performance. In this paper, we consider the problem of finding an optimal topological order on a directed acyclic graph with focus on the memory minimization problem which arises in compilers. We propose an end-to-end machine learning based approach for topological ordering using an encoder-decoder framework. Our encoder is a novel attention based graph neural network architecture called \emph{Topoformer} which uses different topological transforms of a DAG for message passing. The node embeddings produced by the encoder are converted into node priorities which are used by the decoder to generate a probability distribution over topological orders. We train our model on a dataset of synthetically generated graphs called layered graphs. We show that our model outperforms, or is on-par, with several topological ordering baselines while being significantly faster on synthetic graphs with up to 2k nodes. We also train and test our model on a set of real-world computation graphs, showing performance improvements.

cs.LG

Scaling Up Machine Learning For Quantum Field Theory with Equivariant Continuous Flows

We propose a continuous normalizing flow for sampling from the high-dimensional probability distributions of Quantum Field Theories in Physics. In contrast to the deep architectures used so far for this task, our proposal is based on a shallow design and incorporates the symmetries of the problem. We test our model on the $ϕ^4$ theory, showing that it systematically outperforms a realNVP baseline in sampling efficiency, with the difference between the two increasing for larger lattices. On the largest lattice we consider, of size $32\times 32$, we improve a key metric, the effective sample size, from 1% to 66% w.r.t. the realNVP baseline.

cs.LG

Low-frequency vibrational spectrum of mean-field disordered systems

We study a recently introduced and exactly solvable mean-field model for the density of vibrational states $\mathcal{D}(ω)$ of a structurally disordered system. The model is formulated as a collection of disordered anharmonic oscillators, with random stiffness $κ$ drawn from a distribution $p(κ)$, subjected to a constant field $h$ and interacting bilinearly with a coupling of strength $J$. We investigate the vibrational properties of its ground state at zero temperature. When $p(κ)$ is gapped, the emergent $\mathcal{D}(ω)$ is also gapped, for small $J$. Upon increasing $J$, the gap vanishes on a critical line in the $(h,J)$ phase diagram, whereupon replica symmetry is broken. At small $h$, the form of this pseudogap is quadratic, $\mathcal{D}(ω)\simω^2$, and its modes are delocalized, as expected from previously investigated mean-field spin glass models. However, we determine that for large enough $h$, a quartic pseudogap $\mathcal{D}(ω)\simω^4$, populated by localized modes, emerges, the two regimes being separated by a special point on the critical line. We thus uncover that mean-field disordered systems can generically display both a quadratic-delocalized and a quartic-localized spectrum at the glass transition.

cond-mat.dis-nn

Mean-field model of interacting quasilocalized excitations in glasses

Structural glasses feature quasilocalized excitations whose frequencies $ω$ follow a universal density of states ${\cal D}(ω)\!\sim\!ω^4$. Yet, the underlying physics behind this universality is not fully understood. Here we study a mean-field model of quasilocalized excitations in glasses, viewed as groups of particles embedded inside an elastic medium and described collectively as anharmonic oscillators. The oscillators, whose harmonic stiffness is taken from a rather featureless probability distribution (of upper cutoff $κ_0$) in the absence of interactions, interact among themselves through random couplings (characterized by strength $J$) and with the surrounding elastic medium (an interaction characterized by a constant force $h$). We first show that the model gives rise to a gapless density of states ${\cal D}(ω)\!=\!A_{\rm g}\,ω^4$ for a broad range of model parameters, expressed in terms of the strength of stabilizing anharmonicity, which plays a decisive role in the model. Then -- using scaling theory and numerical simulations -- we provide a complete understanding of the non-universal prefactor $A_{\rm g}(h,J,κ_0)$, of the oscillators' interaction-induced mean square displacement and of an emerging characteristic frequency, all in terms of properly identified dimensionless quantities. In particular, we show that $A_{\rm g}(h,J,κ_0)$ is a nonmonotonic function of $J$ for a fixed $h$, varying predominantly exponentially with $-(κ_0 h^{2/3}\!/J^2)$ in the weak interactions (small $J$) regime -- reminiscent of recent observations in computer glasses -- and predominantly decaying as a power-law for larger $J$, in a regime where $h$ plays no role. We discuss the physical interpretation of the model and its possible relations to available observations in structural glasses, along with delineating some future research directions.

cond-mat.dis-nn

Pinching a glass reveals key properties of its soft spots

It is now well established that glasses feature quasilocalized nonphononic excitations --- coined "soft spots"---, which follow a universal $ω^4$ density of states in the limit of low frequencies $ω$. All glass-specific properties, such as the dependence on the preparation protocol or composition, are encapsulated in the non-universal prefactor of the universal $ω^4$ law. The prefactor, however, is a composite quantity that incorporates information both about the number of quasilocalized nonphononic excitations and their characteristic stiffness, in an apparently inseparable manner. We show that by pinching a glass, i.e. by probing its response to force dipoles, one can disentangle and independently extract these two fundamental pieces of physical information. This analysis reveals that the number of quasilocalized nonphononic excitations follows a Boltzmann-like law in terms of the parent temperature from which the glass is quenched. The latter, sometimes termed the fictive (or effective) temperature, plays important roles in non-equilibrium thermodynamic approaches to the relaxation, flow and deformation of glasses. The analysis also shows that the characteristic stiffness of quasilocalized nonphononic excitations can be related to their characteristic size, a long sought-for length scale. These results show that important physical information, which is relevant for various key questions in glass physics, can be obtained through pinching a glass.

cond-mat.soft

Statistical mechanics of local force dipole responses in computer glasses

Soft quasilocalized modes (QLMs) are universally featured by structural glasses quenched from a melt, and are supposedly involved in a number of glassy anomalies such as the low temperature scaling of their thermal conductivity and specific heat, and sound attenuation at intermediate frequencies. In computer glasses, QLMs may assume the form of harmonic vibrational modes under a narrow set of circumstances, however direct access to their full distribution over frequency is hindered by hybridizations of QLMs with other low-frequency modes (e.g.~phonons). Previous studies to overcome this issue have demonstrated that the response of a glass to local force dipoles serves as a good proxy for its QLMs; we therefore study here the statistical-mechanical properties of the responses to local force dipoles in computer glasses, over a large range of glass stabilities and in various spatial dimensions, with the goal of revealing properties of the yet-inaccessible full distribution of QLMs' frequencies. We find that, as opposed to the spatial-dimension-independent universal distribution of QLMs' frequencies $ω$ (and, consequently, also of their stiffness $κ\!=\!ω^2$), the distribution of stiffnesses associated with responses to local force dipoles features a (weak) dependence on spatial dimension. We rationalize this dependence by introducing a lattice model that incorporates both the real-space profiles of QLMs --- associated with dimension-dependent long-range elastic fields --- and the universal statistical properties of their frequencies. Finally, we discuss possible connections between our findings and basic aspects of the glass transition problem, and to finite-size effects in plastic activity of ultrastable glasses.

cond-mat.soft

Particle pinning suppresses spinodal criticality in the shear banding instability

Strained amorphous solids often fail mechanically by creating a shear-band. It had been understood that the shear banding instability is usefully described as crossing a spinodal point (with disorder) in an appropriate thermodynamic description. It remained contested however whether the spinodal is critical (with divergent correlation length) or not. Here we offer evidence for critical spinodal by using particle pinning. For a finite concentration of pinned particles the correlation length is bounded by the average distance between pinned particles, but without pinning it is bounded by the system size.

cond-mat.soft

Wave attenuation in glasses: Rayleigh and generalized-Rayleigh scattering scaling

The attenuation of long-wavelength phonons (waves) by glassy disorder plays a central role in various glass anomalies, yet it is neither fully characterized, nor fully understood. Of particular importance is the scaling of the attenuation rate $Γ(k)$ with small wavenumbers $k\!\to\!0$ in the thermodynamic limit of macroscopic glasses. Here we use a combination of theory and extensive computer simulations to show that the macroscopic low-frequency behavior emerges at intermediate frequencies in finite-size glasses, above a recently identified crossover wavenumber $k_\dagger$, where phonons are no longer quantized into bands. For $k\!<\!k_\dagger$, finite-size effects dominate $Γ(k)$, which is quantitatively described by a theory of disordered phonon bands. For $k\!>\!k_\dagger$, we find that $Γ(k)$ is affected by the number of quasilocalized nonphononic excitations, a generic signature of glasses that feature a universal density of states. In particular, we show that in a frequency range in which this number is small, $Γ(k)$ follows a Rayleigh scattering scaling $\sim\!k^{d+1}$ ($d$ is the spatial dimension), and that in a frequency range in which this number is sufficiently large, the recently observed generalized-Rayleigh scaling of the form $\sim\!k^{d+1}\log\!{(k_0/k)}$ emerges ($k_0\!>k_\dagger$ is a characteristic wavenumber). Our results suggest that macroscopic glasses --- and, in particular, glasses generated by conventional laboratory quenches that are known to strongly suppress quasilocalized nonphononic excitations --- exhibit Rayleigh scaling at the lowest wavenumbers $k$ and a crossover to generalized-Rayleigh scaling at higher $k$. Some supporting experimental evidence from recent literature is presented.

cond-mat.soft

On the protocol dependence of plasticity in ultra-stable amorphous solids

While perfect crystals may exhibit a purely elastic response to shear all the way to yielding, the response of amorphous solids is punctuated by plastic events. The prevalence of this plasticity depends on the number of particles $N$ of the system, with the average strain interval before the first plastic event, $\overline{Δγ}$, scaling like $N^α$ with $α$ negative: larger samples are more susceptible to plasticity due to more numerous disorder-induced soft spots. In this paper we examine this scaling relation in ultra-stable glasses prepared with the Swap Monte Carlo algorithm, with regard to the possibility of protocol-dependent scaling exponent, which would also imply a protocol dependence in the distribution of local yield stresses in the glass. We show that, while a superficial analysis seems to corroborate this hypothesis, this is only a pre-asymptotic effect and in fact our data can be well explained by a simple model wherein such protocol dependence is absent.

cond-mat.soft

How Generic are the Robust Theoretical Aspects of Jamming in Hard Sphere Models?

In very recent work the mean field theory of the jamming transition in infinite dimensional hard spheres models was presented. Surprisingly, this theory predicts quantitatively numerically determined characteristics of jamming in two and three dimensions. This is a rare and unusual finding. Here we argue that this agreement in non-generic: only for hard sphere models it happens that sufficiently close to jamming the effective interactions are in agreement with mean-field theory, justifying the truncation of many body interactions (which is the exact protocol in infinite dimensions). Any softening of the bare hard sphere interactions results in effective interactions that are not mean-field all the way to jamming, making the discussed phenomenon non generic.

cond-mat.soft

Mechanical Failure in Amorphous Solids: Scale Free Spinodal Criticality

The mechanical failure of amorphous media is a ubiquitous phenomenon from material engineering to geology. It has been noticed for a long time that the phenomenon is "scale-free", indicating some type of criticality. In spite of attempts to invoke "Self-Organized Criticality", the physical origin of this criticality, and also its universal nature, being quite insensitive to the nature of microscopic interactions, remained elusive. Recently we proposed that the precise nature of this critical behavior is manifested by a spinodal point of a thermodynamic phase transition. Moreover, at the spinodal point there exists a divergent correlation length which is associated with the system-spanning instabilities (known also as shear bands) which are typical to the mechanical yield. Demonstrating this requires the introduction of an "order parameter" that is suitable for distinguishing between disordered amorphous systems, and an associated correlation function, suitable for picking up the growing correlation length. The theory, the order parameter, and the correlation functions used are universal in nature and can be applied to any amorphous solid that undergoes mechanical yield. Critical exponents for the correlation length divergence and the system size dependence are estimated. The phenomenon is seen at its sharpest in athermal systems, as is explained below; in this paper we extend the discussion also to thermal systems, showing that at sufficiently high temperatures the spinodal phenomenon is destroyed by thermal fluctuations.

cond-mat.dis-nn