CORTEXA
← Browse
arxivcs.LGstat.ML2026-07-15

DAGR: State-Conditioned Goal Representations via Difference-Aware Goal Cross-Attention

Xing Lei, Wenyan Yang, Xuetao Zhang, Donglin Wang

Goal-conditioned reinforcement learning hinges on how the goal is encoded. Contrastive, metric, temporal-distance, and information-theoretic encoders differ in objective. They still share one trait. None of them sees the current state. Such a state-independent embedding cannot mark which part of the goal still needs action. The policy must then recover that cue by inverting both encoders. We propose DAGR. It refines the static embedding of any late-fusion encoder into a state-conditioned one through multi-scale gated cross-attention. A near-identity gated residual preserves the base representation. Difference-aware Goal Cross-Attention then biases the attention scores using a per-token state-goal discrepancy map. On OGBench, DAGR improves navigation. Our ablations trace the gain to the gated residual, not to the difference bias that names the method. On manipulation and puzzle tasks it matches or falls below the base. DAGR is a structured refinement, not a universal improvement.

View free PDFSource page

Related papers

arxivcs.LGstat.ML2026-07-09

AutoAnchor: Stable Diffusion Unlearning Using Cross-Attention as a Manifold Surrogate

Siyuan Wen, Jiahao Zeng, Ningning Ding

Diffusion unlearning is essential for mitigating the generation of harmful or copyrighted content in text-to-image models. Current diffusion unlearning techniques determine the model update direction by either using alternatives of the target concept as an anchor or using empty p…

View free PDFSource page
arxivcs.LGstat.ML2026-07-13

Bet on Features: Anytime-Valid and Feature-Aware Auditing of Conditional Quantile Forecasters

Ivane Antonov, Sohom Mukherjee, Richard Pibernik, Yo Joong Choe

Black-box conditional quantile forecasts are widely used for sequential decisions under asymmetric costs, such as inventory planning in supply chain management. Once deployed, such forecasters must be monitored continuously as data streams drift and regimes change; this invalidat…

View free PDFSource page
arxivstat.MLcs.LGmath.NAstat.ME2026-07-17

Cluster-Aware Matching via Laplacian Optimal Transport

Gabriel Samberg, YoonHaeng Hur, Yuehaw Khoo, Nir Sharon

In many applications of matching, the point clouds to be matched are not merely unstructured sets of points but rather samples from distributions with an intrinsic cluster structure. In such cases, as individual points are often interchangeable within a coherent region, finding a…

View free PDFSource page
arxivcs.LGstat.ML2026-07-06

Geometry-Aware Bayesian Quantification via Compositional Data Analysis

Alejandro Moreo, Pablo González, Juan José del Coz

Accurately estimating the unknown target label distribution is the critical first step for adapting to label shift. This task, widely known as quantification or class prevalence estimation, has recently seen significant advances through continuous KDE-based methods which model th…

View free PDFSource page
arxivstat.MLcs.LG2026-07-18

Semi-Supervised Conditional Generative Learning through Stochastic Interpolation and Sufficient Representations

Changyu Liu, Yuling Jiao, Jian Huang

Conditional generative modeling remains a challenging problem in semi-supervised settings where labeled data is scarce but unlabeled samples are abundant. To effectively leverage structural information embedded within the unlabeled dataset and compensate for sparse conditioning s…

View free PDFSource page