works on

From the 1 of 13 linked papers with an AI index.

collaborators
Showing cs.LGShow all

9 papers · 1 filter

cs.LG2026

Momentum as Residual-Driven Multiplier Correction for Deep Learning Optimization

Zhixin Ren, Yau Lyu, Congrong Li +2

Momentum-based optimizers are widely used in modern deep learning, yet the relations among momentum recursion, update geometry, and acceleration remain only partially understood. W…

cs.LG2026

On the Identifiability of Controlled World Models

Xiangteng Zhang, Yang Guan, Bo Zhang +3

World model serves as a promising tool to infer environment dynamics under high-dimensional observations and candidate actions. Recently, LeCun's JEPA provides a compelling framewo…

cs.LG2026

Distributional Soft Bellman Operator under the Cramér Geometry

Keru Wang, Yixin Deng, Yao Lyu +2

Distributional soft policy iteration (DSPI) provides an important framework for combining distributional reinforcement learning (DRL) with maximum-entropy control, in which the pol…

cs.LG2026

FAST: A Framework for Aligned Sampling and Training in Parallel Reinforcement Learning for Autonomous Driving

Bonan Wang, Letian Tao, Bin Shuai +7

The paper introduces FAST, a synchronous parallel framework that improves sampling efficiency for deep reinforcement learning in autonomous driving by aligning parallel simulations…

cs.LG2026

Augmented Lagrangian Multiplier Network for State-wise Safety in Reinforcement Learning

Jiaming Zhang, Yujie Yang, Yao Lyu +2

Safety is a primary challenge in real-world reinforcement learning (RL). Formulating safety requirements as state-wise constraints has become a prominent paradigm. Handling state-w…

cs.LG2026

A Spectral Revisit of the Distributional Bellman Operator under the Cramér Metric

Keru Wang, Yixin Deng, Yao Lyu +2

Distributional reinforcement learning (DRL) studies the evolution of full return distributions under Bellman updates rather than focusing on expected values. A classical result is…