← Search

Jason Cong

9 accepted papers

2026

ARLArena: Demystifying Policy Gradient Stability in Agentic Reinforcement Learning

ICML 2026poster

Agentic reinforcement learning (ARL) has rapidly gained attention as a promising paradigm for training agents to solve complex, multi-step interactive tasks. In this paper, we first propose $\textbf{ARLArena}$, a fair and systematic analysis framework that encompasses a broad spectrum of ARL algorit…

Cited by 0SourceScholar
2026

Understand and Accelerate Memory Processing Pipeline for Large Language Model Inference

ICML 2026poster

Modern large language model (LLM) serving increasingly depends on efficient long-context processing and generation mechanisms, including sparse attention, retrieval-augmented generation (RAG), and compressed contextual memory, to support complex reasoning. We show that these optimizations can be uni…

Cited by 0SourceScholar
2025

Dynamic-Width Speculative Beam Decoding for LLM Inference

AAAI 2025technical

Large language models (LLMs) based on transformer architecture have shown outstanding performance across numerous real-world tasks. However, the autoregressive nature of these models makes the inference process slow and costly. Speculative decoding has emerged as a promising solution, leveraging a…

2025

HMT: Hierarchical Memory Transformer for Efficient Long Context Language Processing

NAACL 2025long

Transformer-based large language models (LLM) have been widely used in language processing applications. However, due to the memory constraints of the devices, most of them restrict the context window. Even though recurrent models in previous works can memorize past tokens to enable unlimited contex…

2025

Hierarchical Mixture of Experts: Generalizable Learning for High-Level Synthesis

AAAI 2025technical

High-level synthesis (HLS) is a widely used tool in designing Field Programmable Gate Array (FPGA). HLS enables FPGA design with software programming languages by compiling the source code into an FPGA circuit. The source code includes a program (called ``kernel'') and several pragmas that instruct…

2025

Optimized Multi-Token Joint Decoding With Auxiliary Model for LLM Inference

ICLR 2025poster

Large language models (LLMs) have achieved remarkable success across diverse tasks, yet their inference processes are hindered by substantial time and energy demands due to single-token generation at each decoding step. While previous methods such as speculative decoding mitigate these inefficiencie…

Cited by 3SourcePDFScholar
2023

Towards a Comprehensive Benchmark for High-Level Synthesis Targeted to FPGAs

NeurIPS 2023poster

High-level synthesis (HLS) aims to raise the abstraction layer in hardware design, enabling the design of domain-specific accelerators (DSAs) like field-programmable gate arrays (FPGAs) using C/C++ instead of hardware description languages (HDLs). Compiler directives in the form of pragmas play a cr…

2017

A cache-based bandwidth optimized motion compensation architecture for video decoder

ICASSP 2017accepted

In video decoder applications, motion compensation (MC) is bandwidth consuming because of the non-regular memory access. Especially with the popularity of UHD video and the development of new coding standard (HEVC), external memory bandwidth becomes a crucial bottleneck. In this paper, we propose an…

Cited by 0SourceScholar