5 papers
MemWM: Memory-Augmented Text-Based World Model
Yujun Wang, Tao Zhang, Jinhe Bi +9
World models are increasingly used to support planning in agents by predicting how environment states evolve in response to agent actions. Yet fluent next-state predictions can sti…
Representation Interventions Enable Lifelong Knowledge Memory Control in LLMs
Xuyuan Liu, Shengyu Chen, Xinshuai Dong +6
Large language models (LLMs) often produce incorrect or outdated content after being employed. Efficient and accurate knowledge updates without costly retraining are a major challe…
One-Step Bellman Alignment Enables Provably Efficient Transfer in Online RL
Elynn Chen, Enpei Zhang, Jinhang Chai +1
We study online transfer reinforcement learning (RL) in episodic Markov decision processes, where experience from related source tasks is available during learning on a target task…
HalluGuard: Demystifying Data-Driven and Reasoning-Driven Hallucinations in LLMs
Xinyue Zeng, Junhong Lin, Yujun Yan +4
The reliability of Large Language Models (LLMs) in high-stakes domains such as healthcare, law, and scientific discovery is often compromised by hallucinations. These failures typi…
Low-Rank Plus Sparse Matrix Transfer Learning under Growing Representations and Ambient Dimensions
Jinhang Chai, Xuyuan Liu, Elynn Chen +1
Learning systems often expand their ambient features or latent representations over time, embedding earlier representations into larger spaces with limited new latent structure. We…