1 citations · 2 across the 12 of their papers we have counts for
10 papers · 1 filter
Traj-LeWM: Path-Aware World-Model Planning via Latent Trajectory Cost
Xiaodi Huang, Ziyi Ding, Jingtian Wan +6
LeWM is a lightweight visual world model that learns latent dynamics end-to-end from pixels and ranks candidate action sequences by the distance between their predicted endpoints a…
Offline Reinforcement Learning for Rotation Profile Control in Tokamaks
Rohit Sonker, Hiro Josep Farre Kaga, Jiayu Chen +5
Tokamaks remain leading candidates for achieving practical fusion energy, yet many important control problems inside these devices are still difficult or unsolved. One such challen…
Future Forcing: Future-aware Training-free KV Cache Policy for Autoregressive Video Generation
Jiayi Luo, Qiyan Liu, Tengyang Wang +8
Autoregressive (AR) video generation has emerged as a promising paradigm for long-horizon video synthesis, where each frame is generated conditioned on previously generated tokens.…
Offline Reinforcement Learning for Plasma Control in Nuclear Fusion: Codebase and Benchmark
Yang Fu, Haomin Bao, Rohit Sonker +4
Offline reinforcement learning (RL) offers a promising route for developing plasma controllers from historical tokamak data, since online trial-and-error on real devices is costly…
Core-Halo Decomposition: Decentralizing Large-Scale Fixed-Point Problems
Haixiang, Yang Xu, Jiefu Zhang +4
We study solving large-scale fixed-point equation \(x^\star=\bar F(x^\star)\) with decomposition. Standard strict decomposition assigns each agent a disjoint block and evaluates up…
ANO: Robust Policy Optimization via Bounded, Redescending Gain Fields
Yiheng Zhang, Yiming Wang, Kaiyan Zhao +3
Proximal Policy Optimization (PPO) dominates reinforcement learning and LLM alignment, yet its hard-clipping mechanism and unconstrained alternatives (e.g., SPO) sit at two extreme…
Occupancy Reward Shaping: Improving Credit Assignment for Offline Goal-Conditioned Reinforcement Learning
Aravind Venugopal, Jiayu Chen, Xudong Wu +3
The temporal lag between actions and their long-term consequences makes credit assignment a challenge when learning goal-directed behaviors from data. Generative world models captu…
CausalVAE as a Plug-in for World Models: Towards Reliable Counterfactual Dynamics
Ziyi Ding, Xianxin Lai, Weiyu Chen +2
In this work, CausalVAE is introduced as a plug-in structural module for latent world models and is attached to diverse encoder-transition backbones. Across the reported benchmarks…
Retrospective In-Context Learning for Temporal Credit Assignment with Large Language Models
Wen-Tse Chen, Jiayu Chen, Fahim Tajwar +4
Learning from self-sampled data and sparse environmental feedback remains a fundamental challenge in training self-evolving agents. Temporal credit assignment mitigates this issue…
Continual Policy Consolidation for Lifelong Robot Learning
Qijun He, Yuxuan Li, Mingqi Yuan +4
Building a generalist robot policy requires continuously integrating new skills while preserving previously acquired behaviors. Directly optimizing a single policy over a growing t…