activity
20212025
most citedLearning Task-relevant Representations for Generalization via Characteristic Functions of Reward Sequence Distributions

14 citations · 15 across the 9 of their papers we have counts for

collaborators

9 papers

cs.LG2025

Accurate and Scalable Graph Neural Networks via Message Invariance

Zhihao Shi, Jie Wang, Zhiwei Zhuang +3

Message passing-based graph neural networks (GNNs) have achieved great success in many real-world applications. For a sampled mini-batch of target nodes, the message passing proces…

cs.LG2024★ 1 cited

Uncertainty-based Offline Variational Bayesian Reinforcement Learning for Robustness under Diverse Data Corruptions

Rui Yang, Jie Wang, Guoping Wu +1

Real-world offline datasets are often subject to data corruptions (such as noise or adversarial attacks) due to sensor failures or malicious attacks. Despite advances in robust off…

cs.LG2024

Deep Symbolic Optimization for Combinatorial Optimization: Accelerating Node Selection by Discovering Potential Heuristics

Hongyu Liu, Haoyang Liu, Yufei Kuang +2

Combinatorial optimization (CO) is one of the most fundamental mathematical models in real-world applications. Traditional CO solvers, such as Branch-and-Bound (B&B) solvers, heavi…

cs.LG2024

Accelerating Data Generation for Neural Operators via Krylov Subspace Recycling

Hong Wang, Zhongkai Hao, Jie Wang +4

Learning neural operators for solving partial differential equations (PDEs) has attracted great attention due to its high inference efficiency. However, training such operators req…

cs.LG2023

Provably Convergent Subgraph-wise Sampling for Fast GNN Training

Jie Wang, Zhihao Shi, Xize Liang +5

Subgraph-wise sampling -- a promising class of mini-batch training techniques for graph neural networks (GNNs -- is critical for real-world applications. During the message passing…

cs.LG2023

Generalization in Visual Reinforcement Learning with the Reward Sequence Distribution

Jie Wang, Rui Yang, Zijie Geng +7

Generalization in partially observed markov decision processes (POMDPs) is critical for successful applications of visual reinforcement learning (VRL) in real scenarios. A widely u…

cs.LG2022★ 14 cited

Learning Task-relevant Representations for Generalization via Characteristic Functions of Reward Sequence Distributions

Rui Yang, Jie Wang, Zijie Geng +4

Generalization across different environments with the same tasks is critical for successful applications of visual reinforcement learning (RL) in real scenarios. However, visual di…

cs.LG2021

Learning Robust Policy against Disturbance in Transition Dynamics via State-Conservative Policy Optimization

Yufei Kuang, Miao Lu, Jie Wang +3

Deep reinforcement learning algorithms can perform poorly in real-world tasks due to the discrepancy between source and target environments. This discrepancy is commonly viewed as…

cs.LG2021

Sample-Efficient Reinforcement Learning via Conservative Model-Based Actor-Critic

Zhihai Wang, Jie Wang, Qi Zhou +2

Model-based reinforcement learning algorithms, which aim to learn a model of the environment to make decisions, are more sample efficient than their model-free counterparts. The sa…