← Search

Hengjie Cao

5 accepted papers

2026

Metis: Training LLMs with FP4 Quantization

ICLR 2026poster

This work identifies anisotropy in the singular value spectra of parameters, activations, and gradients as the fundamental barrier to low-bit training of large language models (LLMs). These spectra are dominated by a small fraction of large singular values, inducing wide numerical ranges that cause…

Cited by 0SourceScholar
2026

Multi-Head Attention as a Source of Catastrophic Forgetting in MoE Transformers

ICML 2026poster

Mixture-of-Experts (MoE) architectures are often considered a natural fit for continual learning because sparse routing should localize updates and reduce interference, yet MoE Transformers still forget substantially even with sparse, well-balanced expert utilization. We attribute this gap to a pre-…

Cited by 0SourceScholar
2026

SD-MoE: Spectral Decomposition for Effective Expert Specialization

ICML 2026poster

Mixture-of-Experts (MoE) architectures scale Large Language Models via expert specialization induced by conditional computation. In practice, however, expert specialization often fails: some experts become functionally similar, while others functioning as de facto shared experts, limiting the effect…

Cited by 0SourceScholar
2026

Spectra: Rethinking Optimizers for LLMs Under Spectral Anisotropy

ICML 2026poster

Gradient signals in LLM training are highly anisotropic: recurrent linguistic structure concentrates energy into a small set of dominant spectral directions, while context-specific information resides in a long tail. We show that this spike–tail separation persists throughout training, with the spik…

Cited by 0SourceScholar
2025

Oracle-MoE: Locality-preserving Routing in the Oracle Space for Memory-constrained Large Language Model Inference

ICML 2025poster

Mixture-of-Experts (MoE) is widely adopted to deploy Large Language Models (LLMs) on edge devices with limited memory budgets. Although MoE is, in theory, an inborn memory-friendly architecture requiring only a few activated experts to reside in the memory for inference, current MoE architectures ca…

Cited by 0SourcePDFScholar