Searcharxiv⌕ Search

arXiv subjects

Yuheng Zhang

Publications and source records attributed to Yuheng Zhang.

At least 55 records · Page 3Linked to original sources

Improving Deepfake Detection with Reinforcement Learning-Based Adaptive Data Augmentation

The generalization capability of deepfake detectors is critical for real-world use. Data augmentation via synthetic fake face generation effectively enhances generalization, yet current SoTA methods rely on fixed strategies-raising a key question: Is a single static augmentation sufficient, or does the diversity of forgery features demand dynamic approaches? We argue existing methods overlook the evolving complexity of real-world forgeries (e.g., facial warping, expression manipulation), which fixed policies cannot fully simulate. To address this, we propose CRDA (Curriculum Reinforcement-Learning Data Augmentation), a novel framework guiding detectors to progressively master multi-domain forgery features from simple to complex. CRDA synthesizes augmented samples via a configurable pool of forgery operations and dynamically generates adversarial samples tailored to the detector's current learning state. Central to our approach is integrating reinforcement learning (RL) and causal inference. An RL agent dynamically selects augmentation actions based on detector performance to efficiently explore the vast augmentation space, adapting to increasingly challenging forgeries. Simultaneously, the agent introduces action space variations to generate heterogeneous forgery patterns, guided by causal inference to mitigate spurious correlations-suppressing task-irrelevant biases and focusing on causally invariant features. This integration ensures robust generalization by decoupling synthetic augmentation patterns from the model's learned representations. Extensive experiments show our method significantly improves detector generalizability, outperforming SOTA methods across multiple cross-domain datasets.

cs.CV↗

LineBreaker: Finding Token-Inconsistency Bugs with Large Language Models

Token-inconsistency bugs (TIBs) involve the misuse of syntactically valid yet incorrect code tokens, such as misused variables and erroneous function invocations, which can often lead to software bugs. Unlike simple syntactic bugs, TIBs occur at the semantic level and are subtle - sometimes they remain undetected for years. Traditional detection methods, such as static analysis and dynamic testing, often struggle with TIBs due to their versatile and context-dependent nature. However, advancements in large language models (LLMs) like GPT-4 present new opportunities for automating TIB detection by leveraging these models' semantic understanding capabilities. This paper reports the first systematic measurement of LLMs' capabilities in detecting TIBs, revealing that while GPT-4 shows promise, it exhibits limitations in precision and scalability. Specifically, its detection capability is undermined by the model's tendency to focus on the code snippets that do not contain TIBs; its scalability concern arises from GPT-4's high cost and the massive amount of code requiring inspection. To address these challenges, we introduce \name, a novel and cascaded TIB detection system. \name leverages smaller, code-specific, and highly efficient language models to filter out large numbers of code snippets unlikely to contain TIBs, thereby significantly enhancing the system's performance in terms of precision, recall, and scalability. We evaluated \name on 154 Python and C GitHub repositories, each with over 1,000 stars, uncovering 123 new flaws, 45\% of which could be exploited to disrupt program functionalities. Out of our 69 submitted fixes, 41 have already been confirmed or merged.

cs.CR↗

Improving Sampling Efficiency in RLVR through Adaptive Rollout and Response Reuse

Large language models (LLMs) have achieved impressive reasoning performance, with reinforcement learning with verifiable rewards (RLVR) emerging as a standard paradigm for post-training. A representative algorithm, group relative policy optimization (GRPO) (Shao et al., 2024), computes advantages by normalizing outcome rewards within response groups, but suffers from a vanishing advantage issue when all responses in a group receive identical rewards. To address this issue, we propose Adaptive Rollout and Response Reuse Policy Optimization (AR3PO), a sampling efficient RLVR algorithm that introduces two novel techniques: adaptive rollout, which dynamically allocates more responses to difficult prompts while saving computation on easier ones, and response reuse, which leverages previously generated correct responses to provide useful training signals. We compare AR3PO with strong RLVR baselines on multiple representative benchmarks using two different families of base models. Across the 7B and 8B models, AR3PO consistently outperforms GRPO and matches or surpasses DAPO (Yu et al., 2025), reducing rollout cost by up to 4.2x. On the larger 32B model, AR3PO achieves comparable performance to DAPO at similar training steps while maintaining substantially lower rollout cost.

cs.LG↗

Wafer-Scale Squeezed-Light Chips

Squeezed-light generation in photonic integrated circuits (PICs) is essential for scalable continuous-variable (CV) quantum information processing. By suppressing quantum fluctuations below the shot-noise limit, squeezed states enable quantum-enhanced sensing and serve as a standard resource for CV quantum information processing. While chip-level squeezed-light sources have been demonstrated, extending this capability to the wafer level with reproducible strong squeezing to bolster large-scale quantum-enhanced sensing and information processing has been hindered by squeezed light's extreme susceptibility to device imperfections. Here, we report wafer-scale fabrication, generation, and characterization of two-mode squeezed-vacuum states on a fully complementary metal-oxide-semiconductor (CMOS)-compatible silicon nitride (Si$_3$N$_4$) PIC platform. Across a 4-inch wafer, 8 dies yield 2.9-3.1 dB directly measured quadrature squeezing with $< 0.2$ dB variation, demonstrating excellent uniformity. This performance is enabled by co-integrating ultralow-loss, strongly overcoupled high-$Q$ microresonators, cascaded pump-rejection filters, and low-loss inverse-tapered edge couplers. The measurements agree with a first-principles theoretical model parameterized solely by independently extracted device parameters and experimental settings. The measured squeezing level can be further improved by enhancing the efficiencies of off-chip detection and chip-to-fiber coupling. These results establish a reproducible, wafer-scale route to nonclassical-light generation in integrated photonics and lay the groundwork for scalable CV processors, multiplexed entanglement sources, and quantum-enhanced sensing.

quant-ph↗

Identifying and Calibrating Overconfidence in Noisy Speech Recognition

Modern end-to-end automatic speech recognition (ASR) models like Whisper not only suffer from reduced recognition accuracy in noise, but also exhibit overconfidence - assigning high confidence to wrong predictions. We conduct a systematic analysis of Whisper's behavior in additive noise conditions and find that overconfident errors increase dramatically at low signal-to-noise ratios, with 10-20% of tokens incorrectly predicted with confidence above 0.7. To mitigate this, we propose a lightweight, post-hoc calibration framework that detects potential overconfidence and applies temperature scaling selectively to those tokens, without altering the underlying ASR model. Evaluations on the R-SPIN dataset demonstrate that, in the low signal-to-noise ratio range (-18 to -5 dB), our method reduces the expected calibration error (ECE) by 58% and triples the normalized cross entropy (NCE), yielding more reliable confidence estimates under severe noise conditions.

eess.AS↗

Near-Duplicate Text Alignment under Weighted Jaccard Similarity

Near-duplicate text alignment is the task of identifying, among the texts in a corpus, all the subsequences (substrings) that are similar to a given query. Traditional approaches rely on seeding-extension-filtering heuristics, which lack accuracy guarantees and require many hard-to-tune parameters. Recent methods leverage min-hash techniques under a hash-based framework: group subsequences by their min-hash, and for any query, find all sketches similar to the query's sketch. These methods guarantee to report all subsequences whose estimated unweighted Jaccard similarity with the query exceeds a user-provided threshold and are efficient. However, they fail to account for token importance or frequency, which limits their use in real scenarios where tokens carry weights, such as TF-IDF. To address this, we propose MONO, an approach that supports weighted Jaccard similarity using consistent weighted sampling. MONO achieves optimality within the hash-based framework. For example, when token weights are proportional to frequencies, MONO generates O(n + n log f) groups in expectation for a text of length n, where f is the maximum token frequency. Each group takes O(1) space and represents a few subsequences sharing the same sampling. We prove this bound is tight: any algorithm must produce Omega(n + n log f) groups in expectation in the worst case. Experiments show that MONO outperforms the state of the art by up to 26x in index construction time, reduces index size by up to 30 percent, and improves query latency by up to 3x, while scaling well.

cs.DB↗

Hybrid Scandium Aluminum Nitride/Silicon Nitride Integrated Photonic Circuits

Scandium-doped aluminum nitride has recently emerged as a promising material for quantum photonic integrated circuits (PICs) due to its unique combination of strong second-order nonlinearity, ferroelectricity, piezoelectricity, and complementary metal-oxide-semiconductor (CMOS) compatibility. However, the relatively high optical loss reported to date-typically above 2.4 dB/cm-remains a key challenge that limits its widespread application in low-loss PICs. Here, we present a monolithically integrated $\mathrm{Si}_3\mathrm{N}_4$-ScAlN waveguide platform that overcomes this limitation. By confining light within an etched $\mathrm{Si}_3\mathrm{N}_4$ waveguide while preserving the functional properties of the underlying ScAlN layer, we achieve an intrinsic quality factor of $Q_{\mathrm{i}} = 3.35 \times 10^5$, corresponding to a propagation loss of 1.03 dB/cm-comparable to that of commercial single-mode silicon-on-insulator (SOI) waveguides. This hybrid architecture enables low-loss and scalable fabrication while retaining the advanced functionalities offered by ScAlN, such as ferroelectricity and piezoelectricity. Our results establish a new pathway for ScAlN-based PICs with potential applications in high-speed optical communication, modulation, sensing, nonlinear optics, and quantum optics within CMOS-compatible platforms.

physics.optics↗

A Glimpse of Satellite Galaxies in the Milky Way with the 2.5-meter Wide Field Survey Telescope (WFST): Bootes III and Draco

We carry out deep imaging of the Milky Way satellite galaxies, Bootes III and Draco, with WFST as one pilot observing program to demonstrate the capability of WFST. Combining catalogs with PS1 DR2 and Gaia DR3, we derive proper motions for candidate member stars in these two satellite galaxies over a 12-year time baseline, yielding uncertainties of ~1.8 mas/yr at 21 mag and ~3.0 mas/yr at 22 mag in the r band. The proper motions derived from bright and faint stars are consistent, indicating no significant variation in proper motion across stellar luminosity as these galaxies undergo tidal interactions with the MW. Meanwhile, we suggest that Bootes III represents the bound remnant of the progenitor galaxy that gave rise to the Styx stream, as evidenced by its elongated density profile and overdensity in both spatial and kinematic space. This is the first paper to use WFST to measure the proper motions of faint stars in Milky Way satellite galaxies. More detailed analyses will be presented in forthcoming papers from the wide field survey (WFS) program.

astro-ph.GA↗

Iterative Nash Policy Optimization: Aligning LLMs with General Preferences via No-Regret Learning

Reinforcement Learning with Human Feedback (RLHF) has achieved great success in aligning large language models (LLMs) with human preferences. Prevalent RLHF approaches are reward-based, following the Bradley-Terry (BT) model assumption, which may not fully capture the complexity of human preferences. In this paper, we explore RLHF under a general preference framework and approach it from a game-theoretic perspective. Specifically, we formulate the problem as a two-player game and propose a novel online algorithm, iterative Nash policy optimization (INPO). The key idea is to let the policy play against itself via no-regret learning, thereby approximating the Nash policy. Unlike previous methods, INPO bypasses the need for estimating the expected win rate for individual responses, which typically incurs high computational or annotation costs. Instead, we introduce a new loss objective that is directly minimized over a preference dataset. We provide theoretical analysis for our approach and demonstrate its effectiveness through experiments on various representative benchmarks. With an LLaMA-3-8B-based SFT model, INPO achieves a 42.6% length-controlled win rate on AlpacaEval 2.0 and a 37.8% win rate on Arena-Hard, showing substantial improvement over the state-of-the-art online RLHF algorithms.

cs.LG↗

Statistical Tractability of Off-policy Evaluation of History-dependent Policies in POMDPs

We investigate off-policy evaluation (OPE), a central and fundamental problem in reinforcement learning (RL), in the challenging setting of Partially Observable Markov Decision Processes (POMDPs) with large observation spaces. Recent works of Uehara et al. (2023a); Zhang & Jiang (2024) developed a model-free framework and identified important coverage assumptions (called belief and outcome coverage) that enable accurate OPE of memoryless policies with polynomial sample complexities, but handling more general target policies that depend on the entire observable history remained an open problem. In this work, we prove information-theoretic hardness for model-free OPE of history-dependent policies in several settings, characterized by additional assumptions imposed on the behavior policy (memoryless vs. history-dependent) and/or the state-revealing property of the POMDP (single-step vs. multi-step revealing). We further show that some hardness can be circumvented by a natural model-based algorithm -- whose analysis has surprisingly eluded the literature despite the algorithm's simplicity -- demonstrating provable separation between model-free and model-based OPE in POMDPs.

cs.LG↗

Improving LLM General Preference Alignment via Optimistic Online Mirror Descent

Reinforcement learning from human feedback (RLHF) has demonstrated remarkable effectiveness in aligning large language models (LLMs) with human preferences. Many existing alignment approaches rely on the Bradley-Terry (BT) model assumption, which assumes the existence of a ground-truth reward for each prompt-response pair. However, this assumption can be overly restrictive when modeling complex human preferences. In this paper, we drop the BT model assumption and study LLM alignment under general preferences, formulated as a two-player game. Drawing on theoretical insights from learning in games, we integrate optimistic online mirror descent into our alignment framework to approximate the Nash policy. Theoretically, we demonstrate that our approach achieves an $O(T^{-1})$ bound on the duality gap, improving upon the previous $O(T^{-1/2})$ result. More importantly, we implement our method and show through experiments that it outperforms state-of-the-art RLHF algorithms across multiple representative benchmarks.

cs.LG↗

LCSim: A Large-Scale Controllable Traffic Simulator

With the rapid growth of urban transportation and the continuous progress in autonomous driving, a demand for robust benchmarking autonomous driving algorithms has emerged, calling for accurate modeling of large-scale urban traffic scenarios with diverse vehicle driving styles. Traditional traffic simulators, such as SUMO, often depend on hand-crafted scenarios and rule-based models, where vehicle actions are limited to speed adjustment and lane changes, making it difficult for them to create realistic traffic environments. In recent years, real-world traffic scenario datasets have been developed alongside advancements in autonomous driving, facilitating the rise of data-driven simulators and learning-based simulation methods. However, current data-driven simulators are often restricted to replicating the traffic scenarios and driving styles within the datasets they rely on, limiting their ability to model multi-style driving behaviors observed in the real world. We propose \textit{LCSim}, a large-scale controllable traffic simulator. First, we define a unified data format for traffic scenarios and provide tools to construct them from multiple data sources, enabling large-scale traffic simulation. Furthermore, we integrate a diffusion-based vehicle motion planner into LCSim to facilitate realistic and diverse vehicle modeling. Under specific guidance, this allows for the creation of traffic scenarios that reflect various driving styles. Leveraging these features, LCSim can provide large-scale, realistic, and controllable virtual traffic environments. Codes and demos are available at https://tsinghua-fib-lab.github.io/LCSim.

cs.RO↗

Noise Matters: Diffusion Model-based Urban Mobility Generation with Collaborative Noise Priors

With global urbanization, the focus on sustainable cities has largely grown, driving research into equity, resilience, and urban planning, which often relies on mobility data. The rise of web-based apps and mobile devices has provided valuable user data for mobility-related research. However, real-world mobility data is costly and raises privacy concerns. To protect privacy while retaining key features of real-world movement, the demand for synthetic data has steadily increased. Recent advances in diffusion models have shown great potential for mobility trajectory generation due to their ability to model randomness and uncertainty. However, existing approaches often directly apply identically distributed (i.i.d.) noise sampling from image generation techniques, which fail to account for the spatiotemporal correlations and social interactions that shape urban mobility patterns. In this paper, we propose CoDiffMob, a diffusion model for urban mobility generation with collaborative noise priors, we emphasize the critical role of noise in diffusion models for generating mobility data. By leveraging both individual movement characteristics and population-wide dynamics, we construct novel collaborative noise priors that provide richer and more informative guidance throughout the generation process. Extensive experiments demonstrate the superiority of our method, with generated data accurately capturing both individual preferences and collective patterns, achieving an improvement of over 32%. Furthermore, it can effectively replace web-derived mobility data to better support downstream applications, while safeguarding user privacy and fostering a more secure and ethical web. This highlights its tremendous potential for applications in sustainable city-related research. The code and data are available at https://github.com/tsinghua-fib-lab/CoDiffMob.

cs.LG↗

Ultralow-loss photonic integrated chips on 8-inch anomalous-dispersion Si$_3$N$_4$-SiO$_2$-Si Wafer

We report the fabrication of 8-inch crack-free, dispersion-engineered Si$_3$N$_4$-SiO$_2$-Si wafers fully compatible with industrial foundry silicon photonics fabrication lines. By combining these wafers with a developed amorphous silicon (a-Si) hardmask etching technique, we achieve ultra-low-loss Si$_3$N$_4$ photonic integrated circuits (PICs) with intrinsic quality factors exceeding $25 \times 10^6$ using electron beam lithography and $24 \times 10^6$ using standard ultraviolet stepper photolithography. Frequency-comb generation is demonstrated on these high-quality Si$_3$N$_4$ PICs, corroborating the designed anomalous dispersion. These results establish the feasibility of mass-manufacturing high-performance, dispersion-engineered Si$_3$N$_4$ PICs using standard foundry-grade processes, opening new pathways for applications in optical communications, nonlinear optics, and quantum optics.

physics.optics↗

Teaching LLMs to Refine with Tools

Large language models (LLMs) can refine their responses based on feedback, enabling self-improvement through iterative training or test-time refinement. However, existing methods predominantly focus on refinement within the same reasoning format, which may lead to non-correcting behaviors. We propose CaP, a novel approach that uses external tools to refine chain-of-thought (CoT) responses generated by the same or other LLMs. CaP employs a two-stage training process: supervised fine-tuning followed by preference optimization with DPO variants. Our observations highlight the critical role of preference optimization in enabling effective refinement. Additionally, we compare several sampling strategies to leverage CoT and tools at inference time. Experimental results demonstrate CaP's potential for effective cross-reasoning refinement and efficient inference.

cs.CL↗

A first measurement of galaxy merger rate increasing in dynamically colder protoclusters at cosmic noon

The process of galaxy cluster formation likely leaves an imprint on the properties of its individual member galaxies. Understanding this process is essential for uncovering the evolutionary connections between galaxies and cosmic structures. Here we study a sample of ten protoclusters at z~2-3 in different dynamical states that we estimate based on spectroscopic data of their members. We combine the dynamical information with HST imaging to measure galaxy sizes and pair fractions. Our analysis reveals a clear anti-correlation between the velocity dispersion of the protocluster and its galaxy pair fractions (indicative of merger rates). The velocity dispersion also anti-correlates with the dispersion in size among of the member galaxies. These correlations may be explained by protoclusters in colder dynamical states maintaining a velocity dispersion and galaxy number density that boosts galaxy mergers, which in turn contributes to the structural expansion and compaction of galaxies. Our findings offer constraints for cosmological models regarding the evolution of galaxy morphology across different stages in the assembly of protoclusters.

astro-ph.GA↗

Revealing the quiescent galaxy population in the Spiderweb protocluster at z=2.16 with deep HST/WFC3 slitless spectroscopy

We report the HST WFC3 G141 grism slitless spectroscopy observation of the core region of the Spiderweb protocluster at $z=2.16$. We analyzed the spectra of all objects in a $\sim 2 \times 2 \text{ arcmin}^2$ field of view and identified 40 protocluster members, recovering 19 previously identified H$α$-emitters in addition to revealing 21 new members. The spectra allowed us to identify 11 galaxies with quiescent spectra. Three galaxies with quiescent spectra are possibly still star-forming according to SED fitting, indicating a possible left-over or dust-obscured star formation. We estimate a quiescent fraction of $\sim 50\%$ for $M_\star > 10^{11} M_\odot$. About half of the quiescent galaxies possibly host AGN, hinting at AGN's key role in quenching galaxies in the protocluster environment. These quiescent galaxies have relatively more compact and concentrated light profiles than the star-forming members, but they are not yet as bulge-dominated as local ellipticals. These results are consistent with previous studies that indicate the Spiderweb protocluster is in the maturing stage, with a red sequence that has begun forming.

astro-ph.GA↗

Online Iterative Reinforcement Learning from Human Feedback with General Preference Model

We investigate Reinforcement Learning from Human Feedback (RLHF) in the context of a general preference oracle. In particular, we do not assume the existence of a reward function and an oracle preference signal drawn from the Bradley-Terry model as most of the prior works do. We consider a standard mathematical formulation, the reverse-KL regularized minimax game between two LLMs for RLHF under general preference oracle. The learning objective of this formulation is to find a policy so that it is consistently preferred by the KL-regularized preference oracle over any competing LLMs. We show that this framework is strictly more general than the reward-based one, and propose sample-efficient algorithms for both the offline learning from a pre-collected preference dataset and online learning where we can query the preference oracle along the way of training. Empirical studies verify the effectiveness of the proposed framework.

cs.LG↗