← Search

Songzhu Mei

7 accepted papers

2026

MAGE: Multi-scale Autoregressive Generation for Offline Reinforcement Learning

ICLR 2026poster

Generative models have gained significant traction in offline reinforcement learning (RL) due to their ability to model complex trajectory distributions. However, existing generation-based approaches still struggle with long-horizon tasks characterized by sparse rewards. Some hierarchical generation…

Cited by 0SourcecodeScholar
2025

Dovetail: A CPU/GPU Heterogeneous Speculative Decoding for LLM inference

EMNLP 2025

With the continuous advancement in the performance of large language models (LLMs), their demand for computational resources and memory has significantly increased, which poses major challenges for efficient inference on consumer-grade devices and legacy servers. These devices typically feature rela

2025

GradPS: Resolving Futile Neurons in Parameter Sharing Network for Multi-Agent Reinforcement Learning

ICML 2025poster

Parameter-sharing (PS) techniques have been widely adopted in cooperative Multi-Agent Reinforcement Learning (MARL). In PS, all the agents share a policy network with identical parameters, which enjoys good sample efficiency. However, PS could lead to homogeneous policies that limit MARL performance…

2025

LLM-based Rumor Detection via Influence Guided Sample Selection and Game-based Perspective Analysis

ACL 2025long

Rumor detection on social media has become an emerging topic. Traditional deep learning-based methods model rumors based on content, propagation structure, or user behavior, but these approaches are constrained by limited modeling capacity and insufficient training corpora. Recent studies have explo…

Cited by 0SourcePDFScholar
2025

PlanU: Large Language Model Reasoning through Planning under Uncertainty

NeurIPS 2025poster

Large Language Models (LLMs) are increasingly being explored across a range of reasoning tasks. However, LLMs sometimes struggle with reasoning tasks under uncertainty that are relatively easy for humans, such as planning actions in stochastic environments. The adoption of LLMs for reasoning is impe…

Cited by 0SourceScholar
2024

The Dormant Neuron Phenomenon in Multi-Agent Reinforcement Learning Value Factorization

NeurIPS 2024poster

In this work, we study the dormant neuron phenomenon in multi-agent reinforcement learning value factorization, where the mixing network suffers from reduced network expressivity caused by an increasing number of inactive neurons. We demonstrate the presence of the dormant neuron phenomenon across m…

2023

RiskQ: Risk-sensitive Multi-Agent Reinforcement Learning Value Factorization

NeurIPS 2023poster

Multi-agent systems are characterized by environmental uncertainty, varying policies of agents, and partial observability, which result in significant risks. In the context of Multi-Agent Reinforcement Learning (MARL), learning coordinated and decentralized policies that are sensitive to risk is cha…