EXPLORE THE ARCHIVE

Rui Wu

Publications and source records attributed to Rui Wu.

3 recordsLinked to original sources

arXiv2026

Dense Process Supervision for Search Agents via Fact Utility Estimation

Reinforcement learning (RL) for search agents typically relies on outcome rewards. However, it often fails to achieve effective credit assignment, due to the unclear value of intermediate steps. It is hard to separate their contributions from the final result. In this paper, we propose a dense process supervision method based on fact utility estimation, which models the reasoning process as the accumulation of discrete evidence facts. We first extract structured facts from raw observations and organize them into an explicit fact store. To support credit assignment, we then cluster semantically equivalent facts and infer the posterior utility of each fact cluster using Bayesian estimation over group rollouts. Finally, we convert the estimated fact utilities into dense step-level rewards to guide RL training. Experiments on seven single-hop and multi-hop QA benchmarks show that our method consistently outperforms existing baselines. Ablation studies validate clear relative improvements on multi-hop QA compared to outcome reward-only training.

cs.CL↗

arXiv2026

Learning Generated Controls under Fractured Geometry: Projective Residualization and Variation-Allocation Frontiers

Many two-stage estimators assess the first-stage learner by prediction error, even when the next stage uses its residual. In control-function instrumental variables, that residual must preserve the latent control direction without removing the treatment variation that identifies the structural response. A scalar prediction score does not reveal how the learner allocates this variation. Under piecewise-smooth graph geometry, interpolation can suppress the control, whereas isotropic smoothing can leak systematic variation across boundaries. We formulate this as a variation-allocation problem and introduce Adaptive Anisotropic Instrumental Heat Flow (A-IHF). The method uses pilot treatment contrasts to adapt edge conductance, takes the complement of a sparse graph resolvent as the generated control, and selects candidates without consulting outcomes. For a linear control-function regression, the generated control is identified only by its span. Working in that projective geometry, we derive an exact finite-sample fidelity--relevance frontier, spectral identities for remaining treatment variation and coefficient distortion, and a lower bound for monotone fixed-graph residual filters. A connected construction proves that adapting conductance can remove the corresponding fixed-graph obstruction. In a 54-cell benchmark, the A-IHF family wins 32 cells; its guarded observational variant lowers mean nonlinear response error by 8.3%, with the largest gains in fractured designs. Controlled rewiring explains when the graph should be used, replaced by a fallback, or rejected. The resulting lesson is task-specific: a first stage for generated controls should be judged by control fidelity, downstream relevance, and graph compatibility together.

cs.LG↗

arXiv2026

Support Selection Beyond Smooth DAG Exactness: Completion Geometry,Score Margins, and Selective Certificates

Smooth acyclicity constraints answer whether a weighted support is a DAG, whereas structure learning asks which support change should be made. Existing analyses establish degeneracy for particular constraint formulas but do not isolate what follows from smooth exactness itself. At a DAG boundary, we show that minimal cycle completions generate a squarefree monomial ideal containing every restricted Taylor jet of an exact representation. If the smallest completion has $q$ edges, the first possible response has order $q$ for a vector residual and $2q$ for a nonnegative scalar. Exponentially many constant-scale cyclic manifolds exhibit the same lack of ranking away from the boundary for NOTEARS and DAGMA. We derive the exact selection time for an isolated cycle. When $Ψ'(h)\asymp h^ν$, the feasibility-only time is $T_0(\varepsilon)=Θ(\varepsilon^{-(2ν+1)})$; a score margin changes the leading dynamics at scale $T_0^{-1}$ for $ν>0$, while $ν=0$ has a logarithmic boundary layer requiring $γT_0\log(1/\varepsilon)\to0$. Experiments verify this law, and a truth-free separation statistic predicts selection time on 320 official NOTEARS/DAGMA trajectories (Spearman $-0.52$ and $-0.66$, permutation $p<10^{-4}$). For finite samples, a parent-set confidence family and forced-opposite queries certify skeleton and unshielded-collider labels shared by every population optimum of a frozen score. Across 320 runs, every regret bound covers an independent oracle-score audit. None of 3,042 certified skeleton or 2,396 collider labels disagrees with the oracle-score optimum, although 4.4% and 5.5%, respectively, disagree with the generating graph. These results separate DAG feasibility, score-based support selection, and causal identification.

cs.LG↗