← Search

Jiamang Wang

8 accepted papers

2026

Attention Illuminates LLM Reasoning: The Uncovered Preplan-and-Anchor Rhythm Enables Fine-Grained Policy Optimization

ICML 2026poster

The reasoning patterns of large language models (LLMs) remain opaque, and Reinforcement learning (RL) typically assigns uniform credit across an entire generation, blurring the distinction between pivotal and routine steps. This work treats attention as a natural substrate for interpreting LLM reaso…

Cited by 0SourceScholar
2026

Reconstructing KV Caches with Cross-Layer Fusion for Enhanced Transformers

ICLR 2026poster

Transformer decoders have achieved strong results across tasks, but the memory required for the KV cache becomes prohibitive at long sequence lengths. Although Cross-layer KV Cache sharing (e.g., YOCO, CLA) offers a path to mitigate KV Cache bottleneck, it typically underperforms within-layer method…

Cited by 0SourcecodeScholar
2026

Tricks or Traps? A Deep Dive into RL for LLM Reasoning

ICLR 2026poster

Reinforcement learning (RL) for LLM reasoning has rapidly emerged as a prominent research area, marked by a significant surge in related studies on both algorithmic innovations and practical applications. Despite this progress, several critical challenges remain, including the absence of standardize…

Cited by 0SourcecodeScholar
2025

CombatVLA: An Efficient Vision-Language-Action Model for Combat Tasks in 3D Action Role-Playing Games

ICCV 2025poster

Recent advances in Vision-Language-Action models (VLAs) have expanded the capabilities of embodied intelligence. However, significant challenges remain in real-time decision-making in complex 3D environments, which demand second-level responses, high-resolution perception, and tactical reasoning und…

2025

Learning Grouped Lattice Vector Quantizers for Low-Bit LLM Compression

NeurIPS 2025poster

Large Language Models (LLMs) have demonstrated remarkable capabilities but typically require extensive computational resources and memory for inference. Post-training quantization (PTQ) can effectively reduce these demands by storing weights in lower bit-width formats. However, standard uniform quan…

Cited by 0SourcecodeScholar
2024

D-CPT Law: Domain-specific Continual Pre-Training Scaling Law for Large Language Models

NeurIPS 2024poster

Continual Pre-Training (CPT) on Large Language Models (LLMs) has been widely used to expand the model’s fundamental understanding of specific downstream domains (e.g., math and code). For the CPT on domain-specific LLMs, one important question is how to choose the optimal mixture ratio between the g…

Cited by 18SourcePDFScholar
2024

DDK: Distilling Domain Knowledge for Efficient Large Language Models

NeurIPS 2024poster

Despite the advanced intelligence abilities of large language models (LLMs) in various applications, they still face significant computational and storage demands. Knowledge Distillation (KD) has emerged as an effective strategy to improve the performance of a smaller LLM (i.e., the student model)…

Cited by 10SourcePDFScholar