SearcharxivSearch

arXiv subjects

Zheng Zhong

Publications and source records attributed to Zheng Zhong.

At least 19 recordsLinked to original sources

Towards Full Pipeline FP8 Reinforcement Learning for LLMs

Reinforcement learning (RL) has become a key technique for improving the reasoning and agentic abilities of large language models (LLMs). Although FP8 quantization can accelerate RL training, maintaining stability throughout an FP8 RL pipeline remains challenging. While previous works have focused on resolving train-inference mismatches using correction techniques like TIS, we reveal that full-pipeline FP8 RL still suffers from severe training instability, manifesting as anomalous mid-training entropy surges and garbled outputs. We trace this instability to a previously overlooked cause: compounded FP8 quantization noise distorts the importance ratio, disproportionately pushing negative-advantage tokens outside the trust region and erroneously zeroing out their gradients. As a result, pathological outputs are not properly penalized and accumulate over the course of training. To address this, we propose Calibrated Clipping, a dynamic method that aligns the FP8 clipping bounds with high-precision BF16 distributions by matching the lower-bound clipping quantile and rebalancing the upper bound accordingly. Extensive experiments across GRPO and DAPO algorithms, model scales from 8B to 32B, and multiple FP8 scaling granularities demonstrate that our approach successfully eliminates entropy surges and restores performance comparable to the BF16 baseline.

cs.LG

DisagMoE: Computation-Communication overlapped MoE Training via Disaggregated AF-Pipe Parallelism

Mixture-of-experts (MoE) architectures enable trillion-parameter LLMs with sparsely activated experts. Expert parallelism (EP) is a widely adopted MoE training strategy, but it suffers from severe all-to-all communication bottlenecks, which is exaggerated by the limited inter-node network bandwidth as the growing model size requires distributing experts across GPU nodes. Prior work focused on overlapping these all-to-all communications with feed-forward network (FFN) and self-attention computations, which often leaves residual network-bound stalls due to inherent imbalance in attention and FFN layers' computation-communication ratios. We present DisagMoE, a disaggregated MoE training system that jointly optimizes model placement and scheduling for maximal efficiency. DisagMoE separates attention and FFN layers into disjoint GPU groups, introduces a multi-stage pipeline with uni-directional, many-to-many communications, and employs a computation-communication roofline model to balance GPU and network bandwidth allocation among the attention and FFN groups. DisagMoE is implemented on Megatron-LM, and evaluation shows that DisagMoE improves training efficiency across multiple MoE models with up to 1.8x speedup on 16-node 8xH800 clusters.

cs.LG

Thermodynamics of stacking faults and phase stability in cobalt alloys: A combined computational and experimental study

Stacking fault energy dictates phase stability and deformation behavior in Co alloys and WC-Co cemented carbides, yet a quantitative assessment of alloying effects at finite temperatures remains poorly established. By integrating first-principles thermodynamics with microstructural characterization, we provide a rigorous evaluation of these influences across atomic and macroscopic scales. We show that stacking fault energetics at 0K for transition metal solutes are primarily governed by atomic misfit volume. While 4d and 5d elements follow a consistent linear trend, specific 3d solutes exhibit significant deviations due to non-negligible magnetic contributions. By incorporating phonon, electronic, longitudinal spin-fluctuation, and magnetic free-energy contributions, the model accurately captures the fcc-hcp transformation and quantifies how diverse solutes modulate the phase landscape. We demonstrate that V, Ni, Fe, Mo, and W lower the transformation temperature by stabilizing fcc phase, while Cr and C exhibit the opposite effect, consistent with experimental phase diagrams. Furthermore, microscopic analysis confirms that higher W content dissolved in the Co suppresses stacking-fault formation by elevating the stacking fault energy at finite temperatures. This work clarifies the physical mechanisms by which alloying regulates stacking fault energy and phase stability in Co-based systems, providing guidance for the design of Co-based alloys and WC-Co cemented carbides.

cond-mat.mtrl-sci

veScale-FSDP: Flexible and High-Performance FSDP at Scale

Fully Sharded Data Parallel (FSDP), also known as Zero Redundancy Optimizer (ZeRO), is widely used for large-scale model training, because of its memory efficiency and minimal intrusion on model code. However, existing FSDP systems rely on fixed element-wise or row-wise sharding formats that conflict with block-structured computations. As a result, they struggle to support modern structure-aware training methods, including block-wise quantization and non-element-wise optimizers such as Shampoo and Muon. In addition, today's implementations incur communication and memory overheads that degrade efficiency at the scale of tens of thousands of GPUs. We introduce veScale-FSDP, a novel FSDP system that combines RaggedShard, a flexible sharding format, with a structure-aware planning algorithm to deliver both flexibility and performance. veScale-FSDP enables zero-copy FSDP communications and natively supports block-wise quantization and non-element-wise optimizers, achieving 5% to 66% higher throughput and 16% to 30% lower memory usage than existing FSDP systems, while scaling efficiently to tens of thousands of GPUs.

cs.DC

Surface elasticity effect on Plateau-Rayleigh instability in soft solids

Soft solids exhibit instability and develop surface undulations due to surface effects, a phenomenon known as the elastic Plateau-Rayleigh (PR) instability, driven by the interplay of surface and bulk elasticity. Previous studies on the PR instability in solids mainly focused on the case of constant surface tension and ignored the effect of surface elasticity. It has been shown by experiments that the surface effects in solid-like materials depend both on the surface tension and surface elasticity, but little is known about the role of the latter in the elasto-capillary instabilities in soft solids. Here, we conduct an in-depth exploration of the effect of surface elasticity on the PR instability in an elastic cylinder by coupling theoretical and numerical methods. We derive an asymptotically consistent one-dimensional (1d) model to characterize the PR instability from three-dimensional (3d) nonlinear bulk-surface elasticity, and develop a new finite-element (FE) scheme for simulating 3d deformations of the bulk-surface system. The initiation and evolution of the PR instability are obtained analytically with the aid of the 1d model. The 1d results are further validated by the 3d FE simulations. By synthesizing the 1d analytic solutions and 3d numerical results, the effects of surface elasticity, surface compressibility, surface tension, axial force and geometrical size on the PR instability are thoroughly elucidated. Our results can be applied to calibrate surface parameters for solid-like materials and develop constitutive models for elastic surfaces.

cond-mat.soft

A dynamic shim approach for correcting eddy current effects in diffusion-prepared MRI acquisition using a multi-coil AC/DC shim-array

Purpose: We developed a dynamic B0 shimming approach using a 46-channel AC/DC shim array to correct phase errors caused by eddy currents from diffusion-encoding gradients in diffusion-prepared MRI, enabling high b-value imaging without the SNR loss from the use of magnitude stabilizer. Methods: A 46-channel AC/DC shim array and corresponding amplifier system were built. Spin echo prescans with and without diffusion preparation were then used to rapidly measure eddy current induced phase differences. These phase maps were used as targets in an optimization framework to compute compensatory shim currents for multi-shot 3D diffusion-prepared acquisitions. Results: The proposed method allows flexible use of the AC/DC shim array to correct undesirable eddy current effects in diffusion-prepared MRI. Phantom and in vivo experiments demonstrate whole-brain, cardiac-gated, multi-shot 3D diffusion-prepared imaging without the use of magnitude stabilizers. The approach enables preservation of full SNR while achieving reliable diffusion encoding at b-values up to 2000 s/mm2. Conclusions: This work demonstrates a new strategy for applying an AC/DC shim array to compensate for eddy current induced phase errors in diffusion-prepared MRI. By eliminating the need for magnitude stabilizer, it enables efficient high-quality diffusion imaging with full signal sensitivity retained.

physics.med-ph

Parallel Loop Transformer for Efficient Test-Time Computation Scaling

Large Language Models (LLMs) are powerful but often too slow and costly for real-world use during inference. Looped transformers save on parameters by reusing the same weights for multiple computational steps, or "loops." However, this approach has a major flaw: the loops run one after another, causing inference latency and memory requirements to increase with each added loop. This makes them impractical for fast applications. To solve this problem, we introduce the Parallel Loop Transformer (PLT). PLT is a new architecture that delivers the performance benefits of a deep, looped model but with the low latency of a standard, non-looped model. PLT works using two key techniques. First, Cross-Loop Parallelism (CLP) breaks the sequential dependency by computing different loops for different tokens at the same time, all within a single pass. Second, to prevent memory costs from growing, we use an Efficient Representation Enhancement strategy. This method shares the memory (KV cache) from the first loop with all other loops. It then uses a Gated Sliding-Window Attention (G-SWA) to combine this shared global information with local information, maintaining high accuracy. Our experiments show that PLT achieves the high accuracy of a traditional looped model but with almost no extra latency or memory cost compared to a standard transformer.

cs.CL

MegaScale-MoE: Large-Scale Communication-Efficient Training of Mixture-of-Experts Models in Production

We present MegaScale-MoE, a production system tailored for the efficient training of large-scale mixture-of-experts (MoE) models. MoE emerges as a promising architecture to scale large language models (LLMs) to unprecedented sizes, thereby enhancing model performance. However, existing MoE training systems experience a degradation in training efficiency, exacerbated by the escalating scale of MoE models and the continuous evolution of hardware. Recognizing the pivotal role of efficient communication in enhancing MoE training, MegaScale-MoE customizes communication-efficient parallelism strategies for attention and FFNs in each MoE layer and adopts a holistic approach to overlap communication with computation at both inter- and intra-operator levels. Additionally, MegaScale-MoE applies communication compression with adjusted communication patterns to lower precision, further improving training efficiency. When training a 352B MoE model on 1,440 NVIDIA Hopper GPUs, MegaScale-MoE achieves a training throughput of 1.41M tokens/s, improving the efficiency by 1.88$\times$ compared to Megatron-LM. We share our operational experience in accelerating MoE training and hope that by offering our insights in system design, this work will motivate future research in MoE systems.

cs.LG

Mechanics of soft-body rolling motion without external torque

The Drosophila larva, a soft-body animal, can bend its body and roll efficiently to escape danger. However, contrary to common belief, this rolling motion is not driven by the imbalance of gravity and ground reaction forces. Through functional imaging and ablation experiments, we demonstrate that the sequential actuation of axial muscles within an appropriate range of angles is critical for generating rolling. We model the interplay between muscle contraction, hydrostatic skeleton deformation, and body-environment interactions, and systematically explain how sequential muscle actuation generates the rolling motion. Additionally, we constructed a pneumatic soft robot to mimic the larval rolling strategy, successfully validating our model. This mechanics model of soft-body rolling motion not only advances the study of related neural circuits, but also holds potential for applications in soft robotics.

cond-mat.soft

Hybrid Spin and Anomalous Spin-Momentum Locking in Surface Elastic Waves

Transverse spin of surface waves is a universal phenomenon which has recently attracted significant attention in optics and acoustics. It appears in gravity water waves, surface plasmon-polaritons, surface acoustic waves, and exhibits remarkable intrinsic spin-momentum locking, which has found useful applications for efficient spin-direction couplers. Here we demonstrate, both theoretically and experimentally, that the transverse spin of surface elastic (Rayleigh) waves has an anomalous sign near the surface, opposite to that in the case of electromagnetic, sound, or water surface waves. This anomalous sign appears due to the hybrid (neither transverse nor longitudinal) nature of elastic surface waves. Furthermore, we show that this sign anomaly can be employed for the selective spin-controlled excitation of symmetric and antisymmetric Lamb modes propagating in opposite directions in an elastic plate. Our results pave the way for spin-controlled manipulation of elastic waves and can be important for a variety of areas, from phononic spin-based devices to seismic waves.

physics.app-ph

Sub-micrometer phonon mean free paths in metal-organic frameworks revealed by machine-learning molecular dynamics simulations

Metal-organic frameworks (MOFs) are a family of materials that have high porosity and structural tunability and hold great potential in various applications, many of which requiring a proper understanding of the thermal transport properties. Molecular dynamics (MD) simulations play an important role in characterizing the thermal transport properties of various materials. However, due to the complexity of the structures, it is difficult to construct accurate empirical interatomic potentials for reliable MD simulations of MOFs. To this end, we develop a set of accurate yet highly efficient machine-learned potentials for three typical MOFs, including MOF-5, HKUST-1, and ZIF-8, using the neuroevolution potential approach as implemented in the GPUMD package, and perform extensive MD simulations to study thermal transport in the three MOFs. Although the lattice thermal conductivity (LTC) values of the three MOFs are all predicted to be smaller than 1 $\rm{W/(m\ K)}$ at room temperature, the phonon mean free paths (MFPs) are found to reach the sub-micrometer scale in the low-frequency region. As a consequence, the apparent LTC only converges to the diffusive limit for micrometer single crystals, which means that the LTC is heavily reduced in nanocrystalline MOFs. The sub-micrometer phonon MFPs are also found to be correlated with a moderate temperature dependence of LTC between those in typical crystalline and amorphous materials. Both the large phonon MFPs and the moderate temperature dependence of LTC fundamentally change our understanding of thermal transport in MOFs.

cond-mat.mtrl-sci

Elastic Valley Spin Controlled Chiral Coupling in Topological Valley Phononic Crystals

Distinct from the phononic valley pseudo-spin, the real physical spin of elastic waves adds a novel tool-kit capable of envisaging the valley-spin physics of topological valley phononic crystals from a local viewpoint. Here, we report the observation of local elastic valley spin as well as the hidden elastic spin-valley locking mechanism overlooked before. We demonstrate that the selective one-way routing of valley phonon states along the topological interface can be reversed by imposing the elastic spin meta-source at different interface locations with opposite valley-spin correspondence. We unveil the physical mechanism of selective directionality as the elastic spin controlled chiral coupling of valley phonon states, through both analytical theory and experimental measurement of the opposite local elastic spin density at different interface locations for different transport directions. The elastic spin of valley topological edge phonons can be extended to other topological states and offers new tool to explore topological metamaterials.

cond-mat.mes-hall

SLfRank: Shinnar-Le-Roux Pulse Design with Reduced Energy and Accurate Phase Profiles using Rank Factorization

The Shinnar-Le-Roux (SLR) algorithm is widely used to design frequency selective pulses with large flip angles. We improve its design process to generate pulses with lower energy (by as much as 26%) and more accurate phase profiles. Concretely, the SLR algorithm consists of two steps: (1) an invertible transform between frequency selective pulses and polynomial pairs that represent Cayley-Klein (CK) parameters and (2) the design of the CK polynomial pair to match the desired magnetization profiles. Because the CK polynomial pair is bi-linearly coupled, the original algorithm sequentially solves for each polynomial instead of jointly. This results in sub-optimal pulses. Instead, we leverage a convex relaxation technique, commonly used for low rank matrix recovery, to address the bi-linearity. Our numerical experiments show that the resulting pulses are almost always globally optimal in practice. For slice excitation, the proposed algorithm results in more accurate linear phase profiles. And in general the improved pulses have lower energy than the original SLR pulses.

physics.med-ph

GPUMD: A package for constructing accurate machine-learned potentials and performing highly efficient atomistic simulations

We present our latest advancements of machine-learned potentials (MLPs) based on the neuroevolution potential (NEP) framework introduced in [Fan et al., Phys. Rev. B 104, 104309 (2021)] and their implementation in the open-source package GPUMD. We increase the accuracy of NEP models both by improving the radial functions in the atomic-environment descriptor using a linear combination of Chebyshev basis functions and by extending the angular descriptor with some four-body and five-body contributions as in the atomic cluster expansion approach. We also detail our efficient implementation of the NEP approach in graphics processing units as well as our workflow for the construction of NEP models, and we demonstrate their application in large-scale atomistic simulations. By comparing to state-of-the-art MLPs, we show that the NEP approach not only achieves above-average accuracy but also is far more computationally efficient. These results demonstrate that the GPUMD package is a promising tool for solving challenging problems requiring highly accurate, large-scale atomistic simulations. To enable the construction of MLPs using a minimal training set, we propose an active-learning scheme based on the latent space of a pre-trained NEP model. Finally, we introduce three separate Python packages, GPYUMD, CALORINE, and PYNEP, which enable the integration of GPUMD into Python workflows.

physics.comp-ph

Variable thermal transport in black, blue, and violet phosphorene from extensive atomistic simulations with a neuroevolution potential

Phosphorus has diverse chemical bonds and even in its two-dimensional form there are three stable allotropes: black phosphorene (Black-P), blue phosphorene (Blue-P), and violet phosphorene (Violet-P). Due to the complexity of these structures, no efficient and accurate classical interatomic potential has been developed for them. In this paper, we develop an efficient machine-learned neuroevolution potential model for these allotropes and apply it to study thermal transport in them via extensive molecular dynamics (MD) simulations. Based on the homogeneous nonequilibrium MD method, the thermal conductivities are predicted to be $12.5 \pm 0.2$ (Black-P in armchair direction), $78.4 \pm 0.4$ (Black-P in zigzag direction), $128 \pm 3$ (Blue-P), and $2.36 \pm 0.05$ (Violet-P) $\mathrm{Wm^{-1}K^{-1}}$. The underlying reasons for the significantly different thermal conductivity values in these allotropes are unraveled through spectral decomposition, phonon eigenmodes, and phonon participation ratio. Under external tensile strain, the thermal conductivity in black-P and violet-P are finite, while that in blue-P appears unbounded due to the linearization of the flexural phonon dispersion that increases the phonon mean free paths in the zero-frequency limit.

physics.comp-ph

HoVer: A Dataset for Many-Hop Fact Extraction And Claim Verification

We introduce HoVer (HOppy VERification), a dataset for many-hop evidence extraction and fact verification. It challenges models to extract facts from several Wikipedia articles that are relevant to a claim and classify whether the claim is Supported or Not-Supported by the facts. In HoVer, the claims require evidence to be extracted from as many as four English Wikipedia articles and embody reasoning graphs of diverse shapes. Moreover, most of the 3/4-hop claims are written in multiple sentences, which adds to the complexity of understanding long-range dependency relations such as coreference. We show that the performance of an existing state-of-the-art semantic-matching model degrades significantly on our dataset as the number of reasoning hops increases, hence demonstrating the necessity of many-hop reasoning to achieve strong results. We hope that the introduction of this challenging dataset and the accompanying evaluation task will encourage research in many-hop fact retrieval and information verification. We make the HoVer dataset publicly available at https://hover-nlp.github.io

cs.CL

Adversarial Attack on Deep Learning-Based Splice Localization

Regarding image forensics, researchers have proposed various approaches to detect and/or localize manipulations, such as splices. Recent best performing image-forensics algorithms greatly benefit from the application of deep learning, but such tools can be vulnerable to adversarial attacks. Due to the fact that most of the proposed adversarial example generation techniques can be used only on end-to-end classifiers, the adversarial robustness of image-forensics methods that utilize deep learning only for feature extraction has not been studied yet. Using a novel algorithm capable of directly adjusting the underlying representations of patches we demonstrate on three non end-to-end deep learning-based splice localization tools that hiding manipulations of images is feasible via adversarial attacks. While the tested image-forensics methods, EXIF-SC, SpliceRadar, and Noiseprint, rely on feature extractors that were trained on different surrogate tasks, we find that the formed adversarial perturbations can be transferable among them regarding the deterioration of their localization performance.

cs.CV

Severity Assessment of Coronavirus Disease 2019 (COVID-19) Using Quantitative Features from Chest CT Images

Background: Chest computed tomography (CT) is recognized as an important tool for COVID-19 severity assessment. As the number of affected patients increase rapidly, manual severity assessment becomes a labor-intensive task, and may lead to delayed treatment. Purpose: Using machine learning method to realize automatic severity assessment (non-severe or severe) of COVID-19 based on chest CT images, and to explore the severity-related features from the resulting assessment model. Materials and Method: Chest CT images of 176 patients (age 45.3$\pm$16.5 years, 96 male and 80 female) with confirmed COVID-19 are used, from which 63 quantitative features, e.g., the infection volume/ratio of the whole lung and the volume of ground-glass opacity (GGO) regions, are calculated. A random forest (RF) model is trained to assess the severity (non-severe or severe) based on quantitative features. Importance of each quantitative feature, which reflects the correlation to the severity of COVID-19, is calculated from the RF model. Results: Using three-fold cross validation, the RF model shows promising results, i.e., 0.933 of true positive rate, 0.745 of true negative rate, 0.875 of accuracy, and 0.91 of area under receiver operating characteristic curve (AUC). The resulting importance of quantitative features shows that the volume and its ratio (with respect to the whole lung volume) of ground glass opacity (GGO) regions are highly related to the severity of COVID-19, and the quantitative features calculated from the right lung are more related to the severity assessment than those of the left lung. Conclusion: The RF based model can achieve automatic severity assessment (non-severe or severe) of COVID-19 infection, and the performance is promising. Several quantitative features, which have the potential to reflect the severity of COVID-19, were revealed.

eess.IV