Search arXiv⌕ Search

arXiv subjects

Yifan Wei

Publications and source records attributed to Yifan Wei.

At least 19 recordsLinked to original sources

DoPR: Reusable Compressed Document Prefixes for Efficient LLM Reranking

Large language models (LLMs) are effective rerankers, but pointwise reranking repeatedly processes the same document across different queries, causing substantial redundant document-side computation. We propose \textbf{DoPR}, a compressed document prefix framework that decouples offline document processing from online reranking. DoPR first selects query-independent document representations and converts them into compressed document prefix states, which are precomputed offline and reused whenever the document is retrieved. During online reranking, the model scores each query-document pair by processing only the query and scoring token, with document information supplied by the stored prefix states. This design reduces online cost through both document-side compression and cross-query prefix-state reuse. Experiments on TREC DL, BEIR, and BRIGHT with Qwen3 models from $0.6$B to $8$B show that DoPR achieves up to 8.0$\times$ online document-side memory reduction and up to 8.04$\times$ latency speedup, while retaining \textbf{97.1\%-99.5\%} of the average NDCG@10 of matched full-document rerankers.

cs.IR↗

An SPH model with physically prescribed parameters for droplet dynamics on complex surfaces

Numerical simulation of droplet dynamics on complex surfaces with varying wettability is of great significance to both engineering applications and fundamental research. However, existing numerical methods still face challenges in accurately capturing interfacial interactions while preserving physical consistency and computational efficiency. In this work, a physically grounded and efficient smoothed particle hydrodynamics (SPH) model is developed for droplet dynamics simulation. To reduce computational cost, a single-phase droplet modeling strategy is employed. At the interface, long-range interactions are approximated using the SPH kernel function, whereas short-range interactions are represented through pressure. Based on this treatment, an explicit relationship between the intermolecular potential energy and the macroscopic surface tension coefficient is further established, thereby reducing reliance on empirical parameter calibration. The proposed method is first validated through static wetting simulations, where the equilibrium contact angles agree well with the Young--Dupré equation. Further simulations of wetting and droplet impact demonstrate that the method is capable of capturing complex dynamic wetting behaviors.

physics.flu-dyn↗

Matrix Points on Varieties

We study the cohomology of $C_n(X)$, the moduli space of commuting $n$-by-$n$ matrices satisfying the equations defining a quasi-projective scheme $X$. This space can be viewed as a non-commutative Weil restriction from the algebra of $n$-by-$n$ matrices to the ground field. We introduce a semi-simple counterpart $S_n(X)$, defined as the quotient of $X^n \times \mathrm{GL}_n/\mathrm{T}_n$ by the diagonal $S_n$ action. We show that there exists a natural map $σ\colon S_n(X) \to C_n(X)$ inducing isomorphism on $\ell$-adic cohomology under mild restrictions on $X$ or the characteristic of the field. This confirms a heuristic derived from Weil restrictions. Furthermore, we provide explicit combinatorial formulae for the Betti numbers of $C_n(X)$ and prove a Macdonald-type generating series. A version for Hermitian matrix point is also proved in the last section.

math.AG↗

Beyond Self-Play: Hierarchical Reasoning for Continuous Motion in Closed-Loop Traffic Simulation

Closed-loop traffic simulation requires agents that are both scalable and behaviorally realistic. Recent self-play reinforcement learning approaches demonstrate strong scalability, but their equilibrium strategies fail to capture the socially aware behaviors of real human drivers. We propose a hierarchical architecture that goes beyond self-play by combining high-level multi-agent interaction reasoning with low-level continuous trajectory realization. Specifically, a Stackelberg-style Multi-Agent Reinforcement Learning (MARL) module generates interaction-aware intention commands. These commands condition a low-level continuous motion module, translating the strategic intent into physically consistent, scene-responsive control sequences. To mitigate distribution shift in closed-loop deployment, we introduce a hybrid co-training scheme combining MARL with auxiliary recovery supervision. Experiments on a SUMO-based urban network demonstrate that the proposed framework achieves superior control smoothness and safety compared to self-play and passive imitation baselines, while maintaining competitive traffic efficiency.

cs.RO↗

DyBBT: Dynamic Balance via Bandit-inspired Targeting for Dialog Policy with Cognitive Dual-Systems

Task oriented dialog systems often rely on static exploration strategies that do not adapt to dynamic dialog contexts, leading to inefficient exploration and suboptimal performance. We propose DyBBT, a novel dialog policy learning framework that formalizes the exploration challenge through a structured cognitive state space capturing dialog progression, user uncertainty, and slot dependency. DyBBT proposes a bandit inspired meta-controller that dynamically switches between a fast intuitive inference (System 1) and a slow deliberative reasoner (System 2) based on real-time cognitive states and visitation counts. Extensive experiments on single- and multi-domain benchmarks show that DyBBT achieves state-of-the-art performance in success rate, efficiency, and generalization, with human evaluations confirming its decisions are well aligned with expert judgment.

cs.CL↗

HiCoLoRA: Addressing Context-Prompt Misalignment via Hierarchical Collaborative LoRA for Zero-Shot DST

Zero-shot Dialog State Tracking (zs-DST) is essential for enabling Task-Oriented Dialog Systems (TODs) to generalize to new domains without costly data annotation. A central challenge lies in the semantic misalignment between dynamic dialog contexts and static prompts, leading to inflexible cross-layer coordination, domain interference, and catastrophic forgetting. To tackle this, we propose Hierarchical Collaborative Low-Rank Adaptation (HiCoLoRA), a framework that enhances zero-shot slot inference through robust prompt alignment. It features a hierarchical LoRA architecture for dynamic layer-specific processing (combining lower-layer heuristic grouping and higher-layer full interaction), integrates Spectral Joint Domain-Slot Clustering to identify transferable associations (feeding an Adaptive Linear Fusion Mechanism), and employs Semantic-Enhanced SVD Initialization (SemSVD-Init) to preserve pre-trained knowledge. Experiments on multi-domain datasets MultiWOZ and SGD show that HiCoLoRA outperforms baselines, achieving SOTA in zs-DST. Code is available at https://github.com/carsonz/HiCoLoRA.

cs.CL↗

AutoFigure: Generating and Refining Publication-Ready Scientific Illustrations

High-quality scientific illustrations are crucial for effectively communicating complex scientific and technical concepts, yet their manual creation remains a well-recognized bottleneck in both academia and industry. We present FigureBench, the first large-scale benchmark for generating scientific illustrations from long-form scientific texts. It contains 3,300 high-quality scientific text-figure pairs, covering diverse text-to-illustration tasks from scientific papers, surveys, blogs, and textbooks. Moreover, we propose AutoFigure, the first agentic framework that automatically generates high-quality scientific illustrations based on long-form scientific text. Specifically, before rendering the final result, AutoFigure engages in extensive thinking, recombination, and validation to produce a layout that is both structurally sound and aesthetically refined, outputting a scientific illustration that achieves both structural completeness and aesthetic appeal. Leveraging the high-quality data from FigureBench, we conduct extensive experiments to test the performance of AutoFigure against various baseline methods. The results demonstrate that AutoFigure consistently surpasses all baseline methods, producing publication-ready scientific illustrations. The code, dataset and huggingface space are released in https://github.com/ResearAI/AutoFigure.

cs.AI↗

Convergence analysis of SPH method on irregular particle distributions for the Poisson equation

The numerical accuracy of particle-based approximations in Smoothed Particle Hydrodynamics (SPH) is significantly affected by the spatial uniformity of particle distributions, especially for second-order derivatives. This study aims to enhance the accuracy of SPH method and analyze its convergence with irregular particle distributions. By establishing regularity conditions for particle distributions, we ensure that the local truncation error of traditional SPH formulations, including first and second derivatives, achieves second-order accuracy. Our proposed method, the volume reconstruction SPH method, guarantees these regularity conditions while preserving the discrete maximum principle. Benefiting from the discrete maximum principle, we conduct a rigorous global error analysis in the $L^\infty$-norm for the Poisson equation with variable coefficients, achieving second-order convergence. Numerical examples are presented to validate the theoretical findings.

math.NA↗

Towards Compositional Generalization of LLMs via Skill Taxonomy Guided Data Synthesis

Large Language Models (LLMs) and agent-based systems often struggle with compositional generalization due to a data bottleneck in which complex skill combinations follow a long-tailed, power-law distribution, limiting both instruction-following performance and generalization in agent-centric tasks. To address this challenge, we propose STEPS, a Skill Taxonomy guided Entropy-based Post-training data Synthesis framework for generating compositionally challenging data. STEPS explicitly targets compositional generalization by uncovering latent relationships among skills and organizing them into an interpretable, hierarchical skill taxonomy using structural information theory. Building on this taxonomy, we formulate data synthesis as a constrained information maximization problem, selecting skill combinations that maximize marginal structural information within the hierarchy while preserving semantic coherence. Experiments on challenging instruction-following benchmarks show that STEPS outperforms existing data synthesis baselines, while also yielding improved compositional generalization in downstream agent-based evaluations.

cs.CL↗

An Effective Model for Droplet Impact Dynamics on Micro-Structured Surfaces: Nonlocal Theory and SPH Simulation of Pancake Bouncing

The accurate mathematical modeling of droplet impact dynamics on micro-structured surfaces is fundamental to understanding and predicting complex fluid behaviors relevant to a wide range of engineering and scientific applications. In particular, the pancake bouncing phenomenon--systematically studied by Liu et al. (Nature Physics, 2014)--on superhydrophobic micro-structured substrates presents significant theoretical challenges. Central to these challenges is the need to construct effective mathematical models that capture the intricate influence of substrate micro/nanostructures on droplet dynamics. This requires the development of robust formulations for surface tension, contact line dynamics, and the interaction forces between fluid and solid structures. In this work, we formulate a nonlocal mathematical framework for the simulation of 3D pancake bouncing on superhydrophobic micro-cone arrays. The model incorporates intermolecular attractive forces to represent droplet surface tension, and we provide a strict theoretical derivation linking these forces quantitatively to the macroscopic surface tension coefficient, thereby circumventing the reliance on empirical parameter tuning. The complex geometry of micro-cone arrays introduces fundamental difficulties in defining local normal directions for contact algorithms. To overcome this, we develop a nonlocal contact repulsion force model that governs fluid-solid interactions and ensures numerical stability under high Weber number conditions. Based on this mathematical foundation, we implement the model using smoothed particle hydrodynamics (SPH), enabling high-precision 3D simulations. Computational experiments, validated against empirical data, confirm the model's accuracy and robustness, while underscoring the key role of numerical simulation in elucidating droplet-microstructure interactions.

physics.flu-dyn↗

Stacking-orientation and twist-angle control on integer and fractional Chern insulators in moiré rhombohedral graphene

Rhombohedral-stacked multilayer graphene aligned with hexagonal boron nitride has emerged as an excellent platform for investigating exotic quantum phenomena arising from the interplay between electron correlations and nontrivial topology. However, the microscopic mechanism governing the emergence of both the integer and fractional Chern insulator states in this system remains an open question. In this work, we systematically investigate the electrical transport properties of RMG/hBN moiré devices with controlled alignment orientations and twist angles. We demonstrate that alignment orientation strongly modulates correlated phenomena in the moiré-proximal regime, while having negligible influence on the formation of integer and fractional Chern insulators in the moiré-distant regime. Instead, the moiré periodicity, tuned by the twist angle, serves as the key parameter controlling the stability of these correlated topological states in the moiré-distant regime. Furthermore, in the moiré-proximal regime of one specific alignment, we observe anomalous Hall effect and a variety of competing phases near ν = 1, including integer Chern insulator states, extended Chern insulator states, and trivial insulators, whose stability is highly sensitive to both the applied displacement electric field and magnetic field. Our results underscore the critical role of stacking-alignment and twist-angle engineering in exploring novel quantum states based on rhombohedral-stacked multilayer graphene moiré systems.

cond-mat.mes-hall↗

AutoTIR: Autonomous Tools Integrated Reasoning via Reinforcement Learning

Large Language Models (LLMs), when enhanced through reasoning-oriented post-training, evolve into powerful Large Reasoning Models (LRMs). Tool-Integrated Reasoning (TIR) further extends their capabilities by incorporating external tools, but existing methods often rely on rigid, predefined tool-use patterns that risk degrading core language competence. Inspired by the human ability to adaptively select tools, we introduce AutoTIR, a reinforcement learning framework that enables LLMs to autonomously decide whether and which tool to invoke during the reasoning process, rather than following static tool-use strategies. AutoTIR leverages a hybrid reward mechanism that jointly optimizes for task-specific answer correctness, structured output adherence, and penalization of incorrect tool usage, thereby encouraging both precise reasoning and efficient tool integration. Extensive evaluations across diverse knowledge-intensive, mathematical, and general language modeling tasks demonstrate that AutoTIR achieves superior overall performance, significantly outperforming baselines and exhibits superior generalization in tool-use behavior. These results highlight the promise of reinforcement learning in building truly generalizable and scalable TIR capabilities in LLMs. The code and data are available at https://github.com/weiyifan1023/AutoTIR.

cs.CL↗

Feature Construction Using Network Control Theory and Rank Encoding for Graph Machine Learning

In this article, we utilize the concept of average controllability in graphs, along with a novel rank encoding method, to enhance the performance of Graph Neural Networks (GNNs) in social network classification tasks. GNNs have proven highly effective in various network-based learning applications and require some form of node features to function. However, their performance is heavily influenced by the expressiveness of these features. In social networks, node features are often unavailable due to privacy constraints or the absence of inherent attributes, making it challenging for GNNs to achieve optimal performance. To address this limitation, we propose two strategies for constructing expressive node features. First, we introduce average controllability along with other centrality metrics (denoted as NCT-EFA) as node-level metrics that capture critical aspects of network topology. Building on this, we develop a rank encoding method that transforms average controllability or any other graph-theoretic metric into a fixed-dimensional feature space, thereby improving feature representation. We conduct extensive numerical evaluations using six benchmark GNN models across four social network datasets to compare different node feature construction methods. Our results demonstrate that incorporating average controllability into the feature space significantly improves GNN performance. Moreover, the proposed rank encoding method outperforms traditional one-hot degree encoding, improving the ROC AUC from 68.7% to 73.9% using GraphSAGE on the GitHub Stargazers dataset, underscoring its effectiveness in generating expressive and efficient node representations.

cs.LG↗

Robust Anomaly Detection with Graph Neural Networks using Controllability

Anomaly detection in complex domains poses significant challenges due to the need for extensive labeled data and the inherently imbalanced nature of anomalous versus benign samples. Graph-based machine learning models have emerged as a promising solution that combines attribute and relational data to uncover intricate patterns. However, the scarcity of anomalous data exacerbates the challenge, which requires innovative strategies to enhance model learning with limited information. In this paper, we hypothesize that the incorporation of the influence of the nodes, quantified through average controllability, can significantly improve the performance of anomaly detection. We propose two novel approaches to integrate average controllability into graph-based frameworks: (1) using average controllability as an edge weight and (2) encoding it as a one-hot edge attribute vector. Through rigorous evaluation on real-world and synthetic networks with six state-of-the-art baselines, our proposed methods demonstrate improved performance in identifying anomalies, highlighting the critical role of controllability measures in enhancing the performance of graph machine learning models. This work underscores the potential of integrating average controllability as additional metrics to address the challenges of anomaly detection in sparse and imbalanced datasets.

cs.LG↗

A Chebotarev Density Theorem over Local Fields

We compute the $p$-adic densities of points with a given splitting type along a (generically) finite map, analogous to the classical Chebotarev theorem over number fields and function fields. Under some mild hypotheses, we prove that these densities satisfy a functional equation in the size of the residue field. This functional equation is a direct reflection of Poincaré duality in étale cohomology. As a consequence, we prove a conjecture of Bhargava, Cremona, Fisher, and Gajović on factorization densities of p-adic polynomials. The key tool is the notion of admissible pairs associated to a group, which we use as an invariant of the inertia and decomposition action of a local field on the fibers of the finite map. We compute the splitting densities by Möbius inverting certain p-adic integrals along the poset of admissible pairs. The conjecture on factorization densities follows immediately for tamely ramified primes from our general results. We reduce the complete conjecture (including the wild primes) to the existence of an explicit "Tate-type" resolution of the "resultant locus" over the integers and complete the proof of the conjecture by constructing this resolution.

math.NT↗

Structural Entropy Guided Agent for Detecting and Repairing Knowledge Deficiencies in LLMs

Large language models (LLMs) have achieved unprecedented performance by leveraging vast pretraining corpora, yet their performance remains suboptimal in knowledge-intensive domains such as medicine and scientific research, where high factual precision is required. While synthetic data provides a promising avenue for augmenting domain knowledge, existing methods frequently generate redundant samples that do not align with the model's true knowledge gaps. To overcome this limitation, we propose a novel Structural Entropy-guided Knowledge Navigator (SENATOR) framework that addresses the intrinsic knowledge deficiencies of LLMs. Our approach employs the Structure Entropy (SE) metric to quantify uncertainty along knowledge graph paths and leverages Monte Carlo Tree Search (MCTS) to selectively explore regions where the model lacks domain-specific knowledge. Guided by these insights, the framework generates targeted synthetic data for supervised fine-tuning, enabling continuous self-improvement. Experimental results on LLaMA-3 and Qwen2 across multiple domain-specific benchmarks show that SENATOR effectively detects and repairs knowledge deficiencies, achieving notable performance improvements. The code and data for our methods and experiments are available at https://github.com/weiyifan1023/senator.

cs.CL↗

SetKE: Knowledge Editing for Knowledge Elements Overlap

Large Language Models (LLMs) excel in tasks such as retrieval and question answering but require updates to incorporate new knowledge and reduce inaccuracies and hallucinations. Traditional updating methods, like fine-tuning and incremental learning, face challenges such as overfitting and high computational costs. Knowledge Editing (KE) provides a promising alternative but often overlooks the Knowledge Element Overlap (KEO) phenomenon, where multiple triplets share common elements, leading to editing conflicts. We identify the prevalence of KEO in existing KE datasets and show its significant impact on current KE methods, causing performance degradation in handling such triplets. To address this, we propose a new formulation, Knowledge Set Editing (KSE), and introduce SetKE, a method that edits sets of triplets simultaneously. Experimental results demonstrate that SetKE outperforms existing methods in KEO scenarios on mainstream LLMs. Additionally, we introduce EditSet, a dataset containing KEO triplets, providing a comprehensive benchmark.

cs.CL↗

TeraSim: Uncovering Unknown Unsafe Events for Autonomous Vehicles through Generative Simulation

Traffic simulation is essential for autonomous vehicle (AV) development, enabling comprehensive safety evaluation across diverse driving conditions. However, traditional rule-based simulators struggle to capture complex human interactions, while data-driven approaches often fail to maintain long-term behavioral realism or generate diverse safety-critical events. To address these challenges, we propose TeraSim, an open-source, high-fidelity traffic simulation platform designed to uncover unknown unsafe events and efficiently estimate AV statistical performance metrics, such as crash rates. TeraSim is designed for seamless integration with third-party physics simulators and standalone AV stacks, to construct a complete AV simulation system. Experimental results demonstrate its effectiveness in generating diverse safety-critical events involving both static and dynamic agents, identifying hidden deficiencies in AV systems, and enabling statistical performance evaluation. These findings highlight TeraSim's potential as a practical tool for AV safety assessment, benefiting researchers, developers, and policymakers. The code is available at https://github.com/mcity/TeraSim.

cs.RO↗