arXiv ScienceSearch

arXiv subjects

Liang Zheng

Publications and source records attributed to Liang Zheng.

At least 19 recordsLinked to original sources

Positive Sample Propagation along the Audio-Visual Event Line

Visual and audio signals often coexist in natural environments, forming audio-visual events (AVEs). Given a video, we aim to localize video segments containing an AVE and identify its category. In order to learn discriminative features for a classifier, it is pivotal to identify the helpful (or positive) audio-visual segment pairs while filtering out the irrelevant ones, regardless whether they are synchronized or not. To this end, we propose a new positive sample propagation (PSP) module to discover and exploit the closely related audio-visual pairs by evaluating the relationship within every possible pair. It can be done by constructing an all-pair similarity map between each audio and visual segment, and only aggregating the features from the pairs with high similarity scores. To encourage the network to extract high correlated features for positive samples, a new audio-visual pair similarity loss is proposed. We also propose a new weighting branch to better exploit the temporal correlations in weakly supervised setting. We perform extensive experiments on the public AVE dataset and achieve new state-of-the-art accuracy in both fully and weakly supervised settings, thus verifying the effectiveness of our method.

cs.CV

Aesthetic Post-Training Diffusion Models from Generic Preferences with Step-by-step Preference Optimization

Generating visually appealing images is fundamental to modern text-to-image generation models. A potential solution to better aesthetics is direct preference optimization (DPO), which has been applied to diffusion models to improve general image quality including prompt alignment and aesthetics. Popular DPO methods propagate preference labels from clean image pairs to all the intermediate steps along the two generation trajectories. However, preference labels provided in existing datasets are blended with layout and aesthetic opinions, which would disagree with aesthetic preference. Even if aesthetic labels were provided (at substantial cost), it would be hard for the two-trajectory methods to capture nuanced visual differences at different steps. To improve aesthetics economically, this paper uses existing generic preference data and introduces step-by-step preference optimization (SPO) that discards the propagation strategy and allows fine-grained image details to be assessed. Specifically, at each denoising step, we 1) sample a pool of candidates by denoising from a shared noise latent, 2) use a step-aware preference model to find a suitable win-lose pair to supervise the diffusion model, and 3) randomly select one from the pool to initialize the next denoising step. This strategy ensures that diffusion models focus on the subtle, fine-grained visual differences instead of layout aspect. We find that aesthetics can be significantly enhanced by accumulating these improved minor differences. When fine-tuning Stable Diffusion v1.5 and SDXL, SPO yields significant improvements in aesthetics compared with existing DPO methods while not sacrificing image-text alignment compared with vanilla models. Moreover, SPO converges much faster than DPO methods due to the use of more correct preference labels provided by the step-aware preference model.

cs.CV

LeapAlign: Post-Training Flow Matching Models at Any Generation Step by Building Two-Step Trajectories

This paper focuses on the alignment of flow matching models with human preferences. A promising way is fine-tuning by directly backpropagating reward gradients through the differentiable generation process of flow matching. However, backpropagating through long trajectories results in prohibitive memory costs and gradient explosion. Therefore, direct-gradient methods struggle to update early generation steps, which are crucial for determining the global structure of the final image. To address this issue, we introduce LeapAlign, a fine-tuning method that reduces computational cost and enables direct gradient propagation from reward to early generation steps. Specifically, we shorten the long trajectory into only two steps by designing two consecutive leaps, each skipping multiple ODE sampling steps and predicting future latents in a single step. By randomizing the start and end timesteps of the leaps, LeapAlign leads to efficient and stable model updates at any generation step. To better use such shortened trajectories, we assign higher training weights to those that are more consistent with the long generation path. To further enhance gradient stability, we reduce the weights of gradient terms with large magnitude, instead of completely removing them as done in previous works. When fine-tuning the Flux model, LeapAlign consistently outperforms state-of-the-art GRPO-based and direct-gradient methods across various metrics, achieving superior image quality and image-text alignment.

cs.CV

CGC-py: A Monte Carlo Event Generator for Gluon Saturation Physics

We develop CGC-py, a Monte Carlo event generator for deep-inelastic scattering. It couples the full Color Glass Condensate (CGC) cross section for $γ^*p(A)\to q\bar q+X$ to a Parton-Branching transverse-momentum-dependent backward initial-state shower, while \textsc{Pythia}~8 handles final-state radiation and hadronization. CGC-py retains the complete target-elastic and target-inelastic contributions without taking the back-to-back correlation limit, allowing single- and di-hadron observables to be generated consistently from the same event sample. We validate the generator through an analytic closure test of the single-inclusive quark spectrum and a comparison of charged-hadron spectra in $ep$ collisions with H1 data, finding excellent agreement. The predicted nuclear modification factor $R_{e\mathrm{Au}}^h$ shows the expected saturation pattern: suppression at low $p_T^*$ followed by a rise toward unity at higher $p_T^*$. A comparison with a \textsc{Pythia}~6 baseline, together with an $x_g$-rescaling study, indicates that small-$x$ CGC evolution and collinear DGLAP dynamics contribute comparably to the growth of the dihadron away-side width with energy. Genuine saturation-driven broadening emerges only at the highest energies considered. Within CGC-py, $e\mathrm{Au}$ collisions exhibit an enhanced away-side width and a suppressed back-to-back yield relative to $ep$ collisions. These nuclear effects remain modest over EIC kinematics, motivating measurements at the most forward accessible kinematics and the use of complementary observables to maximize sensitivity to gluon saturation.

hep-ph

AsymSpec: Efficient Cloud-Edge Speculative Decoding over Asymmetric Networks

Cloud-edge speculative decoding places a lightweight draft model at an edge gateway and a higher-quality target model in the cloud, but inserts communication into every speculative block. Under a constrained uplink, candidate messages may queue while the verifier is idle. Stop-and-wait scheduling leaves edge compute underutilized; optimistic same-request runahead can waste work when a rejection or an unexpected bonus token invalidates dependent drafts. We present AsymSpec, which addresses uplink-gated verification and invalid dependent work with two corresponding mechanisms. Its asymmetric verification protocol keeps the common-path acceptance upload compact and moves richer, rejection-only correction information to the downlink. A total-variation (TV) certificate for the residual distribution determines whether a small target top-K response suffices; if not, the protocol progressively escalates through proposal-based exact recovery before falling back to the full distribution. Its confirmed-prefix pipeline exposes only independent, valid requests to the edge scheduler and lets the cloud re-batch arrived blocks, hiding verification waits when another confirmed-prefix request is ready without using same-request runahead. Across three draft-target pairs, two workloads, and three asymmetric network profiles, our end-to-end evaluation shows that AsymSpec delivers 2.82-28.03$\times$ the output-token throughput of the strongest baseline.

cs.DC

Investigating forward-backward asymmetry in D-meson production and anisotropic flow in p-Pb collisions at the LHC

We investigate the forward--backward asymmetry in the production and elliptic flow of prompt D0 mesons in proton--lead (p--Pb) collisions at$\sqrt{s_{\mathrm{NN}}}=8.16$ TeV using the heavy-flavor improved string-melting version of the AMPT model. The model calculations provide a simultaneous description of nuclear modification factor $R_{\mathrm{pPb}}$ and $v_2$ in forward and backward rapidities. We find that the observed asymmetry arises from the interplay of initial-state cold nuclear matter effects and final-state partonic interactions, with the competition between coalescence and fragmentation playing a critical role in shaping the transverse momentum and rapidity dependence of both observables. This work suggests that a partonic medium is formed in high-multiplicity p-Pb collisions at LHC energies.

nucl-th

DiffusionBench: On Holistic Evaluation of Diffusion Transformers

Diffusion transformer (DiT) research on image generation has converged to a single evaluation setup: class-conditional generation on ImageNet. While methods improve the FID and related metrics, it is increasingly unclear whether they reflect real progress in generative modeling. The natural alternative, i.e., text-to-image (T2I) generation, is perceived as too costly or inconvenient to train and evaluate and is often skipped. We argue that this perception no longer holds. We introduce NanoGen, a unified DiT training and evaluation framework. NanoGen matches state-of-the-art DiT baselines on ImageNet and, with 12 lines of configuration change, also trains competitive text-to-image models. It currently supports RAE, VAE, pixel-space, and MeanFlow diffusion methods under both ImageNet and T2I setups. Under NanoGen, training T2I requires comparable compute to ImageNet. After training 21 latent diffusion models with NanoGen, we observe that method ranking shows no strong correlation between ImageNet and T2I generation: Pearson correlation is between -0.377 and -0.580 across three metrics. This suggests that a method which improves class-conditional ImageNet FID may show no corresponding improvement on T2I, clearly indicating the necessity of evaluating DiTs on both tasks. To this end, we summarize ImageNet and text-to-image results, which yields DiffusionBench, a holistic benchmark for DiT research. We recommend reporting DiffusionBench in place of ImageNet alone: methods that improve DiffusionBench are more likely to reflect broader progress.

cs.CV

Gecko: A Simulation Environment with Stateful Feedback for Refining Agent Tool Calls

The ability to use tools is fundamental for large language model (LLM) agents. Given a task, existing systems use LLMs to plan and generate tool calls, which are executed by real-world tools to complete the task. However, tool calls are prone to errors because they are generated primarily from the intrinsic capabilities of LLMs. Moreover, while it is useful to let LLMs iteratively refine the tool-call sequence using execution results from real tools, this process can be expensive and may cause unsafe side effects. To improve LLM tool calls and address issues caused by using real tools for refinement, we introduce Gecko, a stateful simulation environment that provides informative feedback for refining LLM tool calls before real execution. Specifically, Gecko combines rules and LLMs to check the validity of tool names and arguments, synthesize schema-conforming and state-consistent responses, and judge task completion against the user objective. These three types of feedback allow LLMs to refine their tool calls in simulation, forming a simple yet effective test-time scaling method named GATS. On BFCLv3 and $τ^2$-bench, GATS consistently improves the performance of various LLMs.

cs.SE

RE-SAC: Disentangling aleatoric and epistemic risks in bus fleet control: A stable and robust ensemble DRL approach

Bus holding control is challenging due to stochastic traffic and passenger demand. While deep reinforcement learning (DRL) shows promise, standard actor-critic algorithms suffer from Q-value instability in volatile environments. A key source of this instability is the conflation of two distinct uncertainties: aleatoric uncertainty (irreducible noise) and epistemic uncertainty (data insufficiency). Treating these as a single risk leads to value underestimation in noisy states, causing catastrophic policy collapse. We propose a robust ensemble soft actor-critic (RE-SAC) framework to explicitly disentangle these uncertainties. RE-SAC applies Integral Probability Metric (IPM)-based weight regularization to the critic network to hedge against aleatoric risk, providing a smooth analytical lower bound for the robust Bellman operator without expensive inner-loop perturbations. To address epistemic risk, a diversified Q-ensemble penalizes overconfident value estimates in sparsely covered regions. This dual mechanism prevents the ensemble variance from misidentifying noise as a data gap, a failure mode identified in our ablation study. Experiments in a realistic bidirectional bus corridor simulation demonstrate that RE-SAC achieves the highest cumulative reward (approx. -0.4e6) compared to vanilla SAC (-0.55e6). Mahalanobis rareness analysis confirms that RE-SAC reduces Oracle Q-value estimation error by up to 62% in rare out-of-distribution states (MAE of 1647 vs. 4343), demonstrating superior robustness under high traffic variability.

cs.LG

BAPR: Bayesian amnesic piecewise-robust reinforcement learning for non-stationary continuous control

Real-world control systems frequently operate under \emph{piecewise stationary} conditions, where dynamics remain stable for extended periods before undergoing abrupt regime changes. Standard robust RL methods face a fundamental dilemma: a globally conservative policy wastes performance during stable periods, while a locally adaptive policy risks catastrophic failure when the regime changes undetected. We propose \textbf{BAPR} (Bayesian Amnesic Piecewise-Robust SAC), which unifies Bayesian Online Change Detection (BOCD) with robust ensemble RL. The BAPR operator -- a convex combination of mode-conditional Bellman operators weighted by a frozen belief distribution -- is a $γ$-contraction. A complementary counterexample, machine-verified in Lean~4, establishes a \emph{sharp boundary}: when beliefs depend on the Q-function, the contraction factor becomes $γ+ λΔ$ (where $Δ$ is the mode reward gap), and contraction fails exactly when $γ+ λΔ\geq 1$. We derive a \emph{component-wise} formal error budget for the abstract operator -- every component machine-verified -- bounding post-switch recovery; the budget applies to the abstract mode-mixture operator and inherits to the implemented shared-critic algorithm only through the frozen-parameter design intuition. All results are formally verified with no \texttt{sorry} (1,145 lines across 3 Lean~4 files, 22 machine-verified theorems). BOCD drives an adaptive conservatism mechanism: the policy becomes maximally conservative after detected change-points and smoothly relaxes as confidence grows, with detection delay $O(\log(1/δ))$. A context-conditioning module trained via RMDM loss provides mode-aware representations from simulator-provided mode IDs at training time and requires no mode labels at deployment.

cs.LG

WARDEN: Endangered Indigenous Language Transcription and Translation with 6 Hours of Training Data

This paper introduces WARDEN, an early language model system capable of transcribing and translating Wardaman, an endangered Australian indigenous language into English. The significant challenge we face is the lack of large-scale training data: in fact, we only have 6 hours of annotated audio. Therefore, while it is common practice to train a single model for transcription and translation using large datasets (like English to French), this practice is no longer viable in the Wardaman to English context. To tackle the low-resource challenge, we design WARDEN to have separate transcription and translation models: WARDEN first turns a Wardaman audio input into phonemic transcription, and then the transcription into English translation. Further, we propose two useful techniques to enhance performance. For transcription, we initialize the Wardaman token from Sundanese, a language that shares similar phonemes with Wardaman, to accelerate fine-tuning of the transcription model. For translation, we compile a Wardaman-English dictionary from expert annotations, and provide this domain-specific knowledge to a large language model (LLM) to reason and decide the final output. We empirically demonstrate that this two-stage design works better than data-hungry unified approaches in extremely low data settings. Using a mere 6 hours of annotated data, WARDEN outperforms larger open-source and proprietary models and establishes a strong baseline. Data and code are available.

cs.CL

Studies of beauty hadron and non-prompt charm hadron production in pp collisions at $\sqrt{s}$=13 TeV within a transport model approach

In high-energy proton proton ($pp$) collisions at the LHC, non-prompt charm hadrons, originating from beauty hadron decays, provide a valuable probe for beauty quark dynamics, particularly at low transverse momentum where direct beauty measurements are challenging. We employ A Multi-Phase Transport Model (AMPT) of string melting version coupled with PYTHIA8 initial conditions to study the beauty hadron and non-prompt charm hadron productions in $pp$ collisions at $\sqrt{s} = 13$ TeV. In this work, the beauty quark mass during the generation stage has been increased to reproduce the measured $b\bar{b}$ cross section, and a beauty flavor specific coalescence parameter $r_{BM}^b$ is introduced to match LHCb measurements of beauty baryon to meson ratios. With these refinements, AMPT achieves a reasonable agreement with experimental data on beauty hadron yields and non-prompt charm hadron production from ALICE and LHCb. We present the transverse momentum and multiplicity dependence of non-prompt to prompt charm hadron ratios, providing new insights into the interplay between beauty quark production and hadronization process. We emphasize that the multiplicity dependence of the non-prompt to prompt charm hadron productions can be useful to constrain the flavor dependences of the coalescence dynamics. This work establishes a unified framework for future studies of heavy quark transport and collective flow behavior in small collision systems.

hep-ph

Gromov Wasserstein Optimal Transport for Semantic Correspondences

Establishing correspondences between image pairs is a long studied problem in computer vision. With recent large-scale foundation models showing strong zero-shot performance on downstream tasks including classification and segmentation, there has been interest in using the internal feature maps of these models for the semantic correspondence task. Recent works observe that features from DINOv2 and Stable Diffusion (SD) are complementary, the former producing accurate but sparse correspondences, while the latter produces spatially consistent correspondences. As a result, current state-of-the-art methods for semantic correspondence involve combining features from both models in an ensemble. While the performance of these methods is impressive, they are computationally expensive, requiring evaluating feature maps from large-scale foundation models. In this work we take a different approach, instead replacing SD features with a superior matching algorithm which is imbued with the desirable spatial consistency property. Specifically, we replace the standard nearest neighbours matching with an optimal transport algorithm that includes a Gromov Wasserstein spatial smoothness prior. We show that we can significantly boost the performance of the DINOv2 baseline, and be competitive and sometimes surpassing state-of-the-art methods using Stable Diffusion features, while being 5--10x more efficient. We make code available at https://github.com/fsnelgar/semantic_matching_gwot .

cs.CV

Studying Energy-Energy Correlators in pp Collisions at the LHC with a Jet-Free Event-Topology Method

We present a jet-free approach for measuring energy-energy correlators (EEC) in proton-proton (pp) collisions at the Large Hadron Collider (LHC), employing an event-topology method that does not rely on explicit jet reconstruction. Using the leading charged hadron as a reference axis, the azimuthal plane is divided into Toward and Transverse regions, enabling a robust background subtraction and extending EEC measurements into the low $p_T$ regime where conventional jet-based approaches become unreliable. The method is validated through comparisons with conventional jet reconstruction results. We systematically explore the dependence of the EEC on the leading-particle transverse momentum and parton flavor. The observed scaling between the EEC peak position and the hard scale suggests that this topology-based EEC captures effectively the transition between perturbative and non-perturbative QCD regimes. Distinct differences are found between quark- and gluon-initiated events, reflecting their different color charges and radiation patterns. Extending the analysis to heavy flavor, EECs triggered by leading charm mesons exhibit a suppressed magnitude and a peak shifted toward larger angular separations relative to inclusive charged-particle triggers, providing a direct manifestation of the dead-cone effect. This jet-free EEC framework offers a simple and experimentally robust tool for studying the scale and flavor dependence of the QCD dynamics, with promising applications to proton-nucleus and heavy-ion collisions at the LHC.

hep-ph

Flexible Geometric Guidance for Probabilistic Human Pose Estimation with Diffusion Models

3D human pose estimation from 2D images is a challenging problem due to depth ambiguity and occlusion. Because of these challenges the task is underdetermined, where there exists multiple -- possibly infinite -- poses that are plausible given the image. Despite this, many prior works assume the existence of a deterministic mapping and estimate a single pose given an image. Furthermore, methods based on machine learning require a large amount of paired 2D-3D data to train and suffer from generalization issues to unseen scenarios. To address both of these issues, we propose a framework for pose estimation using diffusion models, which enables sampling from a probability distribution over plausible poses which are consistent with a 2D image. Our approach falls under the guidance framework for conditional generation, and guides samples from an unconditional diffusion model, trained only on 3D data, using the gradients of the heatmaps from a 2D keypoint detector. We evaluate our method on the Human 3.6M dataset under best-of-$m$ multiple hypothesis evaluation, showing state-of-the-art performance among methods which do not require paired 2D-3D data for training. We additionally evaluate the generalization ability using the MPI-INF-3DHP and 3DPW datasets and demonstrate competitive performance. Finally, we demonstrate the flexibility of our framework by using it for novel tasks including pose generation and pose completion, without the need to train bespoke conditional models. We make code available at https://github.com/fsnelgar/diffusion_pose .

cs.CV

Investigating the transverse-momentum- and pseudorapidity-dependent flow vector decorrelation in p--Pb collisions with a Multi-Phase Transport model

The event-by-event fluctuations in the initial energy density of the nuclear collisions lead to the decorrelation of second order flow vector, as known as its transverse-momentum ($p_{\mathrm{T}}$) and pseudorapidity ($η$) dependence as observed in high-energy heavy-ion collisions. Existing measurements at the CERN Large Hadron Collider shown that these decorrelations are also observed in small collision systems. In this work, a systematic study of the transverse-momentum- and pseudorapidity-dependent flow vector decorrelation is performed in p--Pb collisions at the 5.02 TeV with A Multi-Phase Transport (AMPT) model using different tunings of the initial conditions, partonic and hadronic interactions. It is found that the string-melting version of the AMPT model provides a reasonable description of the measured flow vector decorrelation as a function of $p_{\mathrm{T}}$ and $η$. We demonstrate that the hadronic scatterings do not have significant impact on decorrelation in p--Pb collisions for different centrality selections, while both initial conditions and partonic interactions influence the magnitude of the decorrelations. In addition, we found that the subtraction of the nonflow, especially the long-range jet correlation, is crucial for the accurate extraction of the flow vector decorrelation in small collision systems. The comparison of data and model presented in this paper provide further insights in understanding the fluctuations of the flow vector with $p_{\mathrm{T}}$ and $η$ in small collision systems and has referential value for future measurements.

hep-ph

Open charm production and $Λ_{c}^{+}/D^{0}$ ratio in pp and Au+Au collisions at the RHIC

We study open charm hadrons production in pp and Au+Au collisions at $\sqrt{s_{\mathrm{NN}}} = 200$~GeV using an improved a multi-phase transport (AMPT) model. Specifically, we show the transverse-momentum spectra and nuclear modification factors $R_{\mathrm{AA}}$ of $D^{0}$ mesons and $Λ_{c}^{+}$ baryons, as well as the $Λ_{c}^{+}/D^{0}$ ratio in pp and Au+Au collisions. The results obtained from the AMPT model simulations are compared with the STAR experimental data and found to be consistent. We further investigate the $Λ_{c}^{+}/D^{0}$ ratio by evaluating contributions from coalescence, fragmentation, and the combined coalescence+fragmentation mechanisms, and we find that fragmentation alone underestimates the pronounced enhancement in Au+Au relative to pp at low and intermediate $p_{\mathrm{T}}$, whereas the coalescence+fragmentation mechanism reproduces the observed trend significantly better. These results indicate that coalescence plays a key role in charm baryon productions and helps constrain the relative importance of different hadronization mechanisms in the ultra-relativistic nuclear collisions.

hep-ph

Mosaic: Unlocking Long-Context Inference for Diffusion LLMs via Global Memory Planning and Dynamic Peak Taming

Diffusion-based large language models (dLLMs) have emerged as a promising paradigm, utilizing simultaneous denoising to enable global planning and iterative refinement. While these capabilities are particularly advantageous for long-context generation, deploying such models faces a prohibitive memory capacity barrier stemming from severe system inefficiencies. We identify that existing inference systems are ill-suited for this paradigm: unlike autoregressive models constrained by the cumulative KV-cache, dLLMs are bottlenecked by transient activations recomputed at every step. Furthermore, general-purpose memory reuse mechanisms lack the global visibility to adapt to dLLMs' dynamic memory peaks, which toggle between logits and FFNs. To address these mismatches, we propose Mosaic, a memory-efficient inference system that shifts from local, static management to a global, dynamic paradigm. Mosaic integrates a mask-only logits kernel to eliminate redundancy, a lazy chunking optimizer driven by an online heuristic search to adaptively mitigate dynamic peaks, and a global memory manager to resolve fragmentation via virtual addressing. Extensive evaluations demonstrate that Mosaic achieves an average 2.71$\times$ reduction in the memory peak-to-average ratio and increases the maximum inference sequence length supportable on identical hardware by 15.89-32.98$\times$. This scalability is achieved without compromising accuracy and speed, and in fact reducing latency by 4.12%-23.26%.

cs.LG