← Search

Rui Kong

7 accepted papers

2026

AdaFuse: Accelerating Dynamic Adapter Inference via Token-Level Pre-Gating and Fused Kernel Optimization

AAAI 2026technical

The integration of dynamic, sparse structures like Mixture-of-Experts (MoE) with parameter-efficient adapters (e.g., LoRA) is a powerful technique for enhancing Large Language Models (LLMs). However, this architectural enhancement comes at a steep cost: despite minimal increases in computational loa

Cited by 0SourcePDFScholar
2024

ACT: Empowering Decision Transformer with Dynamic Programming via Advantage Conditioning

AAAI 2024technical

Decision Transformer (DT), which employs expressive sequence modeling techniques to perform action generation, has emerged as a promising approach to offline policy optimization. However, DT generates actions conditioned on a desired future return, which is known to bear some weaknesses such as the…

2024

Efficient and Stable Offline-to-online Reinforcement Learning via Continual Policy Revitalization

IJCAI 2024poster

In offline Reinforcement Learning (RL), the pre-trained policies are utilized for initialization and subsequent online fine-tuning. However, existing methods suffer from instability and low sample efficiency compared to pure online learning. This paper identifies these limitations stemming from dire…

2024

Generalizable Policy Improvement via Reinforcement Sampling (Student Abstract)

AAAI 2024technical

Current policy gradient techniques excel in refining policies over sampled states but falter when generalizing to unseen states. To address this, we introduce Reinforcement Sampling (RS), a novel method leveraging a generalizable action value function to sample improved decisions. RS is able to impr…

Cited by 0SourcePDFScholar
2024

SwapMoE: Serving Off-the-shelf MoE-based Large Language Models with Tunable Memory Budget

ACL 2024long

Mixture of experts (MoE) is a popular technique to improve capacity of Large Language Models (LLMs) with conditionally-activated parallel experts. However, serving MoE models on memory-constrained devices is challenging due to the large parameter size. Typical solutions such as memory swapping or ex…

Cited by 7SourcePDFScholar